[u/mrichter/AliRoot.git] / TRD / AliTRDmcTrack.h

#ifndef ALITRDMCTRACK_H
#define ALITRDMCTRACK_H  

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */ 

#include <TObject.h> 

class AliTRDgeometry;

const Int_t kMAX_CLUSTERS_PER_MC_TRACK=210;  

class AliTRDmcTrack : public TObject {

// Represents TRD related info about generated track

public:

  AliTRDmcTrack();
  AliTRDmcTrack(Int_t label, Bool_t primary, Float_t mass, Int_t charge, Int_t pdg); 

  void SetPin(Int_t plane, Double_t px, Double_t py, Double_t pz)
              { Pin[plane][0] = px; Pin[plane][1] = py; Pin[plane][2] = pz; }

  void SetPout(Int_t plane, Double_t px, Double_t py, Double_t pz)
              {Pout[plane][0] = px; Pout[plane][1] = py; Pout[plane][2] = pz;}

  void SetXYZin(Int_t plane, Double_t x, Double_t y, Double_t z)
    { XYZin[plane][0] = x; XYZin[plane][1] = y; XYZin[plane][2] = z; }

  void SetXYZout(Int_t plane, Double_t x, Double_t y, Double_t z)
    { XYZout[plane][0] = x; XYZout[plane][1] = y; XYZout[plane][2] = z; }

  void GetPxPyPzXYZ(Double_t &px, Double_t &py, Double_t &pz, 
		    Double_t &x,  Double_t &y,  Double_t &z, 
		    Int_t opt = 0) const;

  void GetPlanePxPyPz(Double_t &px, Double_t &py, Double_t &pz
		     ,Int_t plane, Int_t opt = 0) const;

  void GetXYZin(Int_t plane, Double_t &x, Double_t &y, Double_t &z) const 
    { x = XYZin[plane][0]; y = XYZin[plane][1]; z = XYZin[plane][2]; return; }

  void GetXYZout(Int_t plane, Double_t &x, Double_t &y, Double_t &z) const
    {x = XYZout[plane][0]; y = XYZout[plane][1]; z = XYZout[plane][2]; return;}

  void Update(Int_t index) { 
    if (fN < kMAX_CLUSTERS_PER_MC_TRACK-1) fIndex[fN++] = index; }

  Int_t   GetTrackIndex()          const { return fLab;      }
  Bool_t  IsPrimary()              const { return fPrimary;  }
  Float_t GetMass()                const { return fMass;     }
  Int_t   GetCharge()              const { return fCharge;   }
  Int_t   GetPdgCode()             const { return fPDG;      }

  Int_t   GetNumberOfClusters()    const { return fN;        }
  Int_t   GetClusterIndex(Int_t i) const { return fIndex[i]; }  

protected:

   Int_t    fLab;             // Track index  
   Bool_t   fPrimary;         // TRUE if it's a primary particle
   Float_t  fMass;            // Mass of the MC track
   Int_t    fCharge;          // Charge of the MC track
   Int_t    fPDG;             // PDG code of the MC track

   Int_t  fN;               // Number of TRD clusters associated with the track
   Int_t  fIndex[kMAX_CLUSTERS_PER_MC_TRACK]; // Indices of these clusters  
			   
   Double_t Pin[6][3];        // Px,Py,Pz at the entrance of each TRD plane   
   Double_t Pout[6][3];       // Px,Py,Pz at the exit of each TRD plane

   Double_t XYZin[6][3];        // x,y,z at the entrance of the TRD  
   Double_t XYZout[6][3];        // x,y,z at the exit of the TRD    

   ClassDef(AliTRDmcTrack,1)  // TRD MC track

};                   

#endif
Commit	Line	Data
16bf9884	1	#ifndef ALITRDMCTRACK_H
	2	#define ALITRDMCTRACK_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	#include <TObject.h>
	8
	9	class AliTRDgeometry;
	10
fd621f36	11	const Int_t kMAX_CLUSTERS_PER_MC_TRACK=210;
fd621f36	12
16bf9884	13	class AliTRDmcTrack : public TObject {
	14
	15	// Represents TRD related info about generated track
	16
	17	public:
	18
	19	AliTRDmcTrack();
	20	AliTRDmcTrack(Int_t label, Bool_t primary, Float_t mass, Int_t charge, Int_t pdg);
	21
	22	void SetPin(Int_t plane, Double_t px, Double_t py, Double_t pz)
fd621f36	23	{ Pin[plane][0] = px; Pin[plane][1] = py; Pin[plane][2] = pz; }
16bf9884	24
16bf9884	25	void SetPout(Int_t plane, Double_t px, Double_t py, Double_t pz)
fd621f36	26	{Pout[plane][0] = px; Pout[plane][1] = py; Pout[plane][2] = pz;}
	27
	28	void SetXYZin(Int_t plane, Double_t x, Double_t y, Double_t z)
	29	{ XYZin[plane][0] = x; XYZin[plane][1] = y; XYZin[plane][2] = z; }
	30
	31	void SetXYZout(Int_t plane, Double_t x, Double_t y, Double_t z)
	32	{ XYZout[plane][0] = x; XYZout[plane][1] = y; XYZout[plane][2] = z; }
	33
	34	void GetPxPyPzXYZ(Double_t &px, Double_t &py, Double_t &pz,
	35	Double_t &x, Double_t &y, Double_t &z,
	36	Int_t opt = 0) const;
16bf9884	37
16bf9884	38	void GetPlanePxPyPz(Double_t &px, Double_t &py, Double_t &pz
	39	,Int_t plane, Int_t opt = 0) const;
	40
fd621f36	41	void GetXYZin(Int_t plane, Double_t &x, Double_t &y, Double_t &z) const
	42	{ x = XYZin[plane][0]; y = XYZin[plane][1]; z = XYZin[plane][2]; return; }
	43
	44	void GetXYZout(Int_t plane, Double_t &x, Double_t &y, Double_t &z) const
	45	{x = XYZout[plane][0]; y = XYZout[plane][1]; z = XYZout[plane][2]; return;}
	46
	47	void Update(Int_t index) {
	48	if (fN < kMAX_CLUSTERS_PER_MC_TRACK-1) fIndex[fN++] = index; }
16bf9884	49
	50	Int_t GetTrackIndex() const { return fLab; }
	51	Bool_t IsPrimary() const { return fPrimary; }
	52	Float_t GetMass() const { return fMass; }
	53	Int_t GetCharge() const { return fCharge; }
	54	Int_t GetPdgCode() const { return fPDG; }
	55
	56	Int_t GetNumberOfClusters() const { return fN; }
	57	Int_t GetClusterIndex(Int_t i) const { return fIndex[i]; }
	58
	59	protected:
	60
	61	Int_t fLab; // Track index
	62	Bool_t fPrimary; // TRUE if it's a primary particle
	63	Float_t fMass; // Mass of the MC track
	64	Int_t fCharge; // Charge of the MC track
	65	Int_t fPDG; // PDG code of the MC track
	66
fd621f36	67	Int_t fN; // Number of TRD clusters associated with the track
fd621f36	68	Int_t fIndex[kMAX_CLUSTERS_PER_MC_TRACK]; // Indices of these clusters
16bf9884	69
fd621f36	70	Double_t Pin[6][3]; // Px,Py,Pz at the entrance of each TRD plane
	71	Double_t Pout[6][3]; // Px,Py,Pz at the exit of each TRD plane
	72
	73	Double_t XYZin[6][3]; // x,y,z at the entrance of the TRD
	74	Double_t XYZout[6][3]; // x,y,z at the exit of the TRD
16bf9884	75
	76	ClassDef(AliTRDmcTrack,1) // TRD MC track
	77
	78	};
	79
	80	#endif