Changeset 785 in ETALON

Timestamp:

Jul 19, 2018, 4:23:43 PM (6 years ago)

Author:

moutardier

Message:

change print_data to read data from libera

File:

• BPM/print_datas.py (modified) (5 diffs)

BPM/print_datas.py

@@ -23 +23 @@
     Data_list = [[[] for i in range(9)] for j in range(3)]
     value = ""
-    index = 0
+    index = -1
     for i in Data:
         if (i == " ") :
             if index == -1: 
                 bpm_index = int(value)
-                Data_list[0].append(value)
                 index +=1
-            elif index == 1:
-                Data_list[1].append(value)
-                #print(value)
-                index += 1
-            elif index == 2:
-                Data_list[2].append(value)
-                index += 1
-            elif index == 3:
-                Data_list[3].append(value)
-                index += 1
             else:
-                Data_list[4].append(value)
-                index = 0
+                Data_list[bpm_index][index].append(value)
+                index += 1
             value = ""
         elif (i == "\n") :
-            value = ""
+            if value != "":
+                Data_list[bpm_index][index].append(value)
+                value = ""
             index = -1
         else:
             value += i
     fichier.close()
-    for i in range(5):
-        Data_list[i] = float_list(Data_list[i])
-    Data_list[0] = product_list(Data_list[0],10**-6)
+    for bpm in Data_list:
+        for i in range(len(bpm)):
+            bpm[i] = float_list(bpm[i])
+    return(Data_list)
+"""
+
+        bpm[0] = product_list(Data_list[0],10**-6)
+
     for i in range(1,5):
         Data_list[i] = product_list(Data_list[i],1000)
-    return(Data_list)
-
+"""
 
 
@@ -100 +95 @@
 
 
-def print_graph(L): # program to print graph of X_libera vs X_motor with a list L = [[[x_motor_position][y_motor_position][Va][Vb][Vc][Vd][sum][x_libera_position][y_libera_position]][same for BPM1][same for BPM2]], also do a linear regretion and print the equation
+def print_graph_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(data,lin_reg = None): # program to print graph of X_libera vs X_motor with a list data = [[[x_motor_position][y_motor_position][Va][Vb][Vc][Vd][sum][x_libera_position][y_libera_position]][same for BPM1][same for BPM2]],also do a linear regretion and print the equation
     global global_index_figure # to creat different figure
     fig1 = plt.figure(global_index_figure)
     global_index_figure += 1
     plt.clf()
-    for bpm_data in range(L):
-        lr1 = stats.linregress(bpm_data[0],bpm_data[7]) # return tuple (pente,ordonnee a l'origine, coef de correlation, p-value, erreur standard de l'estimation)
-        plt.plot(bpm_data[0],bpm_data[7], "r", label = "BMP_ref")
-        plt.plot(bpm_data[0],[lr1.slope*i + lr1.intercept for i in bpm_data[0]], "r--", label = "BMP_ref regression line, \nerror = "+str(round(lr1.stderr,3))+",\ncorrelation coefficient = "+str(round(lr1.rvalue,3)))
-
-
-
-
-
-
-
-    """
-    lr1 = stats.linregress(L[1],L[1]) # return tuple (pente,ordonnee a l'origine, coef de correlation, p-value, erreur standard de l'estimation)
+    color = ["r","g","b"]
+    legende = ["BPM soleil","BPM impr","BPM ref"]
+    for bpm_number in range(len(data)):
+        plt.plot(data[bpm_number][0],data[bpm_number][7], color[bpm_number]+".", label = legende[bpm_number])
+        if lin_reg == "yes":
+            lr1 = stats.linregress(data[bpm_number][0],data[bpm_number][7]) # return tuple (pente,ordonnee a l'origine, coef de correlation, p-value, erreur standard de l'estimation)
+            plt.plot(data[bpm_number][0],[lr1.slope*i + lr1.intercept for i in data[bpm_number][0]], color[bpm_number]+"--", label = legende[bpm_number] + " regression line, \nerror = "+str(round(lr1.stderr,3))+",\ncorrelation coefficient = "+str(round(lr1.rvalue,3)))
+    plt.legend()
+    plt.xlabel("length in million motor step") #, fontsize=20)
+    plt.ylabel("tension in mV")
+    plt.gca().yaxis.set_tick_params(labelsize = 8)
+        
+
+
+
+def print_graph_position_Va_Vb_Vc_Vd(L): # program to print graph of X_libera vs X_motor with a list L = [[position][Va][Vb][Vc][Vd]], also do a linear regretion and print the equation
+    global global_index_figure # to creat different figure
+    fig1 = plt.figure(global_index_figure)
+    global_index_figure += 1
+    plt.clf()
+    
+    lr1 = stats.linregress(L[0],L[1]) # return tuple (pente,ordonnee a l'origine, coef de correlation, p-value, erreur standard de l'estimation)
     plt.plot(L[0],L[1], "r", label = "BMP_ref")
     plt.plot(L[0],[lr1.slope*i + lr1.intercept for i in L[0]], "r--", label = "BMP_ref regression line, \nerror = "+str(round(lr1.stderr,3))+",\ncorrelation coefficient = "+str(round(lr1.rvalue,3)))
@@ -138 +142 @@
     plt.gca().yaxis.set_tick_params(labelsize = 8)
 #    plt.show()
-"""
+
 def residu_position_Va_Vb_Vc_Vd(L): # program to do graph of residu of data give in a list with [[position][Va][Vb][Vc][Vd]]
     global global_index_figure
@@ -177 +181 @@
 """
 
-def min_max(L,mini = None ,maxi = None): #program to cut a list L = [[position][Va][Vb][Vc][Vd]] with minimum value of position = mini and maximum value of position = maxi 
-    if mini is None :
-        mini = L[0][0]
-    if maxi is None :
-        maxi = L[0][len(L[0])-1]
-    if mini > maxi:
-        mini,maxi = maxi,mini
+
+
+
+
+
+
+def residu_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(data): # program to print graph of X_libera vs X_motor with a list data = [[[x_motor_position][y_motor_position][Va][Vb][Vc][Vd][sum][x_libera_position][y_libera_position]][same for BPM1][same for BPM2]],also do a linear regretion and print the equation
+    global global_index_figure # to creat different figure
+    fig1 = plt.figure(global_index_figure)
+    global_index_figure += 1
+    plt.clf()
+    color = ["r","g","b"]
+    legende = ["BPM soleil","BPM impr","BPM ref"]
+    for bpm_number in range(len(data)):
+        lr1 = stats.linregress(data[bpm_number][0],data[bpm_number][7]) # return tuple (pente,ordonnee a l'origine, coef de correlation, p-value, erreur standard de l'estimation)
+    #    residu = product_list(data[bpm_number][7], lr1.slope, b =  lr1.intercept)
+        residu = []
+        for i in range(len(data[bpm_number][0])):
+            residu.append(abs(data[bpm_number][7][i] - data[bpm_number][0][i]*lr1.slope - lr1.intercept))
+        plt.plot(data[bpm_number][0], residu, color[bpm_number]+".", label = legende[bpm_number])
+    plt.legend()
+    plt.title("Residu en fonction de la position")
+    plt.xlabel("length in million motor step")
+    plt.ylabel("tension in mV")
+    plt.gca().yaxis.set_tick_params(labelsize = 8)
+#    plt.s
+
+
+
+
+
+def min_max(data, cut_index, mini = None ,maxi = None): #program to cut a list data = [[position][Va][Vb][Vc][Vd]] with minimum value of position = mini and maximum value of position = maxi
+    if mini is None and maxi is None :
+        return(data)
+    elif mini is None :
+        for i in data[cut_index]:
+            if i < mini :
+                mini = i
+    elif maxi is None:
+        for i in data[cut_index]:
+            if maxi < i :
+                maxi = i
     index_list = []
-    for i in range(len(L[0])):
-        if  L[0][i] >= mini and L[0][i] <= maxi :
+    for i in range(len(data[cut_index])):
+        if  data[cut_index][i] >= mini and data[cut_index][i] <= maxi :
             index_list.append(i)
-    return([[L[0][i] for i in index_list],[L[1][i] for i in index_list],[L[2][i] for i in index_list],[L[3][i] for i in index_list],[L[4][i] for i in index_list]])
-
-
-def convertion_step_mm(L,begin_mm,end_mm): # length convertion 
-    begin_step = L[0]
-    end_step = L[len(L)-1]
+    return([[data[j][i] for i in index_list] for j in range(len(data))])
+
+            #return([[data[0][i] for i in index_list],[data[1][i] for i in index_list],[data[2][i] for i in index_list],[data[3][i] for i in index_list],[data[4][i] for i in index_list]])
+
+
+def convertion_step_mm(data,begin_mm,end_mm): # length convertion 
+#    print(data)
+    if begin_mm is None:
+        begin_mm = data[0]
+    if end_mm is None:
+        end_mm = data[len(data)-1]
+    begin_step = data[0]
+    end_step = data[len(data)-1]
     a = float((end_mm - begin_mm))/(end_step - begin_step)
     b = begin_mm - a*begin_step
     print(a,b)
-    print(L)
-    L = product_list(L, a, b)
-    return(L)
+    data = product_list(data, a, b)
+    return(data)
     
 
@@ -207 +252 @@
     
 def print_data_position_Va_Vb_Vc_Vd(fichier, min = None, max = None):
-    L = min_max(read_data_position_Va_Vb_Vc_Vd(fichier), mini = min, maxi = max)
-    print_graph(L)
+    L = min_max(read_data_position_Va_Vb_Vc_Vd(fichier), 0, mini = min, maxi = max)
+    print_graph_position_Va_Vb_Vc_Vd(L)
     #print(L)
 
-def print_residu_position_Va_Vb_Vc_Vd(fichier, min = None, max = None):
-    L = min_max(read_data_position_Va_Vb_Vc_Vd(fichier), mini = min, maxi = max)
+def print_residu_position_Va_Vb_Vc_Vd(fichier, min_position = None, max_position = None):
+    L = min_max(read_data_position_Va_Vb_Vc_Vd(fichier), 0, mini = min_position, maxi = max_position)
     residu_position_Va_Vb_Vc_Vd(L)
-
-print_data_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_20180611_160153_0.txt",0.1,0.3)
-print_residu_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_20180611_160153_0.txt",0.1,0.3)
-
+    
+def print_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(fichier, lin_reg = "yes", minx = None, maxx = None, miny = None, maxy = None, begin_mm = None, end_mm = None): 
+    data = read_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(fichier)
+    if (not minx is None) or (not maxx is None):
+        data =  [min_max(bpm, 0, mini = minx, maxi = maxx) for bpm in data]
+    if (not miny is None) or (not maxy is None):
+        data =  [min_max(bpm, 1, mini = miny, maxi = maxy) for bpm in data]
+    if (not begin_mm is None) or (not end_mm is None):
+        for bpm_index in range(len(data)):
+            data[bpm_index][0] = convertion_step_mm(data[bpm_index][0], (begin_mm - end_mm)/2, (-begin_mm +  end_mm)/2)
+    print_graph_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(data,lin_reg)
+
+def print_residu_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(fichier, minx = None, maxx = None, miny = None, maxy = None): 
+    data = read_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(fichier)
+    if (not minx is None) or (not maxx is None):
+        data =  [min_max(bpm, 0, mini = minx, maxi = maxx) for bpm in data]
+    if (not miny is None) or (not maxy is None):
+        data =  [min_max(bpm, 1, mini = miny, maxi = maxy) for bpm in data]
+    residu_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y(data)
+
+#print_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/BPM-number_x-motor_y-motor_Va_Vb_Vc_Vd_Sum_X_Y_BPM0-soleil_BPM1-impr_BPM2-ref_20180718_1.txt",  miny = 0, maxy = 57)
+#print_residu_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/BPM-number_x-motor_y-motor_Va_Vb_Vc_Vd_Sum_X_Y_BPM0-soleil_BPM1-impr_BPM2-ref_20180718_1.txt")
+
+print_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/BPM-number_x-motor_y-motor_Va_Vb_Vc_Vd_Sum_X_Y_BPM0-soleil_BPM1-impr_BPM2-ref_20180718_2_convert-100000_94,36_100000_78,72.txt", begin_mm = 78.72, end_mm = 94.36)
+#print_residu_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/BPM-number_x-motor_y-motor_Va_Vb_Vc_Vd_Sum_X_Y_BPM0-soleil_BPM1-impr_BPM2-ref_20180718_2_convert-100000_94,36_100000_78,72.txt")
+
+#print_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/test.txt", begin_mm = 100)
+#print_data_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/test.txt, begin_mm = 0, end_mm = 1")
+#print_residu_xmotor_ymotor_Va_Vb_Vc_Vd_sum_x_y("data/test.txt")
+
+
+#print_data_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_20180611_160153_0.txt",0.1,0.3)
+#print_residu_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_20180611_160153_0.txt",0.1,0.3)
 #print_data_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_20180611_163036_0.txt")
-
 #print_data_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_vertical_acquisition_20180612_0.txt")
 #print_data_position_Va_Vb_Vc_Vd("data/position_vs_tension_ch1-3_bpm_ref_ch2-4_bpm_impr_vertical_acquisition_20180612_0.txt")