#ifndef ALIEMCALJETTASK_H
#define ALIEMCALJETTASK_H
-// $Id$
-
class TClonesArray;
class AliVEvent;
+class AliRhoParameter;
+
+namespace fastjet {
+ class PseudoJet;
+}
+#include "AliLog.h"
#include "AliAnalysisTaskSE.h"
+#include "AliFJWrapper.h"
+#include "AliAODMCParticle.h"
+#include "AliEmcalJet.h"
class AliEmcalJetTask : public AliAnalysisTaskSE {
public:
+
+ enum JetType {
+ kNone=0,
+ kKT=1<<0,
+ kAKT=1<<1,
+ kFullJet=1<<2,
+ kChargedJet=1<<3,
+ kNeutralJet=1<<4,
+ kR020Jet=1<<5,
+ kR030Jet=1<<6,
+ kR040Jet=1<<7,
+ kRX1Jet=1<<8, // user defined radii, use SetRadius(Double_t)
+ kRX2Jet=1<<9,
+ kRX3Jet=1<<10
+ };
+
AliEmcalJetTask();
AliEmcalJetTask(const char *name);
virtual ~AliEmcalJetTask();
void UserCreateOutputObjects();
void UserExec(Option_t *option);
void Terminate(Option_t *option);
+
+ Bool_t IsLocked() { if(fLocked) {AliFatal("Jet finder task is locked! Changing properties is not allowed."); return kTRUE;} else return kFALSE;};
+ void SetLocked() { fLocked = kTRUE;}
+ void SelectConstituents(UInt_t constSel, UInt_t MCconstSel) { fConstSel = constSel; fMCConstSel = MCconstSel; };
+ void SetAlgo(Int_t a) { if(IsLocked()) return; if (a==0) fJetType |= kKT; else fJetType |= kAKT; } // for backward compatibility only
+ void SetClusName(const char *n) { if(IsLocked()) return; fCaloName = n ; }
+ void SetJetsName(const char *n) { if(IsLocked()) return; fJetsName = n ; }
+ void SetJetsSubName(const char *n) { fJetsSubName = n ; }
+ void SetTracksSubName(const char *n) { fTracksSubName = n ; }
+ void SetClusSubName(const char *n) { fClusSubName = n ; }
+ void SetJetType(UInt_t t) { if(IsLocked()) return; fJetType = t ; }
+ void SetMarkConstituents(UInt_t m) { if(IsLocked()) return; fMarkConst = m ; }
+ void SetMinJetArea(Double_t a) { if(IsLocked()) return; fMinJetArea = a ; }
+ void SetMinJetClusPt(Double_t min) { if(IsLocked()) return; fMinJetClusPt = min ; }
+ void SetMinJetPt(Double_t j) { if(IsLocked()) return; fMinJetPt = j ; }
+ void SetMinJetTrackPt(Double_t min) { if(IsLocked()) return; fMinJetTrackPt = min ; }
+ void SetRadius(Double_t r) { if(IsLocked()) return; fRadius = r ; if ((fJetType & (kRX1Jet|kRX2Jet|kRX3Jet)) == 0) AliWarning("Radius value will be ignored if jet type is not set to a user defined radius (kRX1Jet,kRX2Jet,kRX3Jet)."); }
+ void SetTrackEfficiency(Double_t t) { if(IsLocked()) return; fTrackEfficiency = t ; }
+ void SetTracksName(const char *n) { if(IsLocked()) return; fTracksName = n ; }
+ void SetType(Int_t t) { if(IsLocked()) return;
+ if (t==0) fJetType |= kFullJet;
+ else if (t==1) fJetType |= kChargedJet;
+ else if (t==2) fJetType |= kNeutralJet; } // for backward compatibility only
+ void SetEtaRange(Double_t emi, Double_t ema) {if(IsLocked()) return; fEtaMin = emi; fEtaMax = ema; }
+ void SetPhiRange(Double_t pmi, Double_t pma) {if(IsLocked()) return; fPhiMin = pmi; fPhiMax = pma; }
+ void SetJetEtaRange(Double_t emi, Double_t ema) {if(IsLocked()) return; fJetEtaMin = emi; fJetEtaMax = ema; }
+ void SetJetPhiRange(Double_t pmi, Double_t pma) {if(IsLocked()) return; fJetPhiMin = pmi; fJetPhiMax = pma; }
+ void SetGhostArea(Double_t gharea) { if(IsLocked()) return; fGhostArea = gharea; }
+ void SetMinMCLabel(Int_t s) { if(IsLocked()) return; fMinMCLabel = s ; }
+ void SetRecombScheme(Int_t scheme) { if(IsLocked()) return; fRecombScheme = scheme; }
+ void SelectCollisionCandidates(UInt_t offlineTriggerMask = AliVEvent::kMB)
+ {
+ if(!fIsPSelSet)
+ {
+ fIsPSelSet = kTRUE;
+ fOfflineTriggerMask = offlineTriggerMask;
+ }
+ else
+ {
+ AliWarning("Manually setting the event selection for jet finders is not allowed! Using trigger=old_trigger | your_trigger");
+ fOfflineTriggerMask = fOfflineTriggerMask | offlineTriggerMask;
+ }
+ }
+ void SetLegacyMode(Bool_t mode) { if(IsLocked()) return; fLegacyMode = mode; }
+ void SetMCFlag(UInt_t m) { if(IsLocked()) return; fMCFlag = m ; }
+ void SelectHIJING(Bool_t s) { if(IsLocked()) return; if (s) fGeneratorIndex = 0; else fGeneratorIndex = -1; }
+ void SetGeneratorIndex(Short_t i) { if(IsLocked()) return; fGeneratorIndex = i; }
+ void SelectPhysicalPrimaries(Bool_t s) { if(IsLocked()) return;
+ if (s) fMCFlag |= AliAODMCParticle::kPhysicalPrim ;
+ else fMCFlag &= ~AliAODMCParticle::kPhysicalPrim ; }
+
+ void SetCodeDebug(Bool_t val) { fCodeDebug = val ; }
+ void SetForceIsMcPart(Bool_t b) { fIsMcPart = b ; }
+ void SetPionMassForClusters(Bool_t b) { fPionMassClusters = b ; }
- void SetAlgo(Int_t a) { fAlgo = a ; }
- void SetClusName(const char *n) { fCaloName = n ; }
- void SetJetsName(const char *n) { fJetsName = n ; }
- void SetMinJetArea(Double_t a) { fMinJetArea = a ; }
- void SetMinJetClusPt(Double_t min) { fMinJetClusPt = min ; }
- void SetMinJetPt(Double_t j) { fMinJetPt = j ; }
- void SetMinJetTrackPt(Double_t min) { fMinJetTrackPt = min ; }
- void SetRadius(Double_t r) { fRadius = r ; }
- void SetTracksName(const char *n) { fTracksName = n ; }
- void SetType(Int_t t) { fType = t ; }
- void SetEtaRange(Double_t emi, Double_t ema) {fEtaMin = emi; fEtaMax = ema; }
- void SetPhiRange(Double_t pmi, Double_t pma) {fPhiMin = pmi; fPhiMax = pma; }
+ void SetRhoName(const char *n) { fRhoName = n ; }
+ void SetRhomName(const char *n) { fRhomName = n ; }
+ void SetGenericSubtractionJetMass(Bool_t b) { fDoGenericSubtractionJetMass = b; }
+ void SetGenericSubtractionGR(Bool_t b, Double_t rmax = 2., Double_t dr = 0.04, Double_t ptmin = 0.) { fDoGenericSubtractionGR = b; fRMax=rmax; fDRStep=dr; fPtMinGR=ptmin;}
+ void SetConstituentSubtraction(Bool_t b) { fDoConstituentSubtraction = b; }
+ void SetGenericSubtractionExtraJetShapes(Bool_t b) { fDoGenericSubtractionExtraJetShapes =b;}
+ void SetUseExternalBkg(Bool_t b, Double_t rho, Double_t rhom) { fUseExternalBkg = b; fRho = rho; fRhom = rhom;}
+
+ UInt_t GetJetType() { return fJetType; }
+ Bool_t GetLegacyMode() { return fLegacyMode; }
+ TString GetJetsSubName() { return fJetsSubName; }
+
+ const char* GetJetsName() { return fJetsName.Data(); }
+ Double_t GetRadius() { return fRadius; }
+ const char* GetTracksName() { return fTracksName.Data(); }
+ const char* GetClusName() { return fCaloName.Data(); }
+ const char* GetRhoName() { return fRhoName.Data(); }
+ UInt_t GetMarkConstituents() { return fMarkConst; }
+ Double_t GetMinJetArea() { return fMinJetArea; }
+ Double_t GetMinJetClusPt() { return fMinJetClusPt; }
+ Double_t GetMinJetPt() { return fMinJetPt; }
+ Double_t GetMinJetTrackPt() { return fMinJetTrackPt; }
+ Double_t GetTrackEfficiency() { return fTrackEfficiency; }
+ Double_t GetGhostArea() { return fGhostArea; }
+ Int_t GetMinMCLabel() { return fMinMCLabel; }
+ Int_t GetRecombScheme() { return fRecombScheme; }
+ Double_t GetEtaMin() { return fEtaMin; }
+ Double_t GetEtaMax() { return fEtaMax; }
+ Double_t GetPhiMin() { return fPhiMin; }
+ Double_t GetPhiMax() { return fPhiMax; }
+ Double_t GetJetEtaMin() { return fJetEtaMin; }
+ Double_t GetJetEtaMax() { return fJetEtaMax; }
+ Double_t GetJetPhiMin() { return fJetPhiMin; }
+ Double_t GetJetPhiMax() { return fJetPhiMax; }
+
+ AliVEvent* GetEvent() { return fEvent;}
+ TClonesArray* GetClusters() { return fClus;}
+ TClonesArray* GetTracks() { return fTracks;}
+ TClonesArray* GetJets() { return fJets;}
protected:
void FindJets();
Bool_t DoInit();
+ Bool_t GetSortedArray(Int_t indexes[], std::vector<fastjet::PseudoJet> array) const;
+
+ void FillJetConstituents(std::vector<fastjet::PseudoJet>& constituents,AliEmcalJet *jet,Double_t vertex[3],UInt_t jetCount,Int_t& nt,Int_t& nc,Double_t& maxCh,Double_t& maxNe,Int_t& ncharged,Int_t& nneutral,Double_t& neutralE,Double_t& mcpt,Int_t& cemc,Double_t& emcpt,Int_t& gall,Int_t& gemc,std::vector<fastjet::PseudoJet>& constituentsunsub,Int_t flag);
+ Int_t GetIndexSub(Double_t phisub,std::vector<fastjet::PseudoJet>& constituentsunsub);
TString fTracksName; // name of track collection
TString fCaloName; // name of calo cluster collection
TString fJetsName; // name of jet collection
- Int_t fAlgo; // algo (0==kt, 1==antikt)
+ TString fJetsSubName; // name of subtracted jet collection
+ TString fTracksSubName; // name of subtracted track collection
+ TString fClusSubName; // name of subtracted clusters collection
+ UInt_t fJetType; // jet type (algorithm, radius, constituents)
+ UInt_t fConstSel; // select constituents from a previous jet finding
+ UInt_t fMCConstSel; // select MC constituents (label!=0) from a previous jet finding
+ UInt_t fMarkConst; // constituents are marked (via TObject::SetBit) as belonging to the # leading jets
Double_t fRadius; // jet radius
- Int_t fType; // jet type (0=all, 1=ch, 2=neutral)
Double_t fMinJetTrackPt; // min jet track momentum (applied before clustering)
Double_t fMinJetClusPt; // min jet cluster momentum (applied before clustering)
Double_t fPhiMin; // minimum phi for constituents (applied before clustering)
Double_t fEtaMax; // maximum eta for constituents (applied before clustering)
Double_t fMinJetArea; // min area to keep jet in output
Double_t fMinJetPt; // min jet pt to keep jet in output
+ Double_t fJetPhiMin; // minimum phi to keep jet in output
+ Double_t fJetPhiMax; // maximum phi to keep jet in output
+ Double_t fJetEtaMin; // minimum eta to keep jet in output
+ Double_t fJetEtaMax; // maximum eta to keep jet in output
+ Double_t fGhostArea; // ghost area
+ Int_t fMinMCLabel; // minimum MC label value for the tracks/clusters being considered MC particles
+ Int_t fRecombScheme; // recombination scheme used by fastjet
+ Double_t fTrackEfficiency; // artificial tracking inefficiency (0...1)
+ UInt_t fMCFlag; // select MC particles with flags
+ Short_t fGeneratorIndex; // select MC particles with generator index (default = -1 = switch off selection)
Bool_t fIsInit; //!=true if already initialized
+ Bool_t fLocked; // true if lock is set
+ Bool_t fIsPSelSet; //!=true if physics selection was set
Bool_t fIsMcPart; //!=true if MC particles are given as input
Bool_t fIsEmcPart; //!=true if emcal particles are given as input (for clusters)
+ Bool_t fIsPicoTrack; //!=true if pico tracks are given as input
+ Bool_t fIsEsdClus; //!=true if AliESDCaloCluster is given as input
+ Bool_t fLegacyMode; //! if true, enable FJ 2.x behavior
+ Bool_t fCodeDebug; // use nontested code changes
+ Bool_t fPionMassClusters; // assume pion mass for clusters
+ Bool_t fDoGenericSubtractionJetMass; // calculate generic subtraction
+ Bool_t fDoGenericSubtractionGR; // calculate generic subtraction for angular structure function GR
+ Bool_t fDoGenericSubtractionExtraJetShapes; // calculate generic subtraction for other jet shapes like radialmoment,pTD etc
+ Bool_t fDoConstituentSubtraction; // calculate constituent subtraction
+ Bool_t fUseExternalBkg; // use external background for generic subtractor
+ TString fRhoName; // name of rho
+ TString fRhomName; // name of rhom
+ Double_t fRho; // pT background density
+ Double_t fRhom; // mT background density
+ Double_t fRMax; // R max for GR calculation
+ Double_t fDRStep; // step width for GR calculation
+ Double_t fPtMinGR; // min pT for GR calculation
+
+
TClonesArray *fJets; //!jet collection
+ TClonesArray *fJetsSub; //!subtracted jet collection
+ TClonesArray *fTracksSub; //!subtracted track collection
+ TClonesArray *fClusSub; //!subtracted cluster collection
AliVEvent *fEvent; //!current event
TClonesArray *fTracks; //!tracks collection
TClonesArray *fClus; //!cluster collection
+ AliRhoParameter *fRhoParam; //!event rho
+ AliRhoParameter *fRhomParam; //!event rhom
private:
AliEmcalJetTask(const AliEmcalJetTask&); // not implemented
AliEmcalJetTask &operator=(const AliEmcalJetTask&); // not implemented
- ClassDef(AliEmcalJetTask, 4) // Jet producing task
+ ClassDef(AliEmcalJetTask, 17) // Jet producing task
};
#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff