[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;
    Int_t                         fInputVectorIndex;
    Int_t                         fJetCount;
    std::vector<AliEmcalJet*>     fJetArray;
    // Settings for fastjet
    Double_t                      fGhostArea;
    Double_t                      fRadius;
    Int_t                         fJetAlgorithm;
    Double_t                      fTrackMaxEta;
    // Jet cuts
    Double_t                      fJetMaxEta;
    Double_t                      fJetMinPt;
    Double_t                      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
	41	AliFJWrapper* fFastjetWrapper;
	42	Int_t fInputVectorIndex;
	43	Int_t fJetCount;
	44	std::vector<AliEmcalJet*> fJetArray;
	45	// Settings for fastjet
	46	Double_t fGhostArea;
	47	Double_t fRadius;
	48	Int_t fJetAlgorithm;
	49	Double_t fTrackMaxEta;
	50	// Jet cuts
	51	Double_t fJetMaxEta;
	52	Double_t fJetMinPt;
	53	Double_t fJetMinArea;
	54
	55	ClassDef(AliEmcalJetFinder, 1); // Lightweight fastjet implementation outside analysis tasks
	56	};
	57
	58	#endif