SQPrimaryVertexGen.h

Go to the documentation of this file.

/*==========================================================================
  Author: Abinash Pun, Kun Liu
  Sep, 2019
  Goal: Import the primary vertex generator of E906 experiment(DPVertexGenerator)
  from Kun to E1039 experiment in Fun4All framework
  ============================================================================*/
 
#ifndef __SQPRIMARYVERTEXGEN_H__
#define __SQPRIMARYVERTEXGEN_H__
 
#include <vector>
#include <memory>
#include <TString.h>
#include <TGeoManager.h>
#include <TF2.h>
#include <TVector3.h>
#include <TRandom1.h>
#include "SQBeamlineObject.h"
 
class PHCompositeNode;
 
 
class MaterialProfile
{
public:
  MaterialProfile();
  int findInteractingPiece(double rndm);
  void calcProb();
 
public:
  unsigned int nPieces;
  double probSum;
  double accumulatedProbs[100];
  std::vector<SQBeamlineObject> interactables;
};
 
 
class SQPrimaryVertexGen
{
  static constexpr double Z_MIN = -800.0;
  static constexpr double Z_MAX =  503.0;
 
public:
  SQPrimaryVertexGen();
  ~SQPrimaryVertexGen();
 
  int InitRun(PHCompositeNode* node);
  int InitRun(TString filename);
 
  void fillMaterialProfile(MaterialProfile* prof, double xvtx, double yvtx);
 
  TVector3 generateVertex();
 
  void generateVtxPerp(double& x, double& y);
 
  double getPARatio() { return pARatio; }
 
  double getLuminosity() { return probSum; }
 
  static double funcBeamProfile(double* val, double* par);
  
private:
  void init();
 
  void FindMaterialBoundaries(const double xpos, const double ypos, const double z_min, const double z_max, int& n_b, double* z_b);
 
  void FindMaterialRange(double& z1, double& z2, const std::string name, const double xpos=0, const double ypos=0);
 
  MaterialProfile* defaultMatProf;
  
  double pARatio;
  double probSum;
 
  TF2* beamProfile;
 
  TGeoManager* geoManager;
 
  TRandom1 rndm;
 
  PHCompositeNode* topNode;
 
  bool inited;
  
  double targetX0;
  double targetY0;
  double targetSX;
  double targetSY;
 
  bool targetOnlyMode;
  std::string material_mode;
 
  double z_start;
  double z_stop;
};
 
#endif