#include "IDatum.hxx"
#include "IRealDatum.hxx"
#include "TPrincipal.h"

#include "IECALTestbeamModule.hxx"

ClassImp(COMET::IECALTestbeamModule);

#define CVSTAG "\
  $Name: v5r19 $"
#define CVSID  "\
  $Id: IECALTestbeamModule.cxx,v 1.3 2010/05/24 08:07:58 lwhitehead Exp $"

COMET::IECALTestbeamModule::IECALTestbeamModule(const char *name,
const char *title)
{

    SetNameTitle(name, title);
    // Enable this module by default:
    fIsEnabled = kTRUE;
    fDescription = "ECALTestbeam recon output";
    fCVSTagName = CVSTAG;
    fCVSID = CVSID;

}

COMET::IECALTestbeamModule::~IECALTestbeamModule() {}

Bool_t COMET::IECALTestbeamModule::ProcessFirstEvent(COMET::ICOMETEvent& event)
{
    return true;
}

void COMET::IECALTestbeamModule::InitializeModule()
{

}

void COMET::IECALTestbeamModule::InitializeBranches()
{
    fOutputTree->Branch("Cerenkov1Lo", &Cerenkov1Lo,"Cerenkov1Lo[23]/I");
    fOutputTree->Branch("Cerenkov1Hi", &Cerenkov1Hi,"Cerenkov1Hi[23]/I");
    fOutputTree->Branch("Cerenkov2Lo", &Cerenkov2Lo,"Cerenkov2Lo[23]/I");
    fOutputTree->Branch("Cerenkov2Hi", &Cerenkov2Hi,"Cerenkov2Hi[23]/I");
    fOutputTree->Branch("TOF", &TOF,"TOF[23]/I");
    fOutputTree->Branch("TriggerWord", &TriggerWord, "TriggerWord/I");

    fOutputTree->Branch("TestBeamPID", &PIDResult, "PIDResult/I");
    fOutputTree->Branch("TestBeamMomentum" , &Momentum, "Momentum/D");
    fOutputTree->Branch("TestBeamAngle" , &Angle, "Angle/I");
}

bool COMET::IECALTestbeamModule::FillTree(COMET::ICOMETEvent& event)
{

    COMET::IHandle<COMET::IIntegerDatum> Ch1Lo = event.Get<COMET::IIntegerDatum>("Cerenkov1Lo");
    COMET::IHandle<COMET::IIntegerDatum> Ch2Lo = event.Get<COMET::IIntegerDatum>("Cerenkov2Lo");
    COMET::IHandle<COMET::IIntegerDatum> Ch1Hi = event.Get<COMET::IIntegerDatum>("Cerenkov1Hi");
    COMET::IHandle<COMET::IIntegerDatum> Ch2Hi = event.Get<COMET::IIntegerDatum>("Cerenkov2Hi");
    COMET::IHandle<COMET::IIntegerDatum> TOFLo = event.Get<COMET::IIntegerDatum>("TOFHighGain");
    COMET::IHandle<COMET::IIntegerDatum> TOFHi = event.Get<COMET::IIntegerDatum>("TOFLowGain");
    COMET::IHandle<COMET::IIntegerDatum> TWord = event.Get<COMET::IIntegerDatum>("TriggerWord");

    for(int i = 0; i < 23; ++i) {
        Cerenkov1Lo[i] = (Ch1Lo->GetVector())[i];
        Cerenkov1Hi[i] = (Ch2Lo->GetVector())[i];
        Cerenkov2Lo[i] = (Ch1Hi->GetVector())[i];
        Cerenkov2Hi[i] = (Ch2Hi->GetVector())[i];
        TOF[i] =  (TOFLo->GetVector())[i];
    }

    TriggerWord = TWord->GetValue();

    COMET::IHandle<COMET::IIntegerDatum> pidResultDatum = event.Get<COMET::IIntegerDatum>("testBeamPID");
    COMET::IHandle<COMET::IIntegerDatum> angleDatum = event.Get<COMET::IIntegerDatum>("testBeamAngle");
    COMET::IHandle<COMET::IRealDatum> momentumDatum = event.Get<COMET::IRealDatum>("testBeamMomentum");

    PIDResult = pidResultDatum->GetValue();
    Momentum = momentumDatum->GetValue();
    Angle = angleDatum->GetValue();

    return true;

}