298 lines
11 KiB
C#
298 lines
11 KiB
C#
|
using System.Globalization;
|
|||
|
|
using System.IO.Compression;
|
|||
|
|
using System.Security.Cryptography;
|
|||
|
|
using System.Text;
|
|||
|
|
using System.Xml.Linq;
|
|||
|
|
|
|||
|
|
namespace MAMEHashTool
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
class roomHashInfo
|
|||
|
|
{
|
|||
|
|
public int RomID { get; set; }
|
|||
|
|
public string RomDirName { get; set; }
|
|||
|
|
public string RomAllFileHash { get; set; }
|
|||
|
|
public string ParentName { get; set; }
|
|||
|
|
public int ParentID { get; set; }
|
|||
|
|
public string desc { get; set; }
|
|||
|
|
public string board { get; set; }
|
|||
|
|
public string year { get; set; }
|
|||
|
|
public string manufacturer { get; set; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
internal class Program
|
|||
|
|
{
|
|||
|
|
static Dictionary<int, roomHashInfo> dict = new Dictionary<int, roomHashInfo>();
|
|||
|
|
static Dictionary<string, int> dictName2RomID = new Dictionary<string, int>();
|
|||
|
|
static string romRootDir = "G:\\MAME.Core\\roms";
|
|||
|
|
static HashSet<string> NoInDB = new HashSet<string>();
|
|||
|
|
static HashSet<string> NotfindParent = new HashSet<string>();
|
|||
|
|
static void Main(string[] args)
|
|||
|
|
{
|
|||
|
|
int RoomSeed = 10000;
|
|||
|
|
MAMEDBHelper.LoadROMXML(File.ReadAllText("./mame.xml"));
|
|||
|
|
// 获取当前程序所在目录
|
|||
|
|
string currentDirectory = romRootDir;
|
|||
|
|
|
|||
|
|
// 列出当前目录下的所有文件夹
|
|||
|
|
string[] directories = Directory.GetDirectories(currentDirectory);
|
|||
|
|
|
|||
|
|
// 文件夹名称
|
|||
|
|
foreach (string directory in directories)
|
|||
|
|
{
|
|||
|
|
// 使用Path.GetFileName来获取文件夹名称（不包含路径）
|
|||
|
|
string directoryName = Path.GetFileName(directory);
|
|||
|
|
Console.WriteLine($"正在处理，{RoomSeed}|{directoryName}");
|
|||
|
|
|
|||
|
|
|
|||
|
|
string[] files = Directory.GetFiles(directory);
|
|||
|
|
List<byte[]> filedatas = new List<byte[]>();
|
|||
|
|
foreach (string file in files)
|
|||
|
|
{
|
|||
|
|
filedatas.Add(File.ReadAllBytes(file));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!RomInfo.dictName2Rom.ContainsKey(directoryName))
|
|||
|
|
{
|
|||
|
|
if (!NoInDB.Contains(directoryName))
|
|||
|
|
NoInDB.Add(directoryName);
|
|||
|
|
|
|||
|
|
Console.WriteLine($"数据库中不存在|{directoryName}");
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
RomInfo xmlinfo = RomInfo.dictName2Rom[directoryName];
|
|||
|
|
roomHashInfo hashrom = new roomHashInfo()
|
|||
|
|
{
|
|||
|
|
RomID = RoomSeed,
|
|||
|
|
ParentName = xmlinfo.Parent,
|
|||
|
|
RomDirName = directoryName,
|
|||
|
|
RomAllFileHash = FileMD5Hash(filedatas),
|
|||
|
|
board = xmlinfo.Board,
|
|||
|
|
desc = xmlinfo.Description,
|
|||
|
|
manufacturer = xmlinfo.Manufacturer,
|
|||
|
|
year = xmlinfo.Year,
|
|||
|
|
};
|
|||
|
|
dict[RoomSeed] = hashrom;
|
|||
|
|
dictName2RomID[hashrom.RomDirName] = RoomSeed;
|
|||
|
|
//StartZip(romRootDir, directory);
|
|||
|
|
RoomSeed++;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
foreach (var rom in dict.Values)
|
|||
|
|
{
|
|||
|
|
if (string.IsNullOrEmpty(rom.ParentName))
|
|||
|
|
continue;
|
|||
|
|
if (!dictName2RomID.ContainsKey(rom.ParentName))
|
|||
|
|
{
|
|||
|
|
Console.WriteLine($"父级{rom.ParentName}无法找到");
|
|||
|
|
if (!NotfindParent.Contains(rom.ParentName))
|
|||
|
|
NotfindParent.Add(rom.ParentName);
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
rom.ParentID = dictName2RomID[rom.ParentName];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
List<string> tempOutInfoList = new List<string>();
|
|||
|
|
foreach (var rom in dict.Values)
|
|||
|
|
{
|
|||
|
|
string ParentID = rom.ParentID == 0 ? "" : rom.ParentID.ToString();
|
|||
|
|
tempOutInfoList.Add($"{rom.RomID}|{rom.RomDirName}|{rom.board}|{rom.year}|{rom.manufacturer}|{rom.desc}|{rom.RomAllFileHash}|{ParentID}");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Console.WriteLine($"共{NoInDB.Count}个数据库中不存在");
|
|||
|
|
tempOutInfoList.Add($"共{NoInDB.Count}个数据库中不存在");
|
|||
|
|
foreach (var name in NoInDB)
|
|||
|
|
{
|
|||
|
|
tempOutInfoList.Add(name);
|
|||
|
|
}
|
|||
|
|
Console.WriteLine($"共{NotfindParent.Count}个父级没找到");
|
|||
|
|
tempOutInfoList.Add($"共{NotfindParent.Count}个父级没找到");
|
|||
|
|
foreach (var name in NotfindParent)
|
|||
|
|
{
|
|||
|
|
tempOutInfoList.Add(name);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
File.WriteAllLines(romRootDir + "//axiromInfolist.txt", tempOutInfoList);
|
|||
|
|
Console.WriteLine($"写入文件完毕");
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
Console.WriteLine($"写入文件失败：{ex}");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Console.ReadLine();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static void StartZip(string rootPath, string subDirectory)
|
|||
|
|
{
|
|||
|
|
Console.WriteLine($"压缩{subDirectory}");
|
|||
|
|
string subDirectoryName = Path.GetFileName(subDirectory); // 获取子文件夹名称
|
|||
|
|
string zipFileName = Path.Combine(rootPath, subDirectoryName + ".zip"); // 构建ZIP文件名
|
|||
|
|
|
|||
|
|
// 创建ZIP文件
|
|||
|
|
using (FileStream zipStream = new FileStream(zipFileName, FileMode.Create))
|
|||
|
|
using (ZipArchive archive = new ZipArchive(zipStream, ZipArchiveMode.Create))
|
|||
|
|
{
|
|||
|
|
// 遍历子文件夹中的所有文件，并将它们添加到ZIP文件中
|
|||
|
|
foreach (string filePath in Directory.GetFiles(subDirectory, "*.*", SearchOption.AllDirectories))
|
|||
|
|
{
|
|||
|
|
// 获取相对于子文件夹的路径，以便在ZIP文件中保持正确的目录结构
|
|||
|
|
string relativePath = filePath.Substring(subDirectory.Length + 1); // +1 是为了排除子文件夹路径末尾的反斜杠
|
|||
|
|
|
|||
|
|
// 创建ZIP条目，使用相对于子文件夹的路径作为条目名称
|
|||
|
|
ZipArchiveEntry entry = archive.CreateEntry(relativePath, CompressionLevel.SmallestSize);
|
|||
|
|
|
|||
|
|
// 打开ZIP条目以进行写入
|
|||
|
|
using (Stream entryStream = entry.Open())
|
|||
|
|
{
|
|||
|
|
// 读取文件内容并写入ZIP条目
|
|||
|
|
using (FileStream fileStream = new FileStream(filePath, FileMode.Open, FileAccess.Read))
|
|||
|
|
{
|
|||
|
|
fileStream.CopyTo(entryStream);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Console.WriteLine($"Compressed {subDirectoryName} to {zipFileName}");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static byte[] FileMD5HashByte(byte[] data)
|
|||
|
|
{
|
|||
|
|
using (var md5 = MD5.Create())
|
|||
|
|
{
|
|||
|
|
using (var stream = new MemoryStream(data))
|
|||
|
|
{
|
|||
|
|
return md5.ComputeHash(stream);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 单个文件hash
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="data"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static string FileMD5Hash(byte[] data)
|
|||
|
|
{
|
|||
|
|
byte[] hash = FileMD5HashByte(data);
|
|||
|
|
var sb = new StringBuilder(hash.Length * 2);
|
|||
|
|
foreach (var b in hash)
|
|||
|
|
sb.AppendFormat("{0:x2}", b);
|
|||
|
|
return sb.ToString();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 多文件总hash (顺序不影响结果)
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="dataList"></param>
|
|||
|
|
/// <returns></returns>
|
|||
|
|
public static string FileMD5Hash(List<byte[]> dataList)
|
|||
|
|
{
|
|||
|
|
string allhash = string.Empty;
|
|||
|
|
List<string> temp = new List<string>();
|
|||
|
|
for (int i = 0; i < dataList.Count; i++)
|
|||
|
|
{
|
|||
|
|
if (dataList[i] == null)
|
|||
|
|
continue;
|
|||
|
|
temp.Add(FileMD5Hash(dataList[i]));
|
|||
|
|
}
|
|||
|
|
temp.Sort();
|
|||
|
|
var sb = new StringBuilder();
|
|||
|
|
for (int i = 0; i < temp.Count; i++)
|
|||
|
|
{
|
|||
|
|
sb.AppendLine(temp[i].ToString());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//这里用Ascll
|
|||
|
|
return FileMD5Hash(Encoding.ASCII.GetBytes(sb.ToString()));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
public class MAMEDBHelper
|
|||
|
|
{
|
|||
|
|
public static void LoadROMXML(string xmbString)
|
|||
|
|
{
|
|||
|
|
XElement xe = XElement.Parse(xmbString);
|
|||
|
|
IEnumerable<XElement> elements = from ele in xe.Elements("game") select ele;
|
|||
|
|
showInfoByElements(elements);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static void showInfoByElements(IEnumerable<XElement> elements)
|
|||
|
|
{
|
|||
|
|
RomInfo.romList = new List<RomInfo>();
|
|||
|
|
RomInfo.dictName2Rom = new Dictionary<string, RomInfo>();
|
|||
|
|
foreach (var ele in elements)
|
|||
|
|
{
|
|||
|
|
RomInfo rom = new RomInfo();
|
|||
|
|
rom.Name = ele.Attribute("name").Value;
|
|||
|
|
rom.Board = ele.Attribute("board").Value;
|
|||
|
|
rom.Parent = ele.Element("parent").Value;
|
|||
|
|
rom.Direction = ele.Element("direction").Value;
|
|||
|
|
rom.Description = ele.Element("description").Value;
|
|||
|
|
rom.Year = ele.Element("year").Value;
|
|||
|
|
rom.Manufacturer = ele.Element("manufacturer").Value;
|
|||
|
|
RomInfo.romList.Add(rom);
|
|||
|
|
RomInfo.dictName2Rom[rom.Name] = rom;
|
|||
|
|
//loadform.listView1.Items.Add(new ListViewItem(new string[] { rom.Description, rom.Year, rom.Name, rom.Parent, rom.Direction, rom.Manufacturer, rom.Board }));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
public class RomInfo
|
|||
|
|
{
|
|||
|
|
public static List<RomInfo> romList = new List<RomInfo>();
|
|||
|
|
public static Dictionary<string, RomInfo> dictName2Rom = new Dictionary<string, RomInfo>();
|
|||
|
|
public static RomInfo Rom;
|
|||
|
|
public string Name, Board;
|
|||
|
|
public string Parent;
|
|||
|
|
public string Direction;
|
|||
|
|
public string Description;
|
|||
|
|
public string Year;
|
|||
|
|
public string Manufacturer;
|
|||
|
|
public string M1Default, M1Stop, M1Min, M1Max, M1Subtype;
|
|||
|
|
public static ushort IStop;
|
|||
|
|
public RomInfo()
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
public static RomInfo GetRomByName(string s1)
|
|||
|
|
{
|
|||
|
|
if (!dictName2Rom.TryGetValue(s1, out RomInfo info))
|
|||
|
|
return null;
|
|||
|
|
return info;
|
|||
|
|
}
|
|||
|
|
public static string GetParent(string s1)
|
|||
|
|
{
|
|||
|
|
string sParent = "";
|
|||
|
|
foreach (RomInfo ri in romList)
|
|||
|
|
{
|
|||
|
|
if (s1 == ri.Name)
|
|||
|
|
{
|
|||
|
|
sParent = ri.Parent;
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return sParent;
|
|||
|
|
}
|
|||
|
|
public static List<string> GetParents(string s1)
|
|||
|
|
{
|
|||
|
|
string sChild, sParent;
|
|||
|
|
List<string> ls1 = new List<string>();
|
|||
|
|
sChild = s1;
|
|||
|
|
while (sChild != "")
|
|||
|
|
{
|
|||
|
|
ls1.Add(sChild);
|
|||
|
|
sParent = GetParent(sChild);
|
|||
|
|
sChild = sParent;
|
|||
|
|
}
|
|||
|
|
return ls1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|