From 4c64681238e08c3e91bc8e20d77097e8550f926c Mon Sep 17 00:00:00 2001
From: sfop <sfop@sf-cons-01.psi.ch>
Date: Tue, 5 Jul 2016 16:20:27 +0200
Subject: [PATCH] Script execution

---
 script/camtool.py | 45 +++++++++++++++++++++++++++++++++++----------
 1 file changed, 35 insertions(+), 10 deletions(-)

diff --git a/script/camtool.py b/script/camtool.py
index 3898ac5..55dc516 100644
--- a/script/camtool.py
+++ b/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)
\ No newline at end of file