TAnalysisModuleBase.cxx

Go to the documentation of this file.

#include "TAnalysisModuleBase.hxx"
#include <TObjString.h>

ND::TAnalysisModuleBase::TAnalysisModuleBase() :
      fFilledConfigTree(kFALSE),
      fIsEnabled(kFALSE),
      fIsUsedForPreselection(kFALSE),
      fOutputTree(NULL),
      fBufferSize(32000),
      fSplitLevel(99),
      fDescription("Undefined"),
      fCVSTagName("Undefined: Problem in derived class!"),
      fCVSID("Undefined: Problem in derived class!") {}

ND::TAnalysisModuleBase::~TAnalysisModuleBase() {}

Bool_t ND::TAnalysisModuleBase::Configure(std::string& option) {
  ND280Error("No configurable options defined for this class");
  return kFALSE;
}

std::string const ND::TAnalysisModuleBase::GetDirectoryName() const {
  switch (GetTreeType()) {
    case kHeader:
      return "HeaderDir";
    case kTruth:
      return "TruthDir";
    case kRecon:
      return "ReconDir";
    case kLowLevel:
      return "LowLevelDir";
    default:
      throw ND::EUndefinedTreeType();
  }
}

void ND::TAnalysisModuleBase::Initialize(TTree* tree) {
  fOutputTree = tree;

  // Fix to bug 386
  fOutputTree->SetAutoFlush(3000000);

  InitializeModule();
  fOutputTree->Branch("RunID", &fRunID, "RunID/I", fBufferSize);
  fOutputTree->Branch("EventID", &fEventID, "EventID/I", fBufferSize);
  fOutputTree->Branch("Preselected", &fPreselected, "Preselected/B",
                      fBufferSize);
  fOutputTree->Branch("SubrunID", &fSubrunID, "SubrunID/I", fBufferSize);
  InitializeBranches();
}

void ND::TAnalysisModuleBase::FillConfigTree(TTree* configTree){

  if(fFilledConfigTree){
    ND280NamedWarn("TAnalysisModuleBase", "Module: " << this->GetName()
      << " has already written the config tree this run.");
  }

  TObjString * Description = 0;
  TObjString * CVSTagName = 0;
  TObjString * CVSID = 0;

  configTree->Branch("IsUsedForPreselection", &fIsUsedForPreselection,
    "IsUsedForPreselection/O");
  configTree->Branch("Description", &Description);
  configTree->Branch("CVSTagName", &CVSTagName);
  configTree->Branch("CVSID", &CVSID);

  (*Description) = fDescription.c_str();
  (*CVSTagName) = fCVSTagName.c_str();
  (*CVSID) = fCVSID.c_str();

  configTree->Fill();
  fFilledConfigTree = true;
}

bool ND::TAnalysisModuleBase::Process(ND::TND280Event& event) {
  fRunID = event.GetRunId();
  fEventID = event.GetEventId();
  fSubrunID = event.GetContext().GetSubRun();
  fIsMC = event.GetContext().IsMC();
  if (!FillTree(event)) {
    throw EAnalysisFailure();
  }
  fPreselected = IsFullEventWorthSaving(event);
  fOutputTree->Fill();
  return fPreselected;
}

bool ND::TAnalysisModuleBase::IsFullEventWorthSaving(ND::TND280Event& event) {
  return false;  // Do not save the main event tree entry by default, just the
                 // analysis one
}

Bool_t ND::TAnalysisModuleBase::ProcessFirstEvent(ND::TND280Event&) {
  return true;
}

void ND::TAnalysisModuleBase::Print() {
  // override this function if more details need to be printed out
  ND280Log("Module: " << GetName()
                      << (IsEnabled() ? " (Enabled)  " : " (Disabled) ")
                      << (IsUsedForPreselection() ? " (Preselecting)  "
                                                  : " (Not Preselecting) "));
  ND280Log("Description: " << GetDescription());
}

void ND::TAnalysisModuleBase::SetNameTitle(char const* name,
                                                char const* title) {
  fName = name;
  fTitle = title;
}

char const * ND::TAnalysisModuleBase::GetName(){
  return fName.c_str();
}

char const * ND::TAnalysisModuleBase::GetTitle(){
  return fTitle.c_str();
}