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(Int_t iDet, Int_t  np);
  
  TLorentzVector Get4Momentum() const;


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;

  ClassDef(PndMCTrack,1);

};



#endif