luke
/
OpenRAM
mirror of https://github.com/VLSIDA/OpenRAM.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

PEP8 cleanup

This commit is contained in:
mrg 2021-02-23 13:32:13 -08:00
parent 1c6de4591d
commit 549112fcf8
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
compiler/router/direction.py
View File

@ -7,6 +7,8 @@
# #
from enum import Enum from enum import Enum
from vector3d import vector3d from vector3d import vector3d
import debug
class direction(Enum): class direction(Enum):
NORTH = 1 NORTH = 1
@ -20,31 +22,30 @@ class direction(Enum):
SOUTHEAST = 9 SOUTHEAST = 9
SOUTHWEST = 10 SOUTHWEST = 10
def get_offset(direct): def get_offset(direct):
""" """
Returns the vector offset for a given direction. Returns the vector offset for a given direction.
""" """
if direct==direction.NORTH: if direct==direction.NORTH:
offset = vector3d(0,1,0) offset = vector3d(0, 1, 0)
elif direct==direction.SOUTH: elif direct==direction.SOUTH:
offset = vector3d(0,-1,0) offset = vector3d(0, -1 ,0)
elif direct==direction.EAST: elif direct==direction.EAST:
offset = vector3d(1,0,0) offset = vector3d(1, 0, 0)
elif direct==direction.WEST: elif direct==direction.WEST:
offset = vector3d(-1,0,0) offset = vector3d(-1, 0, 0)
elif direct==direction.UP: elif direct==direction.UP:
offset = vector3d(0,0,1) offset = vector3d(0, 0, 1)
elif direct==direction.DOWN: elif direct==direction.DOWN:
offset = vector3d(0,0,-1) offset = vector3d(0, 0, -1)
elif direct==direction.NORTHEAST: elif direct==direction.NORTHEAST:
offset = vector3d(1,1,0) offset = vector3d(1, 1, 0)
elif direct==direction.NORTHWEST: elif direct==direction.NORTHWEST:
offset = vector3d(-1,1,0) offset = vector3d(-1, 1, 0)
elif direct==direction.SOUTHEAST: elif direct==direction.SOUTHEAST:
offset = vector3d(1,-1,0) offset = vector3d(1, -1, 0)
elif direct==direction.SOUTHWEST: elif direct==direction.SOUTHWEST:
offset = vector3d(-1,-1,0) offset = vector3d(-1, -1, 0)
else: else:
debug.error("Invalid direction {}".format(direct)) debug.error("Invalid direction {}".format(direct))
@ -67,8 +68,8 @@ class direction(Enum):
return [direction.get_offset(d) for d in direction.all_directions()] return [direction.get_offset(d) for d in direction.all_directions()]
def all_neighbors(cell): def all_neighbors(cell):
return [cell+x for x in direction.all_offsets()] return [cell + x for x in direction.all_offsets()]
def cardinal_neighbors(cell): def cardinal_neighbors(cell):
return [cell+x for x in direction.cardinal_offsets()] return [cell + x for x in direction.cardinal_offsets()]