mirror of https://github.com/VLSIDA/OpenRAM.git
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
from enum import Enum
|
|
from vector3d import vector3d
|
|
|
|
class direction(Enum):
|
|
NORTH = 1
|
|
SOUTH = 2
|
|
EAST = 3
|
|
WEST = 4
|
|
UP = 5
|
|
DOWN = 6
|
|
|
|
|
|
def get_offset(direct):
|
|
"""
|
|
Returns the vector offset for a given direction.
|
|
"""
|
|
if direct==direction.NORTH:
|
|
offset = vector3d(0,1,0)
|
|
elif direct==direction.SOUTH:
|
|
offset = vector3d(0,-1,0)
|
|
elif direct==direction.EAST:
|
|
offset = vector3d(1,0,0)
|
|
elif direct==direction.WEST:
|
|
offset = vector3d(-1,0,0)
|
|
elif direct==direction.UP:
|
|
offset = vector3d(0,0,1)
|
|
elif direct==direction.DOWN:
|
|
offset = vector3d(0,0,-1)
|
|
else:
|
|
debug.error("Invalid direction {}".format(dirct))
|
|
|
|
return offset
|
|
|
|
def cardinal_directions():
|
|
return [direction.NORTH, direction.EAST, direction.SOUTH, direction.WEST]
|
|
|
|
def cardinal_offsets():
|
|
return [direction.get_offset(d) for d in direction.cardinal_directions()]
|
|
|