#ifndef AliEveV0_H
#define AliEveV0_H
#include "AliEveTrack.h"
#include <TEveVSDStructs.h>
#include <TPolyLine3D.h>
#include <TPDGCode.h>
class TH1F;
class TH2F;
class AliEveV0List;
class AliEveV0 : public TEvePointSet
{
friend class AliEveV0List;
friend class AliEveV0Editor;
public:
AliEveV0();
AliEveV0(TEveRecTrack* tNeg, TEveRecTrack* tPos, TEveRecV0* v0,
TEveTrackPropagator* rs);
virtual ~AliEveV0();
void MakeV0();
virtual void SetMainColor(Color_t col)
{
TEvePointSet::SetMainColor(col);
fPointingLine->SetLineColor(fMarkerColor);
}
void SetRnrStyle(TEveTrackPropagator* rs) { fRnrStyle = rs; }
Float_t GetDaughterDCA() const { return fDaughterDCA; }
void SetDaughterDCA(Float_t dca) { fDaughterDCA = dca; }
Float_t GetPhi() const { return fRecDecayP.Phi(); }
Float_t GetEta() const { return fRecDecayP.Eta(); }
Float_t GetRadius() const { return fRecDecayV.Perp(); }
Float_t GetPt() const { return fRecDecayP.Perp(); }
Float_t GetMomentum() const { return fRecDecayP.Mag(); }
Float_t GetInvMass(Int_t nPdgCode, Int_t pPdgCode) const;
Float_t GetK0sInvMass() const { return GetInvMass(kPiMinus,kPiPlus); }
Float_t GetLambdaInvMass() const { return GetInvMass(kPiMinus,kProton); }
Float_t GetAntiLambdaInvMass() const { return GetInvMass(kProton,kPiPlus); }
Bool_t GetOnFlyStatus() const { return fOnFlyStatus; }
void SetOnFlyStatus(Bool_t fs) { fOnFlyStatus = fs; }
void SetMaxProbPdgPid(Int_t iDaughter, Int_t rPdg, Float_t rPid);
Int_t GetNegMaxProbPdg() const { return fNegMaxProbPdg; }
Int_t GetPosMaxProbPdg() const { return fPosMaxProbPdg; }
Float_t GetNegMaxProbPid() const { return fNegMaxProbPid; }
Float_t GetPosMaxProbPid() const { return fPosMaxProbPid; }
Int_t GetESDIndex() const { return fESDIndex; }
void SetESDIndex(Int_t ind) { fESDIndex = ind;}
virtual const Text_t* GetName() const { return Form("ESDv0_%i",fESDIndex); }
virtual const Text_t* GetTitle() const { return Form("ESDv0_%i",fESDIndex); }
TEveTrackPropagator* GetPropagator() const { return fRnrStyle; }
AliEveTrack* GetNegTrack() const { return fNegTrack; }
AliEveTrack* GetPosTrack() const { return fPosTrack; }
TEveLine* GetPointingLine() const { return fPointingLine; }
protected:
TEveVector fRecBirthV;
TEveVector fRecDecayV;
TEveVector fRecDecayP;
AliEveTrack *fNegTrack;
AliEveTrack *fPosTrack;
TEveTrackPropagator *fRnrStyle;
TEveLine *fPointingLine;
Int_t fESDIndex;
Bool_t fOnFlyStatus;
Float_t fDaughterDCA;
Float_t fChi2V0;
Int_t fNegMaxProbPdg;
Int_t fPosMaxProbPdg;
Float_t fNegMaxProbPid;
Float_t fPosMaxProbPid;
private:
AliEveV0(const AliEveV0&);
AliEveV0& operator=(const AliEveV0&);
ClassDef(AliEveV0, 0);
};
class AliEveV0List : public TEveElementList
{
friend class AliEveV0ListEditor;
public:
AliEveV0List();
AliEveV0List(TEveTrackPropagator* rs);
AliEveV0List(const Text_t* name, TEveTrackPropagator* rs=0);
virtual ~AliEveV0List() {}
virtual const Text_t* GetTitle() const { return fTitle; }
virtual void SetTitle(const Text_t* t) { fTitle = t; }
virtual void SetTracksColor(Color_t cNeg, Color_t cPos) {
fNegColor = cNeg; fPosColor = cPos;}
virtual Bool_t CanEditMainColor() const { return kTRUE; }
void SetRnrStyle(TEveTrackPropagator* rst) { fRnrStyle = rst; }
TEveTrackPropagator* GetPropagator() { return fRnrStyle; }
Bool_t GetRnrV0vtx() const { return fRnrV0vtx; }
Bool_t GetRnrV0path() const { return fRnrV0path; }
Bool_t GetRnrDaughters() const { return fRnrDaughters; }
void MakeV0s();
void FilterByRadius(Float_t minR, Float_t maxR);
void FilterByDaughterDCA(Float_t minDaughterDCA, Float_t maxDaughterDCA);
void FilterByPt(Float_t minPt, Float_t maxPt);
void FilterByCheckedPidMinProb(Int_t rFlag, Int_t rDaughter, Int_t rPid, Float_t rProb);
void SetNegCheckedPid(Int_t rNegCheckedPid) {fNegCheckedPid = rNegCheckedPid;}
void SetPosCheckedPid(Int_t rPosCheckedPid) {fPosCheckedPid = rPosCheckedPid;}
Int_t GetNegCheckedPid() const { return fNegCheckedPid; }
Int_t GetPosCheckedPid() const { return fPosCheckedPid; }
void SetNegCheckedProb(Float_t rNegCheckedProb) {fNegCheckedProb = rNegCheckedProb;}
void SetPosCheckedProb(Float_t rPosCheckedProb) {fPosCheckedProb = rPosCheckedProb;}
Float_t GetNegCheckedProb() const { return fNegCheckedProb; }
Float_t GetPosCheckedProb() const { return fPosCheckedProb; }
void FilterByInvariantMass(Float_t minPt, Float_t maxPt, Int_t nPdgCode, Int_t pPdgCode);
protected:
TString fTitle;
TEveTrackPropagator *fRnrStyle;
Bool_t fRnrDaughters;
Bool_t fRnrV0vtx;
Bool_t fRnrV0path;
Color_t fNegColor;
Color_t fPosColor;
Float_t fMinRCut;
Float_t fMaxRCut;
Float_t fMinDaughterDCA;
Float_t fMaxDaughterDCA;
Float_t fMinPt;
Float_t fMaxPt;
Int_t fNegCheckedPid;
Int_t fPosCheckedPid;
Float_t fNegCheckedProb;
Float_t fPosCheckedProb;
Float_t fMinInvariantMass;
Float_t fMaxInvariantMass;
private:
void Init();
AliEveV0List(const AliEveV0List&);
AliEveV0List& operator=(const AliEveV0List&);
ClassDef(AliEveV0List, 0);
};
#endif