AtSimpleSimulation.h

Go to the documentation of this file.

#ifndef AT_SIMPLE_SIMULATION_H
#define AT_SIMPLE_SIMULATION_H
 
#include "AtMCPoint.h"
 
#include <Math/Point3D.h>
#include <Math/Point3Dfwd.h> // for XYZPoint
#include <Math/Vector3D.h>
#include <Math/Vector3Dfwd.h> // for XYZVector
#include <Math/Vector4D.h>
#include <Math/Vector4Dfwd.h> // for PxPyPzEVector
#include <TClonesArray.h>
#include <TObject.h>
 
#include <map>
#include <memory>
#include <mutex>
#include <string> // for string
namespace AtTools {
class AtELossModel;
}
class TGeoVolume;
class AtSpaceChargeModel;
 
class AtSimpleSimulation {
protected:
   struct ParticleID {
      int A;
      int Z;
 
      bool operator<(const ParticleID &other) const;
   };
   using SpaceChargeModel = std::shared_ptr<AtSpaceChargeModel>;
   using ModelPtr = std::shared_ptr<AtTools::AtELossModel>;
   using XYZPoint = ROOT::Math::XYZPoint;
   using XYZVector = ROOT::Math::XYZVector;
   using PxPyPzEVector = ROOT::Math::PxPyPzEVector;
 
   std::map<ParticleID, ModelPtr> fModels;
   SpaceChargeModel fSCModel{nullptr};
   double fDistStep{1.}; // Distance step in mm for particles
   std::mutex fGeoMutex;
 
   // Variables to across an entire event
   static thread_local int fTrackID;
   static thread_local TClonesArray fMCPoints;
 
public:
   AtSimpleSimulation(std::string geoFile);
   AtSimpleSimulation();
   AtSimpleSimulation(const AtSimpleSimulation &other) = default;
   ~AtSimpleSimulation() = default;
 
   void RegisterBranch(std::string branchName = "AtTpcPoint", bool pers = true);
   void AddModel(int Z, int A, ModelPtr model);
   void SetSpaceChargeModel(SpaceChargeModel model) { fSCModel = model; }
   void SetDistanceStep(double step) { fDistStep = step; } //<In mm
 
   void NewEvent();
   void SimulateParticle(int Z, int A, const XYZPoint &iniPos, const PxPyPzEVector &iniMom);
 
   AtMCPoint &GetMcPoint(int i) { return dynamic_cast<AtMCPoint &>(*fMCPoints.At(i)); }
   int GetNumPoints() { return fMCPoints.GetEntries(); }
   TClonesArray &GetPointsArray() { return fMCPoints; }
   SpaceChargeModel GetSpaceChargeModel() { return fSCModel; }
 
protected:
   bool IsInVolume(const std::string &volName, const XYZPoint &point);
   std::string GetVolumeName(const XYZPoint &point);
 
   void SimulateParticle(ModelPtr model, const XYZPoint &iniPos, const PxPyPzEVector &iniMom);
   void AddHit(double ELoss, const XYZPoint &pos, const PxPyPzEVector &mom, double length);
   TGeoVolume *GetVolume(const XYZPoint &pos);
};
 
#endif // AT_SIMPLE_SIMULATION_H