mirror of
https://github.com/chibicitiberiu/rainmeter-studio.git
synced 2024-02-24 04:33:31 +00:00
Fixed line endings and applied gitignore
This commit is contained in:
@ -1,245 +1,245 @@
|
||||
/*
|
||||
Copyright (C) 2001 Kimmo Pekkola
|
||||
|
||||
This program is free software; you can redistribute it and/or
|
||||
modify it under the terms of the GNU General Public License
|
||||
as published by the Free Software Foundation; either version 2
|
||||
of the License, or (at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
*/
|
||||
|
||||
#include "StdAfx.h"
|
||||
#include "MeasureDiskSpace.h"
|
||||
#include "Rainmeter.h"
|
||||
#include "System.h"
|
||||
#include "../Common/PathUtil.h"
|
||||
|
||||
enum DRIVETYPE
|
||||
{
|
||||
DRIVETYPE_ERROR = 0,
|
||||
DRIVETYPE_REMOVED = 1,
|
||||
DRIVETYPE_REMOVABLE = 3,
|
||||
DRIVETYPE_FIXED = 4,
|
||||
DRIVETYPE_NETWORK = 5,
|
||||
DRIVETYPE_CDROM = 6,
|
||||
DRIVETYPE_RAM = 7,
|
||||
|
||||
DRIVETYPE_MAX = DRIVETYPE_RAM
|
||||
};
|
||||
|
||||
/*
|
||||
** The constructor
|
||||
**
|
||||
*/
|
||||
MeasureDiskSpace::MeasureDiskSpace(MeterWindow* meterWindow, const WCHAR* name) : Measure(meterWindow, name),
|
||||
m_Type(false),
|
||||
m_Total(false),
|
||||
m_Label(false),
|
||||
m_IgnoreRemovable(true),
|
||||
m_DiskQuota(true),
|
||||
m_OldTotalBytes()
|
||||
{
|
||||
}
|
||||
|
||||
/*
|
||||
** The destructor
|
||||
**
|
||||
*/
|
||||
MeasureDiskSpace::~MeasureDiskSpace()
|
||||
{
|
||||
}
|
||||
|
||||
/*
|
||||
** Updates the current disk free space value.
|
||||
**
|
||||
*/
|
||||
void MeasureDiskSpace::UpdateValue()
|
||||
{
|
||||
if (!m_Drive.empty())
|
||||
{
|
||||
const WCHAR* drive = m_Drive.c_str();
|
||||
UINT type = GetDriveType(drive);
|
||||
|
||||
if (m_Type)
|
||||
{
|
||||
switch (type)
|
||||
{
|
||||
case DRIVE_UNKNOWN:
|
||||
case DRIVE_NO_ROOT_DIR:
|
||||
m_Value = DRIVETYPE_REMOVED;
|
||||
m_StringValue = L"Removed";
|
||||
break;
|
||||
case DRIVE_REMOVABLE:
|
||||
m_Value = DRIVETYPE_REMOVABLE;
|
||||
m_StringValue = L"Removable";
|
||||
break;
|
||||
case DRIVE_FIXED:
|
||||
m_Value = DRIVETYPE_FIXED;
|
||||
m_StringValue = L"Fixed";
|
||||
break;
|
||||
case DRIVE_REMOTE:
|
||||
m_Value = DRIVETYPE_NETWORK;
|
||||
m_StringValue = L"Network";
|
||||
break;
|
||||
case DRIVE_CDROM:
|
||||
m_Value = DRIVETYPE_CDROM;
|
||||
m_StringValue = L"CDRom";
|
||||
break;
|
||||
case DRIVE_RAMDISK:
|
||||
m_Value = DRIVETYPE_RAM;
|
||||
m_StringValue = L"Ram";
|
||||
break;
|
||||
default:
|
||||
m_Value = DRIVETYPE_ERROR;
|
||||
m_StringValue = L"Error";
|
||||
break;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
BOOL sizeResult = FALSE;
|
||||
ULONGLONG i64TotalBytes, i64FreeBytes;
|
||||
|
||||
if (type != DRIVE_NO_ROOT_DIR &&
|
||||
type != DRIVE_CDROM &&
|
||||
(!m_IgnoreRemovable || type != DRIVE_REMOVABLE)) // Ignore CD-ROMS and removable drives
|
||||
{
|
||||
if (!m_DiskQuota)
|
||||
{
|
||||
sizeResult = GetDiskFreeSpaceEx(drive, nullptr, (PULARGE_INTEGER)&i64TotalBytes, (PULARGE_INTEGER)&i64FreeBytes);
|
||||
}
|
||||
else
|
||||
{
|
||||
sizeResult = GetDiskFreeSpaceEx(drive, (PULARGE_INTEGER)&i64FreeBytes, (PULARGE_INTEGER)&i64TotalBytes, nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
if (sizeResult)
|
||||
{
|
||||
m_Value = (double)(__int64)((m_Total) ? i64TotalBytes : i64FreeBytes);
|
||||
|
||||
if (i64TotalBytes != m_OldTotalBytes)
|
||||
{
|
||||
// Total size was changed, so set new max value.
|
||||
m_MaxValue = (double)(__int64)i64TotalBytes;
|
||||
m_OldTotalBytes = i64TotalBytes;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
m_Value = 0.0;
|
||||
m_MaxValue = 0.0;
|
||||
m_OldTotalBytes = 0;
|
||||
}
|
||||
|
||||
if (m_Label)
|
||||
{
|
||||
BOOL labelResult = FALSE;
|
||||
WCHAR volumeName[MAX_PATH + 1];
|
||||
|
||||
if (type != DRIVE_NO_ROOT_DIR &&
|
||||
(!m_IgnoreRemovable || type != DRIVE_REMOVABLE)) // Ignore removable drives
|
||||
{
|
||||
labelResult = GetVolumeInformation(drive, volumeName, MAX_PATH + 1, nullptr, nullptr, nullptr, nullptr, 0);
|
||||
}
|
||||
|
||||
m_StringValue = (labelResult) ? volumeName : L"";
|
||||
}
|
||||
else if (!m_StringValue.empty())
|
||||
{
|
||||
m_StringValue.clear();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
** Returns the time as string.
|
||||
**
|
||||
*/
|
||||
const WCHAR* MeasureDiskSpace::GetStringValue()
|
||||
{
|
||||
return (m_Type || m_Label) ? CheckSubstitute(m_StringValue.c_str()) : nullptr;
|
||||
}
|
||||
|
||||
/*
|
||||
** Read the options specified in the ini file.
|
||||
**
|
||||
*/
|
||||
void MeasureDiskSpace::ReadOptions(ConfigParser& parser, const WCHAR* section)
|
||||
{
|
||||
double oldMaxValue = m_MaxValue;
|
||||
|
||||
Measure::ReadOptions(parser, section);
|
||||
|
||||
m_Drive = parser.ReadString(section, L"Drive", L"C:\\");
|
||||
if (m_Drive.empty())
|
||||
{
|
||||
LogWarningF(this, L"FreeDiskSpace: Drive= empty");
|
||||
m_Value = 0.0;
|
||||
m_MaxValue = 0.0;
|
||||
m_OldTotalBytes = 0;
|
||||
m_StringValue.clear();
|
||||
}
|
||||
else
|
||||
{
|
||||
// A trailing backslash is required for GetDiskFreeSpaceEx().
|
||||
PathUtil::AppendBacklashIfMissing(m_Drive);
|
||||
}
|
||||
|
||||
m_Type = parser.ReadBool(section, L"Type", false);
|
||||
m_Total = parser.ReadBool(section, L"Total", false);
|
||||
m_Label = parser.ReadBool(section, L"Label", false);
|
||||
m_IgnoreRemovable = parser.ReadBool(section, L"IgnoreRemovable", true);
|
||||
m_DiskQuota = parser.ReadBool(section, L"DiskQuota", true);
|
||||
|
||||
// Set the m_MaxValue
|
||||
if (!m_Initialized)
|
||||
{
|
||||
BOOL result = FALSE;
|
||||
ULONGLONG i64TotalBytes;
|
||||
|
||||
if (!m_Drive.empty())
|
||||
{
|
||||
const WCHAR* drive = m_Drive.c_str();
|
||||
UINT type = GetDriveType(drive);
|
||||
if (type != DRIVE_NO_ROOT_DIR &&
|
||||
type != DRIVE_CDROM &&
|
||||
(!m_IgnoreRemovable || type != DRIVE_REMOVABLE)) // Ignore CD-ROMS and removable drives
|
||||
{
|
||||
result = GetDiskFreeSpaceEx(drive, nullptr, (PULARGE_INTEGER)&i64TotalBytes, nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
if (result)
|
||||
{
|
||||
m_MaxValue = (double)(__int64)i64TotalBytes;
|
||||
m_OldTotalBytes = i64TotalBytes;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_MaxValue = 0.0;
|
||||
m_OldTotalBytes = 0;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (m_Type)
|
||||
{
|
||||
m_MaxValue = DRIVETYPE_MAX;
|
||||
m_OldTotalBytes = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_MaxValue = oldMaxValue;
|
||||
}
|
||||
}
|
||||
}
|
||||
/*
|
||||
Copyright (C) 2001 Kimmo Pekkola
|
||||
|
||||
This program is free software; you can redistribute it and/or
|
||||
modify it under the terms of the GNU General Public License
|
||||
as published by the Free Software Foundation; either version 2
|
||||
of the License, or (at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program; if not, write to the Free Software
|
||||
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
*/
|
||||
|
||||
#include "StdAfx.h"
|
||||
#include "MeasureDiskSpace.h"
|
||||
#include "Rainmeter.h"
|
||||
#include "System.h"
|
||||
#include "../Common/PathUtil.h"
|
||||
|
||||
enum DRIVETYPE
|
||||
{
|
||||
DRIVETYPE_ERROR = 0,
|
||||
DRIVETYPE_REMOVED = 1,
|
||||
DRIVETYPE_REMOVABLE = 3,
|
||||
DRIVETYPE_FIXED = 4,
|
||||
DRIVETYPE_NETWORK = 5,
|
||||
DRIVETYPE_CDROM = 6,
|
||||
DRIVETYPE_RAM = 7,
|
||||
|
||||
DRIVETYPE_MAX = DRIVETYPE_RAM
|
||||
};
|
||||
|
||||
/*
|
||||
** The constructor
|
||||
**
|
||||
*/
|
||||
MeasureDiskSpace::MeasureDiskSpace(MeterWindow* meterWindow, const WCHAR* name) : Measure(meterWindow, name),
|
||||
m_Type(false),
|
||||
m_Total(false),
|
||||
m_Label(false),
|
||||
m_IgnoreRemovable(true),
|
||||
m_DiskQuota(true),
|
||||
m_OldTotalBytes()
|
||||
{
|
||||
}
|
||||
|
||||
/*
|
||||
** The destructor
|
||||
**
|
||||
*/
|
||||
MeasureDiskSpace::~MeasureDiskSpace()
|
||||
{
|
||||
}
|
||||
|
||||
/*
|
||||
** Updates the current disk free space value.
|
||||
**
|
||||
*/
|
||||
void MeasureDiskSpace::UpdateValue()
|
||||
{
|
||||
if (!m_Drive.empty())
|
||||
{
|
||||
const WCHAR* drive = m_Drive.c_str();
|
||||
UINT type = GetDriveType(drive);
|
||||
|
||||
if (m_Type)
|
||||
{
|
||||
switch (type)
|
||||
{
|
||||
case DRIVE_UNKNOWN:
|
||||
case DRIVE_NO_ROOT_DIR:
|
||||
m_Value = DRIVETYPE_REMOVED;
|
||||
m_StringValue = L"Removed";
|
||||
break;
|
||||
case DRIVE_REMOVABLE:
|
||||
m_Value = DRIVETYPE_REMOVABLE;
|
||||
m_StringValue = L"Removable";
|
||||
break;
|
||||
case DRIVE_FIXED:
|
||||
m_Value = DRIVETYPE_FIXED;
|
||||
m_StringValue = L"Fixed";
|
||||
break;
|
||||
case DRIVE_REMOTE:
|
||||
m_Value = DRIVETYPE_NETWORK;
|
||||
m_StringValue = L"Network";
|
||||
break;
|
||||
case DRIVE_CDROM:
|
||||
m_Value = DRIVETYPE_CDROM;
|
||||
m_StringValue = L"CDRom";
|
||||
break;
|
||||
case DRIVE_RAMDISK:
|
||||
m_Value = DRIVETYPE_RAM;
|
||||
m_StringValue = L"Ram";
|
||||
break;
|
||||
default:
|
||||
m_Value = DRIVETYPE_ERROR;
|
||||
m_StringValue = L"Error";
|
||||
break;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
BOOL sizeResult = FALSE;
|
||||
ULONGLONG i64TotalBytes, i64FreeBytes;
|
||||
|
||||
if (type != DRIVE_NO_ROOT_DIR &&
|
||||
type != DRIVE_CDROM &&
|
||||
(!m_IgnoreRemovable || type != DRIVE_REMOVABLE)) // Ignore CD-ROMS and removable drives
|
||||
{
|
||||
if (!m_DiskQuota)
|
||||
{
|
||||
sizeResult = GetDiskFreeSpaceEx(drive, nullptr, (PULARGE_INTEGER)&i64TotalBytes, (PULARGE_INTEGER)&i64FreeBytes);
|
||||
}
|
||||
else
|
||||
{
|
||||
sizeResult = GetDiskFreeSpaceEx(drive, (PULARGE_INTEGER)&i64FreeBytes, (PULARGE_INTEGER)&i64TotalBytes, nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
if (sizeResult)
|
||||
{
|
||||
m_Value = (double)(__int64)((m_Total) ? i64TotalBytes : i64FreeBytes);
|
||||
|
||||
if (i64TotalBytes != m_OldTotalBytes)
|
||||
{
|
||||
// Total size was changed, so set new max value.
|
||||
m_MaxValue = (double)(__int64)i64TotalBytes;
|
||||
m_OldTotalBytes = i64TotalBytes;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
m_Value = 0.0;
|
||||
m_MaxValue = 0.0;
|
||||
m_OldTotalBytes = 0;
|
||||
}
|
||||
|
||||
if (m_Label)
|
||||
{
|
||||
BOOL labelResult = FALSE;
|
||||
WCHAR volumeName[MAX_PATH + 1];
|
||||
|
||||
if (type != DRIVE_NO_ROOT_DIR &&
|
||||
(!m_IgnoreRemovable || type != DRIVE_REMOVABLE)) // Ignore removable drives
|
||||
{
|
||||
labelResult = GetVolumeInformation(drive, volumeName, MAX_PATH + 1, nullptr, nullptr, nullptr, nullptr, 0);
|
||||
}
|
||||
|
||||
m_StringValue = (labelResult) ? volumeName : L"";
|
||||
}
|
||||
else if (!m_StringValue.empty())
|
||||
{
|
||||
m_StringValue.clear();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
** Returns the time as string.
|
||||
**
|
||||
*/
|
||||
const WCHAR* MeasureDiskSpace::GetStringValue()
|
||||
{
|
||||
return (m_Type || m_Label) ? CheckSubstitute(m_StringValue.c_str()) : nullptr;
|
||||
}
|
||||
|
||||
/*
|
||||
** Read the options specified in the ini file.
|
||||
**
|
||||
*/
|
||||
void MeasureDiskSpace::ReadOptions(ConfigParser& parser, const WCHAR* section)
|
||||
{
|
||||
double oldMaxValue = m_MaxValue;
|
||||
|
||||
Measure::ReadOptions(parser, section);
|
||||
|
||||
m_Drive = parser.ReadString(section, L"Drive", L"C:\\");
|
||||
if (m_Drive.empty())
|
||||
{
|
||||
LogWarningF(this, L"FreeDiskSpace: Drive= empty");
|
||||
m_Value = 0.0;
|
||||
m_MaxValue = 0.0;
|
||||
m_OldTotalBytes = 0;
|
||||
m_StringValue.clear();
|
||||
}
|
||||
else
|
||||
{
|
||||
// A trailing backslash is required for GetDiskFreeSpaceEx().
|
||||
PathUtil::AppendBacklashIfMissing(m_Drive);
|
||||
}
|
||||
|
||||
m_Type = parser.ReadBool(section, L"Type", false);
|
||||
m_Total = parser.ReadBool(section, L"Total", false);
|
||||
m_Label = parser.ReadBool(section, L"Label", false);
|
||||
m_IgnoreRemovable = parser.ReadBool(section, L"IgnoreRemovable", true);
|
||||
m_DiskQuota = parser.ReadBool(section, L"DiskQuota", true);
|
||||
|
||||
// Set the m_MaxValue
|
||||
if (!m_Initialized)
|
||||
{
|
||||
BOOL result = FALSE;
|
||||
ULONGLONG i64TotalBytes;
|
||||
|
||||
if (!m_Drive.empty())
|
||||
{
|
||||
const WCHAR* drive = m_Drive.c_str();
|
||||
UINT type = GetDriveType(drive);
|
||||
if (type != DRIVE_NO_ROOT_DIR &&
|
||||
type != DRIVE_CDROM &&
|
||||
(!m_IgnoreRemovable || type != DRIVE_REMOVABLE)) // Ignore CD-ROMS and removable drives
|
||||
{
|
||||
result = GetDiskFreeSpaceEx(drive, nullptr, (PULARGE_INTEGER)&i64TotalBytes, nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
if (result)
|
||||
{
|
||||
m_MaxValue = (double)(__int64)i64TotalBytes;
|
||||
m_OldTotalBytes = i64TotalBytes;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_MaxValue = 0.0;
|
||||
m_OldTotalBytes = 0;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (m_Type)
|
||||
{
|
||||
m_MaxValue = DRIVETYPE_MAX;
|
||||
m_OldTotalBytes = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
m_MaxValue = oldMaxValue;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user