GFluxFileConfigI.h

Go to the documentation of this file.

//____________________________________________________________________________
/*!
 
\class   genie::flux::GFluxFileConfigI
 
\brief
 
\author  Robert Hatcher <rhatcher \at fnal.gov>
         Fermi National Accelerator Laboratory
 
\created 2015-03-17
 
\cpright Copyright (c) 2003-2025, The GENIE Collaboration
         for the full text of the license visit http://copyright.genie-mc.org
*/
//____________________________________________________________________________
 
#ifndef GENIE_FLUX_GFLUXFILECONFIGI_H
#define GENIE_FLUX_GFLUXFILECONFIGI_H
 
#include <string>
#include <vector>
#include <set>
 
#include "Framework/ParticleData/PDGCodeList.h"
class TTree;
 
namespace genie {
namespace flux {
 
  class GFluxFileConfigI {
 
  public:
 
    GFluxFileConfigI();
    virtual ~GFluxFileConfigI();
 
    //
    // define the GFluxFileConfigI interface:
    //
 
    /// first is primary method for loading root flux ntuple files and config
    /// others are alternatives that can be overloaded but have
    /// sensible defaults to fall back to calling the vector version
 
    virtual void  LoadBeamSimData(const std::vector<std::string>& filenames,
                                  const std::string&              det_loc) = 0;
 
    virtual void  LoadBeamSimData(const std::set<std::string>&    filenames,
                                  const std::string&              det_loc);
 
    virtual void  LoadBeamSimData(const std::string&              filename,
                                  const std::string&              det_loc);
 
    virtual void         SetXMLFileBase(std::string xmlbasename="");
    virtual std::string  GetXMLFileBase() const { return fXMLbasename; }
 
    /// allow caller to copy current status / ntuple entry info
    /// in the output file by providing copies of internal info
    ///
    /// Assumes that branch object pointers will not change
    /// which may require either a copy be made or, if using the class
    /// directly for reading the branch, one must force ROOT to
    /// not autodelete:
    ///   myns::MyClassType* fCurrMyClass = new myns::MyClassType;
    ///   myTree->SetBranchAddress("bname",&fCurMyClass);
    ///   //? TBranch* b = myTree->GetBranch("bname");
    ///   //? b->SetAutoDelete(false);
    ///
    /// ensure vectors are sized sufficiently (or use .push_back())
    ///  branchNames[i]       = "bname"
    ///  branchClassNames[i]  = "myns::MyClassType"
    ///  branchObjPointers[i] = (void**)&fCurMyClass;
 
    virtual void  GetBranchInfo(std::vector<std::string>& branchNames,
                                std::vector<std::string>& branchClassNames,
                                std::vector<void**>&      branchObjPointers);
 
    virtual TTree* GetMetaDataTree(); //
 
    /// print the current configuration
    virtual void         PrintConfig() = 0;
 
    /// specify list of flux neutrino species
    virtual void         SetFluxParticles(const PDGCodeList & particles);
 
    /// set flux neutrino initial z position (upstream of the detector)
    /// pushed back from the normal flux window
    virtual void         SetUpstreamZ(double z0);
 
    /// limit cycling through input files
    virtual void         SetNumOfCycles(long int ncycle);
 
  protected:  // visible to derived classes
 
    PDGCodeList * fPdgCList;     ///< list of neutrino pdg-codes to generate
    PDGCodeList * fPdgCListRej;  ///< list of nu pdg-codes seen but rejected
    std::string   fXMLbasename;  ///< XML file that might hold config param_sets
    long int      fNCycles;      ///< # times to cycle through the ntuple(s)
    long int      fICycle;       ///< current file cycle
                                 ///< default 0 = infinitely
    double        fZ0;           ///< configurable starting z position for
                                 ///< each flux neutrino (in detector coord system)
  };
 
} // namespace flux
} // namespace genie
 
#endif //GENIE_FLUX_GFLUXFILECONFIGI_H