#ifndef AliEmcalJet_H
#define AliEmcalJet_H
-// $Id$
-
+#include <vector>
+#include <algorithm>
+#include <utility>
#include <TArrayS.h>
#include <TLorentzVector.h>
#include <TMath.h>
#include <TClonesArray.h>
+#include <TVector2.h>
#include "AliVParticle.h"
#include "AliVCluster.h"
+#include "AliVEvent.h"
class AliEmcalJet : public AliVParticle
{
public:
+ enum EFlavourTag{
+ kDStar = 1<<0,
+ kD0 = 1<<1,
+ kSig1 = 1<<2,
+ kSig2 = 1<<3,
+ kBckgrd1 = 1<<4,
+ kBckgrd2 = 1<<5,
+ kBckgrd3 = 1<<6
+ //.....
+ };
+
AliEmcalJet();
AliEmcalJet(Double_t px, Double_t py, Double_t pz);
AliEmcalJet(Double_t pt, Double_t eta, Double_t phi, Double_t m);
Double_t Phi() const { return fPhi; }
Double_t Theta() const { return 2*TMath::ATan(TMath::Exp(-fEta)); }
Double_t E() const { Double_t p=P(); return TMath::Sqrt(M()*M()+p*p); }
- Double_t M() const { return 0.13957; }
+ Double_t M() const { return fM; }
Double_t Eta() const { return fEta; }
Double_t Y() const { return 0.5*TMath::Log((E()+Pz())/(E()-Pz())); }
Short_t Charge() const { return 0; }
- Int_t GetLabel() const { return -1; }
+ Int_t GetLabel() const { return fLabel; }
Int_t PdgCode() const { return 0; }
const Double_t *PID() const { return 0; }
void GetMom(TLorentzVector &vec) const;
Int_t Compare(const TObject* obj) const;
Short_t ClusterAt(Int_t idx) const { return fClusterIDs.At(idx); }
AliVCluster *ClusterAt(Int_t idx, TClonesArray *ca) const { if (!ca) return 0; return dynamic_cast<AliVCluster*>(ca->At(ClusterAt(idx))); }
- UShort_t GetNumberOfClusters() const { return fClusterIDs.GetSize(); }
+ AliVCluster *GetLeadingCluster(TClonesArray *clusters) const;
+ Int_t GetNumberOfClusters() const { return fClusterIDs.GetSize(); }
UShort_t GetNumberOfTracks() const { return fTrackIDs.GetSize(); }
UShort_t GetNumberOfConstituents() const { return GetNumberOfClusters()+GetNumberOfTracks(); }
Double_t FracEmcalArea() const { return fAreaEmc/fArea; }
Double_t PtSub(Double_t rho) const { return fPt - fArea*rho; }
Double_t PtSubVect(Double_t rho) const;
Short_t TrackAt(Int_t idx) const { return fTrackIDs.At(idx); }
- AliVParticle *TrackAt(Int_t idx, TClonesArray *ta) const { if (!ta) return 0; return dynamic_cast<AliVParticle*>(ta->At(TrackAt(idx))); }
-
+ AliVParticle *TrackAt(Int_t idx, TClonesArray *ta) const { if (!ta) return 0; return dynamic_cast<AliVParticle*>(ta->At(TrackAt(idx))); }
+ AliVParticle *GetLeadingTrack(TClonesArray *tracks) const;
+ Int_t GetFlavour() const { return fFlavourTagging; }
+
void AddClusterAt(Int_t clus, Int_t idx){ fClusterIDs.AddAt(clus, idx); }
+ void AddFlavourTag(Int_t tag) { fFlavourTagging |= tag; }
void AddTrackAt(Int_t track, Int_t idx) { fTrackIDs.AddAt(track, idx); }
void Clear(Option_t */*option*/="") { fClusterIDs.Set(0); fTrackIDs.Set(0); fClosestJets[0] = 0; fClosestJets[1] = 0;
fClosestJetsDist[0] = 0; fClosestJetsDist[1] = 0; fMatched = 0; fPtSub = 0; }
+ Double_t DeltaR(const AliVParticle* part) const;
+ Double_t GetZ ( const Double_t trkPx, const Double_t trkPy, const Double_t trkPz ) const; // Get Z of constituent trk
+ Double_t GetZ ( const AliVParticle* trk ) const; // Get Z of constituent trk
+ Double_t GetXi ( const AliVParticle* trk ) const { return TMath::Log ( 1/GetZ (trk) ); } // Get Xi of constituent trk
+ Double_t GetXi ( const Double_t trkPx, const Double_t trkPy, const Double_t trkPz ) const { return TMath::Log ( 1/GetZ (trkPx, trkPy, trkPz ) ); } // Get Xi of constituent trk
+
+ void SetLabel(Int_t l) { fLabel = l; }
void SetArea(Double_t a) { fArea = a; }
void SetAreaEta(Double_t a) { fAreaEta = a; }
void SetAreaPhi(Double_t a) { fAreaPhi = a; }
void SetAreaEmc(Double_t a) { fAreaEmc = a; }
void SetAxisInEmcal(Bool_t b) { fAxisInEmcal = b; }
+ void SetFlavour(Int_t flavour) { fFlavourTagging = flavour; }
void SetMaxNeutralPt(Double32_t t) { fMaxNPt = t; }
void SetMaxChargedPt(Double32_t t) { fMaxCPt = t; }
void SetNEF(Double_t nef) { fNEF = nef; }
void SetNumberOfNeutrals(Int_t n) { fNn = n; }
void SetMCPt(Double_t p) { fMCPt = p; }
void SortConstituents();
- void SetNEmc(Int_t n) { fNEmc = n; }
- void SetPtEmc(Double_t pt) { fPtEmc = pt; }
- void SetPtSub(Double_t ps) { fPtSub = ps; }
- void SetPtSubVect(Double_t ps) { fPtVectSub = ps; }
+ std::vector<int> SortConstituentsPt(TClonesArray *tracks) const;
+ void SetNEmc(Int_t n) { fNEmc = n; }
+ void SetPtEmc(Double_t pt) { fPtEmc = pt; }
+ void SetPtSub(Double_t ps) { fPtSub = ps; }
+ void SetPtSubVect(Double_t ps) { fPtVectSub = ps; }
+ Bool_t TestFlavourTag(Int_t tag) { return (Bool_t)((tag & fFlavourTagging) !=0); }
+
+ // Trigger
+ Bool_t IsTriggerJet(UInt_t trigger=AliVEvent::kEMCEJE) const { return (Bool_t)((fTriggers & trigger) != 0); }
+ void SetTrigger(UInt_t trigger) { fTriggers = trigger; }
+ void AddTrigger(UInt_t trigger) { fTriggers |= trigger; }
// Matching
- void SetClosestJet(AliEmcalJet *j, Double_t d) { fClosestJets[0] = j; fClosestJetsDist[0] = d ; }
+ void SetClosestJet(AliEmcalJet *j, Double_t d) { fClosestJets[0] = j; fClosestJetsDist[0] = d ; }
void SetSecondClosestJet(AliEmcalJet *j, Double_t d) { fClosestJets[1] = j; fClosestJetsDist[1] = d ; }
void SetMatchedToClosest(UShort_t m) { fMatched = 0; fMatchingType = m ; }
void SetMatchedToSecondClosest(UShort_t m) { fMatched = 1; fMatchingType = m ; }
+ void ResetMatching();
AliEmcalJet* ClosestJet() const { return fClosestJets[0] ; }
Double_t ClosestJetDistance() const { return fClosestJetsDist[0] ; }
AliEmcalJet* SecondClosestJet() const { return fClosestJets[1] ; }
AliEmcalJet* MatchedJet() const { return fMatched < 2 ? fClosestJets[fMatched] : 0; }
UShort_t GetMatchingType() const { return fMatchingType ; }
+ void SetTaggedJet(AliEmcalJet *j) { fTaggedJet = j ; }
+ void SetTagStatus(Int_t i) { fTagStatus = i ; }
+ AliEmcalJet* GetTaggedJet() const { return fTaggedJet ; }
+ Int_t GetTagStatus() const { return fTagStatus ; }
+
+ //jet shape derivatives
+ //jet mass
+ void SetFirstDerivative(Double_t d) { fJetShapeMassFirstDer = d ; }
+ void SetSecondDerivative(Double_t d) { fJetShapeMassSecondDer = d ; }
+ void SetFirstOrderSubtracted(Double_t d) { fJetShapeMassFirstSub = d ; }
+ void SetSecondOrderSubtracted(Double_t d) { fJetShapeMassSecondSub = d ; }
+ Double_t GetFirstDerivative() const { return fJetShapeMassFirstDer ; }
+ Double_t GetSecondDerivative() const { return fJetShapeMassSecondDer ; }
+ Double_t GetFirstOrderSubtracted() const { return fJetShapeMassFirstSub ; }
+ Double_t GetSecondOrderSubtracted() const { return fJetShapeMassSecondSub ; }
+
+ //jet structure function
+ TArrayF GetGRNumerator() const { return fGRNumerator ; }
+ TArrayF GetGRDenominator() const { return fGRDenominator ; }
+ TArrayF GetGRNumeratorSub() const { return fGRNumeratorSub ; }
+ TArrayF GetGRDenominatorSub() const { return fGRDenominatorSub ; }
+ void AddGRNumAt(Float_t num, Int_t idx) { fGRNumerator.AddAt(num, idx) ; }
+ void AddGRDenAt(Float_t den, Int_t idx) { fGRDenominator.AddAt(den, idx) ; }
+ void SetGRNumSize(UInt_t s) { fGRNumerator.Set(s) ; }
+ void SetGRDenSize(UInt_t s) { fGRDenominator.Set(s) ; }
+
+ void AddGRNumSubAt(Float_t num, Int_t idx) { fGRNumeratorSub.AddAt(num, idx) ; }
+ void AddGRDenSubAt(Float_t den, Int_t idx) { fGRDenominatorSub.AddAt(den, idx) ; }
+ void SetGRNumSubSize(UInt_t s) { fGRNumeratorSub.Set(s) ; }
+ void SetGRDenSubSize(UInt_t s) { fGRDenominatorSub.Set(s) ; }
+ void PrintGR();
+
+ //Angularity
+ void SetFirstDerivativeAngularity(Double_t d) { fJetShapeAngularityFirstDer = d ; }
+ void SetSecondDerivativeAngularity(Double_t d) { fJetShapeAngularitySecondDer = d ; }
+ void SetFirstOrderSubtractedAngularity(Double_t d) { fJetShapeAngularityFirstSub = d ; }
+ void SetSecondOrderSubtractedAngularity(Double_t d) { fJetShapeAngularitySecondSub = d ; }
+ Double_t GetFirstDerivativeAngularity() const { return fJetShapeAngularityFirstDer ; }
+ Double_t GetSecondDerivativeAngularity() const { return fJetShapeAngularitySecondDer ; }
+ Double_t GetFirstOrderSubtractedAngularity() const { return fJetShapeAngularityFirstSub ; }
+ Double_t GetSecondOrderSubtractedAngularity() const { return fJetShapeAngularitySecondSub ; }
+
+ //pTD
+ void SetFirstDerivativepTD(Double_t d) { fJetShapepTDFirstDer = d ; }
+ void SetSecondDerivativepTD(Double_t d) { fJetShapepTDSecondDer = d ; }
+ void SetFirstOrderSubtractedpTD(Double_t d) { fJetShapepTDFirstSub = d ; }
+ void SetSecondOrderSubtractedpTD(Double_t d) { fJetShapepTDSecondSub = d ; }
+ Double_t GetFirstDerivativepTD() const { return fJetShapepTDFirstDer ; }
+ Double_t GetSecondDerivativepTD() const { return fJetShapepTDSecondDer ; }
+ Double_t GetFirstOrderSubtractedpTD() const { return fJetShapepTDFirstSub ; }
+ Double_t GetSecondOrderSubtractedpTD() const { return fJetShapepTDSecondSub ; }
+
+ //Circularity
+ void SetFirstDerivativeCircularity(Double_t d) { fJetShapeCircularityFirstDer = d ; }
+ void SetSecondDerivativeCircularity(Double_t d) { fJetShapeCircularitySecondDer = d ; }
+ void SetFirstOrderSubtractedCircularity(Double_t d) { fJetShapeCircularityFirstSub = d ; }
+ void SetSecondOrderSubtractedCircularity(Double_t d) { fJetShapeCircularitySecondSub = d ; }
+ Double_t GetFirstDerivativeCircularity() const { return fJetShapeCircularityFirstDer ; }
+ Double_t GetSecondDerivativeCircularity() const { return fJetShapeCircularitySecondDer ; }
+ Double_t GetFirstOrderSubtractedCircularity() const { return fJetShapeCircularityFirstSub ; }
+ Double_t GetSecondOrderSubtractedCircularity() const { return fJetShapeCircularitySecondSub ; }
+
+ //Sigma2
+ void SetFirstDerivativeSigma2(Double_t d) { fJetShapeSigma2FirstDer = d ; }
+ void SetSecondDerivativeSigma2(Double_t d) { fJetShapeSigma2SecondDer = d ; }
+ void SetFirstOrderSubtractedSigma2(Double_t d) { fJetShapeSigma2FirstSub = d ; }
+ void SetSecondOrderSubtractedSigma2(Double_t d) { fJetShapeSigma2SecondSub = d ; }
+ Double_t GetFirstDerivativeSigma2() const { return fJetShapeSigma2FirstDer ; }
+ Double_t GetSecondDerivativeSigma2() const { return fJetShapeSigma2SecondDer ; }
+ Double_t GetFirstOrderSubtractedSigma2() const { return fJetShapeSigma2FirstSub ; }
+ Double_t GetSecondOrderSubtractedSigma2() const { return fJetShapeSigma2SecondSub ; }
+
+
+ //number of contituents
+ void SetFirstDerivativeConstituent(Double_t d) { fJetShapeConstituentFirstDer = d ; }
+ void SetSecondDerivativeConstituent(Double_t d) { fJetShapeConstituentSecondDer = d ; }
+ void SetFirstOrderSubtractedConstituent(Double_t d) { fJetShapeConstituentFirstSub = d ; }
+ void SetSecondOrderSubtractedConstituent(Double_t d) { fJetShapeConstituentSecondSub = d ; }
+ Double_t GetFirstDerivativeConstituent() const { return fJetShapeConstituentFirstDer ; }
+ Double_t GetSecondDerivativeConstituent() const { return fJetShapeConstituentSecondDer ; }
+ Double_t GetFirstOrderSubtractedConstituent() const { return fJetShapeConstituentFirstSub ; }
+ Double_t GetSecondOrderSubtractedConstituent() const { return fJetShapeConstituentSecondSub ; }
+
+ //leading minus subleading constituent
+ void SetFirstDerivativeLeSub(Double_t d) { fJetShapeLeSubFirstDer = d ; }
+ void SetSecondDerivativeLeSub(Double_t d) { fJetShapeLeSubSecondDer = d ; }
+ void SetFirstOrderSubtractedLeSub(Double_t d) { fJetShapeLeSubFirstSub = d ; }
+ void SetSecondOrderSubtractedLeSub(Double_t d) { fJetShapeLeSubSecondSub = d ; }
+ Double_t GetFirstDerivativeLeSub() const { return fJetShapeLeSubFirstDer ; }
+ Double_t GetSecondDerivativeLeSub() const { return fJetShapeLeSubSecondDer ; }
+ Double_t GetFirstOrderSubtractedLeSub() const { return fJetShapeLeSubFirstSub ; }
+ Double_t GetSecondOrderSubtractedLeSub() const { return fJetShapeLeSubSecondSub ; }
+
protected:
Double32_t fPt; //[0,0,12] pt
Double32_t fEta; //[-1,1,12] eta
Double32_t fAreaPhi; //[0,0,12] area phi
Double32_t fAreaEmc; //[0,0,12] area on EMCAL surface (determined from ghosts)
Bool_t fAxisInEmcal; // =true if jet axis inside EMCAL acceptance
+ Int_t fFlavourTagging; // tag jet with a falvour, bit 0 = no tag; bit 1= Dstar; bit 2 = D0
Double32_t fMaxCPt; //[0,0,12] pt of maximum charged constituent
Double32_t fMaxNPt; //[0,0,12] pt of maximum neutral constituent
Double32_t fMCPt; // pt from MC particles contributing to the jet
Int_t fNch; // number of charged constituents
Double32_t fPtEmc; //[0,0,12] pt in EMCAL acceptance
Int_t fNEmc; // number of constituents in EMCAL acceptance
- TArrayS fClusterIDs; // array of cluster constituents
- TArrayS fTrackIDs; // array of track constituents
+ TArrayI fClusterIDs; // array containing ids of cluster constituents
+ TArrayI fTrackIDs; // array containing ids of track constituents
AliEmcalJet *fClosestJets[2]; //! if this is MC it contains the two closest detector level jets in order of distance and viceversa
Double32_t fClosestJetsDist[2]; //! distance to closest jets (see above)
UShort_t fMatched; //! 0,1 if it is matched with one of the closest jets; 2 if it is not matched
UShort_t fMatchingType; //! matching type
+ AliEmcalJet *fTaggedJet; //! jet tagged to this jet
+ Int_t fTagStatus; //! status of tagging -1: NA 0: not tagged 1: tagged
Double_t fPtSub; //! background subtracted pt (not stored set from outside)
Double_t fPtVectSub; //! background vector subtracted pt (not stored set from outside)
+ UInt_t fTriggers; //! triggers that the jet might have fired (AliVEvent::EOfflineTriggerTypes)
+
+ Double_t fJetShapeMassFirstDer; //! result from shape derivatives for jet mass: 1st derivative
+ Double_t fJetShapeMassSecondDer; //! result from shape derivatives for jet mass: 2nd derivative
+ Double_t fJetShapeMassFirstSub; //! result from shape derivatives for jet mass: 1st order subtracted
+ Double_t fJetShapeMassSecondSub; //! result from shape derivatives for jet mass: 2nd order subtracted
+ Int_t fLabel; // label to inclusive jet for constituent subtracted jet
+
+ TArrayF fGRNumerator; //! array with angular structure function numerator
+ TArrayF fGRDenominator; //! array with angular structure function denominator
+ TArrayF fGRNumeratorSub; //! array with angular structure function numerator
+ TArrayF fGRDenominatorSub; //! array with angular structure function denominator
+
+ Double_t fJetShapeAngularityFirstDer; //! result from shape derivatives for jet Angularity: 1st derivative
+ Double_t fJetShapeAngularitySecondDer; //! result from shape derivatives for jet Angularity: 2nd derivative
+ Double_t fJetShapeAngularityFirstSub; //! result from shape derivatives for jet Angularity: 1st order subtracted
+ Double_t fJetShapeAngularitySecondSub; //! result from shape derivatives for jet Angularity: 2nd order subtracted
+
+ Double_t fJetShapepTDFirstDer; //! result from shape derivatives for jet pTD: 1st derivative
+ Double_t fJetShapepTDSecondDer; //! result from shape derivatives for jet pTD: 2nd derivative
+ Double_t fJetShapepTDFirstSub; //! result from shape derivatives for jet pTD: 1st order subtracted
+ Double_t fJetShapepTDSecondSub; //! result from shape derivatives for jet pTD: 2nd order subtracted
+
+ Double_t fJetShapeCircularityFirstDer; //! result from shape derivatives for jet circularity: 1st derivative
+ Double_t fJetShapeCircularitySecondDer; //! result from shape derivatives for jet circularity: 2nd derivative
+ Double_t fJetShapeCircularityFirstSub; //! result from shape derivatives for jet circularity: 1st order subtracted
+ Double_t fJetShapeCircularitySecondSub; //! result from shape derivatives for jetcircularity: 2nd order subtracted
+
+ Double_t fJetShapeSigma2FirstDer; //! result from shape derivatives for jet sigma2: 1st derivative
+ Double_t fJetShapeSigma2SecondDer; //! result from shape derivatives for jet sigma2: 2nd derivative
+ Double_t fJetShapeSigma2FirstSub; //! result from shape derivatives for jet sigma2: 1st order subtracted
+ Double_t fJetShapeSigma2SecondSub; //! result from shape derivatives for jetsigma2: 2nd order subtracted
+
+ Double_t fJetShapeConstituentFirstDer; //! result from shape derivatives for jet const: 1st derivative
+ Double_t fJetShapeConstituentSecondDer; //! result from shape derivatives for jet const: 2nd derivative
+ Double_t fJetShapeConstituentFirstSub; //! result from shape derivatives for jet const: 1st order subtracted
+ Double_t fJetShapeConstituentSecondSub; //! result from shape derivatives for jet const: 2nd order subtracted
+
+ Double_t fJetShapeLeSubFirstDer; //! result from shape derivatives for jet LeSub: 1st derivative
+ Double_t fJetShapeLeSubSecondDer; //! result from shape derivatives for jet LeSub: 2nd derivative
+ Double_t fJetShapeLeSubFirstSub; //! result from shape derivatives for jet LeSub: 1st order subtracted
+ Double_t fJetShapeLeSubSecondSub; //! result from shape derivatives for jet LeSub: 2nd order subtracted
+
+ private:
+ struct sort_descend
+ { // sort in decreasing order
+ // first value of the pair is Pt and the second is entry index
+ bool operator () (const std::pair<Double_t, Int_t>& p1, const std::pair<Double_t, Int_t>& p2) { return p1.first > p2.first ; }
+ };
- ClassDef(AliEmcalJet,9) // Emcal jet class in cylindrical coordinates
+ ClassDef(AliEmcalJet,16) // Emcal jet class in cylindrical coordinates
};
#endif