Script execution

script/camtool.py

@@ -16,20 +16,25 @@ class CamTool:
         self.cam_image_shape = Channel(self.data_prefix + "image.shape", alias = "cam_image_shape") #[D1, D2, ...]
         self.cam_bg_image = Channel(self.data_prefix + "bg_image", alias = "cam_bg_image") #[D1, D2, ...]                
         
-        self.cam_bg_en = Channel(self.prefix + "bg.enabled", alias = "cam_bg_en")
-        self.cam_bg_capture = Channel(self.prefix + "bg.capture", alias = "cam_bg_capture")
-        self.cam_bg_capture_remain = Channel(self.prefix + "bg.capture", alias = "cam_bg_capture_remain")        
-        if latch:
-            self.start_cam()
+        self.cam_bg_en = Channel(self.prefix + "bg.enabled", type = 'l', alias = "cam_bg_en")
+        self.cam_bg_capture = Channel(self.prefix + "bg.capture", type = 'l', alias = "cam_bg_capture")
+        self.cam_bg_capture_remain = Channel(self.prefix + "bg.capture_remain", alias = "cam_bg_capture_remain")        
+
+        self.init()
+
+    def init(self):
+        if self.latch:
+            self.start()
         else:
-            self.stop_cam()
-        
-    def start_cam(self):
+            self.stop()
+    
+    def start(self):
         self.cam_run.write(-1)
 
-    def stop_cam(self):
+    def stop(self):
         self.cam_run.write(0)    
     
+
     def capture(self):
         timestamp = self.timestamp.read()
         if self.latch:
@@ -38,8 +43,28 @@ class CamTool:
             self.cam_run.write(1)
         while(timestamp == self.timestamp.read()):
             time.sleep(0.001)
-            
 
+    def enable_bg(self, value):
+        self.cam_bg_en.write(1 if value else 0)
+        
+    def capture_bg(self, images):        
+        self.start()
+        self.cam_bg_capture.write(images)
+        sleep(0.1)
+        while( self.cam_bg_capture_remain.read() > 0):
+            time.sleep(0.001)                
+        #self.stop()        
+        #self.cam_bg_capture.write(images)
+        #while( self.cam_bg_capture_remain.read() > 0):
+        #    self.capture()
+        #self.init()
+
+#if __name__ == "__builtin__":
 camera_tool = CamTool()
 camera_tool.capture()
 print camera_tool.cam_x.read(), camera_tool.cam_y.read() 
+
+camera_tool.enable_bg(True)
+camera_tool.enable_bg(False)
+
+camera_tool.capture_bg(5)