import psana.detector.utils_calib_components as ucc

np, info_ndarr, table_nxn_epix10ka_from_ndarr =\
  ucc.np, ucc.info_ndarr, ucc.psu.table_nxn_epix10ka_from_ndarr


class EventProcessor():

    def __init__(self):
        self.counter = 0
        self.cc = None

    def begin_run(self):
        pass

    def end_run(self):
        pass

    def begin_step(self, step):
        pass

    def end_step(self, step):
        pass

    def on_event(self, raw, config, calibconst, *args, **kwargs):
        self.counter += 1
        if self.cc is None:
            self.cc = ucc.calib_components_epix(calibconst, config)
        cc = self.cc

        kwa = {'status': True}
        cmpars = (0, 7, 300, 10)

        ctypes   = cc.calib_types() # list of calibration types
        npanels  = cc.number_of_panels() # number of panels/segments in the detector
        peds     = cc.pedestals() # OR cc.calib_constants('pedestals')
        gain     = cc.gain()      # OR cc.calib_constants('pixel_gain') # ADU/keV
        gfactor  = cc.gain_factor() # keV/ADU
        status   = cc.status() # 4-d array of pixel_status constants
        comode   = cc.common_mode()  # tuple of common mode correction parameters
        trbit_p0 = cc.trbit_for_panel(0)  # list of per-ASIC trbit for panel 0
        ascfg_p0 = cc.asicPixelConfig_for_panel(0) # matrix of asicPixelConfig for panel 0
        mask     = cc.mask(**kwa) # mask defined by kwa
        dettype  = cc.dettype() # detector type, e.g.  "epix10ka" or "epixhr"
        cbitscfg = cc.cbits_config_detector()
        cbitstot = cc.cbits_config_and_data_detector(raw, cbitscfg)
        gmaps    = cc.gain_maps_epix(raw) # gain map
        pedest   = cc.event_pedestals(raw) # per-pixel array of pedestals in the event
        factor   = cc.event_gain_factor(raw) # per-pixel array of gain factors in the event
        calib0   = cc.calib(raw, cmpars, **kwa) # method calib
        cmcorr   = cc.common_mode_correction(raw, cmpars=cmpars, **kwa) # common mode correction
        arrf = np.array(raw & cc.data_bit_mask(), dtype=np.float32) - pedest

        print('== Event %04d ==' % self.counter)
#        print('config', cc.config)
#        print('calib_metadata', cc.calib_metadata('pedestals'))
        print('calib_types', ctypes)
        print(info_ndarr(peds, 'pedestals'))
        print(info_ndarr(cc.gain(), 'gain'))
        print(info_ndarr(gfactor, 'gain_factor'))
        print(info_ndarr(status, 'status'))
        print('common_mode from caliconst', str(comode))
        print('number_of_panels', npanels)
        print('trbit_for_panel(0)', trbit_p0)
        print(info_ndarr(ascfg_p0, 'asicPixelConfig_for_panel(0)'))
        print(info_ndarr(raw, 'raw'))
        print(info_ndarr(mask, 'mask'))
        print('dettype', dettype)
        print(info_ndarr(cbitscfg, 'cbitscfg'))
        print(info_ndarr(cbitstot, 'cbitstot'))
        print(info_ndarr(gmaps, 'gmaps'))
        print(info_ndarr(pedest, 'pedest'))
        print(info_ndarr(factor, 'factor'))
        print(info_ndarr(cmcorr, 'cmcorr'))
        print(info_ndarr(arrf, 'raw(data bits)-peds'))
        print(info_ndarr(calib0, 'calib0'))

        # det.raw.calib(...) algorithm close reproduction
        calib1 = None
        if True:
            arrf1 = arrf.copy()
            cc.common_mode_apply(arrf1, gmaps, cmpars=cmpars, **kwa)
            calib1 = arrf1 * factor if mask is None else arrf1 * factor * mask
            print(info_ndarr(calib1, 'calib1'))

        # det.raw.calib(...) - effective algorithm
        calib2 = None
        if True:
            arrf2 = arrf.copy() + cmcorr
            calib2 = arrf2 * factor if mask is None else arrf2 * factor * mask
            print(info_ndarr(calib2, 'calib2'))

        # img = cmcorr[0, 144:, :192]
        # img = table_nxn_epix10ka_from_ndarr(cmcorr)
        img = table_nxn_epix10ka_from_ndarr(calib1)
        print(info_ndarr(img, 'img'))
        return img