evgen::ldm::NuMiKaonGen Class Reference

NuMiKaonGen class definiton. More...

Inheritance diagram for evgen::ldm::NuMiKaonGen:

Public Member Functions
	NuMiKaonGen (fhicl::ParameterSet const &pset)
	Constructor. More...
	~NuMiKaonGen ()
	Destructor. More...
double	GetPOT () override
simb::MCFlux	GetNext () override
void	configure (const fhicl::ParameterSet &) override
	Interface for configuring the particular algorithm tool. More...
const bsim::Dk2Nu *	GetNextEntry ()
std::vector< std::string >	LoadFluxFiles ()
simb::MCFlux	MakeMCFlux (const bsim::Dk2Nu &dk2nu)
double	LoadPOT ()
double	MaxWeight () override
Public Member Functions inherited from evgen::ldm::IMesonGen
virtual	~IMesonGen () noexcept=default
	Virtual Destructor. More...
Public Member Functions inherited from evgen::ldm::IMeVPrtlStage
virtual	~IMeVPrtlStage () noexcept
	Virtual Destructor. More...
	IMeVPrtlStage (const char *name)
TVector3	RandomUnitVector ()
double	GetRandom ()
const char *	Name ()

Private Attributes
std::string	fSearchPath
std::vector< std::string >	fSearchPatterns
unsigned long	fMaxFluxFileMB
std::string	fFluxCopyMethod
bool	fRandomizeFiles
std::string	fTreeName
std::string	fMetaTreeName
unsigned	fFileIndex
bool	fNewFile
std::vector< std::string >	fFluxFiles
unsigned	fEntry
unsigned	fEntryStart
TTree *	fFluxTree
TFile *	fFluxFile
bsim::Dk2Nu *	fDk2Nu
double	fAccumulatedPOT
double	fThisFilePOT

Additional Inherited Members
Protected Attributes inherited from evgen::ldm::IMeVPrtlStage
CLHEP::HepRandomEngine *	fEngine
const char *	fName

Detailed Description

NuMiKaonGen class definiton.

Definition at line 48 of file NuMiKaonGen_tool.cc.

Constructor & Destructor Documentation

evgen::ldm::NuMiKaonGen::NuMiKaonGen

(

fhicl::ParameterSet const &

pset

)

Constructor.

Definition at line 105 of file NuMiKaonGen_tool.cc.

                                                     :
  IMeVPrtlStage("NuMiKaonGen") 
{
  configure(pset);

  // copy the flux files locally
  fFluxFiles = LoadFluxFiles();

  // setup indices
  fFileIndex = 0;
  fEntry = 0;
  fEntryStart = 0;
  fNewFile = true;
  fFluxTree = NULL;
  fFluxFile = NULL;
  fDk2Nu = new bsim::Dk2Nu;

  fAccumulatedPOT = 0.;
  fThisFilePOT = 0.;
    
}

evgen::ldm::NuMiKaonGen::~NuMiKaonGen

(

)

Destructor.

Definition at line 129 of file NuMiKaonGen_tool.cc.

{

  if (fDk2Nu) delete fDk2Nu;
}

Member Function Documentation

void evgen::ldm::NuMiKaonGen::configure ( const fhicl::ParameterSet &  )

overridevirtual

Interface for configuring the particular algorithm tool.

Parameters

ParameterSet

The input set of parameters for configuration

Implements evgen::ldm::IMeVPrtlStage.

Definition at line 136 of file NuMiKaonGen_tool.cc.

{
  fSearchPath = pset.get<std::string>("SearchPath");
  fSearchPatterns = pset.get<std::vector<std::string>>("FluxFiles");
  fMaxFluxFileMB = pset.get<unsigned long>("MaxFluxFileMB", 2 * 1024);
  fFluxCopyMethod = pset.get<std::string>("FluxCopyMethod", "IFDH");
  fTreeName = pset.get<std::string>("TreeName");
  fMetaTreeName = pset.get<std::string>("MetaTreeName");
  fRandomizeFiles = pset.get<bool>("RandomizeFiles");

  std::cout << "Searching for flux files at path: " << fSearchPath << std::endl;
  std::cout << "With patterns:\n";
  for (const std::string &s: fSearchPatterns) std::cout << s << std::endl;
  std::cout << "With copy method: " << fFluxCopyMethod << std::endl;
}

simb::MCFlux evgen::ldm::NuMiKaonGen::GetNext ( )

overridevirtual

Implements evgen::ldm::IMesonGen.

Definition at line 269 of file NuMiKaonGen_tool.cc.

                                {
  const bsim::Dk2Nu *flux = GetNextEntry();
  return MakeMCFlux(*flux);
}

const bsim::Dk2Nu * evgen::ldm::NuMiKaonGen::GetNextEntry

(

)

Definition at line 227 of file NuMiKaonGen_tool.cc.

                                           {
  // new file -- set the start entry 
  if (fNewFile) {
    // wrap file index around
    if (fFileIndex >= fFluxFiles.size()) {
      fFileIndex = 0;
    }
    // if (fFileIndex >= fFluxFiles.size()) {
    //  throw cet::exception("FluxReader Out of Files", 
    //                       "At file index (" + std::to_string(fFileIndex) + ") of available files (" + std::to_string(fFluxFiles.size()) + ").");
    // }

    std::cout << "New file: " << fFluxFiles[fFileIndex] << " at index: " << fFileIndex << " of: " << fFluxFiles.size() << std::endl;
    if (fFluxFile) delete fFluxFile;
    fFluxFile = new TFile(fFluxFiles[fFileIndex].c_str());
    fFluxTree = (TTree*)fFluxFile->Get(fTreeName.c_str());
    fFluxTree->SetBranchAddress("dk2nu",&fDk2Nu);

    // Start at a random index in this file
    fEntryStart = CLHEP::RandFlat::shootInt(fEngine, fFluxTree->GetEntries()-1);
    fEntry = fEntryStart;

    // load the POT in this file
    fThisFilePOT = LoadPOT();
    fNewFile = false;
  }
  else {
    fEntry = (fEntry + 1) % fFluxTree->GetEntries();
    // if this is the last entry, get ready for the next file
    if ((fEntry + 1) % fFluxTree->GetEntries() == fEntryStart) {
      fFileIndex ++;
      fNewFile = true;
    }
  }

  // count the POT
  fAccumulatedPOT += fThisFilePOT / fFluxTree->GetEntries();
    
  fFluxTree->GetEntry(fEntry);
  return fDk2Nu;
}

double evgen::ldm::NuMiKaonGen::GetPOT ( )

overridevirtual

Implements evgen::ldm::IMesonGen.

Definition at line 221 of file NuMiKaonGen_tool.cc.

                           {
  double ret = fAccumulatedPOT;
  fAccumulatedPOT = 0.;
  return ret;
}

std::vector< std::string > evgen::ldm::NuMiKaonGen::LoadFluxFiles

(

)

Definition at line 152 of file NuMiKaonGen_tool.cc.

                                                {
  art::ServiceHandle<IFDH> ifdhp;

  std::vector<std::pair<std::string, long>> allFiles;

  // find the flux files
  for (unsigned i = 0; i < fSearchPatterns.size(); i++) {
    std::vector<std::pair<std::string, long>> thisList = ifdhp->findMatchingFiles(fSearchPath, fSearchPatterns[i]);
    std::copy (thisList.begin(), thisList.end(), std::back_inserter(allFiles));
  }

  // first randomize the flux files
  std::vector<unsigned long> order(allFiles.size(), 0);
  if (fRandomizeFiles) {
    std::vector<double> rand(allFiles.size(), 0.);
    CLHEP::RandFlat::shootArray(fEngine, rand.size(), &rand[0]);
    TMath::Sort(allFiles.size(), &rand[0], &order[0], false);
  }
  else {
    for (unsigned i = 0; i < order.size(); i++) {
      order[i] = i;
    }
  }

  // If we are directly accessing the files, no need to copy
  if (fFluxCopyMethod == "DIRECT") {
    std::cout << "DIRECTLY ACCESSING FLUX FILES.\n";
    std::vector<std::string> files(allFiles.size());
    for (unsigned i = 0; i < order.size(); i++) {
      files[i] = allFiles[order[i]].first;
    }
    return files;
  }

  // copy over up to the provided limit
  std::vector<std::pair<std::string, long>> selected;
  unsigned long totalBytes = 0;
  unsigned ind = 0;
  while (totalBytes < (fMaxFluxFileMB * 1024 * 1024) && ind < allFiles.size()) {
    selected.push_back(allFiles[order[ind]]);
    totalBytes += allFiles[order[ind]].second;
    ind ++;
  } 

  // copy the files locally
  std::vector<std::pair<std::string, long>> localFiles = ifdhp->fetchSharedFiles(selected, fFluxCopyMethod);

  std::vector<std::string> files(localFiles.size());
  for (unsigned i = 0; i < localFiles.size(); i++) {
    files[i] = localFiles[i].first;
  }

  return files;
}

double evgen::ldm::NuMiKaonGen::LoadPOT

(

)

Definition at line 207 of file NuMiKaonGen_tool.cc.

                            {
  TTreeReader metaReader(fMetaTreeName.c_str(), fFluxFile);
  TTreeReaderValue<double> pot(metaReader, "pots");
  
  double total_pot = 0.;

  while (metaReader.Next()) {
    total_pot += *pot;
  }

  return total_pot;
}

simb::MCFlux evgen::ldm::NuMiKaonGen::MakeMCFlux

(

const bsim::Dk2Nu &

dk2nu

)

Definition at line 274 of file NuMiKaonGen_tool.cc.

                                                         {
  simb::MCFlux flux;
  
  flux.fFluxType = simb::kDk2Nu;
  flux.fntype    = dk2nu.decay.ntype;
  flux.fnimpwt   = dk2nu.decay.nimpwt;
  flux.fvx       = dk2nu.decay.vx;
  flux.fvy       = dk2nu.decay.vy;
  flux.fvz       = dk2nu.decay.vz;
  flux.fpdpx     = dk2nu.decay.pdpx;
  flux.fpdpy     = dk2nu.decay.pdpy;
  flux.fpdpz     = dk2nu.decay.pdpz;
  flux.fppdxdz   = dk2nu.decay.ppdxdz;
  flux.fppdydz   = dk2nu.decay.ppdydz;
  flux.fpppz     = dk2nu.decay.pppz;   
  flux.fppenergy  = dk2nu.decay.ppenergy;
  flux.fppmedium = dk2nu.decay.ppmedium;
  flux.fptype    = dk2nu.decay.ptype;
  flux.fndecay   = dk2nu.decay.ndecay;
  flux.fmuparpx  = dk2nu.decay.muparpx;
  flux.fmuparpy  = dk2nu.decay.muparpy;
  flux.fmuparpz  = dk2nu.decay.muparpz;
  flux.fmupare   = dk2nu.decay.mupare;
  flux.fnecm     = dk2nu.decay.necm;
  
  flux.fppvx = dk2nu.ppvx;
  flux.fppvy = dk2nu.ppvy;
  flux.fppvz = dk2nu.ppvz;
  
  flux.ftvx      = dk2nu.tgtexit.tvx;
  flux.ftvy      = dk2nu.tgtexit.tvy;
  flux.ftvz      = dk2nu.tgtexit.tvz;
  flux.ftpx      = dk2nu.tgtexit.tpx;
  flux.ftpy      = dk2nu.tgtexit.tpy;
  flux.ftpz      = dk2nu.tgtexit.tpz;
  flux.ftptype   = dk2nu.tgtexit.tptype;
  flux.ftgen     = dk2nu.tgtexit.tgen;
  
  flux.frun      = dk2nu.job;
  flux.fevtno    = dk2nu.potnum;
  flux.ftgptype  = dk2nu.ancestor[1].pdg;

  // flux.fnenergyn = flux.fnenergyf = enu;
  // flux.fnwtnear  = flux.fnwtfar = wgt; 
  // ignore variables dealing with the neutrino
  flux.fnenergyn = -1;
  flux.fnwtnear = flux.fnwtfar = -1;
  flux.fdk2gen   = -1;

  // placeholder for time
  flux.fxpoint = dk2nu.ancestor.back().startt;

  return flux;  
}

double evgen::ldm::NuMiKaonGen::MaxWeight ( )

inlineoverridevirtual

Implements evgen::ldm::IMeVPrtlStage.

Definition at line 72 of file NuMiKaonGen_tool.cc.

{ return -1.; }

Member Data Documentation

double evgen::ldm::NuMiKaonGen::fAccumulatedPOT

private

Definition at line 100 of file NuMiKaonGen_tool.cc.

bsim::Dk2Nu* evgen::ldm::NuMiKaonGen::fDk2Nu

private

Definition at line 97 of file NuMiKaonGen_tool.cc.

unsigned evgen::ldm::NuMiKaonGen::fEntry

private

Definition at line 91 of file NuMiKaonGen_tool.cc.

unsigned evgen::ldm::NuMiKaonGen::fEntryStart

private

Definition at line 92 of file NuMiKaonGen_tool.cc.

unsigned evgen::ldm::NuMiKaonGen::fFileIndex

private

Definition at line 86 of file NuMiKaonGen_tool.cc.

std::string evgen::ldm::NuMiKaonGen::fFluxCopyMethod

private

Definition at line 79 of file NuMiKaonGen_tool.cc.

TFile* evgen::ldm::NuMiKaonGen::fFluxFile

private

Definition at line 96 of file NuMiKaonGen_tool.cc.

std::vector<std::string> evgen::ldm::NuMiKaonGen::fFluxFiles

private

Definition at line 88 of file NuMiKaonGen_tool.cc.

TTree* evgen::ldm::NuMiKaonGen::fFluxTree

private

Definition at line 95 of file NuMiKaonGen_tool.cc.

unsigned long evgen::ldm::NuMiKaonGen::fMaxFluxFileMB

private

Definition at line 78 of file NuMiKaonGen_tool.cc.

std::string evgen::ldm::NuMiKaonGen::fMetaTreeName

private

Definition at line 83 of file NuMiKaonGen_tool.cc.

bool evgen::ldm::NuMiKaonGen::fNewFile

private

Definition at line 87 of file NuMiKaonGen_tool.cc.

bool evgen::ldm::NuMiKaonGen::fRandomizeFiles

private

Definition at line 80 of file NuMiKaonGen_tool.cc.

std::string evgen::ldm::NuMiKaonGen::fSearchPath

private

Definition at line 76 of file NuMiKaonGen_tool.cc.

std::vector<std::string> evgen::ldm::NuMiKaonGen::fSearchPatterns

private

Definition at line 77 of file NuMiKaonGen_tool.cc.

double evgen::ldm::NuMiKaonGen::fThisFilePOT

private

Definition at line 101 of file NuMiKaonGen_tool.cc.

std::string evgen::ldm::NuMiKaonGen::fTreeName

private

Definition at line 82 of file NuMiKaonGen_tool.cc.

---

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

• NuMiKaonGen_tool.cc