示例#1
0
文件: main.py 项目: fmgc/abss
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()