SabreTools/SabreTools.Helper/Objects/Stats.cs

using System;
using System.Collections.Generic;

namespace SabreTools.Helper
{
	/// <summary>
	/// Get statistics on one or more DAT files
	/// </summary>
	public class Stats
	{
		// Private instance variables
		private List<String> _inputs;
		private bool _single;
		private Logger _logger;

		/// <summary>
		/// Create a new UncompressedSize object
		/// </summary>
		/// <param name="inputs">List of files and folders to parse</param>
		/// <param name="single">True if single DAT stats are output, false otherwise</param>
		/// <param name="logger">Logger object for file and console output</param>
		public Stats(List<String> inputs, bool single, Logger logger)
		{
			_inputs = inputs;
			_single = single;
			_logger = logger;
		}

		/// <summary>
		/// Output all requested statistics
		/// </summary>
		/// <returns>True if output succeeded, false otherwise</returns>
		public bool Process()
		{
			// Init all total variables
			long totalSize = 0;
			long totalGame = 0;
			long totalRom = 0;
			long totalDisk = 0;
			long totalCRC = 0;
			long totalMD5 = 0;
			long totalSHA1 = 0;
			long totalNodump = 0;

			/// Now process each of the input files
			foreach (string filename in _inputs)
			{
				_logger.Log("Beginning stat collection for '" + filename + "'");
				List<String> games = new List<String>();
				DatFile datdata = new DatFile();
				DatFile.Parse(filename, 0, 0, ref datdata, _logger);
				SortedDictionary<string, List<DatItem>> newroms = DatFile.BucketByGame(datdata.Files, false, true, _logger, false);

				// Output single DAT stats (if asked)
				if (_single)
				{
					_logger.User(@"\nFor file '" + filename + @"':
--------------------------------------------------");
					OutputStats(datdata, _logger);
				}
				else
				{
					_logger.User("Adding stats for file '" + filename + "'\n");
				}

				// Add single DAT stats to totals
				totalSize += datdata.TotalSize;
				totalGame += newroms.Count;
				totalRom += datdata.RomCount;
				totalDisk += datdata.DiskCount;
				totalCRC += datdata.CRCCount;
				totalMD5 += datdata.MD5Count;
				totalSHA1 += datdata.SHA1Count;
				totalNodump += datdata.NodumpCount;
			}

			// Output total DAT stats
			if (!_single) { _logger.User(""); }
			DatFile totaldata = new DatFile
			{
				TotalSize = totalSize,
				RomCount = totalRom,
				DiskCount = totalDisk,
				CRCCount = totalCRC,
				MD5Count = totalMD5,
				SHA1Count = totalSHA1,
				NodumpCount = totalNodump,
			};
			_logger.User(@"For ALL DATs found
--------------------------------------------------");
			OutputStats(totaldata, _logger, game:totalGame);
			_logger.User(@"
Please check the log folder if the stats scrolled offscreen");

			return true;
		}

		/// <summary>
		/// Output the stats in a human-readable format
		/// </summary>
		/// <param name="datdata">DatData object to read stats from</param>
		/// <param name="logger">Logger object for file and console writing</param>
		/// <param name="recalculate">True if numbers should be recalculated for the DAT, false otherwise (default)</param>
		/// <param name="game">Number of games to use, -1 means recalculate games (default)</param>
		public static void OutputStats(DatFile datdata, Logger logger, bool recalculate = false, long game = -1)
		{
			if (recalculate)
			{
				// Wipe out any stats already there
				datdata.RomCount = 0;
				datdata.DiskCount = 0;
				datdata.TotalSize = 0;
				datdata.CRCCount = 0;
				datdata.MD5Count = 0;
				datdata.SHA1Count = 0;
				datdata.NodumpCount = 0;

				// Loop through and add
				foreach (List<DatItem> roms in datdata.Files.Values)
				{
					foreach (Rom rom in roms)
					{
						datdata.RomCount += (rom.Type == ItemType.Rom ? 1 : 0);
						datdata.DiskCount += (rom.Type == ItemType.Disk ? 1 : 0);
						datdata.TotalSize += (rom.Nodump ? 0 : rom.HashData.Size);
						datdata.CRCCount += (String.IsNullOrEmpty(rom.HashData.CRC) ? 0 : 1);
						datdata.MD5Count += (String.IsNullOrEmpty(rom.HashData.MD5) ? 0 : 1);
						datdata.SHA1Count += (String.IsNullOrEmpty(rom.HashData.SHA1) ? 0 : 1);
						datdata.NodumpCount += (rom.Nodump ? 1 : 0);
					}
				}
			}

			SortedDictionary<string, List<DatItem>> newroms = DatFile.BucketByGame(datdata.Files, false, true, logger, false);
			if (datdata.TotalSize < 0)
			{
				datdata.TotalSize = Int64.MaxValue + datdata.TotalSize;
			}
			logger.User("    Uncompressed size:       " + Style.GetBytesReadable(datdata.TotalSize) + @"
    Games found:             " + (game == -1 ? newroms.Count : game) + @"
    Roms found:              " + datdata.RomCount + @"
    Disks found:             " + datdata.DiskCount + @"
    Roms with CRC:           " + datdata.CRCCount + @"
    Roms with MD5:           " + datdata.MD5Count + @"
    Roms with SHA-1:         " + datdata.SHA1Count + @"
    Roms with Nodump status: " + datdata.NodumpCount + @"
");
		}
	}
}