"""
Arguments:

VECTOR (Double[][], Scan vector: Eph,Elow,Ehigh or Eph,Ecenter)
SENSORS (list)
LATENCY (double)
MODE ('fixed' or 'swept')
TYPE ('CIS' or 'CFS')
STEP (double)
"""

LATENCY = 0.0


#################### Si 2p PhD scan ##############################
VECTOR = [ 170.0]
#for nn in range(63):
#    VECTOR.append(VECTOR[-1]+4)
nn = 0


initial_energy_range = [[63, 67, 0.1], 
                        [67.5, 77, 0.5]]
pass_energy_hold = 20
angle_time = [0.6, 0.7]
#angle_range = [-9.2, -9.2+30.]
#azi_range = [-90., -45.]

#################### C 1s PhD scan ##############################
#VECTOR = [ 340.0]
#for nn in range(63):
#    VECTOR.append(VECTOR[-1]+4)
#nn = 0


#initial_energy_range = [[45, 52.4, 0.1], 
#                        [52.5, 62.5, 0.5]]
#pass_energy_hold = 50
#angle_time = [0.6, 0.7]
#angle_range = [-9.2, -9.2+30.]
#azi_range = [-90., -45.]

#################### O 1s PhD scan ##############################
#VECTOR = [ 630.0]
#for nn in range(63):
#    VECTOR.append(VECTOR[-1]+4)

#pass_energy_hold = 100

#initial_energy_range = [[87.2, 95.5, 0.1], 
#                        [95.0, 105.0, 0.5]]

#angle_time = [0.3, 0.4, 0.5, 0.6]
#angle_range = [-9.2+20., -9.2+40., -9.2+60.]
#azi_range = [-90., -45.]


class SpectrumReader(ReadonlyRegisterBase, ReadonlyRegisterArray):
    def doRead(self):
        global VECTOR
        self.offset = Eph.getSetpoint().take() - VECTOR[0]
        self.setup()
        trig_scienta()
        time.sleep(0.5)
        return Scienta.getSpectrum().read()

    def getSize(self):
        return len(Scienta.getSpectrumX())

energy_scale_array = []
                        
class SpectrumReader1(SpectrumReader):
    def setup(self):
        global energy_scale_array, initial_energy_range               
        Scienta.getLowEnergy().write(initial_energy_range[0][0] + self.offset)
        Scienta.getHighEnergy().write(initial_energy_range[0][1] + self.offset)
        Scienta.getStepSize().write(initial_energy_range[0][2])        
        energy_scale_array = [Scienta.getLowEnergy().take(), Scienta.getHighEnergy().take(), Scienta.getStepSize().take()]

class SpectrumReader2(SpectrumReader):
    def setup(self):        
        global energy_scale_array, initial_energy_range
        Scienta.getLowEnergy().write(initial_energy_range[1][0] + self.offset)
        Scienta.getHighEnergy().write(initial_energy_range[1][1] + self.offset)
        Scienta.getStepSize().write(initial_energy_range[1][2])        
        energy_scale_array = energy_scale_array + [Scienta.getLowEnergy().take(), Scienta.getHighEnergy().take(), Scienta.getStepSize().take()]
        append_dataset(energy_scale_table,energy_scale_array)   

class SpectrumReader3(SpectrumReader):
    def setup(self):        
        global initial_energy_range,energy_scale_array, energy_scale_table
        Scienta.getLowEnergy().write(initial_energy_range[2][0] + self.offset)
        Scienta.getHighEnergy().write(initial_energy_range[2][1] + self.offset)        
        Scienta.getStepSize().write(initial_energy_range[2][2])        
        energy_scale_array = energy_scale_array + [Scienta.getLowEnergy().take(), Scienta.getHighEnergy().take(), Scienta.getStepSize().take()]
        append_dataset(energy_scale_table,energy_scale_array)    


class ImageReader(ReadonlyRegisterBase, ReadonlyRegisterMatrix):   
    def doRead(self):
        return Scienta.getDataMatrix().read()

    def getWidth(self):
        global initial_energy_range 
        return int( (initial_energy_range[self.index][1] - initial_energy_range[self.index][0]) / initial_energy_range[self.index][2])

    def getHeight(self):
        return Scienta.getDataMatrix().getHeight()

reader1 = SpectrumReader1(); reader1.initialize()
reader2 = SpectrumReader2(); reader2.initialize()
#reader3 = SpectrumReader3(); reader3.initialize()
image1 = ImageReader(); image1.initialize(); image1.index=0; set_device_alias(image1, "Image1")
image2 = ImageReader(); image2.initialize(); image2.index=1; set_device_alias(image2, "Image2")
#image3 = ImageReader(); image3.initialize(); image3.index=2; set_device_alias(image3, "Image3")


Scienta.getDataMatrix()

SENSORS = [RefCurrent, reader1, image1, reader2, image2]#, reader3, image3]

Scienta.setAcquisitionMode(ch.psi.pshell.epics.Scienta.AcquisitionMode.Swept)


energy_scale_table = "scan 1/EnergyScale"
create_dataset(energy_scale_table, 'd', dimensions=[0,9])


adjust_sensors()
set_adc_averaging()
set_preference(Preference.PLOT_TYPES, {'Scienta spectrum':1})

try:
#    for ll in range(len(azi_range)):
#        ManipulatorPhi.move(azi_range[ll])
#        for nn in range(len(angle_range)):
            tic = time.time()
#            ManipulatorTheta.move(angle_range[nn])
#            print 'current theta is ', ManipulatorTheta.take(), ' and current azimuth is ' , ManipulatorPhi.take()
            Scienta.setPassEnergy(pass_energy_hold)
            Scienta.getStepTime().write(angle_time[nn])
            sleep(1)
            vscan(Eph, SENSORS, VECTOR, True, LATENCY,False, before_read=wait_beam, after_read = after_readout)
            toc = time.time()
            tictoc = round((toc-tic)/60*100)/100
            print 'time taken is ', tictoc, ' minutes'
finally:
    if ENDSCAN:
        after_scan()    

toc = time.time()