Closedown

script/local.py

@@ -272,8 +272,8 @@ def get_diag_channels():
     return diag_channels
                     
 def get_diag_name(diag):
-    return ch.psi.utils.Str.toTitleCase(diag.getName()).replace(" ", "")
-    
+    return ch.psi.utils.Str.toTitleCase(diag if is_string(diag) else diag.getName()).replace(" ", "")
+        
 def print_diag():
     for f in get_diag_channels():
         print "%-25s  %s" % (get_diag_name(f) ,  str(f.read()))
@@ -378,6 +378,100 @@ def wait_for_file_size(filepath, size, timeout = None):
         time.sleep(wait_time)
 
 
+###################################################################################################
+#  Scan commands
+###################################################################################################
+
+def relscan(motor, start, end, number_of_steps, count_time):
+    """
+    Relative scan
+    """
+    set_count_time(count_time)    
+    detectors = [mythen, mythen.acquire_time, pixel, pixel.image_filename, image.intensity,image.corrected_intensity, image.matrix]    
+    #TODO: Set based on experiment context
+    pixel.set_path("../../expdata/bml_20190703/pixel/", "images")       
+    #TODO: should be set automatically
+    set_roi(173,88,285,136)    
+    show_panel(detector_image)
+    pixel.show()
+    pixel.assert_ready()        
+    return lscan (motor, detectors, start, end, int(number_of_steps), relative=True, before_read=before_sample, after_read=after_sample)    
+    
+def absscan(motor, start, end, number_of_steps, count_time):
+    """
+    Absolute scan
+    """
+    set_count_time(count_time)    
+    detectors = [mythen, mythen.acquire_time, pixel, pixel.image_filename, image.intensity,image.corrected_intensity, image.matrix]    
+    #TODO: Set based on experiment context
+    pixel.set_path("../../expdata/bml_20190703/pixel/", "images")       
+    #TODO: should be set automatically
+    set_roi(173,88,285,136)    
+    show_panel(detector_image)
+    pixel.show()
+    pixel.assert_ready()        
+    return lscan (motor, detectors, start, end, int(number_of_steps), relative=False, before_read=before_sample, after_read=after_sample)    
+
+def abs2scan(motor1, start1, end1, motor2, start2, end2, number_of_steps, count_time):
+    """
+    Absolute scan of 2 motors
+    """
+    set_count_time(count_time)    
+    detectors = [mythen, mythen.acquire_time, pixel, pixel.image_filename, image.intensity,image.corrected_intensity, image.matrix]    
+    #TODO: Set based on experiment context
+    pixel.set_path("../../expdata/bml_20190703/pixel/", "images")       
+    #TODO: should be set automatically
+    set_roi(173,88,285,136)    
+    show_panel(detector_image)
+    pixel.show()
+    pixel.assert_ready()        
+    return lscan ([motor1, motor2], detectors, [start1, start2], [end1, end2], int(number_of_steps), relative=False, before_read=before_sample, after_read=after_sample)    
+
+def hrodscan(start, end, number_of_steps, count_time):
+    """
+    Scan on l
+    """
+    return absscan (h, start, end, number_of_steps, count_time)    
+    
+def krodscan(start, end, number_of_steps, count_time):
+    """
+    Scan on k
+    """
+    return absscan (k, start, end, number_of_steps, count_time)    
+
+def lrodscan(start, end, number_of_steps, count_time):
+    """
+    Scan on l
+    """
+    return absscan (l, start, end, number_of_steps, count_time)    
+
+
+def hkllinscan(hstart, hfinish, kstart, kfinish, lstart, lfinish, number_of_steps, count_time):    
+    """
+    Linear scan on hkl
+    """
+    vector = []   
+    hs = float(hfinish - hstart)/number_of_steps
+    ks = float(kfinish - kstart)/number_of_steps
+    ls = float(lfinish - lstart)/number_of_steps
+    for i in range(number_of_steps+1):
+        hp = hstart + i * hs
+        kp = kstart + i * ks
+        lp = lstart + i * ls
+        vector.append([hp, kp, lp] )
+
+    set_count_time(count_time)    
+    detectors = [mythen, mythen.acquire_time, pixel, pixel.image_filename, image.intensity,image.corrected_intensity, image.matrix]    
+    #TODO: Set based on experiment context
+    pixel.set_path("../../expdata/bml_20190703/pixel/", "images")       
+    #TODO: should be set automatically
+    set_roi(173,88,285,136)    
+    show_panel(detector_image)
+    pixel.show()
+    pixel.assert_ready()   
+    return hklscan(vector, detectors,latency = 0.0, before_read=before_sample, after_read=after_sample)
+        
+        
 
 ###################################################################################################
 #  Configuration