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 "DQMServices/Core/interface/DQMStore.h"

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

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

class ParticlePropagator;
class RandomEngineAndDistribution;
//class DaqMonitorBEInterface;

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