diff --git a/script/setup/Layout.py b/script/setup/Layout.py index cf6e9c9..c8a041b 100644 --- a/script/setup/Layout.py +++ b/script/setup/Layout.py @@ -47,8 +47,18 @@ class Puck: def __str__(self): return "Number: " + str(self.number) + "\nElement: " + str(self.element) + "\nIndex: " + str(self.index) + "\nCenter: " + str(self.center) + "\nLed Unipuck: " + str(self.led_uni) + "\nLed Minispine: " + str(self.led_mini) -pucks = [] +_puck_list = [] for p in(puck_layout): puck = Puck(p[0], p[1], p[3], (p[11],p[12]), (p[9],p[10]), (p[11],p[12])) - pucks.append(puck) + _puck_list.append(puck) + + +def get_pucks(element = None): + ret = [] + for p in _puck_list: + if (element is None) or (element==p.element): + ret.append(p) + return ret + +