from HCDDataReader import HCDDataReader
from HCDDataProcessor import HCDDataProcessor
import matplotlib.pyplot as plt
import numpy as np
from scipy import interpolate
pressure_binsize = 0.5
############################################## KMI 10-2014 ####################################################################
KMI_reader = HCDDataReader("./Datasets/KMI102014", software_version=1)
KMI_processor = HCDDataProcessor()
KMI_processor.remove_peaks(KMI_reader.data)
pressure_KMI = KMI_reader.data["press1"]
gm1_KMI = np.array(KMI_reader.data["gm1"])
gm2_KMI = np.array(KMI_reader.data["gm2"])
gm3_KMI = np.array(KMI_reader.data["gm3"])
gm4_KMI = np.array(KMI_reader.data["gm4"])
gm1_KMI_average, gm1_KMI_error, gm1_KMI_binpoints = KMI_processor.average_counts_over_pressure(gm1_KMI, pressure_KMI, binsize=pressure_binsize)
gm2_KMI_average, gm2_KMI_error, gm2_KMI_binpoints = KMI_processor.average_counts_over_pressure(gm2_KMI, pressure_KMI, binsize=pressure_binsize)
gm3_KMI_average, gm3_KMI_error, gm3_KMI_binpoints = KMI_processor.average_counts_over_pressure(gm3_KMI, pressure_KMI, binsize=pressure_binsize)
gm4_KMI_average, gm4_KMI_error, gm4_KMI_binpoints = KMI_processor.average_counts_over_pressure(gm4_KMI, pressure_KMI, binsize=pressure_binsize)
pressures_KMI_average, pressures_KMI_error = KMI_processor.pressurebin_centers(pressure_KMI, binsize=pressure_binsize)
gm_KMI_weighted_average, gm_KMI_weighted_error = KMI_processor.weighted_average([gm1_KMI_average, gm2_KMI_average, gm3_KMI_average, gm4_KMI_average], [gm1_KMI_error, gm2_KMI_error, gm3_KMI_error, gm4_KMI_error])
peak_boundaries_KMI = KMI_processor.set_peak_boundaries(xmin = 3.043499, xmax = 10.316913)
from HCDDataReader import HCDDataReader
from HCDDataProcessor import HCDDataProcessor
import pylab
import numpy as np
pressure_binsize = 0.5
##############################################################################################
######## BEXUS ############
print "BEXUS"
BEXUS_reader = HCDDataReader("./Datasets/BEXUSDetectorFlightData")
BEXUS_processor = HCDDataProcessor()
BEXUS_data = BEXUS_processor.remove_peaks(BEXUS_reader.data)
pressure1_BX = BEXUS_data["press1"]
pressure2_BX = BEXUS_data["press2"]
pressure3_BX = BEXUS_data["press3"]
loop_time_BX = BEXUS_data["looptimer"]
HV_state_BX = BEXUS_data["hvstate"]
eth_state_BX = BEXUS_data["ethstate"]
gm1_BX_raw = np.array(BEXUS_data["gm1"])
gm2_BX_raw = np.array(BEXUS_data["gm2"])
gm3_BX_raw = np.array(BEXUS_data["gm3"])
gm4_BX_raw = np.array(BEXUS_data["gm4"])
gm1_BX = BEXUS_processor.datacut_based_on_statedata(gm1_BX_raw, HV_state_BX)
gm2_BX = BEXUS_processor.datacut_based_on_statedata(gm2_BX_raw, HV_state_BX)
gm3_BX = BEXUS_processor.datacut_based_on_statedata(gm3_BX_raw, HV_state_BX)
gm4_BX = BEXUS_processor.datacut_based_on_statedata(gm4_BX_raw, HV_state_BX)
