Files
LCD-Game-Shrinker/rom_parser.py
bzhxx 782aeb9616 Update rom_parser.py
Fix hh_sm510.cpp parsing. used a tested commit instead of master.
2021-12-05 15:45:47 +01:00

554 lines
21 KiB
Python
Executable File

#!/usr/bin/env python
# -*- coding: utf-8 -*-
""" ~LCD Game Shrinker~
parser used to automatically recover required parameters to perform
the shrinking process.
This program permits to shrink MAME high resolution artwork and
graphics for protable device running LCD game emulator.
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 3 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, see <http://www.gnu.org/licenses/>.
"""
import lxml
import urllib.request
import sys
import os
import rom_config as rom
__author__ = "bzhxx"
__contact__ = "https://github.com/bzhxx"
__license__ = "GPLv3"
"""
Various parameters for different ROM files
the repository of the original ROM files is composed of:
<rom_name> directory /
- BackgroundNS.png (mandatory)
- <rom name>.svg (mandatory)
- binary program (mandatory)
- melody (optional)
> background_file
This is the background ! BackgroundNS.png.
> segment_file
This is the SVG file where the segments are defined (.svg)
> program_file
the raw binary program file
> melody_file
the raw binary melody file (if it exists)
> full name
The name of the generated file at the end
> CPU to emulate
indicates which CPU need to be emulated
from MAME source code or .ini files, hh_sm510.cpp
The follwing values are supported
CPU_TYPE="SM512__\0"
CPU_TYPE="SM511__\0"
CPU_TYPE="SM510__\0"
CPU_TYPE="SM500__\0"
CPU_TYPE="SM5A___\0"
> background dimensions (applied on backgroundNS.png)
> screen dimensions (applied on .svg)
### Parameters from MAME layout file default.lay
# <view name="Background Only (No Shadow)">
background_x=0
background_y=0
background_width=1290
background_height=854
screen_x=35
screen_y=17
screen_width=1221
screen_height=802
> flag_rendering_lcd_inverted = True or False
graphic rendering (multiply)
If false (default case)
by default the background is fromm the background file
if a segment state is ON, the background and non_white segment pixels are 'RGB multiply".
or
If True ( case "tabletop" or "inverted lcd screen")
by default the background is black
if a segment state is ON, the background and the non_black segment pixels are 'RGB multiply".
> flag_background_jpeg =True or False
JPEG or all the rom is compressed wth LZ4
> jpeg_quality=90
typical values : 80 to 95
The following presets are available by default:
``web_low``, ``web_medium``, ``web_high``, ``web_very_high``, ``web_maximum``,
``low``, ``medium``, ``high``, ``maximum``.
https://github.com/python-pillow/Pillow/blob/master/src/PIL/JpegPresets.py#L71
> keys/buttons mapping
keys/buttons mapping can be found in MAME source file (hh_sm510.cpp)
CPU INPUTS S1..S8, BA, B
GW 8 BUTTONS used for retro-go: 8 bits
K4..K1 : 32 bits
LEFT,RIGHT,UP,DOWN,GAME,SELECT,A,B
>sound configuration
TODO define the appropritate functions to generate the sound piezo buzzer
"""
DEBUG = False
def log(s):
if DEBUG:
print(s)
def warm(s):
print(rom.mame_fullname + " WARNING:"+s + ' ['+str(rom.name) + ']')
def error(s):
print(rom.mame_fullname + " ERROR:"+s + ' [' + str(rom.name) + ']')
exit()
# def set_parameters(rom_name,mame_rom_dir, mame_driver_file):
def set_parameters(rom_name, mame_rom_dir):
rom.mame_rom_dir = mame_rom_dir
rom.name = rom_name
# ROM Reset values
rom.mame_year = 0
rom.mame_name = 0
rom.mame_parent = 0
rom.mame_comp = 0
rom.mame_machine = 0
rom.mame_input = 0
rom.mame_class = "none"
rom.mame_init = 0
rom.mame_company = "none"
rom.mame_fullname = "'unknown game'"
rom.mame_flags = 0
rom.background_file = "BackgroundNS.png"
rom.segments_file = "none"
rom.program_file = "none"
rom.melody_file = "none"
# dual screens
rom.segments_top_file = "none"
rom.segments_bottom_file = "none"
rom.segments_left_file = "none"
rom.segments_right_file = "none"
rom.dual_screen_vert = False
rom.dual_screen_hor = False
rom.CPU_TYPE = 0
rom.background_x = 0
rom.background_y = 0
rom.background_width = 0
rom.background_height = 0
rom.screen_x = 0
rom.screen_y = 0
rom.screen_width = 0
rom.screen_height = 0
rom.border_left_right = 0
rom.border_top_down = 0
rom.flag_rendering_lcd_inverted = False
# Buttons mapping according to the host machine
rom.BTN_SIZE = 10
rom.BTN_DATA = [0]*rom.BTN_SIZE
rom.flag_sound = rom.FLAG_SOUND_R1_PIEZO
rom.found = False
###################################################
mame_driver_file = os.path.join("./build/", "hh_sm510.cpp")
url_master = "https://raw.githubusercontent.com/mamedev/mame/master/src/mame/drivers/hh_sm510.cpp"
url_tested= "https://raw.githubusercontent.com/mamedev/mame/aaef28cd47db02b2b66359a49ca50c4ffaed464c/src/mame/drivers/hh_sm510.cpp"
#print('Beginning file download from MAME github...')
if not os.path.isfile(mame_driver_file):
urllib.request.urlretrieve(url_tested, mame_driver_file)
if not os.path.isfile(mame_driver_file):
error("Please copy hh_sm510.cpp in build directory")
# Get the full name
with open(mame_driver_file, "r", encoding="utf8") as myfile:
myline = myfile.readline()
while myline:
line_mame = myline.split(',')
try:
if (line_mame[0].find('CONS') > -1) & (line_mame[1].strip() == rom_name):
# MAME constructor structure
rom.mame_year = line_mame[0]
rom.mame_name = line_mame[1]
rom.mame_parent = line_mame[2]
rom.mame_comp = line_mame[3]
rom.mame_machine = line_mame[4]
rom.mame_input = line_mame[5]
rom.mame_class = line_mame[6]
rom.mame_init = line_mame[7]
rom.mame_company = line_mame[8]
rom.mame_fullname = myline.split("\"")[3]
rom.found = True
except:
pass
myline = myfile.readline()
if (rom.mame_fullname.find("Panorama Screen") > -1):
rom.flag_rendering_lcd_inverted = True
if (rom.mame_fullname.find("Table Top") > -1):
rom.flag_rendering_lcd_inverted = True
# possible extensions are none, '.program' or '.bin'
program_extension = ['.program', '.bin']
melody_extension = '.melody'
seg_extension = '.svg'
rom.dual_screen_hor = False
rom.dual_screen_vert = False
for root, dirs, files in os.walk(mame_rom_dir):
for f in files:
fileName, fileExtension = os.path.splitext(f)
if fileName.startswith("."):
continue
# maincpu
if fileExtension in program_extension:
rom.program_file = os.path.basename(f)
if not fileExtension:
rom.program_file = os.path.basename(f)
# maincpu:melody
if fileExtension == melody_extension:
rom.melody_file = os.path.basename(f)
# screen
if fileExtension == seg_extension:
# dual screens
if f.find("_top.svg") > -1:
rom.segments_top_file = os.path.basename(f)
rom.dual_screen_vert = True
rom.keep_aspect_ratio = False
if f.find("_bottom.svg") > -1:
rom.segments_bottom_file = os.path.basename(f)
rom.dual_screen_vert = True
rom.keep_aspect_ratio = False
if f.find("_left.svg") > -1:
rom.segments_left_file = os.path.basename(f)
rom.dual_screen_hor = True
rom.keep_aspect_ratio = False
if f.find("_right.svg") > -1:
rom.segments_right_file = os.path.basename(f)
rom.dual_screen_hor = True
rom.keep_aspect_ratio = False
# single screens
rom.segments_file = rom.name+".svg"
# GET CPU SM5A SM510 SM511 SM512 or KB
# Add sm510_tiger, sm511_tiger2bit supported
mame_class_found = False
if rom.found:
with open(mame_driver_file, "r", encoding="utf8") as myfile:
myline = myfile.readline()
while myline:
if (myline.find(rom.mame_class.strip()) > -1) & (myline.find(rom.mame_name.strip()) > -1) & (myline.find('machine_config') > -1):
mame_class_found = True
if mame_class_found:
if (myline.find('sm510_common') > -1):
rom.CPU_TYPE = 'SM510__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm511_common') > -1):
rom.CPU_TYPE = 'SM511__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm512_common') > -1):
rom.CPU_TYPE = 'SM512__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm5a_common') > -1):
rom.flag_lcd_deflicker_level = 2
rom.CPU_TYPE = 'SM5A___\0'
break
if (myline.find('kb1013vk12_common') > -1):
rom.flag_lcd_deflicker_level = 2
rom.CPU_TYPE = 'SM5A___\0'
break
if (myline.find('sm510_tiger') > -1):
rom.CPU_TYPE = 'SM510__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm511_tiger2bit') > -1):
rom.CPU_TYPE = 'SM511__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm510_dualv') > -1):
rom.CPU_TYPE = 'SM510__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm510_dualh') > -1):
rom.CPU_TYPE = 'SM510__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm511_dualv') > -1):
rom.CPU_TYPE = 'SM511__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm511_dualh') > -1):
rom.CPU_TYPE = 'SM511__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm512_dualv') > -1):
rom.CPU_TYPE = 'SM512__\0'
rom.flag_lcd_deflicker_level = 1
break
if (myline.find('sm512_dualh') > -1):
rom.CPU_TYPE = 'SM512__\0'
rom.flag_lcd_deflicker_level = 1
break
myline = myfile.readline()
log(rom.CPU_TYPE)
log(rom.flag_lcd_deflicker_level)
# Get Background and screen dimensions from MAME layout artwork file
layout_file = os.path.join(mame_rom_dir, 'default.lay')
tree = lxml.etree.parse(layout_file)
layout_root = tree.getroot()
rom.layout_found = False
# First tentative look for 'background' & 'only' in the name view
for x in layout_root:
if str(x.tag) == 'view':
if (( (str(x.attrib).upper().find('BACK') > -1) & (str(x.attrib).upper().find('ONLY') > -1) ) | (str(x.attrib).find('External Layout') > -1) ) & (str(x.attrib).upper().find('FAN') == -1):
log('START ----------------------------------------------------')
log(x.attrib)
# log('-BOUND-')
# bounds=x.find('bounds')
# bound_x = int(bounds.get('x'))
# bound_y = int(bounds.get('y'))
# bound_width= int(bounds.get('width'))
# bound_height= int(bounds.get('height'))
log('-SCREEN-')
rom.screen = x.find('screen')
screen_bounds = rom.screen.find('bounds')
rom.screen_x = int(screen_bounds.get('x'))
rom.screen_y = int(screen_bounds.get('y'))
rom.screen_width = int(screen_bounds.get('width'))
rom.screen_height = int(screen_bounds.get('height'))
log('-ELEMENTS/OVERLAY-')
all_element = x.findall('element')
for element in all_element:
element_bounds = element.find('bounds')
log(element.attrib)
log(element_bounds.attrib)
if (str(element.attrib).upper().find('GROUND') > -1) or \
(str(element.attrib).upper().find('BG') > -1) or \
(str(element.attrib).upper().find('OVERLAY') > -1):
background_ref = element.get('ref')
log(element.attrib)
log(element_bounds.attrib)
rom.background_x = int(element_bounds.get('x'))
rom.background_y = int(element_bounds.get('y'))
rom.background_width = int(
element_bounds.get('width'))
rom.background_height = int(
element_bounds.get('height'))
rom.layout_found = True
all_overlay = x.findall('overlay')
for element in all_overlay:
element_bounds = element.find('bounds')
log(element.attrib)
log(element_bounds.attrib)
if (str(element.attrib).upper().find('GROUND') > -1) or \
(str(element.attrib).upper().find('BG') > -1) or \
(str(element.attrib).upper().find('OVERLAY') > -1):
background_ref = element.get('ref')
log(element.attrib)
log(element_bounds.attrib)
rom.background_x = int(element_bounds.get('x'))
rom.background_y = int(element_bounds.get('y'))
rom.background_width = int(
element_bounds.get('width'))
rom.background_height = int(
element_bounds.get('height'))
rom.layout_found = True
log('END -------------------------------------------------------')
# Second tentative look for 'background' in the name view
if not rom.layout_found:
for x in layout_root:
if str(x.tag) == 'view':
if (str(x.attrib).upper().find('BACK') > -1):
log('START ----------------------------------------------------')
log(x.attrib)
# log('-BOUND-')
# bounds=x.find('bounds')
# bound_x = int(bounds.get('x'))
# bound_y = int(bounds.get('y'))
# bound_width= int(bounds.get('width'))
# bound_height= int(bounds.get('height'))
log('-SCREEN-')
rom.screen = x.find('screen')
screen_bounds = rom.screen.find('bounds')
rom.screen_x = int(screen_bounds.get('x'))
rom.screen_y = int(screen_bounds.get('y'))
rom.screen_width = int(screen_bounds.get('width'))
rom.screen_height = int(screen_bounds.get('height'))
log('-ELEMENTS/OVERLAY-')
all_element = x.findall('element')
for element in all_element:
element_bounds = element.find('bounds')
log(element.attrib)
log(element_bounds.attrib)
if (str(element.attrib).upper().find('GROUND') > -1) or \
(str(element.attrib).upper().find('BG') > -1) or \
(str(element.attrib).upper().find('OVERLAY') > -1):
background_ref = element.get('ref')
log(element.attrib)
log(element_bounds.attrib)
rom.background_x = int(element_bounds.get('x'))
rom.background_y = int(element_bounds.get('y'))
rom.background_width = int(
element_bounds.get('width'))
rom.background_height = int(
element_bounds.get('height'))
rom.layout_found = True
all_overlay = x.findall('overlay')
for element in all_overlay:
element_bounds = element.find('bounds')
log(element.attrib)
log(element_bounds.attrib)
if (str(element.attrib).upper().find('GROUND') > -1) or \
(str(element.attrib).upper().find('BG') > -1) or \
(str(element.attrib).upper().find('OVERLAY') > -1):
background_ref = element.get('ref')
if background_ref == None:
background_ref = element.get('element')
log(element.attrib)
log(element_bounds.attrib)
rom.background_x = int(element_bounds.get('x'))
rom.background_y = int(element_bounds.get('y'))
rom.background_width = int(
element_bounds.get('width'))
rom.background_height = int(
element_bounds.get('height'))
rom.layout_found = True
log('END -------------------------------------------------------')
if rom.layout_found:
# look for the artwork file name
log(background_ref)
all_element = layout_root.findall('element')
for element in all_element:
log(element.get('name'))
if str(element.get('name')) == background_ref:
rom.background_file = str(
(element.find('image').get('file')))
log(' background:'+rom.background_file)
if not rom.layout_found:
log('Failed to find a layout...')
log('background:'+rom.background_file)
log('background_x:'+str(rom.background_x))
log('background_y:'+str(rom.background_y))
log('background_width:'+str(rom.background_width))
log('background_height:'+str(rom.background_height))
log('screen_x:'+str(rom.screen_x))
log('screen_y:'+str(rom.screen_y))
log('screen_width:'+str(rom.screen_width))
log('screen_height:'+str(rom.screen_height))
# Rules according to ROM file name used as a python module
# check if there is a custom module to load
custom_script = os.path.join("custom", rom_name+".py")
if os.path.isfile(custom_script):
rom.custom_script_notfound = False
import importlib
module_name = "custom."+rom_name
if not (module_name in sys.modules):
importlib.import_module(module_name)
else:
importlib.reload(sys.modules[module_name])
else:
rom.custom_script_notfound = True