PndMCTrack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      PndMCTrack header file                   -----
// -----                  Created 03/08/04  by V. Friese               -----
// -----                  Adapted to Panda  M. Al-Turany               -----
// -------------------------------------------------------------------------

#ifndef CBMMCTRACK_H
#define CBMMCTRACK_H 1

#include "PndDetectorList.h"
#include "TObject.h"
#include "TParticle.h"
#include "TVector3.h"
#include "TLorentzVector.h"

#include <iostream>

#ifndef ROOT_TParticlePDG
#include "TParticlePDG.h"
#endif
#ifndef ROOT_TDatabasePDG
#include "TDatabasePDG.h"
#endif

class PndMCTrack : public TObject {

 public:
  PndMCTrack();

  //  PndMCTrack(Int_t pdgCode, Int_t motherID, TVector3 startVvertex,
  //         Double_t startTime, TLorentzVector momentum, Int_t  nPoint=0);

  PndMCTrack(const PndMCTrack &track);

  PndMCTrack(TParticle *particle);

  virtual ~PndMCTrack();

  void Print(Int_t iTrack = 0) const;

  friend std::ostream &operator<<(std::ostream &out, PndMCTrack &track)
  {
    track.Print();
    return out;
  }

  Int_t GetPdgCode() const { return fPdgCode; }
  Int_t GetMotherID() const { return fMotherID; }
  Int_t GetSecondMotherID() const { return fSecondMotherID; }
  TVector3 GetStartVertex() const { return TVector3(fStartX, fStartY, fStartZ); }
  Double_t GetStartTime() const { return fStartT; }
  TVector3 GetMomentum() const { return TVector3(fPx, fPy, fPz); }
  Double_t GetPt() const { return TMath::Sqrt(fPx * fPx + fPy * fPy); }
  /*
  Int_t GetStsPoints()  const { return   (fPoints &  15        )        ; }
  */

  Bool_t IsGeneratorCreated(void) const { return (fGeneratorFlags & 0x1) != 0; }
  Bool_t IsGeneratorDecayed(void) const { return (fGeneratorFlags & 0x2) != 0; }
  Bool_t IsGeneratorLast(void) const { return (fGeneratorFlags & 0x1) != 0 && (fGeneratorFlags & 0x2) == 0; }
  void SetGeneratorCreated(void) { fGeneratorFlags |= 0x1; }
  void SetGeneratorDecayed(void) { fGeneratorFlags |= 0x2; }

  Int_t GetNPoints(DetectorId detId) const;
  void SetMotherID(Int_t id) { fMotherID = id; }
  void SetSecondMotherID(Int_t id) { fSecondMotherID = id; }
  /*
  void SetStsPoints(Int_t np);
  */

  /*
   void AddStsPoint()  { SetStsPoints(  GetStsPoints()  + 1 ); }
  */

  void SetNPoints(DetectorId iDet, Int_t np);

  TLorentzVector Get4Momentum() const;

  UInt_t GetProcessID() const { return fProcess; }

 private:
  Int_t fPdgCode;

  Double32_t fPx, fPy, fPz, fE;

  Int_t fMotherID;
  Int_t fSecondMotherID;

  Int_t fGeneratorFlags;

  Double32_t fStartX, fStartY, fStartZ, fStartT;

  Int_t fPoints;

  UInt_t fProcess{}; // should actually be a TMCProcess, but as we abuse the TObject::fUniqueId to store the ProcessID causing the track ...
  ClassDef(PndMCTrack, 2);
};

#endif