import matplotlib.pyplot as plt
import IQMixer.IQCalib as IQ
import numpy as np
import FileReader as reader
from matplotlib.backends.backend_pdf import PdfPages
IQCorrectionfile = 'IQMixer_Calib/20160803_1M_BOX/EllipseFit_0dBm_2000MHz_8000MHz.csv'
"""
Edit folder and file name and measurement frequency in nnnn MHz
"""
cosraydata_folder = "../../../MeasurementResult/20160814_Al_Noguchi/CosmicRay/4579/"
freq = 4579
cosraydata_file = '2000K_0.5S_9624'
### Calibrate IQ data and calculate amplitude of IQ (IQ)
fs, num, I, Q = reader.ReadCosmicRay(cosraydata_folder, cosraydata_file)
paras = IQ.IQ_GetPara(IQCorrectionfile, freq)
I_mixercalibrated, Q_mixercalibrated = IQ.IQ_CorrtBarends(paras, I, Q)
IQ = np.sqrt((I_mixercalibrated - np.mean(I_mixercalibrated))**2 +
(Q_mixercalibrated - np.mean(Q_mixercalibrated))**2)
### Get index of maximum point and
### cut measurement data 500 points left from the peak and
### 2500 points right from the peak
numleft = 500
numright = 2500
index = IQ.argmax()
Ievent = I_mixercalibrated[index - numleft:index + numright]
Qevent = Q_mixercalibrated[index - numleft:index + numright]
amplitude = np.asarray(IQ[index - numleft:index + numright])
phase = np.arctan2(Qevent, Ievent)
