Changeset 748 in ETALON for CLIO/control/read_plot_data.py

Timestamp:

Nov 29, 2017, 8:27:37 AM (7 years ago)

Author:

delerue

Message:

CLIO graphical interface and analysis updated

File:

• CLIO/control/read_plot_data.py (modified) (6 diffs)

CLIO/control/read_plot_data.py

@@ -14 +14 @@
 from scipy.fftpack import fft, ifft
 from scipy import signal
+from clio_constants import *
 
 def read_plot_data(filename):
@@ -68 +69 @@
                 idx_angles_sorted.append(data_mapping.index(angle))
             idx_data_mapping[data_mapping.index(angle)]=int(len(angles_sorted))
+            
     #Check if the filename contains the path or not
     if (len(os.path.dirname(filename))==0):
@@ -159 +161 @@
         move(filename,target_filename)
             
-    #print Sig_mean, Sig_std
+    print Sig_mean, Sig_std
 
  
@@ -205 +207 @@
     imagename_no_path=os.path.basename(imagename)
 
-
-    fid=open(target_directory+'list.html','a');
-    fid.write("<A HREF=")
-    fid.write(imagename_no_path)
-    fid.write("><IMG SRC=")
-    fid.write(imagename_no_path)
-    fid.write(" width=600>")
-    fid.write("</A><BR/>")
-    fid.write(imagename_no_path)
-    fid.write("<BR/><BR/>\n")
-    fid.close()
-    copyfile(imagename,os.path.dirname(imagename)+'/last.png')
+## We no longer create an HTML list
+#    fid=open(target_directory+'list.html','a');
+#    fid.write("<A HREF=")
+#    fid.write(imagename_no_path)
+#    fid.write("><IMG SRC=")
+#    fid.write(imagename_no_path)
+#    fid.write(" width=600>")
+#    fid.write("</A><BR/>")
+#    fid.write(imagename_no_path)
+#    fid.write("<BR/><BR/>\n")
+#    fid.close()
+#    copyfile(imagename,os.path.dirname(imagename)+'/last.png')
 
     if (electrons_found):
@@ -227 +229 @@
         figzoom.savefig(imagenamezoom)
         plt.close(figzoom)
-        copyfile(imagenamezoom,os.path.dirname(imagename)+'/last_zoom.png')
+#        copyfile(imagenamezoom,os.path.dirname(imagename)+'/last_zoom.png')
     
         imagenamezoom_no_path=os.path.basename(imagenamezoom)
-        fid=open('zoom_list.html','a');
-        fid.write("<A HREF=")
-        fid.write(imagenamezoom_no_path)
-        fid.write("><IMG SRC=")
-        fid.write(imagenamezoom_no_path)
-        fid.write(" width=600>")
-        fid.write("</A><BR/>")
-        fid.write(imagenamezoom_no_path)
-        fid.write("<BR/><BR/>\n")
-        fid.close()
-        
+
+## We no longer create an HTML list
+#        fid=open('zoom_list.html','a');
+#        fid.write("<A HREF=")
+#        fid.write(imagenamezoom_no_path)
+#        fid.write("><IMG SRC=")
+#        fid.write(imagenamezoom_no_path)
+#        fid.write(" width=600>")
+#        fid.write("</A><BR/>")
+#        fid.write(imagenamezoom_no_path)
+#        fid.write("<BR/><BR/>\n")
+#        fid.close()
+
+
         plt.figure(3)
-        plt.plot(range(48,132,7), Sig_mean,'r^-')
+        plt.plot(angles_sorted, Sig_mean,'r^-',label='Data')
+        refdata=load_reference_signal()
+        if (len(refdata)>0):
+#            print refdata[:,0]
+#            print refdata[:,1]
+            plt.plot(refdata[:,0],refdata[:,1],'go:',label='Reference')            
         plt.title(key_values['date']+ ' ---  Motor value:  '+key_values['motor_value']+ ' ---  Current:  '+key_values['current_value'])
         plt.ylabel('Signal amplitude [a.u.]')
         plt.xlabel('Detector angle [deg.]')
-        plt.grid(True)        
+        plt.grid(True)
+        plt.legend()
         imagenamesignal=target_filename.replace('.txt','_signal.png')
         figsignal.savefig(imagenamesignal)
         plt.close(figsignal)
-        copyfile(imagenamesignal,os.path.dirname(imagename)+'/last_signal.png')
+
+        if (len(refdata)>0):
+            figref=plt.figure(4,figsize=figure_size,dpi=figure_dpi)
+            plt.figure(4)
+            diffdata=np.zeros((len(angles_sorted),2))
+            for angle in angles_sorted:
+                idx=angles_sorted.index(angle)
+                idxref=np.where(refdata == angle)
+                diffdata[idx,0]=angle
+                diffdata[idx,1]=Sig_mean[idx]-refdata[idxref[0],1]
+            #print diffdata
+            plt.plot(diffdata[:,0], diffdata[:,1],'r^-')
+            plt.title('Diff --- ' +key_values['date']+ ' ---  Motor value:  '+key_values['motor_value']+ ' ---  Current:  '+key_values['current_value'])
+            plt.ylabel('Signal amplitude above reference [a.u.]')
+            plt.xlabel('Detector angle [deg.]')
+            plt.grid(True)        
+            imagenamesignal=target_filename.replace('.txt','_amplitude.png')
+            figref.savefig(imagenamesignal)
+            plt.close(figref)
+
+
+        
+#        copyfile(imagenamesignal,os.path.dirname(imagename)+'/last_signal.png')
 
         imagenamesignal_no_path=os.path.basename(imagenamesignal)
-        fid=open('signal_list.html','a');
-        fid.write("<A HREF=")
-        fid.write(imagenamesignal_no_path)
-        fid.write("><IMG SRC=")
-        fid.write(imagenamesignal_no_path)
-        fid.write(" width=600>")
-        fid.write("</A><BR/>")
-        fid.write(imagenamesignal_no_path)
-        fid.write("<BR/><BR/>\n")
-        fid.close()
+        
+## We no longer create an HTML list
+#        fid=open('signal_list.html','a');
+#        fid.write("<A HREF=")
+#        fid.write(imagenamesignal_no_path)
+#        fid.write("><IMG SRC=")
+#        fid.write(imagenamesignal_no_path)
+#        fid.write(" width=600>")
+#        fid.write("</A><BR/>")
+#        fid.write(imagenamesignal_no_path)
+#        fid.write("<BR/><BR/>\n")
+#        fid.close()
         
     filenamesig=target_filename.replace('.txt','.sig')
@@ -282 +317 @@
     return;
 
+
+def load_reference_signal():
+    global FILE_REF
+
+    try:
+        fid=open(FILE_REF)        
+        refname=fid.readline().rstrip()
+        fid.close()
+    except:
+        print "Unable to read reference file"
+        refname=""
+
+    if (len(refname)>0):
+        #read refrence parameters    
+        ref_key_values={}
+        with open(refname,'r') as fid:
+            fileline=fid.readline()
+            while (len(fileline)>0)and(fileline[0]=='#'):                
+                entries=fileline.replace('#','').replace(' ','').replace('\n','').split('=')
+                ref_key_values[entries[0]]=entries[1]
+                fileline=fid.readline()
+            fid.close()
+        
+            #for p in ref_key_values: print p,'=', ref_key_values[p]
+            
+        allrefdata=np.loadtxt(refname, comments='#', delimiter='\t', usecols=range(0,3))
+#        print allrefdata
+        return allrefdata