1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
|
#author: Bryan Bishop <kanzure@gmail.com>
#date: 2012-01-15
#replace dimensions with constants
import sys #for non-newline-terminated output :/
from add_map_labels_to_map_headers import find_with_start_of_line
from pretty_map_headers import map_name_cleaner, spacing, offset_to_pointer, map_constants
from connection_helper import print_connections
from ctypes import c_int8
# X/Y_Movement_Of_Connection
#~~~~~~~~~~~~~~~~~~~~~~~~~
#
# A X movement is how many map blocks there are to the left of one of your north/south connections.
# A Y movement is how many map blocks there are above your west/east connection.
#===============================================================================
# #4-#5 : Current Map Position
#===============================================================================
#
# This points to the part of the current map (further up in RAM)
# that the connection strips upperleft block is placed on the current map.
#
# ____________________
# Connection |
# Direction | Formula
# ___________|_______
#
# North: C6EB + X_Movement_of_Connection Strip
#
# South: C6EB + (Height of Map + 3) * (Width of Map + 6) +
# X_Movement_of_Connection Strip
#
# West: C6E8 + (Width of Map + 6) * (Y_Movement_of_"Connection Strip" + 3)
#
# East: C6E5 + (Width of Map + 6) * (Y_Movement_of_"Connection Strip" + 4)
asm = None
asm_lines = None
def load_asm():
global asm, asm_lines
asm = open("../main.asm", "r").read()
asm_lines = asm.split("\n")
def get_xy_movement_of_connection_strip(map_id, connection_id):
map1 = extract_maps.map_headers[map_id]
connections = map1["connections"]
connection = connections[connection_id]
direction = connection["direction"]
current_map_location = int(connection["current_map_tile_pointer"], 16)
map2 = extract_maps.map_headers[connection["map_id"]]
map2_height = int(map2["y"], 16)
map2_width = int(map2["x"], 16)
y_mov = None
#if direction == "WEST":
# y_mov = ((current_map_location - 0xC6E8) / (map2_width + 6)) - 3
#elif direction == "EAST":
# y_mov = ((current_map_location - 0xC6E5) / (map2_width + 6)) - 4
if direction in ["WEST", "EAST"]:
y_mov = c_int8(connection["y"]).value / -2
x_mov = None
#if direction == "NORTH":
# x_mov = current_map_location - 0xC6EB
#elif direction == "SOUTH":
# x_mov = current_map_location - 0xC6EB - ((map2_height + 3) * (map2_width + 6))
if direction in ["NORTH", "SOUTH"]:
x_mov = c_int8(connection["x"]).value / -2
return {"y_mov": y_mov, "x_mov": x_mov}
def find_line_starting_with(value):
global asm_lines
id = 0
for line in asm_lines:
if len(line) < len(value): continue
if line[:len(value)] == value:
return asm_lines.index(line)
id += 1
return False #not found
def current_map_position_formula(map_id, connection_id):
map1_id = map_id
map1 = extract_maps.map_headers[map_id]
connections = map1["connections"]
connection = connections[connection_id]
map1_height = int(map1["y"], 16)
map1_width = int(map1["x"], 16)
map1_name = map1["name"]
map1_name = map_name_cleaner(map1_name, None)[:-2]
direction = connection["direction"]
current_map_location = int(connection["current_map_tile_pointer"], 16)
map2_id = connection["map_id"]
map2 = extract_maps.map_headers[map2_id]
map2_name = map2["name"]
map2_name = map_name_cleaner(map2_name, None)[:-2]
map2_height = int(map2["y"], 16)
map2_width = int(map2["x"], 16)
y_mov = None
if direction == "WEST":
y_mov = ((current_map_location - 0xC6E8) / (map1_width + 6)) - 3
elif direction == "EAST":
y_mov = ((current_map_location - 0xC6E5) / (map1_width + 6)) - 4
x_mov = None
if direction == "NORTH":
x_mov = current_map_location - 0xC6EB
elif direction == "SOUTH":
x_mov = current_map_location - 0xC6EB - ((map1_height + 3) * (map1_width + 6))
formula = ""
if direction == "NORTH":
formula = "$C6EB + " + str(x_mov)
elif direction == "SOUTH":
formula = "$C6EB + (" + map1_name + "Height + 3) * (" + map1_name + "Width + 6) + " + str(x_mov)
elif direction == "WEST":
formula = "$C6E8 + (" + map1_name + "Width + 6) * (" + str(y_mov) + " + 3)"
elif direction == "EAST":
formula = "$C6E5 + (" + map1_name + "Width + 6) * (" + str(y_mov) + " + 4)"
return formula
def replace_values():
global asm_lines #0-15 ok
for map_id in [3]: #extract_maps.map_headers.keys():
if map_id in extract_maps.bad_maps: continue #skip
if map_id == 12: continue #skip Route 1
map1 = extract_maps.map_headers[map_id]
label_name = map_name_cleaner(map1["name"], None)
clean_name = label_name[:-2]
line_number = find_line_starting_with(label_name)
if line_number == False: continue #skip, not found
#replace dimensions if necessary
if "dimensions" in asm_lines[line_number + 2] and "$" in asm_lines[line_number + 2] and not "\t" in asm_lines[line_number+2]:
asm_lines[line_number + 2] = spacing + "db " + clean_name + "Height, " + clean_name + "Width ; dimensions (y, x)"
#skip the rest of this if there are no connections
if len(map1["connections"]) == 0: continue
if not "; connections data" in asm_lines[line_number + 6]: continue
connection_offset = line_number + 8
for connection_id in map1["connections"]:
connection = map1["connections"][connection_id]
direction = connection["direction"]
map2_id = connection["map_id"]
map2 = extract_maps.map_headers[map2_id]
map2_name = map_name_cleaner(map2["name"], None)[:-2]
map2_height = int(map2["y"], 16)
map2_width = int(map2["x"], 16)
movements = get_xy_movement_of_connection_strip(map_id, connection_id)
y_mov = movements["y_mov"]
x_mov = movements["x_mov"]
#replace the first two pointers
if " dw " in asm_lines[connection_offset + 1]:
formula = print_connections(map_id, in_connection_id=connection_id)
formula2 = current_map_position_formula(map_id, connection_id)
temp_line = asm_lines[connection_offset + 1]
temp_line = spacing + "dw " + formula + " ; connection strip location\n" #connection strip location
temp_line += spacing + "dw " + formula2 + " ; current map position" #current map position
asm_lines[connection_offset + 1] = temp_line
#bigness, width
if "bigness, width" in asm_lines[connection_offset + 2]:
temp_line = spacing + "db "
if int(connection["bigness"],16) == map2_width:
temp_line += map2_name + "Width"
elif int(connection["bigness"],16) == map2_height:
temp_line += map2_name + "Height"
else: #dunno wtf to do
temp_line += "$" + hex(int(connection["bigness"],16))[2:]
#if direction in ["NORTH", "SOUTH"]:
# temp_line += map2_name + "Width"
#elif direction in ["WEST", "EAST"]:
# temp_line += map2_name + "Height"
temp_line += ", " + map2_name + "Width"
temp_line += " ; bigness, width"
asm_lines[connection_offset + 2] = temp_line
#alignments (y, x)
if "alignments (y, x)" in asm_lines[connection_offset + 3]:
temp_line = spacing + "db "
if direction == "NORTH":
temp_line += "(" + map2_name + "Height * 2) - 1"
elif direction == "SOUTH":
temp_line += "0"
elif direction in ["WEST", "EAST"]:
#TODO: this might be y_mov/4 ??
temp_line += "(" + str(y_mov) + " * -2)"
temp_line += ", "
#Relative X-Position of player after entering connected map.
if direction in ["NORTH", "SOUTH"]:
temp_line += "(" + str(x_mov) + " * -2)"
elif direction == "WEST":
temp_line += "(" + map2_name + "Width * 2) - 1"
elif direction == "EAST":
temp_line += "0"
temp_line += " ; alignments (y, x)"
asm_lines[connection_offset + 3] = temp_line
#window
if "; window" in asm_lines[connection_offset + 4]:
temp_line = spacing + "dw "
if direction == "NORTH":
temp_line += "$C6E9 + " + map2_name + "Height * (" + map2_name + "Width + 6)"
elif direction in ["SOUTH", "EAST"]:
temp_line += "$C6EF + " + map2_name + "Width"
elif direction == "WEST":
temp_line += "$C6EE + 2 * " + map2_name + "Width"
temp_line += " ; window"
asm_lines[connection_offset + 4] = temp_line
#jump to the next connection
connection_offset += 6
if __name__ == "__main__":
import extract_maps
extract_maps.load_rom()
extract_maps.load_map_pointers()
extract_maps.read_all_map_headers()
load_asm()
replace_values()
sys.stdout.write("\n".join(asm_lines))
|
