ref: 85d26e31a5f209aa059250778ec79f762212825f
parent: fcf43b13dabed748eba1737770eeb9f5d55820c5
author: Bryan Bishop <kanzure@gmail.com>
date: Sun Jul 1 04:41:38 EDT 2012
better graph-generating routines
--- a/extras/graph.py
+++ b/extras/graph.py
@@ -28,7 +28,7 @@
start_address = 0x150
# and where is a good place to stop?
- end_address = 0x4000 * 0x01 # only do the first bank? sure..
+ end_address = 0x4000 * 0x03 # only do the first bank? sure..
# where is the rom stored?
rompath = "../baserom.gbc"
@@ -68,10 +68,10 @@
count = 0
while True:
- if count > 100:
+ if count > 3000:
break
- if address < self.end_address and address not in functions.keys():
+ if address < self.end_address and (address not in functions.keys()) and address >= 0x150:
# address is okay to parse at, keep going
pass
elif len(other_addresses) > 0:
@@ -84,6 +84,17 @@
# parse the asm
func = self.rom.to_asm(address)
+ # check if there are any nops (probably not a function)
+ nops = 0
+ for (id, command) in func.asm_commands.items():
+ if command.has_key("id") and command["id"] == 0x0:+ nops += 1
+
+ # skip this function
+ if nops > 1:
+ address = 0
+ continue
+
# store this parsed function
functions[address] = func
@@ -97,7 +108,8 @@
other_addresses.update([used_address])
# add this other address to the graph
- self.add_node(used_address)
+ if used_address > 100:
+ self.add_node(used_address)
# add this as an edge between the two nodes
self.add_edge(address, used_address)
@@ -120,10 +132,15 @@
"""
import networkx.readwrite.json_graph as json_graph
content = json_graph.dumps(self)
- fh = open("graphs.json", "w")+ fh = open("crystal/crystal.json", "w")fh.write(content)
fh.close()
+ def to_gephi(self):
+ """ Generates a gexf file.
+ """
+ nx.write_gexf(self, "graph.gexf")
+
class RedGraph(RomGraph):
""" Not implemented. Go away.
"""
@@ -140,4 +157,4 @@
if __name__ == "__main__":
crygraph = CryGraph()
crygraph.pretty_printer()
- crygraph.to_d3()
+ crygraph.to_gephi()
--
⑨