Script execution

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
+
+