FastSimulation/MaterialEffects/interface/NuclearInteractionSimulator.h - master - githubmirror/cmssw

FastSimulation/MaterialEffects/interface/NuclearInteractionSimulator.h
100 строк3 KB
Kevin Pedro
disable nuclear interaction output file by default, and save it at the correct step in the new propagator
06 дек 2025, 00:42
06 дек 2025, 00:427b3aa6f
#ifndef FastSimulation_MaterialEffects_NuclearInteractionSimulator_H
#define FastSimulation_MaterialEffects_NuclearInteractionSimulator_H

/** 
 * This class computes the probability for hadrons to interact with a 
 * nucleon of the tracker material (inelastically) and then reads a 
 * nuclear interaction randomly from multiple fully simulated files 
 * The fraction of interaction lengths traversed by the particle in this 
 * tracker layer is determined in MaterialEffectsSimulator as a fraction 
 * the radiation lengths. 
 *
 * \author Patrick Janot
 * $Date: 25-Jan-2007
 */

#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:
  /// Constructor
  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,
                              bool saveOutput);

  /// Default Destructor
  ~NuclearInteractionSimulator() override;

  /// Save current nuclear interaction (for later use)
  void save() override;

  /// Read former nuclear interaction (from previous run)
  bool read(std::string inputFile);

private:
  /// Generate a nuclear interaction according to the probability that it happens
  void compute(ParticlePropagator& Particle, RandomEngineAndDistribution const*) override;

  /// Compute distance between secondary and primary
  double distanceToPrimary(const RawParticle& Particle, const RawParticle& aDaughter) const;

  /// Return a hashed index for a given pid
  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;

  bool mySaveOutput;
  std::ofstream myOutputFile;
  unsigned myOutputBuffer;

  bool currentValuesWereSet;
};
#endif