source: ETALON/CLIO/control/plot_compare.py @ 748

# -*- coding: utf-8 -*-
"""
Created on 28 november 2017

@author: delerue
"""


import numpy as np
import matplotlib.pyplot as plt

#from shutil import copyfile,move
import os.path
#from scipy.fftpack import fft, ifft
#from scipy import signal

from clio_constants import *

global FILE_COMPARE
global data_directory_base



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


#figure_size (in inches, that is 100*pixels with dpi=100)
figure_size=(8, 6)
figure_dpi=100

alldata=[]

figcomp=plt.figure(1,figsize=figure_size,dpi=figure_dpi)
figdiff=plt.figure(2,figsize=figure_size,dpi=figure_dpi)
plt.figure(1)

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)
            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()

#    ckey_values={}
#    with open(fname,'r') as fid:
#        fileline=fid.readline()
#        while (len(fileline)>0)and(fileline[0]=='#'):                
#            entries=fileline.replace('#','').replace(' ','').replace('\n','').split('=')
#            ckey_values[entries[0]]=entries[1]
#            fileline=fid.readline()
#        fid.close()                        

cfid.close()


plt.figure(1)
plt.title('Comparison')
plt.ylabel('Signal amplitude [a.u.]')
plt.xlabel('Detector angle [deg.]')
plt.grid(True)        
plt.legend()
imagenamecomp=data_directory_base+'data_comparison.png'
figcomp.savefig(imagenamecomp)
plt.close(figcomp)

plt.figure(2)
plt.title('Comparison (ref substracted)')
plt.ylabel('Signal amplitude above reference [a.u.]')
plt.xlabel('Detector angle [deg.]')
plt.grid(True)        
plt.legend()
imagenamediff=data_directory_base+'data_comparison_with_reference.png'
figdiff.savefig(imagenamediff)
plt.close(figdiff)