def generateBasemap(case): s = SPSim() m = __import__("cases.%s" % case) caseEdges = eval("m.%s.EDGES" % case) caseNames = eval("m.%s.NAMES" % case) max_node = 0 for e in caseEdges: max_node = max(max_node, e[0], e[1]) for i in range(max_node + 1): ag = TestAgent(profileRandom(), profileRandom()) ag.setName(caseNames[i]) s.addAgent(ag) for e in caseEdges: s.addNeighborEdge(e[0], e[1]) f = file("cases/%s.gv" % case, "w") f.write(s.getNeighborsGraphDOT()) f.close()