[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliEmcalJetTask.h

#ifndef ALIEMCALJETTASK_H
#define ALIEMCALJETTASK_H

// $Id$

class TClonesArray;
class AliVEvent;

namespace fastjet {
  class PseudoJet;
}

#include "AliLog.h"
#include "AliAnalysisTaskSE.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,
    kAllJets=kKT|kAKT|kFullJet|kChargedJet|kNeutralJet|kR020Jet|kR030Jet|kR040Jet|kRX1Jet|kRX2Jet|kRX3Jet
  };


  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                   SetGhostArea(Double_t gharea)    { fGhostArea      = gharea;  }
  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;
    }
  }

  UInt_t                 GetJetType()                     { return fJetType; }

  static bool            ComparePseudoJets(fastjet::PseudoJet a, fastjet::PseudoJet b);

 protected:
  void                   FindJets();
  Bool_t                 DoInit();

  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
  Double_t               fRadius;                 // jet radius
  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               fEtaMin;                 // minimum eta 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               fGhostArea;              // ghost area
  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)
  TClonesArray          *fJets;                   //!jet collection
  AliVEvent             *fEvent;                  //!current event
  TClonesArray          *fTracks;                 //!tracks collection
  TClonesArray          *fClus;                   //!cluster collection

 private:
  AliEmcalJetTask(const AliEmcalJetTask&);            // not implemented
  AliEmcalJetTask &operator=(const AliEmcalJetTask&); // not implemented

  ClassDef(AliEmcalJetTask, 7) // Jet producing task
};
#endif
Commit	Line	Data
914d486c	1	#ifndef ALIEMCALJETTASK_H
914d486c	2	#define ALIEMCALJETTASK_H
48d874ff	3
	4	// $Id$
	5
	6	class TClonesArray;
b65fac7a	7	class AliVEvent;
48d874ff	8
7030f36f	9	namespace fastjet {
	10	class PseudoJet;
	11	}
	12
d14d6e2c	13	#include "AliLog.h"
48d874ff	14	#include "AliAnalysisTaskSE.h"
48d874ff	15
914d486c	16	class AliEmcalJetTask : public AliAnalysisTaskSE {
48d874ff	17	public:
6a20534a	18
	19	enum JetType {
	20	kNone=0,
	21	kKT=1<<0,
	22	kAKT=1<<1,
	23	kFullJet=1<<2,
	24	kChargedJet=1<<3,
	25	kNeutralJet=1<<4,
	26	kR020Jet=1<<5,
	27	kR030Jet=1<<6,
	28	kR040Jet=1<<7,
	29	kRX1Jet=1<<8, // user defined radii, use SetRadius(Double_t)
	30	kRX2Jet=1<<9,
	31	kRX3Jet=1<<10,
	32	kAllJets=kKT\|kAKT\|kFullJet\|kChargedJet\|kNeutralJet\|kR020Jet\|kR030Jet\|kR040Jet\|kRX1Jet\|kRX2Jet\|kRX3Jet
	33	};
	34
	35
914d486c	36	AliEmcalJetTask();
	37	AliEmcalJetTask(const char *name);
	38	virtual ~AliEmcalJetTask();
48d874ff	39
b65fac7a	40	void UserCreateOutputObjects();
	41	void UserExec(Option_t *option);
	42	void Terminate(Option_t *option);
	43
6a20534a	44	void SelectConstituents(UInt_t constSel, UInt_t MCconstSel) { fConstSel = constSel; fMCConstSel = MCconstSel; };
6a20534a	45	void SetAlgo(Int_t a) { if (a==0) fJetType \|= kKT; else fJetType \|= kAKT; } // for backward compatibility only
b65fac7a	46	void SetClusName(const char *n) { fCaloName = n ; }
b65fac7a	47	void SetJetsName(const char *n) { fJetsName = n ; }
6a20534a	48	void SetJetType(UInt_t t) { fJetType = t ; }
7030f36f	49	void SetMarkConstituents(UInt_t m) { fMarkConst = m ; }
b65fac7a	50	void SetMinJetArea(Double_t a) { fMinJetArea = a ; }
111318e8	51	void SetMinJetClusPt(Double_t min) { fMinJetClusPt = min ; }
b65fac7a	52	void SetMinJetPt(Double_t j) { fMinJetPt = j ; }
111318e8	53	void SetMinJetTrackPt(Double_t min) { fMinJetTrackPt = min ; }
6a20534a	54	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)."); }
111318e8	55	void SetTracksName(const char *n) { fTracksName = n ; }
6a20534a	56	void SetType(Int_t t) { if (t==0) fJetType \|= kFullJet;
	57	else if (t==1) fJetType \|= kChargedJet;
	58	else if (t==2) fJetType \|= kNeutralJet; } // for backward compatibility only
04905231	59	void SetEtaRange(Double_t emi, Double_t ema) {fEtaMin = emi; fEtaMax = ema; }
04905231	60	void SetPhiRange(Double_t pmi, Double_t pma) {fPhiMin = pmi; fPhiMax = pma; }
1f9c287f	61	void SetGhostArea(Double_t gharea) { fGhostArea = gharea; }
d14d6e2c	62	void SelectCollisionCandidates(UInt_t offlineTriggerMask = AliVEvent::kMB)
	63	{
	64	if(!fIsPSelSet)
	65	{
	66	fIsPSelSet = kTRUE;
	67	fOfflineTriggerMask = offlineTriggerMask;
	68	}
	69	else
	70	{
	71	AliWarning("Manually setting the event selection for jet finders is not allowed! Using trigger=old_trigger \| your_trigger");
	72	fOfflineTriggerMask = fOfflineTriggerMask \| offlineTriggerMask;
	73	}
	74	}
6a20534a	75
	76	UInt_t GetJetType() { return fJetType; }
	77
7030f36f	78	static bool ComparePseudoJets(fastjet::PseudoJet a, fastjet::PseudoJet b);
7030f36f	79
48d874ff	80	protected:
04905231	81	void FindJets();
04905231	82	Bool_t DoInit();
48d874ff	83
900f5135	84	TString fTracksName; // name of track collection
900f5135	85	TString fCaloName; // name of calo cluster collection
7efbea04	86	TString fJetsName; // name of jet collection
6a20534a	87	UInt_t fJetType; // jet type (algorithm, radius, constituents)
	88	UInt_t fConstSel; // select constituents from a previous jet finding
	89	UInt_t fMCConstSel; // select MC constituents (label!=0) from a previous jet finding
7030f36f	90	UInt_t fMarkConst; // constituents are marked (via TObject::SetBit) as belonging to the # leading jets
48d874ff	91	Double_t fRadius; // jet radius
f5f3c8e9	92	Double_t fMinJetTrackPt; // min jet track momentum (applied before clustering)
f5f3c8e9	93	Double_t fMinJetClusPt; // min jet cluster momentum (applied before clustering)
04905231	94	Double_t fPhiMin; // minimum phi for constituents (applied before clustering)
	95	Double_t fPhiMax; // maximum phi for constituents (applied before clustering)
	96	Double_t fEtaMin; // minimum eta for constituents (applied before clustering)
	97	Double_t fEtaMax; // maximum eta for constituents (applied before clustering)
f5f3c8e9	98	Double_t fMinJetArea; // min area to keep jet in output
f5f3c8e9	99	Double_t fMinJetPt; // min jet pt to keep jet in output
1f9c287f	100	Double_t fGhostArea; // ghost area
04905231	101	Bool_t fIsInit; //!=true if already initialized
d14d6e2c	102	Bool_t fIsPSelSet; //!=true if physics selection was set
04905231	103	Bool_t fIsMcPart; //!=true if MC particles are given as input
04905231	104	Bool_t fIsEmcPart; //!=true if emcal particles are given as input (for clusters)
48d874ff	105	TClonesArray *fJets; //!jet collection
b65fac7a	106	AliVEvent *fEvent; //!current event
04905231	107	TClonesArray *fTracks; //!tracks collection
04905231	108	TClonesArray *fClus; //!cluster collection
48d874ff	109
48d874ff	110	private:
914d486c	111	AliEmcalJetTask(const AliEmcalJetTask&); // not implemented
914d486c	112	AliEmcalJetTask &operator=(const AliEmcalJetTask&); // not implemented
48d874ff	113
7030f36f	114	ClassDef(AliEmcalJetTask, 7) // Jet producing task
48d874ff	115	};
48d874ff	116	#endif