template <typename Value> class TVectorT;
-typedef struct TVectorT<Double_t> TVectorD;
+typedef class TVectorT<Double_t> TVectorD;
class AliTRDseedV1;
class AliTRDtrackV1;
class AliTrackReference;
AliESDinfo(const AliESDinfo &esd);
virtual ~AliESDinfo();
AliESDinfo& operator=(const AliESDinfo &esd);
-
- Int_t GetId() const {return fId;}
- ULong_t GetStatus() const {return fStatus;}
- Int_t GetKinkIndex() const {return fKinkIndex;}
- UShort_t GetTPCncls() const {return fTPCncls;}
- UChar_t GetPidQuality() const {return fTRDpidQuality;}
- Int_t GetNSlices() const {return fTRDnSlices;}
- Double32_t* GetSliceIter() const {return fTRDslices;}
- const Double32_t* GetResponseIter() const {return &fTRDr[0];}
+ void Delete(const Option_t *);
+ Bool_t HasV0() const { return fHasV0;}
+ Int_t GetId() const { return fId;}
+ ULong_t GetStatus() const { return fStatus;}
+ Int_t GetKinkIndex() const { return fKinkIndex;}
+ Short_t GetTOFbc() const { return fTOFbc;}
+ UShort_t GetTPCncls() const { return fTPCncls;}
+ UChar_t GetPidQuality() const { return fTRDpidQuality;}
+ Int_t GetNSlices() const { return fTRDnSlices;}
+ Double32_t* GetSliceIter() const { return fTRDslices;}
+ const Double32_t* GetResponseIter() const { return &fTRDr[0];}
AliExternalTrackParam* GetOuterParam() const { return fOP;}
+ AliExternalTrackParam* GetTPCoutParam() const { return fTPCout;}
+ const Int_t* GetV0pid() const { return &fTRDv0pid[0];}
+ Int_t GetV0pid(Int_t i) const { return fTRDv0pid[i];}
protected:
+ UChar_t fHasV0; // v0 bit
Int_t fId; // ESD track id
ULong_t fStatus; // ESD track status
Int_t fKinkIndex; // ESD kink index
UShort_t fTPCncls; // Number of Clusters inside TPC
+ Short_t fTOFbc; // TOF bunch crossing index
Double32_t fTRDr[AliPID::kSPECIES]; // TRD radial position
UChar_t fTRDpidQuality; // TRD PID quality
Int_t fTRDnSlices; // number of slices used for PID
Double32_t *fTRDslices; //[fTRDnSlices]
- AliExternalTrackParam *fOP; // outer param
+ AliExternalTrackParam *fOP; // outer track param
+ AliExternalTrackParam *fTPCout; // outer TPC param
+ Int_t fTRDv0pid[AliPID::kSPECIES]; // PID from v0s
- ClassDef(AliESDinfo, 2) // ESD info related to TRD
+ ClassDef(AliESDinfo, 5) // ESD info related to TRD
};
class AliMCinfo{
Int_t GetLabel() const {return fLabel;}
Int_t GetNTrackRefs() const {return fNTrackRefs;}
Int_t GetPDG() const {return fPDG;}
- Bool_t GetDirections(Float_t &x0, Float_t &y0, Float_t &z0, Float_t &dydx, Float_t &dzdx, Float_t &pt, UChar_t &s) const;
+ Int_t GetPID() const ;
+ Bool_t GetDirections(Float_t &x0, Float_t &y0, Float_t &z0, Float_t &dydx, Float_t &dzdx, Float_t &pt, Float_t &eta, Float_t &phi, UChar_t &s) const;
AliTrackReference const* GetTrackRef(Int_t ref=0) const {return fTrackRefs[ref];}
- void PropagateKalman(
- TVectorD *dx, TVectorD *dy, TVectorD *dz,
- TVectorD *dpt, TVectorD *c, Double_t step = 2.) const;
-
+ static Double_t GetKalmanStep() {return fgKalmanStep;}
+ static Bool_t IsKalmanUpdate() {return fgKalmanUpdate;}
+ Bool_t PropagateKalman(
+ TVectorD *x, TVectorD *y, TVectorD *z,
+ TVectorD *dx, TVectorD *dy, TVectorD *dz,
+ TVectorD *pt, TVectorD *dpt, TVectorD *budget, TVectorD *c, Double_t mass=-1) const;
+ static void SetKalmanStep(Double_t s) {fgKalmanStep = s;}
+ static void SetKalmanUpdate(Bool_t s=kTRUE) {fgKalmanUpdate = s;}
protected:
- Int_t fLabel; // MC label
- Int_t fPDG; // particle code
- Int_t fNTrackRefs; // number of track refs
+ Int_t fLabel; // MC label
+ Int_t fPDG; // particle code
+ Int_t fNTrackRefs; // number of track refs
+ static Double_t fgKalmanStep; // Kalman step propagation
+ static Bool_t fgKalmanUpdate; // Kalman update with TRD tracklets
AliTrackReference *fTrackRefs[kNTrackRefs]; // track refs array
- ClassDef(AliMCinfo, 1) // MC info related to TRD
+
+ ClassDef(AliMCinfo, 2) // MC info related to TRD
};
AliTRDtrackInfo();
void Delete(const Option_t *);
AliTRDtrackInfo& operator=(const AliTRDtrackInfo &trdInfo);
void AddTrackRef(const AliTrackReference *trackRef);
- Int_t GetTrackId() const { return fESD.fId;}
- const AliESDinfo* GetESDinfo() const { return &fESD; }
- const AliMCinfo* GetMCinfo() const { return fMC; }
+ Int_t GetTrackId() const { return fESD.fId;}
+ const AliESDinfo* GetESDinfo() const { return &fESD; }
+ const AliMCinfo* GetMCinfo() const { return fMC; }
Int_t GetNumberOfClusters() const;
- Int_t GetNumberOfClustersRefit() const {return fNClusters;}
+ Int_t GetNumberOfClustersRefit() const { return fNClusters;}
Int_t GetNTracklets() const;
- Int_t GetNTrackRefs() const {return fMC ? fMC->fNTrackRefs:0;}
- Int_t GetLabel() const { return fMC ? fMC->fLabel:0; }
- Int_t GetKinkIndex() const { return fESD.fKinkIndex;}
- UShort_t GetTPCncls() const { return fESD.fTPCncls;}
- Int_t GetPDG() const { return fMC ? fMC->fPDG : 0; }
- ULong_t GetStatus() const {return fESD.fStatus;}
- AliTRDtrackV1* GetTrack() const { return fTRDtrack; }
+ Int_t GetNTrackRefs() const { return fMC ? fMC->fNTrackRefs:0;}
+ Int_t GetLabel() const { return fMC ? fMC->fLabel:0; }
+ Int_t GetKinkIndex() const { return fESD.fKinkIndex;}
+ Short_t GetTOFbc() const { return fESD.fTOFbc;}
+ UShort_t GetTPCncls() const { return fESD.fTPCncls;}
+ Int_t GetPDG() const { return fMC ? fMC->fPDG : 0; }
+ Int_t GetPID() const { return fMC ? fMC->GetPID() : -1; }
+ ULong_t GetStatus() const { return fESD.fStatus;}
+ AliTRDtrackV1* GetTrack() const { return fTRDtrack; }
AliTrackReference* GetTrackRef(Int_t entry) const;
- AliTrackReference* GetTrackRef(AliTRDseedV1* const tracklet) const;
+ AliTrackReference* GetTrackRef(const AliTRDseedV1* const tracklet) const;
- Bool_t IsCurved() const {return TestBit(kCurv);}
- Bool_t IsPrimary() const {return TestBit(kPrim);}
- Bool_t HasESDtrack() const{return ((fTRDtrack != 0x0) ||(fESD.fOP != 0));}
- Bool_t HasMCinfo() const { return (Bool_t)fMC; }
+ Bool_t IsCurved() const { return TestBit(kCurv);}
+ Bool_t IsPrimary() const { return TestBit(kPrim);}
+ Bool_t HasESDtrack() const { return ((fTRDtrack != 0x0) ||(fESD.fOP != 0));}
+ Bool_t HasMCinfo() const { return (Bool_t)fMC; }
- void SetCurved(Bool_t curv = kTRUE) {SetBit(kCurv, curv);}
- void SetLabel(Int_t lab) { SetMC(); fMC->fLabel = lab; }
- void SetNumberOfClustersRefit(Int_t n) {fNClusters = n;}
+ void SetCurved(Bool_t curv = kTRUE) { SetBit(kCurv, curv);}
+ void SetLabel(Int_t lab) { if(fMC) fMC->fLabel = lab; }
+ void SetNumberOfClustersRefit(Int_t n){fNClusters = n;}
inline void SetMC();
- void SetPDG(Int_t pdg) { SetMC(); fMC->fPDG = pdg; }
- void SetPrimary(Bool_t prim = kTRUE) {SetBit(kPrim, prim);}
+ void SetPDG(Int_t pdg) { if(fMC) fMC->fPDG = pdg; }
+ void SetPrimary(Bool_t prim = kTRUE) {SetBit(kPrim, prim);}
void SetOuterParam(const AliExternalTrackParam *op);
- void SetStatus(ULong_t stat) {fESD.fStatus = stat;}
- void SetKinkIndex(Int_t kinkIndex) {fESD.fKinkIndex = kinkIndex;}
- void SetTPCncls(UShort_t TPCncls) {fESD.fTPCncls = TPCncls;}
- void SetTrackId(Int_t id) {fESD.fId = id;}
+ void SetTPCoutParam(const AliExternalTrackParam *op);
+ void SetStatus(ULong_t stat) { fESD.fStatus = stat;}
+ void SetKinkIndex(Int_t kinkIndex) { fESD.fKinkIndex = kinkIndex;}
+ void SetTOFbc(Int_t bc) { fESD.fTOFbc = bc;}
+ void SetTPCncls(UShort_t TPCncls) { fESD.fTPCncls = TPCncls;}
+ void SetTrackId(Int_t id) { fESD.fId = id;}
void SetTrack(const AliTRDtrackV1 *track);
- void SetESDpidQuality(UChar_t q) { fESD.fTRDpidQuality = q;}
+ void SetESDpidQuality(UChar_t q) { fESD.fTRDpidQuality = q;}
void SetSlices(Int_t n, Double32_t *s);
inline void SetESDpid(Double_t *);
+ inline void SetV0pid(Int_t *);
+ void SetV0(Bool_t v0=kTRUE) { fESD.fHasV0 = v0;}
private:
enum{
AliMCinfo *fMC; // MC extract for TRD
AliESDinfo fESD; // ESD extract for TRD
- ClassDef(AliTRDtrackInfo, 3) // TRD track info
+ ClassDef(AliTRDtrackInfo, 4) // TRD track info
};
for(Int_t is = AliPID::kSPECIES; is--;) fESD.fTRDr[is] = r[is];
}
+//________________________________________________________
+inline void AliTRDtrackInfo::SetV0pid(Int_t * const r)
+{
+ for(Int_t is = AliPID::kSPECIES; is--;) fESD.fTRDv0pid[is] = r[is];
+}
+
#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff