[u/mrichter/AliRoot.git] / HLT / EMCAL / AliFJWrapper.h

#ifndef AliFJWrapper_HH
#define AliFJWrapper_HH

#include <vector>
#include <fastjet/PseudoJet.hh>
#include <fastjet/JetDefinition.hh>
#include <fastjet/ClusterSequence.hh>
#include <fastjet/ClusterSequenceArea.hh>
#include <fastjet/AreaDefinition.hh>
#include <fastjet/SISConePlugin.hh>

#include "TNamed.h"

class AliFastJetHeader;

class AliFJWrapper : public TNamed
{
 public:

  AliFJWrapper(const char *_name, const char *_title);

  virtual ~AliFJWrapper();
  
  virtual void CopySettingsFrom (const AliFJWrapper&     _wrapper);
  virtual void SetupFromHeader  (const AliFastJetHeader* _header);

  virtual void AddInputVector (double _px, double _py, double _pz, double _E, int _index = -99999);

  virtual void AddInputVector (const fastjet::PseudoJet& _vec,                int _index = -99999);
  virtual void AddInputVectors(const std::vector<fastjet::PseudoJet>& _vecs,  int _offsetIndex = -99999);

  virtual Int_t  Run();
  virtual void   Clear(const Option_t* _opt = "");

  void SetStrategy     (const fastjet::Strategy _strat)             { fStrategy_ = _strat;  }
  void SetAlgorithm    (const fastjet::JetAlgorithm _algor)         { fAlgor_    = _algor;  }
  void SetRecombScheme (const fastjet::RecombinationScheme _scheme) { fScheme_   = _scheme; }
  void SetAreaType     (const fastjet::AreaType _atype)             { fAreaType_ = _atype;  }

  void SetupAlgorithmfromOpt (const char *option);
  void SetupStrategyfromOpt  (const char *option);
  void SetupSchemefromOpt    (const char *option);
  void SetupAreaTypefromOpt  (const char *option);

  void SetNRepeats      (const int    _nrepeat) { fNGhostRepeats_ = _nrepeat; }
  void SetGhostArea     (const double _gharea)  { fGhostArea_     = _gharea;  }
  void SetMaxRap        (const double _maxrap)  { fMaxRap_        = _maxrap;  }
  void SetR             (const double _r)       { fR_             = _r;       }
  void SetGridScatter   (const double _gridSc)  { fGridScatter_   = _gridSc;  }
  void SetKtScatter     (const double _ktSc)    { fKtScatter_     = _ktSc;    }
  void SetMeanGhostKt   (const double _meankt)  { fMeanGhostKt_   = _meankt;  }
  void SetPluginAlgor   (const int    _plugin)  { fPluginAlgor_   = _plugin;  }
  void SetUseArea4Vector(const bool   _useA4v)  { fUseArea4Vector_ = _useA4v; }

  virtual void   GetMedianAndSigma(double& _median, double& _sigma, int _remove = 0);
  virtual double GetMedianUsedForBgSubtraction() {return fMedianUsedForBgSubtraction_;}

  std::vector<fastjet::PseudoJet> GetInputVectors()    { return fInputVectors_;}
  std::vector<fastjet::PseudoJet> GetInclusiveJets()   { return fInclusiveJets_; }

  std::vector<fastjet::PseudoJet> GetJetConstituents(const unsigned int _idx);

  fastjet::ClusterSequenceArea*   GetClusterSequence() { return fClustSeq_;      }

  std::vector<double>             GetSubtractedJetsPts(const double _median_pt = -1, const bool _sorted = false);

  double                          GetJetArea         (const unsigned int _idx);
  double                          GetJetSubtractedPt (const unsigned int _idx);

 protected:

  AliFJWrapper();
  AliFJWrapper(const AliFJWrapper& _wrapper);
  AliFJWrapper& operator = (const AliFJWrapper& _wrapper);// {return *this;}

  std::vector<fastjet::PseudoJet> fInputVectors_;
  std::vector<fastjet::PseudoJet> fInclusiveJets_;

  std::vector<double> fSubtractedJetsPt_;
  
  fastjet::AreaDefinition        *fAreaDef_;//!
  fastjet::VoronoiAreaSpec       *fVorAreaSpec_;//!
  fastjet::GhostedAreaSpec       *fGhostedAreaSpec_;//!
  fastjet::JetDefinition         *fJetDef_;//!
  fastjet::JetDefinition::Plugin *fPlugin_;//!
  fastjet::RangeDefinition       *fRange_;//!
  fastjet::ClusterSequenceArea   *fClustSeq_;//!

  // same as in the alifjheader
  fastjet::Strategy               fStrategy_;
  fastjet::JetAlgorithm           fAlgor_;
  fastjet::RecombinationScheme    fScheme_;
  fastjet::AreaType               fAreaType_;

  int                             fNGhostRepeats_;
  double                          fGhostArea_;
  double                          fMaxRap_;
  double                          fR_;

  // not included in the header (within AliFastJetHeader)
  // no setters for the moment - used default values in the constructor
  double                          fGridScatter_;
  double                          fKtScatter_;
  double                          fMeanGhostKt_;
  int                             fPluginAlgor_;

  // extra parameters
  double                          fMedianUsedForBgSubtraction_;
  bool                            fUseArea4Vector_;

  virtual void   SubtractBackground(const double _median_pt = -1);

  ClassDef(AliFJWrapper, 0)
};

#endif
Commit	Line	Data
7634d323	1	#ifndef AliFJWrapper_HH
	2	#define AliFJWrapper_HH
	3
	4	#include <vector>
	5	#include <fastjet/PseudoJet.hh>
	6	#include <fastjet/JetDefinition.hh>
	7	#include <fastjet/ClusterSequence.hh>
	8	#include <fastjet/ClusterSequenceArea.hh>
	9	#include <fastjet/AreaDefinition.hh>
	10	#include <fastjet/SISConePlugin.hh>
	11
	12	#include "TNamed.h"
	13
	14	class AliFastJetHeader;
	15
	16	class AliFJWrapper : public TNamed
	17	{
	18	public:
	19
	20	AliFJWrapper(const char _name, const char _title);
	21
	22	virtual ~AliFJWrapper();
	23
	24	virtual void CopySettingsFrom (const AliFJWrapper& _wrapper);
	25	virtual void SetupFromHeader (const AliFastJetHeader* _header);
	26
	27	virtual void AddInputVector (double _px, double _py, double _pz, double _E, int _index = -99999);
	28
	29	virtual void AddInputVector (const fastjet::PseudoJet& _vec, int _index = -99999);
	30	virtual void AddInputVectors(const std::vector<fastjet::PseudoJet>& _vecs, int _offsetIndex = -99999);
	31
	32	virtual Int_t Run();
	33	virtual void Clear(const Option_t* _opt = "");
	34
	35	void SetStrategy (const fastjet::Strategy _strat) { fStrategy_ = _strat; }
	36	void SetAlgorithm (const fastjet::JetAlgorithm _algor) { fAlgor_ = _algor; }
	37	void SetRecombScheme (const fastjet::RecombinationScheme _scheme) { fScheme_ = _scheme; }
	38	void SetAreaType (const fastjet::AreaType _atype) { fAreaType_ = _atype; }
	39
ab9926fa	40	void SetupAlgorithmfromOpt (const char *option);
	41	void SetupStrategyfromOpt (const char *option);
	42	void SetupSchemefromOpt (const char *option);
	43	void SetupAreaTypefromOpt (const char *option);
	44
7634d323	45	void SetNRepeats (const int _nrepeat) { fNGhostRepeats_ = _nrepeat; }
	46	void SetGhostArea (const double _gharea) { fGhostArea_ = _gharea; }
	47	void SetMaxRap (const double _maxrap) { fMaxRap_ = _maxrap; }
	48	void SetR (const double _r) { fR_ = _r; }
	49	void SetGridScatter (const double _gridSc) { fGridScatter_ = _gridSc; }
	50	void SetKtScatter (const double _ktSc) { fKtScatter_ = _ktSc; }
	51	void SetMeanGhostKt (const double _meankt) { fMeanGhostKt_ = _meankt; }
	52	void SetPluginAlgor (const int _plugin) { fPluginAlgor_ = _plugin; }
	53	void SetUseArea4Vector(const bool _useA4v) { fUseArea4Vector_ = _useA4v; }
	54
	55	virtual void GetMedianAndSigma(double& _median, double& _sigma, int _remove = 0);
	56	virtual double GetMedianUsedForBgSubtraction() {return fMedianUsedForBgSubtraction_;}
	57
	58	std::vector<fastjet::PseudoJet> GetInputVectors() { return fInputVectors_;}
	59	std::vector<fastjet::PseudoJet> GetInclusiveJets() { return fInclusiveJets_; }
	60
	61	std::vector<fastjet::PseudoJet> GetJetConstituents(const unsigned int _idx);
	62
	63	fastjet::ClusterSequenceArea* GetClusterSequence() { return fClustSeq_; }
	64
	65	std::vector<double> GetSubtractedJetsPts(const double _median_pt = -1, const bool _sorted = false);
	66
	67	double GetJetArea (const unsigned int _idx);
	68	double GetJetSubtractedPt (const unsigned int _idx);
	69
	70	protected:
	71
	72	AliFJWrapper();
	73	AliFJWrapper(const AliFJWrapper& _wrapper);
	74	AliFJWrapper& operator = (const AliFJWrapper& _wrapper);// {return *this;}
	75
	76	std::vector<fastjet::PseudoJet> fInputVectors_;
	77	std::vector<fastjet::PseudoJet> fInclusiveJets_;
	78
	79	std::vector<double> fSubtractedJetsPt_;
	80
	81	fastjet::AreaDefinition *fAreaDef_;//!
	82	fastjet::VoronoiAreaSpec *fVorAreaSpec_;//!
	83	fastjet::GhostedAreaSpec *fGhostedAreaSpec_;//!
	84	fastjet::JetDefinition *fJetDef_;//!
	85	fastjet::JetDefinition::Plugin *fPlugin_;//!
	86	fastjet::RangeDefinition *fRange_;//!
	87	fastjet::ClusterSequenceArea *fClustSeq_;//!
	88
	89	// same as in the alifjheader
	90	fastjet::Strategy fStrategy_;
	91	fastjet::JetAlgorithm fAlgor_;
	92	fastjet::RecombinationScheme fScheme_;
	93	fastjet::AreaType fAreaType_;
	94
	95	int fNGhostRepeats_;
	96	double fGhostArea_;
	97	double fMaxRap_;
	98	double fR_;
	99
	100	// not included in the header (within AliFastJetHeader)
	101	// no setters for the moment - used default values in the constructor
	102	double fGridScatter_;
	103	double fKtScatter_;
	104	double fMeanGhostKt_;
	105	int fPluginAlgor_;
	106
	107	// extra parameters
	108	double fMedianUsedForBgSubtraction_;
109	bool fUseArea4Vector_;
110
111	virtual void SubtractBackground(const double _median_pt = -1);
112
113	ClassDef(AliFJWrapper, 0)
114	};
115
116	#endif