NuclearInteractionSimulator.h

Go to the documentation of this file.

#ifndef FastSimulation_MaterialEffects_NuclearInteractionSimulator_H
#define FastSimulation_MaterialEffects_NuclearInteractionSimulator_H

#include "FastSimulation/MaterialEffects/interface/MaterialEffectsSimulator.h"

#include <vector>
#include <map>
#include <string>
#include <fstream>

class TFile;
class TTree;
class TBranch;
class NUEvent;

class ParticlePropagator;
class RandomEngineAndDistribution;
//class DaqMonitorBEInterface;
//class MonitorElement;

class NuclearInteractionSimulator : public MaterialEffectsSimulator
{
 public:

  NuclearInteractionSimulator(std::vector<double>& hadronEnergies,
                  std::vector<int>& hadronTypes,
                  std::vector<std::string>& hadronNames,
                  std::vector<double>& hadronMasses,
                  std::vector<double>& hadronPMin,
                  double pionEnergy,
                  std::vector<double>& lengthRatio,
                  std::vector< std::vector<double> >& ratios,
                  std::map<int,int >& idMap,
                  std::string inputFile,
                  unsigned int distAlgo,
                  double distCut);

  ~NuclearInteractionSimulator() override;

  void save() override;

  bool read(std::string inputFile);

 private:

  void compute(ParticlePropagator& Particle, RandomEngineAndDistribution const*) override;

  double distanceToPrimary(const RawParticle& Particle,
               const RawParticle& aDaughter) const;

  unsigned index(int thePid);

  std::vector<double> thePionEN;
  std::vector<int> thePionID;
  std::vector<std::string> thePionNA;
  std::vector<double> thePionMA;
  std::vector<double> thePionPMin;
  double thePionEnergy;
  std::vector<double> theLengthRatio;
  std::vector< std::vector<double> > theRatios;

  TFile* theFile;
  std::vector< std::vector<TTree*> > theTrees;
  std::vector< std::vector<TBranch*> > theBranches;
  std::vector< std::vector<NUEvent*> > theNUEvents;
  std::vector< std::vector<unsigned> > theCurrentEntry;
  std::vector< std::vector<unsigned> > theCurrentInteraction;
  std::vector< std::vector<unsigned> > theNumberOfEntries;
  std::vector< std::vector<unsigned> > theNumberOfInteractions;
  std::vector< std::vector<std::string> > theFileNames;
  std::vector< std::vector<double> > thePionCM;
  
  std::map< int, int > theIDMap;
  unsigned ien4;
  unsigned theDistAlgo;
  double theDistCut;

  std::ofstream myOutputFile;
  unsigned myOutputBuffer;

  bool currentValuesWereSet;

  //  DaqMonitorBEInterface * dbe;
  //  MonitorElement* hAfter;
  //  MonitorElement* hAfter2;
  //  MonitorElement* hAfter3;
  //  MonitorElement* htot;
  //  MonitorElement* helas;
  //  MonitorElement* hinel;
  //  MonitorElement* hscatter;
  //  MonitorElement* hscatter2;

};
#endif