PHG4TpcDirectLaser.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4TpcDirectLaser.h

#ifndef G4TPC_PHG4TPCDIRECTLASER_H
#define G4TPC_PHG4TPCDIRECTLASER_H

#include <fun4all/SubsysReco.h>

#include <phparameter/PHParameterInterface.h>

#include <TVector3.h>
#include <TNtuple.h>

#include <cmath>
#include <string>  // for string, allocator
#include <vector>  // for vector

class PHG4HitContainer;
class SvtxTrackMap;
class PHG4TruthInfoContainer;
class PHCompositeNode;

class PHG4TpcDirectLaser : public SubsysReco, public PHParameterInterface
{
 public:
  PHG4TpcDirectLaser(const std::string &name = "PHG4TpcDirectLaser");

  ~PHG4TpcDirectLaser() override = default;

  int InitRun(PHCompositeNode *) override;

  int process_event(PHCompositeNode *) override;

  void SetDefaultParameters() override;

  void Detector(const std::string &d)
  {
    detector = d;
  }

  void SetPhiStepping(int n, double min, double max);

  void SetThetaStepping(int n, double min, double max);

  void SetFileStepping(int n);


  int GetNpatternSteps() const
  {
    return nPhiSteps * nThetaSteps;
  };

  void SetCurrentPatternStep(int value)
  {
    currentPatternStep = value;
  }

  void SetDirectLaserAuto(bool value)
  {
    m_autoAdvanceDirectLaser = value;
  };

  void SetDirectLaserPatternfromFile(bool value)
  {
    m_steppingpattern = value;
  };

  void SetArbitraryThetaPhi(double theta, double phi)
  {
    arbitrary_theta = theta;
    arbitrary_phi = phi;
  }

 private:
  /* by default there are 4 lasers on each side of the TPC */
  void SetupLasers();

  void AimToThetaPhi(double theta, double phi);

  void AimToPatternStep(int n);

  void AimToPatternStep_File(int n);

  float theta_p, phi_p;
  TNtuple *pattern = nullptr;

  void AimToNextPatternStep();

  class Laser
  {
   public:
    TVector3 m_position;

    double m_phi = 0;

    int m_direction = 1;
  };

  void AppendLaserTrack(double theta, double phi, const Laser &);

  std::string detector = "TPC";

  std::string hitnodename;

  std::vector<Laser> m_lasers;

  int electrons_per_cm = 300;

  // number of electrons per deposited GeV in TPC gas
  double electrons_per_gev = NAN;

  double arbitrary_theta = -30.0;  // degrees
  double arbitrary_phi = -30.0;    // degrees


  int nPhiSteps = 1;
  int nThetaSteps = 1;
  int nTotalSteps = 1;
  double minPhi = 0;
  double maxPhi = 0;
  double minTheta = 0;
  double maxTheta = 0;

  // current patter step
  int currentPatternStep = 0;

  bool m_autoAdvanceDirectLaser = false;

  bool m_steppingpattern = false;


  PHG4HitContainer *m_g4hitcontainer = nullptr;

  PHG4TruthInfoContainer *m_g4truthinfo = nullptr;

  std::string m_track_map_name = "SvtxTrackMap";
  SvtxTrackMap *m_track_map = nullptr;
};

#endif