regriptppass.py 文件源码

def plotGraph(self, pltax, offset = [0,0]):
        """

        :param pltax:
        :param offset: where to plot the graph
        :return:
        """

        # add offset
        for i in self.gnodesplotpos.keys():
            self.gnodesplotpos[i] = map(add, self.gnodesplotpos[i], offset)

        transitedges = []
        transferedges = []
        for nid0, nid1, reggedgedata in self.regg.edges(data=True):
            if reggedgedata['edgetype'] is 'transit':
                transitedges.append([self.gnodesplotpos[nid0][:2], self.gnodesplotpos[nid1][:2]])
            if reggedgedata['edgetype'] is 'transfer':
                transferedges.append([self.gnodesplotpos[nid0][:2], self.gnodesplotpos[nid1][:2]])
        transitec = mc.LineCollection(transitedges, colors=[0,1,1,1], linewidths=1)
        transferec = mc.LineCollection(transferedges, colors=[0,0,0,.1], linewidths=1)
        pltax.add_collection(transferec)
        pltax.add_collection(transitec)