evgendp::NEUTImport Class Reference

Inheritance diagram for evgendp::NEUTImport:

Classes
struct	Config

Public Types
using	Parameters = art::EDProducer::Table< Config >
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer
using	WorkerType = WorkerT< EDProducer >
Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >
Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >

Public Member Functions
	NEUTImport (Parameters const &config)
	NEUTImport (NEUTImport const &)=delete
	NEUTImport (NEUTImport &&)=delete
NEUTImport &	operator= (NEUTImport const &)=delete
NEUTImport &	operator= (NEUTImport &&)=delete
void	produce (art::Event &e) override
void	beginJob () override
Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDProducer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept
	Producer (fhicl::ParameterSet const &)
	Producer (Producer const &)=delete
	Producer (Producer &&)=delete
Producer &	operator= (Producer const &)=delete
Producer &	operator= (Producer &&)=delete
void	doBeginJob (SharedResources const &resources)
void	doEndJob ()
void	doRespondToOpenInputFile (FileBlock const &fb)
void	doRespondToCloseInputFile (FileBlock const &fb)
void	doRespondToOpenOutputFiles (FileBlock const &fb)
void	doRespondToCloseOutputFiles (FileBlock const &fb)
bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)
Public Member Functions inherited from art::Modifier
	~Modifier () noexcept
	Modifier ()
	Modifier (Modifier const &)=delete
	Modifier (Modifier &&)=delete
Modifier &	operator= (Modifier const &)=delete
Modifier &	operator= (Modifier &&)=delete
Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept
	ModuleBase ()
ModuleDescription const &	moduleDescription () const
void	setModuleDescription (ModuleDescription const &)
std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const
void	sortConsumables (std::string const &current_process_name)
template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)
template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Public Attributes
int	fLogLevel
int	fStartEvent
int	fNumberOfEvents
double	fStartPositionX
double	fStartPositionY
double	fStartPositionZ
std::string	fFileName
bool	fUseRotatedStartMomentum
std::map< int, std::vector< RooTrackerParticle * > >	RooTrackerEventMap

Additional Inherited Members
Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)
Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()
template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)
template<typename T , BranchType = InEvent>
void	consumesMany ()
template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)
template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Definition at line 95 of file NEUTImport_module.cc.

Member Typedef Documentation

using evgendp::NEUTImport::Parameters = art::EDProducer::Table<Config>

Definition at line 146 of file NEUTImport_module.cc.

Constructor & Destructor Documentation

evgendp::NEUTImport::NEUTImport ( Parameters const &  config )

explicit

Definition at line 177 of file NEUTImport_module.cc.

 : EDProducer{config},
   fLogLevel                    (config().LogLevel()),
   fStartEvent                  (config().StartEvent()),
   fNumberOfEvents              (config().NumberOfEvents()),
   fStartPositionX              (config().StartPositionX()),
   fStartPositionY              (config().StartPositionY()),
   fStartPositionZ              (config().StartPositionZ()),
   fFileName                    (config().FileName()),
   fUseRotatedStartMomentum     (config().UseRotatedStartMomentum())
{
    produces< std::vector<simb::MCTruth> >();
    produces< sumdata::RunData, art::InRun >();
}

evgendp::NEUTImport::NEUTImport ( NEUTImport const &  )

delete

evgendp::NEUTImport::NEUTImport ( NEUTImport &&  )

delete

Member Function Documentation

void evgendp::NEUTImport::beginJob ( )

overridevirtual

Reimplemented from art::EDProducer.

Definition at line 196 of file NEUTImport_module.cc.

                                {

  const int NMaxParticlesPerEvent = 1000;

  int tStdHepN = 0;
  int tStdHepStatus[NMaxParticlesPerEvent] = {0};
  int tStdHepPdg[NMaxParticlesPerEvent] = {0};
  double tStdHepP4[4*NMaxParticlesPerEvent] = {0.};


  TFile *neutFile = new TFile(fFileName.c_str(), "READ");
  TTree *neutTree = (TTree*)neutFile->Get("nRooTracker");
  int NEvents = (int)neutTree->GetEntries();
  std::cout << "Reading in " << NEvents << " NEUT events. " << std::endl;

  neutTree->SetBranchAddress("StdHepN",&tStdHepN);
  neutTree->SetBranchAddress("StdHepStatus",&tStdHepStatus);
  neutTree->SetBranchAddress("StdHepPdg",&tStdHepPdg);

  TTree *neutTreeRot = NULL;

  if(fUseRotatedStartMomentum)
  {
    neutTreeRot = (TTree*)neutFile->Get("nRooTrackerRot");
    neutTreeRot->SetBranchAddress("StdHepP4Rot",&tStdHepP4);
  }
  else
  {
    neutTree->SetBranchAddress("StdHepP4",&tStdHepP4);
  }


  for(int i=fStartEvent; i<fStartEvent+fNumberOfEvents; i++) //Event loop
  {
    neutTree->GetEntry(i);
    if(fUseRotatedStartMomentum)
    {
      neutTreeRot->GetEntry(i);
    }

    if(fLogLevel == 1)
    {
      std::cout << std::endl;
      std::cout << std::endl;
      std::cout << "Event #" << i+1 << " in LArSoft, event #" << i << " in ROOT file." << std::endl;
      std::cout << "NParticles:\t" << tStdHepN << " (from 0 to " << tStdHepN-1 << ")" << std::endl;
    }

    for(int j=0; j<tStdHepN; j++)
    {
      double fAbsoluteParticleMomentum = sqrt( pow(tStdHepP4[4*j],2) + pow(tStdHepP4[4*j+1],2) + pow(tStdHepP4[4*j+2],2) );
      //Only take particles with real PDG code (!=0), status 1 and momentum > 0.
      //Also ignore first particle (j=0): this is the decaying nucleon in case of nucleon decay or incoming neutrino in case of atmospheric neutrino background.
      if( tStdHepPdg[j] != 0 && tStdHepStatus[j] == 1 && j > 0 && fAbsoluteParticleMomentum > 0 )
      {
        static TDatabasePDG  pdgt;
        TParticlePDG* pdgp = pdgt.GetParticle(tStdHepPdg[j]);

        RooTrackerParticle *rootrackerparticle = new RooTrackerParticle();
        rootrackerparticle->pdg = tStdHepPdg[j];
        rootrackerparticle->mass = pdgp->Mass();
        rootrackerparticle->startX = fStartPositionX;
        rootrackerparticle->startY = fStartPositionY;
        rootrackerparticle->startZ = fStartPositionZ;
        rootrackerparticle->momX = tStdHepP4[4*j];
        rootrackerparticle->momY = tStdHepP4[4*j+1];
        rootrackerparticle->momZ = tStdHepP4[4*j+2];
        rootrackerparticle->energy = tStdHepP4[4*j+3];
        RooTrackerEventMap[i-fStartEvent].push_back(rootrackerparticle);

        if(fLogLevel == 1)
        {
          std::cout << std::endl;
          std::cout << "Particle " << j << ":" << std::endl;
          std::cout << "Status particle " << j << ":\t" << tStdHepStatus[j] << std::endl;
          std::cout << "PDG particle " << j << ":\t" << rootrackerparticle->pdg << std::endl;
          std::cout << "Mass particle " << j << ":\t" << rootrackerparticle->mass << std::endl;
          std::cout << "StartPositionX particle " << j << ":\t" << rootrackerparticle->startX << std::endl;
          std::cout << "StartPositionY particle " << j << ":\t" <<rootrackerparticle->startY << std::endl;
          std::cout << "StartPositionZ particle " << j << ":\t" << rootrackerparticle->startZ << std::endl;
          std::cout << "StartMomentumX particle " << j << ":\t" << rootrackerparticle->momX << std::endl;
          std::cout << "StartMomentumY particle " << j << ":\t" <<rootrackerparticle->momY << std::endl;
          std::cout << "StartMomentumZ particle " << j << ":\t" << rootrackerparticle->momZ << std::endl;
          std::cout << "StartEnergy particle " << j << ":\t" << tStdHepP4[4*j+3] << std::endl;
          std::cout << "Absolute momentum particle " << j << ":\t" << fAbsoluteParticleMomentum << std::endl;
        }
      }
    }
  }
}

NEUTImport& evgendp::NEUTImport::operator= ( NEUTImport const &  )

delete

NEUTImport& evgendp::NEUTImport::operator= ( NEUTImport &&  )

delete

void evgendp::NEUTImport::produce ( art::Event &  e )

overridevirtual

Implements art::EDProducer.

Definition at line 289 of file NEUTImport_module.cc.

                                           {
  std::unique_ptr< std::vector<simb::MCTruth> > truthcol(new std::vector<simb::MCTruth>);

  simb::MCTruth truth;
  truth.SetOrigin(simb::kUnknown);

  //read the particle map and create the MCParticle
  int rootrackerparticlecounter=0;
  for(RooTrackerParticle* rootrackerparticle : RooTrackerEventMap[e.id().event()-1] ){ //-1: art events start at 1, events in EventMap start at 0.

    simb::MCParticle particle( rootrackerparticlecounter, rootrackerparticle->pdg ,"primary",-200, rootrackerparticle->mass , 1 );
    particle.AddTrajectoryPoint( rootrackerparticle->getPosition() , rootrackerparticle->getMomentum() );

    truth.Add(particle);

    rootrackerparticlecounter++;
  }

  truthcol->push_back(truth);
  e.put(std::move(truthcol));

}

Member Data Documentation

std::string evgendp::NEUTImport::fFileName

Definition at line 166 of file NEUTImport_module.cc.

int evgendp::NEUTImport::fLogLevel

Definition at line 160 of file NEUTImport_module.cc.

int evgendp::NEUTImport::fNumberOfEvents

Definition at line 162 of file NEUTImport_module.cc.

int evgendp::NEUTImport::fStartEvent

Definition at line 161 of file NEUTImport_module.cc.

double evgendp::NEUTImport::fStartPositionX

Definition at line 163 of file NEUTImport_module.cc.

double evgendp::NEUTImport::fStartPositionY

Definition at line 164 of file NEUTImport_module.cc.

double evgendp::NEUTImport::fStartPositionZ

Definition at line 165 of file NEUTImport_module.cc.

bool evgendp::NEUTImport::fUseRotatedStartMomentum

Definition at line 167 of file NEUTImport_module.cc.

std::map< int, std::vector< RooTrackerParticle*> > evgendp::NEUTImport::RooTrackerEventMap

Definition at line 169 of file NEUTImport_module.cc.

---

The documentation for this class was generated from the following file:

• dunesim/dunesim/EventGenerator/NEUT/NEUTImport_module.cc