NuModule.cxx

Go to the documentation of this file.

// Coded by Jeff Hartnell Aug/2007 onwards
//
// Contact: j.j.hartnell@rl.ac.uk

#include <cassert>

#include "JobControl/JobCModuleRegistry.h"
#include "JobControl/JobCommand.h"
#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "BeamDataNtuple/NtpBDLiteRecord.h"
#include "CandFitTrackSA/Ntp/NtpFitSARecord.h"
#include "StandardNtuple/NtpStRecord.h"

#include "NuMuBar/NuAnalysis.h"
#include "NuMuBar/NuModule.h"

JOBMODULE(NuModule,
          "NuModule",
          "A module to produce NuMuBar DST ntuples");

CVSID("$Id: NuModule.cxx,v 1.3 2007/10/14 19:59:08 hartnell Exp $");

ClassImp(NuModule)

//......................................................................

NuModule::NuModule() 
{
  MSG("NuModule", Msg::kDebug) 
   <<"Running NuModule constructor..."<<endl;

  //initialise data members
  fNuAnalysis=0;
  fUseOldMethod=0;

  MSG("NuModule", Msg::kDebug) 
   <<"Finished constructor"<<endl;
}

//......................................................................

NuModule::~NuModule()
{
  MSG("NuModule", Msg::kDebug) 
   <<"Running destructor..."<<endl;

  MSG("NuModule", Msg::kDebug) 
   <<"Finished destructor"<<endl;
}

//......................................................................

JobCResult NuModule::Ana(const MomNavigator *mom)
{
  //to test number of entries
  //static Int_t counter=0;
  //counter++;
  //cout<<"counter="<<counter<<endl;

  //set default result to kPassed
  JobCResult result(JobCResult::kPassed);
  
  //check that mom exists.
  assert(mom);
  
  const NtpStRecord* pntp=dynamic_cast<NtpStRecord*>
    (mom->GetFragment("NtpStRecord"));
  if (!pntp){
    MSG("NuModule",Msg::kWarning) 
      << "No NtpStRecord in Mom" << endl;
  }
  
  const NtpBDLiteRecord* pntpBD=dynamic_cast<NtpBDLiteRecord*>
    (mom->GetFragment("NtpBDLiteRecord"));
  if (!pntpBD){
    MAXMSG("NuModule",Msg::kInfo,1)
      <<"No NtpBDLiteRecord in Mom (note: not present for MC)"<<endl;
  }

  //NtpFitSARecord
  const NtpFitSARecord* pntpSA=dynamic_cast<NtpFitSARecord*>
    (mom->GetFragment("NtpFitSARecord"));
  if (!pntpSA){
    MAXMSG("NuModule",Msg::kInfo,1) 
      <<"No NtpFitSARecord in Mom (pntpSA="<<pntpSA<<")"<<endl;
    //mom->Print();
  }
  else {
    MAXMSG("NuModule",Msg::kInfo,1)
      << "Found NtpFitSARecord in Mom" << endl;
  }

  //set the file name
  static Bool_t firstTime=true;
  if (firstTime) {
    firstTime=false;
    NuBase::InputFileName(this->GetCurrentFile());
  }
  
  //run the code to calc the meu values for events in this snarl
  if (fUseOldMethod==1) {
    fNuAnalysis->ChargeSeparationOneSnarl(pntp,pntpBD,pntpSA);
  }
  else {
    fNuAnalysis->MakeFullDST(pntp,pntpBD,pntpSA);
  }
  
  return result;
}

//......................................................................

void NuModule::BeginJob() 
{
  MSG("NuModule",Msg::kInfo)
    <<"Running NuModule::BeginJob()..."<<endl;

  //construct the NuAnalysis object
  fNuAnalysis=new NuAnalysis();

  MSG("NuModule",Msg::kInfo)
    <<"Finished running NuModule::BeginJob()"<<endl;
}

//......................................................................

void NuModule::EndJob() 
{  
  MSG("NuModule",Msg::kInfo)
    <<"Running NuModule::EndJob()..."<<endl;

  //finish the job
  fNuAnalysis->EndJob();
  
  MSG("NuModule",Msg::kInfo)
    <<"Finished running NuModule::EndJob()"<<endl;
}

//......................................................................

const Registry& NuModule::DefaultConfig() const
{
  
  MSG("NuModule", Msg::kDebug)
    << "Running NuModule::DefaultConfig..." << endl;
  
  static Registry r;
  
  string name=this->JobCModule::GetName();
  name+=".config.default";
  r.SetName(name.c_str());
  r.UnLockValues();
  
  r.Set("UseOldMethod",0);  
  //r.Set("LIRunSeqRequired",-1);
  //r.Set("WriteOnlyGainCurves",0);

  r.LockValues();
  
  return r;
}

//......................................................................

void NuModule::Config(const Registry& r)
{

  MSG("NuModule",Msg::kDebug)<<"NuModule::Config"<<endl;

  Int_t useOldMethod=0;
  if (r.Get("UseOldMethod",useOldMethod)) fUseOldMethod=useOldMethod;
  
  /*
    Int_t runSeqReq=-1;
    if (r.Get("LIRunSeqRequired",runSeqReq)) fLIRunSeqReq=runSeqReq;
  
    Int_t writeOnlyGainCurves=0;
    if (r.Get("WriteOnlyGainCurves",
    writeOnlyGainCurves)) fWriteOnlyGainCurves=writeOnlyGainCurves;
  */
}

//......................................................................

---