gtestEventLoop.cxx File Reference

#include <string>
#include <TSystem.h>
#include <TFile.h>
#include <TTree.h>
#include <TIterator.h>
#include "Framework/EventGen/EventRecord.h"
#include "Framework/GHEP/GHepParticle.h"
#include "Framework/Ntuple/NtpMCFormat.h"
#include "Framework/Ntuple/NtpMCTreeHeader.h"
#include "Framework/Ntuple/NtpMCEventRecord.h"
#include "Framework/Messenger/Messenger.h"
#include "Framework/ParticleData/PDGCodes.h"
#include "Framework/Utils/CmdLnArgParser.h"

Include dependency graph for gtestEventLoop.cxx:

Go to the source code of this file.

Functions
void	GetCommandLineArgs (int argc, char **argv)
int	main (int argc, char **argv)

Variables
int	gOptNEvt
string	gOptInpFilename

Function Documentation

void GetCommandLineArgs	(	int	argc,
		char **	argv
	)

int main	(	int	argc,
		char **	argv
	)

Definition at line 44 of file gtestEventLoop.cxx.

References genie::NtpMCEventRecord::Clear(), genie::GHepParticle::E(), genie::NtpMCEventRecord::event, GetCommandLineArgs(), gOptInpFilename, gOptNEvt, genie::kIStStableFinalState, genie::kPdgPi0, genie::kPdgPiM, genie::kPdgPiP, LOG, genie::GHepParticle::Name(), genie::GHepParticle::Pdg(), pNOTICE, and genie::GHepParticle::Status().

{
  GetCommandLineArgs (argc, argv);

  //-- open the ROOT file and get the TTree & its header
  TTree *           tree = 0;
  NtpMCTreeHeader * thdr = 0;

  TFile file(gOptInpFilename.c_str(),"READ");

  tree = dynamic_cast <TTree *>           ( file.Get("gtree")  );
  thdr = dynamic_cast <NtpMCTreeHeader *> ( file.Get("header") );

  if(!tree) return 1;

  NtpMCEventRecord * mcrec = 0;
  tree->SetBranchAddress("gmcrec", &mcrec);

  int nev = (gOptNEvt > 0) ?
        TMath::Min(gOptNEvt, (int)tree->GetEntries()) :
        (int) tree->GetEntries();

  //
  // Loop over all events
  //
  for(int i = 0; i < nev; i++) {

    // get next tree entry
    tree->GetEntry(i);

    // get the GENIE event
    EventRecord &  event = *(mcrec->event);

    LOG("myAnalysis", pNOTICE) << event;

    //
    // Put your event analysis code here 
    //
    // ... ... ... ... ...
    // ... ... ... ... ...
    //
    // 

    

    //
    // Loop over all particles in this event
    //

    GHepParticle * p = 0;
    TIter event_iter(&event);

    while((p=dynamic_cast<GHepParticle *>(event_iter.Next())))
    {
       //
       // Put your event analysis code here 
       //
       // ... ... ... ... ...
       // ... ... ... ... ...
       //
       //

       // EXAMPLE: Print out the energy of all final state pions.
       if (p->Status() == kIStStableFinalState ) {
          if (p->Pdg() == kPdgPi0 ||
              p->Pdg() == kPdgPiP ||
              p->Pdg() == kPdgPiM) 
           {
            LOG("myAnalysis", pNOTICE)  
               << "Got a : " << p->Name() << " with E = " << p->E() << " GeV"; 
          }
       }

    }// end loop over particles 
    
    // clear current mc event record
    mcrec->Clear();

  }//end loop over events

  // close input GHEP event file
  file.Close();

  LOG("myAnalysis", pNOTICE)  << "Done!";

  return 0;
}

Variable Documentation

string gOptInpFilename

Definition at line 41 of file gtestEventLoop.cxx.

int gOptNEvt

Definition at line 40 of file gtestEventLoop.cxx.

Referenced by main().