evgendp::DataGen311 Class Reference

Inheritance diagram for evgendp::DataGen311:

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
	DataGen311 (Parameters const &config)
	DataGen311 (DataGen311 const &)=delete
	DataGen311 (DataGen311 &&)=delete
DataGen311 &	operator= (DataGen311 const &)=delete
DataGen311 &	operator= (DataGen311 &&)=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	fEventsToProcess
int	fStartEvent
int	fPDG
bool	fGetEnergyFromCORSIKA
double	fUniformEnergyMin
double	fUniformEnergyMax
std::string	fTrackFile
std::string	fHistFile
std::map< int, std::vector< Track * > >	fEventTrackMap

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 116 of file DataGen311_module.cc.

Member Typedef Documentation

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

Definition at line 167 of file DataGen311_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 198 of file DataGen311_module.cc.

 : EDProducer{config},
   fEventsToProcess             (config().EventsToProcess()),
   fStartEvent                  (config().StartEvent()),
   fPDG                         (config().PDG()),
   fGetEnergyFromCORSIKA        (config().GetEnergyFromCORSIKA()),
   fUniformEnergyMin            (config().UniformEnergyMin()),
   fUniformEnergyMax            (config().UniformEnergyMax()),
   fTrackFile                   (config().TrackFile()),
   fHistFile                    (config().HistFile())
{
    produces< std::vector<simb::MCTruth> >();
    produces< sumdata::RunData, art::InRun >();
}

evgendp::DataGen311::DataGen311 ( DataGen311 const &  )

delete

evgendp::DataGen311::DataGen311 ( DataGen311 &&  )

delete

Member Function Documentation

void evgendp::DataGen311::beginJob ( )

overridevirtual

Reimplemented from art::EDProducer.

Definition at line 217 of file DataGen311_module.cc.

                                {

  //Random number generator for particle energy
  gRandom = new TRandom3();

  //read the files and store the information on the map
  std::ifstream trackfile;
  trackfile.open( fTrackFile );
  if( !trackfile.is_open() )
    throw cet::exception("DataGen311") << "Can't open file " << fTrackFile <<"\n";

  TFile *chistfile = new TFile(fHistFile.c_str());
  if( chistfile->IsOpen() == kFALSE )
    throw cet::exception("DataGen311") << "Can't open file " << fHistFile <<"\n";

  std::stringstream hetss;
  hetss << "hEnergyTheta" << fPDG;
  TH2D *hEnergyTheta = (TH2D*)chistfile->Get(hetss.str().c_str());

  int eventBefore=0;
  int eventCounter=0;
  int trackCounter=0, skippedTrackCounter=0;
  std::string line;
  std::getline(trackfile, line); // Skip header
  while(std::getline( trackfile, line )){
    std::stringstream sstream(line);

    int eventID = 9999999;
    sstream >> eventID;
    if (eventID < fStartEvent) {
      continue;
    } else if (eventID >= fStartEvent+fEventsToProcess) {
      break;
    }

    Track *track = new Track();
    sstream >> track->run >> track->subrun >> track->event >> track->trackID >> track->startDirectionX >> track->startDirectionY >> track->startDirectionZ >> track->length >> track->startX >> track->startY >> track->startZ;
    track->pdg = fPDG;
    trackCounter++;

    // Start direction components might not be perfectly normalized
    double sdirmag = sqrt(track->startDirectionX*track->startDirectionX + track->startDirectionY*track->startDirectionY + track->startDirectionZ*track->startDirectionZ);
    track->startDirectionX /= sdirmag;
    track->startDirectionY /= sdirmag;
    track->startDirectionZ /= sdirmag;


    if(fGetEnergyFromCORSIKA)
    {
      // Theta from +X to -X
      double theta = acos(track->startDirectionX);

      int biny = hEnergyTheta->GetYaxis()->FindBin(theta);
      TH1D *hEnergyAtTheta = hEnergyTheta->ProjectionX("", biny, biny);

      if( hEnergyAtTheta->GetEntries() < 1 ){
        skippedTrackCounter++;
        delete track;
        continue;
      }

      track->energy = hEnergyAtTheta->GetRandom();
    }
    else if(fUniformEnergyMin == fUniformEnergyMax) //get energy from .fcl parameter
    {
      track->energy = fUniformEnergyMin;
    }  
    else //get energy from .fcl parameter
    {
      track->energy = gRandom->Uniform(fUniformEnergyMin, fUniformEnergyMax);
    }

    if( track->event != eventBefore ){
      eventBefore = track->event;
      eventCounter++;
    }

    //push back in map
    fEventTrackMap[eventCounter].push_back(track);
  }

  trackfile.close();

  //initialize randomn number seed for

}

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

delete

DataGen311& evgendp::DataGen311::operator= ( DataGen311 &&  )

delete

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

overridevirtual

Implements art::EDProducer.

Definition at line 306 of file DataGen311_module.cc.

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

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

  //read the particle map and create the particle
  int trackCtr=0;
  for(Track* track : fEventTrackMap[e.id().event()] ){

    simb::MCParticle particle( trackCtr, track->pdg ,"primary",-200, track->m , 1 );
    particle.AddTrajectoryPoint( track->getPosition() , track->getMomentum() );

    truth.Add(particle);

    trackCtr++;
  }

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

}

Member Data Documentation

int evgendp::DataGen311::fEventsToProcess

Definition at line 181 of file DataGen311_module.cc.

std::map< int, std::vector< Track*> > evgendp::DataGen311::fEventTrackMap

Definition at line 190 of file DataGen311_module.cc.

bool evgendp::DataGen311::fGetEnergyFromCORSIKA

Definition at line 184 of file DataGen311_module.cc.

std::string evgendp::DataGen311::fHistFile

Definition at line 188 of file DataGen311_module.cc.

int evgendp::DataGen311::fPDG

Definition at line 183 of file DataGen311_module.cc.

int evgendp::DataGen311::fStartEvent

Definition at line 182 of file DataGen311_module.cc.

std::string evgendp::DataGen311::fTrackFile

Definition at line 187 of file DataGen311_module.cc.

double evgendp::DataGen311::fUniformEnergyMax

Definition at line 186 of file DataGen311_module.cc.

double evgendp::DataGen311::fUniformEnergyMin

Definition at line 185 of file DataGen311_module.cc.

---

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

• dunesim/dunesim/EventGenerator/CORSIKAprotodunedp/DataGen311_module.cc