#define ALIEMCALJETTASK_H
class TClonesArray;
+class TObjArray;
class AliVEvent;
-class AliRhoParameter;
+class AliEMCALGeometry;
+class AliEmcalJetUtility;
+class AliFJWrapper;
namespace fastjet {
class PseudoJet;
#include "AliFJWrapper.h"
#include "AliAODMCParticle.h"
#include "AliEmcalJet.h"
+
class AliEmcalJetTask : public AliAnalysisTaskSE {
public:
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 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)
+ void UserCreateOutputObjects();
+ void UserExec(Option_t* option);
+ void Terminate(Option_t* /*option*/) {;}
+
+ 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 SetClusLabelRange(Int_t min, Int_t max) { if (IsLocked()) return; fMinLabelClusters = min ; fMaxLabelClusters = max; }
+ void SetEtaRange(Double_t emi, Double_t ema) { if (IsLocked()) return; fEtaMin = emi ; fEtaMax = ema ; }
+ void SetGhostArea(Double_t gharea) { if (IsLocked()) return; fGhostArea = gharea; }
+ void SetJetsName(const char *n) { if (IsLocked()) return; fJetsName = n ; }
+ 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 SetJetType(UInt_t t) { if (IsLocked()) return; fJetType = t ; }
+ void SetLocked() { fLocked = kTRUE;}
+ 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 SetMinMCLabel(Int_t s) { if (IsLocked()) return; fMinMCLabel = s ; }
+ void SetPhiRange(Double_t pmi, Double_t pma) { if (IsLocked()) return; fPhiMin = pmi ; fPhiMax = pma; }
+ void SetRecombScheme(Int_t scheme) { if (IsLocked()) return; fRecombScheme = scheme; }
+ void SetTracksName(const char *n) { if (IsLocked()) return; fTracksName = n ; }
+ void SetTrackEfficiency(Double_t t) { if (IsLocked()) return; fTrackEfficiency = t ; }
+ void SetTrackLabelRange(Int_t min, Int_t max) { if (IsLocked()) return; fMinLabelTracks = min ; fMaxLabelTracks = max; }
+ 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 ; }
+
+ AliEmcalJetUtility* AddUtility(AliEmcalJetUtility* utility);
+
+ const char* GetClusName() { return fCaloName.Data() ; }
+ Double_t GetEtaMin() { return fEtaMin ; }
+ Double_t GetEtaMax() { return fEtaMax ; }
+ Double_t GetGhostArea() { return fGhostArea ; }
+ const char* GetJetsName() { return fJetsName.Data() ; }
+ Double_t GetJetEtaMin() { return fJetEtaMin ; }
+ Double_t GetJetEtaMax() { return fJetEtaMax ; }
+ Double_t GetJetPhiMin() { return fJetPhiMin ; }
+ Double_t GetJetPhiMax() { return fJetPhiMax ; }
+ UInt_t GetJetType() { return fJetType ; }
+ Bool_t GetLegacyMode() { return fLegacyMode ; }
+ Double_t GetMinJetArea() { return fMinJetArea ; }
+ Double_t GetMinJetClusPt() { return fMinJetClusPt ; }
+ Double_t GetMinJetPt() { return fMinJetPt ; }
+ Double_t GetMinJetTrackPt() { return fMinJetTrackPt ; }
+ Int_t GetMinMCLabel() { return fMinMCLabel ; }
+ Double_t GetPhiMin() { return fPhiMin ; }
+ Double_t GetPhiMax() { return fPhiMax ; }
+ Double_t GetRadius() { return fRadius ; }
+ Int_t GetRecombScheme() { return fRecombScheme ; }
+ const char* GetTracksName() { return fTracksName.Data() ; }
+ Double_t GetTrackEfficiency() { return fTrackEfficiency ; }
+
+ AliVEvent* GetEvent() { return fEvent ; }
+ TClonesArray* GetClusters() { return fClus ; }
+ TClonesArray* GetTracks() { return fTracks ; }
+ TClonesArray* GetJets() { return fJets ; }
+ TObjArray* GetUtilities() { return fUtilities ; }
+
+ void FillJetConstituents(AliEmcalJet *jet, std::vector<fastjet::PseudoJet>& constituents, TClonesArray *tracks, TClonesArray *clusters,
+ std::vector<fastjet::PseudoJet>& constituents_sub, Int_t flag = 0, TClonesArray *particles_sub = 0);
+
+ Int_t GetIndexSub(Double_t phi_sub, std::vector<fastjet::PseudoJet>& constituents_unsub);
+
+ Bool_t IsLocked()
+ {
+ if (fLocked) {
+ AliFatal("Jet finder task is locked! Changing properties is not allowed.");
+ return kTRUE;
+ }
+ else {
+ return kFALSE;
+ }
+ }
+
+ void SelectCollisionCandidates(UInt_t offlineTriggerMask = AliVEvent::kMB)
{
if(!fIsPSelSet)
{
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 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;}
+
+ 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 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 SelectPhysicalPrimaries(Bool_t s)
+ {
+ if(IsLocked()) return;
+ if (s) fMCFlag |= AliAODMCParticle::kPhysicalPrim;
+ else fMCFlag &= ~AliAODMCParticle::kPhysicalPrim;
+ }
protected:
+
void FindJets();
+ void FillJetBranch();
Bool_t DoInit();
+ void InitUtilities();
+ void PrepareUtilities();
+ void ExecuteUtilities(AliEmcalJet* jet, Int_t ij);
+ void TerminateUtilities();
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],Int_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);
TString fTracksName; // name of track collection
TString fCaloName; // name of calo cluster collection
TString fJetsName; // name of jet collection
- TString fJetsSubName; // name of subtracted jet 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
+ Int_t fMinLabelTracks; // select track constituents with a minimum label index (track->GetLabel())
+ Int_t fMaxLabelTracks; // select track constituents with a maximum label index (track->GetLabel())
+ Int_t fMinLabelClusters; // select cluster constituents with a minimum label index (cluster->GetLabel())
+ Int_t fMaxLabelClusters; // select cluster constituents with a maximum label index (cluster->GetLabel())
+ Int_t fMinMCLabel; // minimum MC label value for the tracks/clusters being considered MC particles
Double_t fRadius; // jet radius
- Double_t fMinJetTrackPt; // min jet track momentum (applied before clustering)
+ 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 fPhiMax; // maximum phi for constituents (applied before clustering)
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
+ UInt_t fMCFlag; // select MC particles with flags (e.g. to select physical primaries)
+ Short_t fGeneratorIndex; // select MC particles with generator index (default = -1 to switch off selection)
+ TObjArray *fUtilities; // jet utilities (gen subtractor, constituent subtractor etc.)
Bool_t fLocked; // true if lock is set
+
+ Bool_t fIsInit; //!=true if already initialized
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 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
-
+ Bool_t fLegacyMode; //!=true to enable FJ 2.x behavior
+ Double_t fVertex[3]; //!vertex of the current event
+ AliEMCALGeometry *fGeom; //!emcal geometry object
TClonesArray *fJets; //!jet collection
- TClonesArray *fJetsSub; //!subtracted jet collection
AliVEvent *fEvent; //!current event
TClonesArray *fTracks; //!tracks collection
TClonesArray *fClus; //!cluster collection
- AliRhoParameter *fRhoParam; //!event rho
- AliRhoParameter *fRhomParam; //!event rhom
+ AliFJWrapper fFastJetWrapper; //!fastjet wrapper
private:
AliEmcalJetTask(const AliEmcalJetTask&); // not implemented
AliEmcalJetTask &operator=(const AliEmcalJetTask&); // not implemented
- ClassDef(AliEmcalJetTask, 16) // Jet producing task
+ ClassDef(AliEmcalJetTask, 17) // Jet producing task
};
#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff