from __future__ import division
from time import clock
start = clock()
import scipy as sp
from scipy.optimize import leastsq
from numpy.core.defchararray import add
import numpy as np
from math import *
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

from State_tomography import load_tomo_file, tomo_input_analysis, tomography_error_comparison

intensity=[1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1]
# intensity=np.linspace(1,0.9,num=36)

conf={'NQubits':2,'NDetectors':1,'Crosstalk':np.array([[1,0,0.0,0],[0,1,0,0],[0,0,1,0],[0,0,0,1]]),\
      'UseDerivative':0,'DoErrorEstimation':0,'DoDriftCorrection':'no','Window':40,'Efficiency':np.array([1,1,1,1])}
      
# before = load_tomo_file('temp36_nofr_attempt7.dat')
# after_acw = load_tomo_file('temp36_fr_acw4_attempt1.dat')

# tomo_input_analysis(before)
# tomo_input_analysis(after_acw)

tomo_file = '/home/qitlab/projects/clocksync/data/tomography/20190503/20190503_0305_36_settings.dat'
tomo_file_before = '/home/qitlab/projects/clocksync/data/tomography/20190502/20190502_0143_36_settings.dat'

tomo_input = load_tomo_file(tomo_file)
tomo_input_before = load_tomo_file(tomo_file_before)

tomo_input_analysis(tomo_input_before)
tomo_input_analysis(tomo_input)

# tomography_error_comparison(real_tomo_input,real_tomo_input,intensity,intensity,conf,10)
tomography_error_comparison(tomo_input_before,tomo_input,intensity,intensity,conf,100)