PndBoxGenerator.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

// -------------------------------------------------------------------------
// -----                PndBoxGenerator header file                    -----
// -----               Created 22/05/07  by S.Spataro                  -----
// -------------------------------------------------------------------------

#ifndef PND_BOXGENERATOR_H
#define PND_BOXGENERATOR_H

#include "PndTargetGenerator.h"
#include <iostream>

class FairPrimaryGenerator;

class PndBoxGenerator : public PndTargetGenerator {
 public:
  PndBoxGenerator();

  PndBoxGenerator(Int_t pdgid, Int_t mult = 1);

  virtual ~PndBoxGenerator(){};

  void SetPDGType(Int_t pdg) { fPDGType = pdg; };

  void SetMultiplicity(Int_t mult) { fMult = mult; };

  void SetPRange(Double32_t pmin = 0, Double32_t pmax = 10)
  {
    fPMin = pmin;
    fPMax = pmax;
    fPRangeIsSet = kTRUE;
  }

  void SetPtRange(Double32_t ptmin = 0, Double32_t ptmax = 10)
  {
    fPtMin = ptmin;
    fPtMax = ptmax;
    fPtRangeIsSet = kTRUE;
  };

  void SetPhiRange(Double32_t phimin = 0, Double32_t phimax = 360)
  {
    fPhiMin = phimin;
    fPhiMax = phimax;
  };

  void SetEtaRange(Double32_t etamin = -5, Double32_t etamax = 7)
  {
    fEtaMin = etamin;
    fEtaMax = etamax;
    fEtaRangeIsSet = kTRUE;
  };

  void SetYRange(Double32_t ymin = -5, Double32_t ymax = 7)
  {
    fYMin = ymin;
    fYMax = ymax;
    fYRangeIsSet = kTRUE;
  };

  void SetThetaRange(Double32_t thetamin = 0, Double32_t thetamax = 90)
  {
    fThetaMin = thetamin;
    fThetaMax = thetamax;
    fThetaRangeIsSet = kTRUE;
  };

  void SetCosTheta() { fCosThetaIsSet = kTRUE; };

  void SetInverseP() { fInversePIsSet = kTRUE; };

  void SetXYZ(Double32_t x = 0, Double32_t y = 0, Double32_t z = 0)
  {
    fX = x;
    fY = y;
    fZ = z;
    fPointVtxIsSet = kTRUE;
  }

  void SetBoxXYZ(Double32_t x1 = 0, Double32_t y1 = 0, Double32_t x2 = 0, Double32_t y2 = 0, Double32_t z = 0)
  {
    fX1 = x1;
    fY1 = y1;
    fX2 = x2;
    fY2 = y2;
    fZ = z;
    fBoxVtxIsSet = kTRUE;
  }

  void SetDebug(Bool_t debug = 0) { fDebug = debug; }

  Bool_t Init();

  virtual Bool_t ReadEvent(FairPrimaryGenerator *primGen);

  friend std::ostream& operator<< (std::ostream& out, PndBoxGenerator& gen)
  {
      out << "BoxGenerator settings:" << std::endl;
      out << gen.fMult << " * " << gen.fPDGType << std::endl;
      if (gen.fPRangeIsSet) out << "P Range: " << gen.fPMin << " < " << gen.fPMax << std::endl;
      if (gen.fPtRangeIsSet) out << "Pt Range: " << gen.fPtMin << " < " << gen.fPtMax << std::endl;
      if (gen.fInversePIsSet) out << "InverseP is set" << std::endl;
      if (gen.fThetaRangeIsSet) out << "Theta Range: " << gen.fThetaMin << " < " << gen.fThetaMax << std::endl;
      if (gen.fCosThetaIsSet) out << "CosTheta is set" << std::endl;
      if (gen.fEtaRangeIsSet) out << "Eta Range: " << gen.fEtaMin << " < " << gen.fEtaMax << std::endl;
      if (gen.fYRangeIsSet) out << "Y Range: " << gen.fYMin << " < " << gen.fYMax << std::endl;
      out << "Phi Range: " << gen.fPhiMin << " < " << gen.fPhiMax << std::endl;
      if (gen.fPointVtxIsSet) out << "Point Vertex: " << gen.fX << "/" << gen.fY << "/" << gen.fZ << std::endl;
      if (gen.fBoxVtxIsSet) out << "Box Vertex: p1: " << gen.fX1 << "/" << gen.fY1 << " p2: " << gen.fX2 << "/" << gen.fY2 << " z " << gen.fZ << std::endl;

      return out;

  }

 private:
  Int_t fPDGType; // Particle type (PDG encoding)
  Int_t fMult;    // Multiplicity

  Double32_t fPDGMass;             // Particle mass [GeV]
  Double32_t fPtMin, fPtMax;       // Transverse momentum range [GeV]
  Double32_t fPhiMin, fPhiMax;     // Azimuth angle range [degree]
  Double32_t fEtaMin, fEtaMax;     // Pseudorapidity range in lab system
  Double32_t fYMin, fYMax;         // Rapidity range in lab system
  Double32_t fPMin, fPMax;         // Momentum range in lab system
  Double32_t fThetaMin, fThetaMax; // Polar angle range in lab system [degree]
  Double32_t fX, fY, fZ;           // Point vertex coordinates [cm]
  Double32_t fX1, fY1, fX2, fY2;   // Box vertex coords (x1,y1)->(x2,y2)

  Bool_t fEtaRangeIsSet;   // True if eta range is set
  Bool_t fYRangeIsSet;     // True if rapidity range is set
  Bool_t fThetaRangeIsSet; // True if theta range is set
  Bool_t fCosThetaIsSet;   // True if uniform distribution in cos(theta) is set (default -> uniform theta)
  Bool_t fInversePIsSet;   // True if uniform distribution in 1/p
  Bool_t fPtRangeIsSet;    // True if transverse momentum range is set
  Bool_t fPRangeIsSet;     // True if abs.momentum range is set
  Bool_t fPointVtxIsSet;   // True if point vertex is set
  Bool_t fBoxVtxIsSet;     // True if box vertex is set
  Bool_t fDebug;           // Debug switch

  ClassDef(PndBoxGenerator, 1);
};

#endif