Changeset 754 in ETALON for CLIO/control/plot_compare.py

Timestamp:

Jan 19, 2018, 6:51:35 PM (6 years ago)

Author:

delerue

Message:

Graphical interface updated

File:

• CLIO/control/plot_compare.py (modified) (4 diffs)

CLIO/control/plot_compare.py

@@ -19 +19 @@
 global FILE_COMPARE
 global data_directory_base
+
+
+print "Use plot several files instead"
+exit -1
 
 
@@ -61 +65 @@
 plt.figure(1)
 
+filenames=[]
+
+with open(FILE_COMPARE) as cfid:
+    fname=cfid.readline().rstrip()
+    if (len(fname)==0):
+        fname=cfid.readline().rstrip()
+    fnameprev=''
+    while(len(fname)>0):
+        if (fname!=fnameprev):
+            print fname
+            filenames.append(fname)
+            fnameprev=fname
+        fname=cfid.readline().rstrip()
+cfid.close()
+
+print "Plotting"
+
 dataref=load_reference_signal()
 alldata=dataref
 
-with open(FILE_COMPARE) as cfid:
-    fname=cfid.readline().rstrip()
-    while(len(fname)>0):
-        print fname
-        filedata=np.loadtxt(fname, comments='#', delimiter='\t', usecols=range(0,3))
-        alldata=np.hstack((alldata,np.zeros((np.shape(alldata)[0],1),dtype=alldata.dtype)))
-        colref=np.shape(alldata)[1]-1
-        for angle in alldata[:,0]:
-            idx=list(alldata[:,0]).index(angle)
-            idxref=np.where(filedata == angle)
+for fname in filenames:
+    print fname
+    filedata=np.loadtxt(fname, comments='#', delimiter='\t', usecols=range(0,3))
+    alldata=np.hstack((alldata,np.zeros((np.shape(alldata)[0],1),dtype=alldata.dtype)))
+    colref=np.shape(alldata)[1]-1
+    for angle in alldata[:,0]:
+        idx=list(alldata[:,0]).index(angle)
+        idxref=np.where(filedata == angle)
+        if (len(idxref[0])==1):
             alldata[idx,colref]=filedata[idxref[0],1]
             
-        plt.figure(1)
-        plt.plot(alldata[:,0], alldata[:,colref],label=os.path.basename(fname).replace("data_CLIO_","").replace(".sig",""))
-        plt.figure(2)
-        plt.plot(alldata[:,0], alldata[:,colref]-alldata[:,1],label=os.path.basename(fname).replace("data_CLIO_","").replace(".sig",""))
-        fname=cfid.readline().rstrip()
+    plt.figure(1)
+    plt.plot(alldata[:,0], alldata[:,colref],label=(os.path.basename(fname).replace("data_CLIO_","").replace(".sig",""))[-8:])
+#        plt.plot(alldata[:,0], alldata[:,colref],label=(os.path.basename(fname).replace("data_CLIO_","").replace(".sig","")))
+    plt.figure(2)
+    plt.plot(alldata[:,0], alldata[:,colref]-alldata[:,1],label=os.path.basename(fname).replace("data_CLIO_","").replace(".sig","")[-8:])
 
 #    ckey_values={}
@@ -91 +111 @@
 #        fid.close()                        
 
-cfid.close()
 
 
 plt.figure(1)
-plt.title('Comparison')
+plt.title('Comparison of the selected data')
 plt.ylabel('Signal amplitude [a.u.]')
 plt.xlabel('Detector angle [deg.]')
@@ -105 +124 @@
 
 plt.figure(2)
-plt.title('Comparison (ref substracted)')
+plt.title('Comparison (reference substracted)')
 plt.ylabel('Signal amplitude above reference [a.u.]')
 plt.xlabel('Detector angle [deg.]')