In [3]:

    

import math

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from scipy.optimize import curve_fit

import optics_calcs.shgFitting as shg


    

In [4]:

    

#location of data files
inputpath = '/Users/geiger/Box Sync/Science/Data/SHG/2018/122218'

#path of output files
outputpath = '/Users/geiger/Box Sync/Science/Data/SHG/2018/122218/'


    

In [5]:

    

names = ['salt_100_scan2','quartz_181_scan']


    

In [6]:

    

#import data and plot it
data = shg.importSHGdata(names,inputpath,outputpath)


    

In [7]:

    

#fit the data
(initFitParams,initFitValues) = shg.initFit(data)


    

In [8]:

    

#write parameters to file
shg.writeToFile(initFitParams,outputpath,'INTinitFitParams.txt')
shg.writeToFile(initFitValues,outputpath,'INTinitFitValues.txt')


    

In [9]:

    

(fAve,fStd) = shg.calcFAve(initFitParams)


    

    




f = 119.59 +- 3.02

In [10]:

    

(finalFitParams, finalFitValues) = shg.finalFit(data,fAve,fStd)


    

In [12]:

    

#write parameters to file
shg.writeToFile(finalFitParams,outputpath,'INTfinalFitParams.txt')
shg.writeToFile(finalFitValues,outputpath,'INTfinalFitValues.txt')


    

In [13]:

    

plt.figure()
plt.plot(finalFitParams['phideg'])


    

    
Out[13]:





[<matplotlib.lines.Line2D at 0x1021511c88>]






    

In [ ]: