AnalysisModule.cxx

Go to the documentation of this file.

//$Id: AnalysisModule.cxx,v 1.2 2006/02/25 00:47:10 gmieg Exp $
//
//AnalysisModule.cxx
//
// A template module used for doing analysis with the AnalysisNtuples package
//
//B. Rebel 2/2005

#include "AnalysisNtuples/Module/AnalysisModule.h"

#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "JobControl/JobCModuleRegistry.h" // For JOBMODULE macro
#include "JobControl/JobCommand.h"

#include "TFolder.h"
#include "TDirectory.h"
#include "TObjArray.h"
#include "TCanvas.h"
#include "TPad.h"
#include "TArc.h"

#include <cassert>
#include <algorithm>

ClassImp(AnalysisModule)

CVSID("$Id: AnalysisModule.cxx,v 1.2 2006/02/25 00:47:10 gmieg Exp $");

// Declare this module to JobControll. Arguments are:
//  (1) The class name 
//  (2) The human-readable name 
//  (3) A short, human-readable description of what the module does
JOBMODULE(AnalysisModule, 
          "AnalysisModule",
          "A template module used for analyzing data");

//......................................................................
AnalysisModule::AnalysisModule() :
    fFileName("beamTestResults.root"),
    fTreeName("ANtp"),
    fDataPath("data/*.root")
{
    MSG("JobC", Msg::kDebug) << "AnalysisModule::Constructor" << endl;
        
    //make histograms for raw distributions

        
    //histograms for cut distributions
        
    //the info objects
    fHeaderInfo = new ANtpHeaderInfo();
    fEventInfo = new ANtpEventInfo();
    fShowerInfo = new ANtpShowerInfo();
    fTrackInfo = new ANtpTrackInfo();
    
}

//----------------------------------------------------------------------
AnalysisModule::~AnalysisModule()
{

  MSG("JobC", Msg::kDebug) << "AnalysisModule::Destructor" << endl;

}

//......................................................................
void AnalysisModule::BeginJob()
{
    MSG("AnalysisModule", Msg::kDebug) << "start BeginJob" << endl;

    
    return;
}

//......................................................................
//here is where you would loop over the events in the snarl to fill the 
//analysis tree
JobCResult AnalysisModule::Ana(const MomNavigator *mom)
{
    MSG("AnalysisModule", Msg::kDebug) << "start ana" << endl;
    
    JobCResult result(JobCResult::kPassed);
    
    //get the records from MOM
    assert(mom);
        
    return result;
}

//......................................................................
void AnalysisModule::Help() 
{
  MSG("JobC", Msg::kInfo) 
    << "AnalysisModule::Help\n"
    <<"AnalysisModule is a module which analyzes beam data."
    << endl;
}

//----------------------------------------------------------------------
void AnalysisModule::EndJob() 
{

    MSG("AnalysisModule", Msg::kInfo) << "start end job method" << endl;        
    
    //create a chain for the analysis ntuples with name in fTreeName
    fChain = new TChain(fTreeName.c_str());
    
    MSG("AnalysisModule", Msg::kInfo) << "path = " << fDataPath << endl;
    
    //add the files to the chain
    fChain->Add(fDataPath.c_str());
    
    //attach the info objects to the chains
    fChain->SetBranchAddress("header.", &fHeaderInfo);
    fChain->SetBranchAddress("event.", &fEventInfo);
    fChain->SetBranchAddress("shower.", &fShowerInfo);
    fChain->SetBranchAddress("track.", &fTrackInfo);
    fChain->SetBranchAddress("truth.", &fTruthInfo);

    //do your analysis here.  probably best to write methods to do individual bits 
    //of analysis like declaring and filling histograms, drawing histograms, 
    //checking if the event vertex is in the fiducial volume, etc
    
    return;
}

//......................................................................
const Registry& AnalysisModule::DefaultConfig() const
{
    
    int itrue = 1;  // work around for lack of bool in registry
    int ifalse = 0; // work around for lack of bool in registry
    
    static Registry r;
    
    r.UnLockValues();
    
    r.Set("FileName",               "analysisResults.root");
    r.Set("TreeName",               "ANtp");
    r.Set("DataPath",               "data/*.root");
    
    r.LockValues();

    itrue = ifalse;

    return r;
}

//......................................................................
void AnalysisModule::Config(const Registry& r)
{
    int         tmpb;  // a temp bool. See comment under DefaultConfig...
    int         tmpi;  // a temp int.
    double      tmpd;  // a temp double.
    const char* tmps;  // a temp string

    if (r.Get("FileName",               tmps)) fFileName               = tmps;
    if (r.Get("TreeName",               tmps)) fTreeName               = tmps;
    if (r.Get("DataPath",               tmps)) fDataPath               = tmps;

    tmpb = 0;
    tmpi = 0;
    tmpd = 0.;

    return;
}

---