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

#ifndef ALIEMCALJETFINDER_H
#define ALIEMCALJETFINDER_H

// $Id$

namespace fastjet {
  class PseudoJet;
}

class AliEmcalJet;
class AliFJWrapper;
class TNamed;
class TH1;

class AliEmcalJetFinder : public TNamed
{
  public:
    AliEmcalJetFinder();
    AliEmcalJetFinder(const char* name);
    ~AliEmcalJetFinder();
    
    Bool_t                        FindJets();
    void                          AddInputVector(Double_t px, Double_t py, Double_t pz);
    void                          FillPtHistogram(TH1* histogram);
    void                          FillEtaHistogram(TH1* histogram);
    void                          FillPhiHistogram(TH1* histogram);

    Int_t                         GetJets(AliEmcalJet*& jets)     {jets = fJetArray[0]; return fJetCount;}
    AliEmcalJet*                  GetJet(Int_t index)             {return fJetArray[index];}
    void                          SetGhostArea(Double_t val)      {fGhostArea = val;}
    void                          SetRadius(Double_t val)         {fRadius = val;}
    void                          SetJetAlgorithm(Int_t val)      {fJetAlgorithm = val;}
    void                          SetTrackMaxEta(Double_t val)    {fTrackMaxEta = val;}
    void                          SetJetMaxEta(Double_t val)      {fJetMaxEta = val;}
    void                          SetJetMinPt(Double_t val)       {fJetMinPt = val;}
    void                          SetJetMinArea(Double_t val)     {fJetMinArea = val;}

    void                          SetManualIndex(Int_t val)       {fInputVectorIndex = val;}
  private:
    // General properties
    AliFJWrapper*                 fFastjetWrapper;                // Interface object to fastjet
    Int_t                         fInputVectorIndex;              // Current index of input vectors (by default: count of vectors)
    Int_t                         fJetCount;                      // Found jets within the given acceptances
    std::vector<AliEmcalJet*>     fJetArray;                      // Internal array for the jets
    // Settings for fastjet
    Double_t                      fGhostArea;                     // setting for ghost area in FJ
    Double_t                      fRadius;                        // Radius parameter
    Int_t                         fJetAlgorithm;                  // var for algorithm (0=antikt, 1=kt)
    Double_t                      fTrackMaxEta;                   // cut for |track-eta| < fTrackMaxEta
    // Jet cuts
    Double_t                      fJetMaxEta;                     // cut for |jet-eta| < fJetMaxEta
    Double_t                      fJetMinPt;                      // cut for  jet-pT > fJetMinPt
    Double_t                      fJetMinArea;                    // cut for  jet-area > fJetMinArea 

    ClassDef(AliEmcalJetFinder, 1); // Lightweight fastjet implementation outside analysis tasks
};

#endif
Commit	Line	Data
14ca6380	1	#ifndef ALIEMCALJETFINDER_H
	2	#define ALIEMCALJETFINDER_H
	3
	4	// $Id$
	5
	6	namespace fastjet {
	7	class PseudoJet;
	8	}
	9
	10	class AliEmcalJet;
	11	class AliFJWrapper;
	12	class TNamed;
	13	class TH1;
	14
	15	class AliEmcalJetFinder : public TNamed
	16	{
	17	public:
	18	AliEmcalJetFinder();
	19	AliEmcalJetFinder(const char* name);
	20	~AliEmcalJetFinder();
	21
	22	Bool_t FindJets();
	23	void AddInputVector(Double_t px, Double_t py, Double_t pz);
	24	void FillPtHistogram(TH1* histogram);
	25	void FillEtaHistogram(TH1* histogram);
	26	void FillPhiHistogram(TH1* histogram);
	27
	28	Int_t GetJets(AliEmcalJet*& jets) {jets = fJetArray[0]; return fJetCount;}
	29	AliEmcalJet* GetJet(Int_t index) {return fJetArray[index];}
	30	void SetGhostArea(Double_t val) {fGhostArea = val;}
	31	void SetRadius(Double_t val) {fRadius = val;}
	32	void SetJetAlgorithm(Int_t val) {fJetAlgorithm = val;}
	33	void SetTrackMaxEta(Double_t val) {fTrackMaxEta = val;}
	34	void SetJetMaxEta(Double_t val) {fJetMaxEta = val;}
	35	void SetJetMinPt(Double_t val) {fJetMinPt = val;}
	36	void SetJetMinArea(Double_t val) {fJetMinArea = val;}
	37
	38	void SetManualIndex(Int_t val) {fInputVectorIndex = val;}
	39	private:
	40	// General properties
2eedf9a5	41	AliFJWrapper* fFastjetWrapper; // Interface object to fastjet
	42	Int_t fInputVectorIndex; // Current index of input vectors (by default: count of vectors)
	43	Int_t fJetCount; // Found jets within the given acceptances
	44	std::vector<AliEmcalJet*> fJetArray; // Internal array for the jets
14ca6380	45	// Settings for fastjet
2eedf9a5	46	Double_t fGhostArea; // setting for ghost area in FJ
	47	Double_t fRadius; // Radius parameter
	48	Int_t fJetAlgorithm; // var for algorithm (0=antikt, 1=kt)
	49	Double_t fTrackMaxEta; // cut for \|track-eta\| < fTrackMaxEta
14ca6380	50	// Jet cuts
2eedf9a5	51	Double_t fJetMaxEta; // cut for \|jet-eta\| < fJetMaxEta
	52	Double_t fJetMinPt; // cut for jet-pT > fJetMinPt
	53	Double_t fJetMinArea; // cut for jet-area > fJetMinArea
14ca6380	54
	55	ClassDef(AliEmcalJetFinder, 1); // Lightweight fastjet implementation outside analysis tasks
	56	};
	57
	58	#endif