Skip to content
Snippets Groups Projects
Commit 436fac21 authored by hagrid67's avatar hagrid67
Browse files

fixed annoying lint problems

parent dd3df88b
No related branches found
No related tags found
No related merge requests found
...@@ -30,7 +30,6 @@ class RenderTool(object): ...@@ -30,7 +30,6 @@ class RenderTool(object):
def __init__(self, env): def __init__(self, env):
self.env = env self.env = env
#def plotTreeOnRail(self, rcPos, iDir, nDepth=10, color="r"):
def plotTreeOnRail(self, lVisits, color="r"): def plotTreeOnRail(self, lVisits, color="r"):
""" """
Derives and plots a tree of transitions starting at position rcPos Derives and plots a tree of transitions starting at position rcPos
...@@ -69,10 +68,9 @@ class RenderTool(object): ...@@ -69,10 +68,9 @@ class RenderTool(object):
tbTrans = rt.RETrans.get_transitions(oTrans, visit.iDir) tbTrans = rt.RETrans.get_transitions(oTrans, visit.iDir)
giTrans = np.where(tbTrans)[0] # RC list of transitions giTrans = np.where(tbTrans)[0] # RC list of transitions
gTransRCAg = rt.gTransRC[giTrans] gTransRCAg = rt.gTransRC[giTrans]
self.plotTrans(visit.rc, gTransRCAg, depth=str(visit.iDepth), color=color) self.plotTrans(
visit.rc, gTransRCAg,
#lVisits = self.getTreeFromRail(rcPos, iDir, nDepth=nDepth) depth=str(visit.iDepth), color=color)
#return lVisits
def plotAgents(self): def plotAgents(self):
rt = self.__class__ rt = self.__class__
...@@ -152,8 +150,6 @@ class RenderTool(object): ...@@ -152,8 +150,6 @@ class RenderTool(object):
rt = self.__class__ rt = self.__class__
xyPos = np.matmul(rcPos, rt.grc2xy) + rt.xyHalf xyPos = np.matmul(rcPos, rt.grc2xy) + rt.xyHalf
gxyTrans = xyPos + np.matmul(gTransRCAg, rt.grc2xy/2.4) gxyTrans = xyPos + np.matmul(gTransRCAg, rt.grc2xy/2.4)
# print(gxyTrans)
#print(gxyTrans, color)
plt.scatter(*gxyTrans.T, color=color, marker="o", s=50, alpha=0.2) plt.scatter(*gxyTrans.T, color=color, marker="o", s=50, alpha=0.2)
if depth is not None: if depth is not None:
for x, y in gxyTrans: for x, y in gxyTrans:
...@@ -200,7 +196,9 @@ class RenderTool(object): ...@@ -200,7 +196,9 @@ class RenderTool(object):
# plot the available transitions from this node # plot the available transitions from this node
if bPlot: if bPlot:
self.plotTrans(visit.rc, gTransRCAg, depth=str(visit.iDepth)) self.plotTrans(
visit.rc, gTransRCAg,
depth=str(visit.iDepth))
return lVisits return lVisits
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment