From a351ff57e2214c0e134083ecc4a6c8e7407ba600 Mon Sep 17 00:00:00 2001 From: gac-S_Changer Date: Fri, 7 Jul 2017 12:00:50 +0200 Subject: [PATCH] Script execution --- script/setup/Layout.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) 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 + +