Sprinter/mame
2
0
Fork 0
mirror of https://github.com/holub/mame synced 2025-04-19 15:11:37 +03:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

minimaws: identify software list dumps in CLI as well

Browse Source
This commit is contained in:
Vas Crabb 2019-12-16 11:52:14 +11:00
parent 53e3751bca
commit 87d31092e9
1 changed files with 93 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

142
scripts/minimaws/lib/auxverbs.py
View File

@ -18,53 +18,13 @@ class _Identifier(object):
def __init__(self, dbcurs, **kwargs):
super(_Identifier, self).__init__(**kwargs)
self.dbcurs = dbcurs
self.shortnamewitdh = 0
self.pathwidth = 0
self.labelwidth = 0
self.matches = { }
self.machines = { }
self.software = { }
self.unmatched = [ ]
def processRomFile(self, path, f):
crc, sha1 = self.digestRom(f)
matched = False
for shortname, description, label, bad in self.dbcurs.get_rom_dumps(crc, sha1):
matched = True
self.labelwidth = max(len(label), self.labelwidth)
romset = self.matches.get(shortname)
if romset is None:
romset = (description, [])
self.matches[shortname] = romset
romset[1].append((path, label, bad))
if not matched:
self.unmatched.append((path, crc, sha1))
self.pathwidth = max(len(path), self.pathwidth)
def processChd(self, path, sha1):
matched = False
for shortname, description, label, bad in self.dbcurs.get_disk_dumps(sha1):
matched = True
self.labelwidth = max(len(label), self.labelwidth)
romset = self.matches.get(shortname)
if romset is None:
romset = (description, [])
self.matches[shortname] = romset
romset[1].append((path, label, bad))
if not matched:
self.unmatched.append((path, None, sha1))
self.pathwidth = max(len(path), self.pathwidth)
def processFile(self, path):
if os.path.splitext(path)[1].lower() != '.chd':
with open(path, mode='rb', buffering=0) as f:
self.processRomFile(path, f)
else:
with open(path, mode='rb') as f:
sha1 = self.probeChd(f)
if sha1 is None:
f.seek(0)
self.processRomFile(path, f)
else:
self.processChd(path, sha1)
def processPath(self, path, depth=0):
try:
if not os.path.isdir(path):
@ -78,20 +38,30 @@ class _Identifier(object):
sys.stderr.write('Error identifying \'%s\': %s\n' % (path, e))
def printResults(self):
nw = self.shortnamewitdh - (self.shortnamewitdh % 4) + 4
pw = self.pathwidth - (self.pathwidth % 4) + 4
lw = self.labelwidth - (self.labelwidth % 4) + 4
first = True
for shortname, romset in sorted(self.matches.items()):
for shortname, romset in sorted(self.machines.items()):
if first:
first = False
else:
sys.stdout.write('\n')
sys.stdout.write('%-20s%s\n' % (shortname, romset[0]))
for path, label, bad in romset[1]:
if bad:
sys.stdout.write(' %-*s= %-*s(BAD)\n' % (pw, path, lw, label))
sys.stdout.write('%-*s%s\n' % (nw, shortname, romset[0]))
self.printMatches(romset[1], pw, lw)
for softwarelist, listinfo in sorted(self.software.items()):
for shortname, softwareinfo in sorted(listinfo[1].items()):
if first:
first = False
else:
sys.stdout.write(' %-*s= %s\n' % (pw, path, label))
sys.stdout.write('\n')
sys.stdout.write('%-*s%s\n' % (nw, '%s:%s' % (softwarelist, shortname), softwareinfo[0]))
for part, partinfo in sorted(softwareinfo[1].items()):
if partinfo[0] is not None:
sys.stdout.write('%-*s%s\n' % (nw, part, partinfo[0]))
else:
sys.stdout.write('%s\n' % (part, ))
self.printMatches(partinfo[1], pw, lw)
if self.unmatched:
if first:
first = False
@ -104,6 +74,80 @@ class _Identifier(object):
else:
sys.stdout.write(' %-*sSHA1(%s)\n' % (pw, path, sha1))
def getMachineMatches(self, shortname, description):
result = self.machines.get(shortname)
if result is None:
result = (description, [ ])
self.machines[shortname] = result
return result[1]
def getSoftwareMatches(self, softwarelist, softwarelistdescription, shortname, description, part, part_id):
listinfo = self.software.get(softwarelist)
if listinfo is None:
listinfo = (softwarelistdescription, { })
self.software[softwarelist] = listinfo
softwareinfo = listinfo[1].get(shortname)
if softwareinfo is None:
softwareinfo = (description, { })
listinfo[1][shortname] = softwareinfo
partinfo = softwareinfo[1].get(part)
if partinfo is None:
partinfo = (part_id, [ ])
softwareinfo[1][part] = partinfo
return partinfo[1]
def processRomFile(self, path, f):
crc, sha1 = self.digestRom(f)
matched = False
for shortname, description, label, bad in self.dbcurs.get_rom_dumps(crc, sha1):
matched = True
self.shortnamewitdh = max(len(shortname), self.shortnamewitdh)
self.labelwidth = max(len(label), self.labelwidth)
self.getMachineMatches(shortname, description).append((path, label, bad))
for softwarelist, softwarelistdescription, shortname, description, part, part_id, label, bad in self.dbcurs.get_software_rom_dumps(crc, sha1):
matched = True
self.shortnamewitdh = max(len(softwarelist) + 1 + len(shortname), len(part), self.shortnamewitdh)
self.labelwidth = max(len(label), self.labelwidth)
self.getSoftwareMatches(softwarelist, softwarelistdescription, shortname, description, part, part_id).append((path, label, bad))
if not matched:
self.unmatched.append((path, crc, sha1))
def processChd(self, path, sha1):
matched = False
for shortname, description, label, bad in self.dbcurs.get_disk_dumps(sha1):
matched = True
self.shortnamewitdh = max(len(shortname), self.shortnamewitdh)
self.labelwidth = max(len(label), self.labelwidth)
self.getMachineMatches(shortname, description).append((path, label, bad))
for softwarelist, softwarelistdescription, shortname, description, part, part_id, label, bad in self.dbcurs.get_software_disk_dumps(sha1):
matched = True
self.shortnamewitdh = max(len(softwarelist) + 1 + len(shortname), len(part), self.shortnamewitdh)
self.labelwidth = max(len(label), self.labelwidth)
self.getSoftwareMatches(softwarelist, softwarelistdescription, shortname, description, part, part_id).append((path, label, bad))
if not matched:
self.unmatched.append((path, None, sha1))
def processFile(self, path):
if os.path.splitext(path)[1].lower() != '.chd':
with open(path, mode='rb', buffering=0) as f:
self.processRomFile(path, f)
else:
with open(path, mode='rb') as f:
sha1 = self.probeChd(f)
if sha1 is None:
f.seek(0)
self.processRomFile(path, f)
else:
self.processChd(path, sha1)
self.pathwidth = max(len(path), self.pathwidth)
def printMatches(self, matches, pathwidth, labelwidth):
for path, label, bad in matches:
if bad:
sys.stdout.write(' %-*s= %-*s(BAD)\n' % (pathwidth, path, labelwidth, label))
else:
sys.stdout.write(' %-*s= %s\n' % (pathwidth, path, label))
@staticmethod
def iterateBlocks(f, s=65536):
while True: