Files
SabreTools/SabreTools.Library/DatFiles/SeparatedValue.cs
Matt Nadareski 3dc3c627f0 Checkpoint (nw)
2020-08-25 11:20:50 -07:00

607 lines
20 KiB
C#

using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using SabreTools.Library.Data;
using SabreTools.Library.DatItems;
using SabreTools.Library.IO;
using SabreTools.Library.Tools;
namespace SabreTools.Library.DatFiles
{
/// <summary>
/// Represents parsing and writing of a value-separated DAT
/// </summary>
internal class SeparatedValue : DatFile
{
// Private instance variables specific to Separated Value DATs
private readonly char _delim;
/// <summary>
/// Constructor designed for casting a base DatFile
/// </summary>
/// <param name="datFile">Parent DatFile to copy from</param>
/// <param name="delim">Delimiter for parsing individual lines</param>
public SeparatedValue(DatFile datFile, char delim)
: base(datFile)
{
_delim = delim;
}
/// <summary>
/// Parse a character-separated value DAT and return all found games and roms within
/// </summary>
/// <param name="filename">Name of the file to be parsed</param>
/// <param name="indexId">Index ID for the DAT</param>
/// <param name="keep">True if full pathnames are to be kept, false otherwise (default)</param>
protected override void ParseFile(
// Standard Dat parsing
string filename,
int indexId,
// Miscellaneous
bool keep)
{
// Open a file reader
Encoding enc = FileExtensions.GetEncoding(filename);
SeparatedValueReader svr = new SeparatedValueReader(FileExtensions.TryOpenRead(filename), enc)
{
Header = true,
Quotes = true,
Separator = _delim,
VerifyFieldCount = true
};
// If we're somehow at the end of the stream already, we can't do anything
if (svr.EndOfStream)
return;
// Read in the header
svr.ReadHeader();
// Loop through all of the data lines
while (!svr.EndOfStream)
{
try
{
// Get the current line, split and parse
svr.ReadNextLine();
}
catch (InvalidDataException)
{
Globals.Logger.Warning($"Malformed line found in '{filename}' at line {svr.LineNumber}");
continue;
}
// Create mapping dictionary
var mappings = new Dictionary<Field, string>();
// Now we loop through and get values for everything
for (int i = 0; i < svr.HeaderValues.Count; i++)
{
Field key = svr.HeaderValues[i].AsField();
string value = svr.Line[i];
mappings[key] = value;
}
// Set DatHeader fields
DatHeader header = new DatHeader();
header.SetFields(mappings);
Header.ConditionalCopy(header);
// Set Machine and DatItem fields
if (mappings.ContainsKey(Field.DatItem_Type))
{
DatItem datItem = DatItem.Create(mappings[Field.DatItem_Type].AsItemType());
datItem.SetFields(mappings);
datItem.Source = new Source(indexId, filename);
ParseAddHelper(datItem);
}
}
}
/// <summary>
/// Get normalized header value for a given separated value header field
/// </summary>
private string GetNormalizedHeader(string header)
{
switch (header.ToLowerInvariant())
{
#region Machine
#region Logiqx
case "board":
return "Machine.Board";
case "rebuildto":
case "rebuild to":
case "rebuild-to":
return "Machine.RebuildTo";
#endregion
#region Logiqx EmuArc
case "titleid":
case "title id":
case "title-id":
return "Machine.TitleID";
case "developer":
return "Machine.Developer";
case "genre":
return "Machine.Genre";
case "subgenre":
return "Machine.Subgenre";
case "ratings":
return "Machine.Ratings";
case "score":
return "Machine.Score";
case "enabled":
return "Machine.Enabled";
case "hascrc":
case "has crc":
case "has-crc":
return "Machine.HasCrc";
case "relatedto":
case "related to":
case "related-to":
return "Machine.RelatedTo";
#endregion
#region OpenMSX
case "genmsxid":
case "genmsx id":
case "genmsx-id":
case "gen msx id":
case "gen-msx-id":
return "Machine.GenMSXID";
case "msxsystem":
case "msx system":
case "msx-system":
return "Machine.System";
case "country":
return "Machine.Country";
#endregion
#region SoftwareList
case "supported":
return "Machine.Supported";
case "sharedfeat":
case "shared feat":
case "shared-feat":
case "sharedfeature":
case "shared feature":
case "shared-feature":
case "sharedfeatures":
case "shared features":
case "shared-features":
return "Machine.SharedFeatures";
case "dipswitch":
case "dip switch":
case "dip-switch":
case "dipswitches":
case "dip switches":
case "dip-switches":
return "Machine.DipSwitches";
#endregion
#endregion // Machine
#region DatItem
#region Common
case "itemtype":
case "item type":
case "type":
return "DatItem.Type";
case "disk":
case "diskname":
case "disk name":
case "item":
case "itemname":
case "item name":
case "name":
case "rom":
case "romname":
case "rom name":
return "DatItem.Name";
#endregion
#region AttractMode
case "altname":
case "alt name":
case "alt-name":
case "altromname":
case "alt romname":
case "alt-romname":
return "DatItem.AltName";
case "alttitle":
case "alt title":
case "alt-title":
case "altromtitle":
case "alt romtitle":
case "alt-romtitle":
return "DatItem.AltTitle";
#endregion
#region OpenMSX
case "original":
return "DatItem.Original";
case "subtype":
case "sub type":
case "sub-type":
case "openmsx_subtype":
return "DatItem.OpenMSXSubType";
case "openmsx_type":
return "DatItem.OpenMSXType";
case "remark":
return "DatItem.Remark";
case "boot":
return "DatItem.Boot";
#endregion
#region SoftwareList
case "partname":
case "part name":
case "part-name":
return "DatItem.PartName";
case "partinterface":
case "part interface":
case "part-interface":
return "DatItem.PartInterface";
case "features":
return "DatItem.Features";
case "areaname":
case "area name":
case "area-name":
return "DatItem.AreaName";
case "areasize":
case "area size":
case "area-size":
return "DatItem.AreaSize";
case "areawidth":
case "area width":
case "area-width":
return "DatItem.AreaWidth";
case "areaendinanness":
case "area endianness":
case "area-endianness":
return "DatItem.AreaEndianness";
case "value":
return "DatItem.Value";
case "loadflag":
case "load flag":
case "load-flag":
return "DatItem.LoadFlag";
#endregion
case "default":
return "DatItem.Default";
case "biosdescription":
case "bios description":
return "DatItem.Description";
case "itemsize":
case "item size":
case "size":
return "DatItem.Size";
case "crc":
case "crc hash":
return "DatItem.CRC";
case "md5":
case "md5 hash":
return "DatItem.MD5";
case "ripemd":
case "ripemd160":
case "ripemd hash":
case "ripemd160 hash":
return "DatItem.RIPEMD160";
case "sha1":
case "sha-1":
case "sha1 hash":
case "sha-1 hash":
return "DatItem.SHA1";
case "sha256":
case "sha-256":
case "sha256 hash":
case "sha-256 hash":
return "DatItem.SHA256";
case "sha384":
case "sha-384":
case "sha384 hash":
case "sha-384 hash":
return "DatItem.SHA384";
case "sha512":
case "sha-512":
case "sha512 hash":
case "sha-512 hash":
return "DatItem.SHA512";
case "merge":
case "mergetag":
case "merge tag":
return "DatItem.Merge";
case "region":
return "DatItem.Region";
case "index":
return "DatItem.Index";
case "writable":
return "DatItem.Writable";
case "optional":
return "DatItem.Optional";
case "nodump":
case "no dump":
case "status":
case "item status":
return "DatItem.Status";
case "language":
return "DatItem.Language";
case "date":
return "DatItem.Date";
case "bios":
return "DatItem.Bios";
case "offset":
return "DatItem.Offset";
case "inverted":
return "DatItem.Inverted";
#endregion // DatItem
default:
return "INVALID";
}
}
/// <summary>
/// Create and open an output file for writing direct from a dictionary
/// </summary>
/// <param name="outfile">Name of the file to write to</param>
/// <param name="ignoreblanks">True if blank roms should be skipped on output, false otherwise (default)</param>
/// <returns>True if the DAT was written correctly, false otherwise</returns>
public override bool WriteToFile(string outfile, bool ignoreblanks = false)
{
try
{
Globals.Logger.User($"Opening file for writing: {outfile}");
FileStream fs = FileExtensions.TryCreate(outfile);
// If we get back null for some reason, just log and return
if (fs == null)
{
Globals.Logger.Warning($"File '{outfile}' could not be created for writing! Please check to see if the file is writable");
return false;
}
SeparatedValueWriter svw = new SeparatedValueWriter(fs, new UTF8Encoding(false))
{
Quotes = true,
Separator = this._delim,
VerifyFieldCount = true
};
// Write out the header
WriteHeader(svw);
// Use a sorted list of games to output
foreach (string key in Items.SortedKeys)
{
List<DatItem> roms = Items[key];
// Resolve the names in the block
roms = DatItem.ResolveNames(roms);
for (int index = 0; index < roms.Count; index++)
{
DatItem rom = roms[index];
// There are apparently times when a null rom can skip by, skip them
if (rom.Name == null || rom.Machine.Name == null)
{
Globals.Logger.Warning("Null rom found!");
continue;
}
// If we have a "null" game (created by DATFromDir or something similar), log it to file
if (rom.ItemType == ItemType.Rom
&& ((Rom)rom).Size == -1
&& ((Rom)rom).CRC == "null")
{
Globals.Logger.Verbose($"Empty folder found: {rom.Machine.Name}");
}
// Now, output the rom data
WriteDatItem(svw, rom, ignoreblanks);
}
}
Globals.Logger.Verbose("File written!" + Environment.NewLine);
svw.Dispose();
fs.Dispose();
}
catch (Exception ex)
{
Globals.Logger.Error(ex.ToString());
return false;
}
return true;
}
/// <summary>
/// Write out DAT header using the supplied StreamWriter
/// </summary>
/// <param name="svw">SeparatedValueWriter to output to</param>
/// <returns>True if the data was written, false on error</returns>
private bool WriteHeader(SeparatedValueWriter svw)
{
try
{
string[] headers = new string[]
{
"File Name",
"Internal Name",
"Description",
"Game Name",
"Game Description",
"Type",
"Rom Name",
"Disk Name",
"Size",
"CRC",
"MD5",
//"RIPEMD160",
"SHA1",
"SHA256",
//"SHA384",
//"SHA512",
"Nodump",
};
svw.WriteHeader(headers);
svw.Flush();
}
catch (Exception ex)
{
Globals.Logger.Error(ex.ToString());
return false;
}
return true;
}
/// <summary>
/// Write out DatItem using the supplied StreamWriter
/// </summary>
/// <param name="svw">SeparatedValueWriter to output to</param>
/// <param name="datItem">DatItem object to be output</param>
/// <param name="ignoreblanks">True if blank roms should be skipped on output, false otherwise (default)</param>
/// <returns>True if the data was written, false on error</returns>
private bool WriteDatItem(SeparatedValueWriter svw, DatItem datItem, bool ignoreblanks = false)
{
// If we are in ignore blanks mode AND we have a blank (0-size) rom, skip
if (ignoreblanks && (datItem.ItemType == ItemType.Rom && ((datItem as Rom).Size == 0 || (datItem as Rom).Size == -1)))
return true;
try
{
// Separated values should only output Rom and Disk
if (datItem.ItemType != ItemType.Disk && datItem.ItemType != ItemType.Rom)
return true;
// Build the state
// TODO: Can we have some way of saying what fields to write out? Support for read extends to all fields now
string[] fields = new string[14]; // 17;
fields[0] = Header.FileName;
fields[1] = Header.Name;
fields[2] = Header.Description;
fields[3] = datItem.Machine.Name;
fields[4] = datItem.Machine.Description;
switch (datItem.ItemType)
{
case ItemType.Disk:
var disk = datItem as Disk;
fields[5] = "disk";
fields[6] = string.Empty;
fields[7] = disk.Name;
fields[8] = string.Empty;
fields[9] = string.Empty;
fields[10] = disk.MD5.ToLowerInvariant();
//fields[11] = disk.RIPEMD160?.ToLowerInvariant();
fields[11] = disk.SHA1.ToLowerInvariant();
fields[12] = disk.SHA256.ToLowerInvariant();
//fields[13] = disk.SHA384?.ToLowerInvariant();
//fields[14] = disk.SHA512?.ToLowerInvariant();
fields[13] = disk.ItemStatus.ToString();
break;
case ItemType.Rom:
var rom = datItem as Rom;
fields[5] = "rom";
fields[6] = rom.Name;
fields[7] = string.Empty;
fields[8] = rom.Size.ToString();
fields[9] = rom.CRC?.ToLowerInvariant();
fields[10] = rom.MD5?.ToLowerInvariant();
//fields[11] = rom.RIPEMD160?.ToLowerInvariant();
fields[11] = rom.SHA1?.ToLowerInvariant();
fields[12] = rom.SHA256?.ToLowerInvariant();
//fields[13] = rom.SHA384?.ToLowerInvariant();
//fields[14] = rom.SHA512?.ToLowerInvariant();
fields[13] = rom.ItemStatus.ToString();
break;
}
svw.WriteString(CreatePrefixPostfix(datItem, true));
svw.WriteValues(fields, false);
svw.WriteString(CreatePrefixPostfix(datItem, false));
svw.WriteLine();
svw.Flush();
}
catch (Exception ex)
{
Globals.Logger.Error(ex.ToString());
return false;
}
return true;
}
}
}