#include "IBasicDataQualityModule.hxx"
#include "IG4PrimaryParticle.hxx"
#include "IG4PrimaryVertex.hxx"
#include "IDataQuality.hxx"
#include <iostream>  
using namespace std; 

ClassImp(COMET::IBasicDataQualityModule);

#define CVSTAG "\
  $Name: v5r19 $"
#define CVSID  "\
  $Id: IBasicDataQualityModule.cxx,v 1.8 2011/02/28 00:23:32 dilodovico Exp $"

COMET::IBasicDataQualityModule::IBasicDataQualityModule(const char *name,
const char *title) {
  SetNameTitle(name, title);
  // Enable this module by default:
  fIsEnabled = kTRUE;
  //  fDescription = "Standard module which fills the standard header information about an event";
  fCVSTagName = CVSTAG;
  fCVSID = CVSID;
  
  fCOMETOffFlag = -1;
  fTPCFlag = -1;
  fTPC1Flag = -1;
  fTPC2Flag = -1;
  fTPC3Flag = -1;
  fFGDFlag = -1;
  fFGD1Flag = -1;
  fFGD2Flag = -1;
  fECALFlag = -1;
  fDSECALFlag = -1;
  fBarECALFlag = -1;
  fP0DECALFlag = -1;
  fP0DFlag = -1;
  fSMRDFlag = -1;
  fMAGNETFlag = -1;
  fINGRIDFlag = -1;
}


COMET::IBasicDataQualityModule::~IBasicDataQualityModule() {}

void COMET::IBasicDataQualityModule::InitializeBranches() {
  //DATA QUALITY INFO
  fOutputTree->Branch("COMETOffFlag", &fCOMETOffFlag, "COMETOffFlag/Int_t", fBufferSize);
  fOutputTree->Branch("TPCFlag", &fTPCFlag, "TPCFlag/Int_t", fBufferSize);
  fOutputTree->Branch("TPC1Flag", &fTPC1Flag, "TPC1Flag/Int_t", fBufferSize);
  fOutputTree->Branch("TPC2Flag", &fTPC2Flag, "TPC2Flag/Int_t", fBufferSize);
  fOutputTree->Branch("TPC3Flag", &fTPC3Flag, "TPC3Flag/Int_t", fBufferSize);
  fOutputTree->Branch("FGDFlag", &fFGDFlag, "FGDFlag/Int_t", fBufferSize);
  fOutputTree->Branch("FGD1Flag", &fFGD1Flag, "FGD1Flag/Int_t", fBufferSize);
  fOutputTree->Branch("FGD2Flag", &fFGD2Flag, "FGD2Flag/Int_t", fBufferSize);
  fOutputTree->Branch("ECALFlag", &fECALFlag, "ECALFlag/Int_t", fBufferSize);
  fOutputTree->Branch("DSECALFlag", &fDSECALFlag, "DSECALFlag/Int_t", fBufferSize);
  fOutputTree->Branch("BarECALFlag", &fBarECALFlag, "BarECALFlag/Int_t", fBufferSize);
  fOutputTree->Branch("P0DECALFlag", &fP0DECALFlag, "P0DECALFlag/Int_t", fBufferSize);
  fOutputTree->Branch("P0DFlag", &fP0DFlag, "P0DFlag/Int_t", fBufferSize);
  fOutputTree->Branch("SMRDFlag", &fSMRDFlag, "SMRDFlag/Int_t", fBufferSize);
  fOutputTree->Branch("MAGNETFlag", &fMAGNETFlag, "MAGNETFlag/Int_t", fBufferSize);
  fOutputTree->Branch("INGRIDFlag", &fINGRIDFlag, "INGRIDFlag/Int_t", fBufferSize);
}


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

  if(event.GetContext().IsMC()) return true;

  fEventTime = event.GetHeader().GetMCMSecond();

  //COMET::IDataQuality dqObject;

  dqObject.getStatus(fEventTime);

  fCOMETOffFlag = dqObject.AllStatus();
  fTPCFlag = dqObject.TPCStatus();
  fTPC1Flag = dqObject.TPC1Status();
  fTPC2Flag = dqObject.TPC2Status();
  fTPC3Flag = dqObject.TPC3Status();
  fFGDFlag = dqObject.FGDStatus();
  fFGD1Flag = dqObject.FGD1Status();
  fFGD2Flag = dqObject.FGD2Status();
  fECALFlag = dqObject.ECALStatus();
  fDSECALFlag = dqObject.DSECALStatus();
  fBarECALFlag = dqObject.BarECALStatus();
  fP0DECALFlag = dqObject.P0DECALStatus();
  fP0DFlag = dqObject.P0DStatus();
  fSMRDFlag = dqObject.SMRDStatus();
  fMAGNETFlag = dqObject.MAGNETStatus();
  fINGRIDFlag = dqObject.INGRIDStatus();
  
  //  cout <<"Timestamp: "<<fEventTime<<" status ALL:" << fCOMETOffFlag<<" status TPC:" << fTPCFlag<<" "<<fTPC1Flag<<" "<<fTPC2Flag<<" "<<fTPC3Flag<<" FGD: "<<fFGDFlag<<" "<<fFGD1Flag<<" "<<fFGD2Flag<<" ECAL:"<<fECALFlag<<" "<<fDSECALFlag<<" "<<fBarECALFlag<<" "<<fP0DECALFlag<<" P0D:"<<fP0DFlag<<" SMRD:"<<fSMRDFlag<<" MAGNET:"<<fMAGNETFlag<<" INGRID:"<<fINGRIDFlag << endl;  

  return true;
}


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