#ifndef ALIEMCALJETTASK_H
#define ALIEMCALJETTASK_H
-// $Id$
-
class TClonesArray;
+class TObjArray;
class AliVEvent;
+class AliEMCALGeometry;
+class AliEmcalJetUtility;
+class AliFJWrapper;
namespace fastjet {
class PseudoJet;
#include "AliLog.h"
#include "AliAnalysisTaskSE.h"
+#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);
-
- void SelectConstituents(UInt_t constSel, UInt_t MCconstSel) { fConstSel = constSel; fMCConstSel = MCconstSel; };
- void SetAlgo(Int_t a) { if (a==0) fJetType |= kKT; else fJetType |= kAKT; } // for backward compatibility only
- void SetClusName(const char *n) { fCaloName = n ; }
- void SetJetsName(const char *n) { fJetsName = n ; }
- void SetJetType(UInt_t t) { fJetType = t ; }
- void SetMarkConstituents(UInt_t m) { fMarkConst = m ; }
- 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 ; 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 SetTracksName(const char *n) { fTracksName = n ; }
- void SetType(Int_t t) { 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) {fEtaMin = emi; fEtaMax = ema; }
- void SetPhiRange(Double_t pmi, Double_t pma) {fPhiMin = pmi; fPhiMax = pma; }
- void SetJetEtaRange(Double_t emi, Double_t ema) {fJetEtaMin = emi; fJetEtaMax = ema; }
- void SetJetPhiRange(Double_t pmi, Double_t pma) {fJetPhiMin = pmi; fJetPhiMax = pma; }
- void SetGhostArea(Double_t gharea) { fGhostArea = gharea; }
- void SetMinMCLabel(Int_t s) { fMinMCLabel = s ; }
- 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)
{
}
}
- UInt_t GetJetType() { return fJetType; }
+ 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;
TString fTracksName; // name of track collection
TString fCaloName; // name of calo cluster collection
TString fJetsName; // name of 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 (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; //!=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
AliVEvent *fEvent; //!current event
TClonesArray *fTracks; //!tracks collection
TClonesArray *fClus; //!cluster collection
+ AliFJWrapper fFastJetWrapper; //!fastjet wrapper
private:
AliEmcalJetTask(const AliEmcalJetTask&); // not implemented
AliEmcalJetTask &operator=(const AliEmcalJetTask&); // not implemented
- ClassDef(AliEmcalJetTask, 9) // Jet producing task
+ ClassDef(AliEmcalJetTask, 17) // Jet producing task
};
#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff