def __read_and_compare(self, script_fname, bytecode_fname, string_map):
with file(os.path.join('data', script_fname), 'r') as script_file:
input_str = script_file.read()
string_table = StringTable(string_map)
external_parser_context = ParserContext(string_table)
result = compiler.compile_to_string(input_str, external_parser_context)
with file(os.path.join('data', bytecode_fname), 'rb') as bytecode_file:
expected_bytecode = bytecode_file.read()
self.__compare_scripts(input_str, result, script_fname, string_table)
self.assertEqual(len(expected_bytecode), len(result))
for i, (expected, actual) in enumerate(itertools.izip(expected_bytecode, result)):
self.assertEqual(expected, actual, 'Values don\'t match: expected 0x{0:X}, vs actual 0x{1:X}, pos {2} (0x{2:X})'.format(
ord(expected), ord(actual), i
))
def import_file(self, json_file_name):
external_files = []
def decode_objects(dct):
try:
type_val = dct['__type__']
except KeyError:
return dct
if type_val == 'RulesData':
rules = RulesData()
rules.chunk1PointArray = dct['chunk1PointArray']
rules.characters = dct['characters']
rules.strings = dct['strings']
rules.scripts = dct['scripts']
rules.menuScripts = dct['menuScripts']
#rules.gameScriptIndexes = dct['gameScriptIndexes']
rules.gameScripts = dct['gameScripts']
rules.rulesChunk9 = dct['rulesChunk9']
#rules.chunk10Indexes = dct['chunk10Indexes']
rules.rulesChunk11 = dct['rulesChunk11']
rules.rectangles = dct['rectangles']
rules.interfaceTwoStepAction = dct['hotspots']['interfaceTwoStepAction']
rules.interfaceHotspotsX = dct['hotspots']['interfaceHotspotsX']
rules.interfaceHotspotsY = dct['hotspots']['interfaceHotspotsY']
rules.keyboardMapping = dct['hotspots']['keyboardMapping']
return rules
elif type_val == 'PointData':
point = PointData()
point.x = dct['x']
point.y = dct['y']
return point
elif type_val == 'CharacterData':
character = CharacterData()
character.posX = dct['posX']
character.posY = dct['posY']
character.posAltitude = dct['posAltitude']
character.frameArray = dct['frameArray']
character._rulesBuffer2_5 = dct['_rulesBuffer2_5']
character._rulesBuffer2_6 = dct['_rulesBuffer2_6']
character._rulesBuffer2_7 = dct['_rulesBuffer2_7']
character.spriteSize = dct['spriteSize']
character.direction = dct['direction']
character._rulesBuffer2_10 = dct['_rulesBuffer2_10']
character._rulesBuffer2_11 = dct['_rulesBuffer2_11']
character._rulesBuffer2_12 = dct['_rulesBuffer2_12']
character._rulesBuffer2_13_posX = dct['_rulesBuffer2_13_posX']
character._rulesBuffer2_14_posY = dct['_rulesBuffer2_14_posY']
character.variables = dct['variables']
character._rulesBuffer2_16 = dct['_rulesBuffer2_16']
return character
elif type_val == 'RectData':
rect = RectData()
rect.maxX = dct['maxX']
rect.minX = dct['minX']
rect.maxY = dct['maxY']
rect.minY = dct['minY']
rect.topLeftPosY = dct['topLeftPosY']
rect.topLeftPosX = dct['topLeftPosX']
rect.bottomRightPosY = dct['bottomRightPosY']
rect.bottomRightPosX = dct['bottomRightPosX']
return rect
elif type_val == 'RawData':
id = dct['id']
base_name = dct['path']
path_name = os.path.split(json_file_name)[0]
raw_fname = os.path.join(path_name, base_name)
raw_data = RawData(id, None) # Data will be populated later
external_files.append(
(raw_data, raw_fname)
)
return raw_data
elif type_val == 'ScriptData':
id = dct['id']
base_name = dct['path']
path_name = os.path.split(json_file_name)[0]
script_fname = os.path.join(path_name, 'scripts', base_name)
script_data = ScriptData(id, None) # Data will be populated later
external_files.append(
(script_data, script_fname)
)
return script_data
with file(json_file_name, 'r') as json_file:
rules = json.load(json_file, object_hook=decode_objects)
# Process external files after the main JSON. (ugly)
external_parser_context = ParserContext(rules.strings)
for obj, fname in external_files:
with file(fname, 'rb') as ext_file:
data = ext_file.read()
if isinstance(obj, ScriptData): # yuck
data = compiler.compile_to_string(data, external_parser_context)
obj.data = data
return rules
