[u/mrichter/AliRoot.git] / PWGGA / EMCALJetTasks / AliJetEmbeddingTask.h

#ifndef ALIJETEMBEDDINGTASK_H
#define ALIJETEMBEDDINGTASK_H

// $Id$

class TClonesArray;
class AliEMCALGeometry;

#include "AliAnalysisTaskSE.h"

class AliJetEmbeddingTask : public AliAnalysisTaskSE {
 public:
  AliJetEmbeddingTask();
  AliJetEmbeddingTask(const char *name); 
  virtual ~AliJetEmbeddingTask();

  void         Init();
  void         UserExec(Option_t* /*option*/);
  void         Terminate(const Option_t* /*option*/) {;}

  void         SetClusName(const char *n)            { fCaloName     = n;    }
  void         SetTracksName(const char *n)          { fTracksName   = n;    }
  void         SetEtaRange(Float_t min, Float_t max) { fEtaMin       = min;  fEtaMax = max; }
  void         SetPhiRange(Float_t min, Float_t max) { fPhiMin       = min;  fPhiMax = max; }
  void         SetPtRange(Float_t min, Float_t max)  { fPtMin        = min;  fPtMax  = max;  }
  void         SetCopyArray(Bool_t copy)             { fCopyArray    = copy; }
  void         SetNClusters(Int_t n)                 { fNEmbClusters = n;    }
  void         SetNTracks(Int_t n)                   { fNEmbTracks   = n;    }
  void         SetGeometryName(const char *n)        { fGeomName     = n;    }

 protected:

  virtual void           Embed();                 // do embedding

  TString                fGeomName;               // EMCal geometry name
  TString                fTracksName;             // name of track collection
  TString                fOutTracksName;          // name of output track collection
  TString                fCaloName;               // name of calo cluster collection
  TString                fOutCaloName;            // name of output cluster collection
  Float_t                fEtaMin;                 // eta minimum value
  Float_t                fEtaMax;                 // eta maximum value
  Float_t                fPhiMin;                 // phi minimum value
  Float_t                fPhiMax;                 // phi maximum value
  Float_t                fPtMin;                  // pt minimum value
  Float_t                fPtMax;                  // pt maximum value
  Bool_t                 fCopyArray;              // whether or not the array will be copied to a new one before embedding
  Int_t                  fNEmbClusters;           // how many clusters are being embedded
  Int_t                  fNEmbTracks;             // how many tracks are being embedded
  AliEMCALGeometry      *fGeom;                   //!pointer to EMCal geometry
  TClonesArray          *fClusters;               //!cluster collection
  TClonesArray          *fOutClusters;            //!output cluster collection
  TClonesArray          *fTracks;                 //!track collection
  TClonesArray          *fOutTracks;              //!output track collection

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

  ClassDef(AliJetEmbeddingTask, 1) // Jet embedding task
};
#endif
Commit	Line	Data
44629b14	1	#ifndef ALIJETEMBEDDINGTASK_H
	2	#define ALIJETEMBEDDINGTASK_H
	3
	4	// $Id$
	5
	6	class TClonesArray;
	7	class AliEMCALGeometry;
	8
	9	#include "AliAnalysisTaskSE.h"
	10
	11	class AliJetEmbeddingTask : public AliAnalysisTaskSE {
	12	public:
	13	AliJetEmbeddingTask();
	14	AliJetEmbeddingTask(const char *name);
	15	virtual ~AliJetEmbeddingTask();
	16
	17	void Init();
	18	void UserExec(Option_t* /option/);
	19	void Terminate(const Option_t* /option/) {;}
	20
	21	void SetClusName(const char *n) { fCaloName = n; }
	22	void SetTracksName(const char *n) { fTracksName = n; }
	23	void SetEtaRange(Float_t min, Float_t max) { fEtaMin = min; fEtaMax = max; }
	24	void SetPhiRange(Float_t min, Float_t max) { fPhiMin = min; fPhiMax = max; }
	25	void SetPtRange(Float_t min, Float_t max) { fPtMin = min; fPtMax = max; }
	26	void SetCopyArray(Bool_t copy) { fCopyArray = copy; }
	27	void SetNClusters(Int_t n) { fNEmbClusters = n; }
	28	void SetNTracks(Int_t n) { fNEmbTracks = n; }
	29	void SetGeometryName(const char *n) { fGeomName = n; }
	30
	31	protected:
	32
	33	virtual void Embed(); // do embedding
	34
	35	TString fGeomName; // EMCal geometry name
	36	TString fTracksName; // name of track collection
	37	TString fOutTracksName; // name of output track collection
	38	TString fCaloName; // name of calo cluster collection
	39	TString fOutCaloName; // name of output cluster collection
	40	Float_t fEtaMin; // eta minimum value
	41	Float_t fEtaMax; // eta maximum value
	42	Float_t fPhiMin; // phi minimum value
	43	Float_t fPhiMax; // phi maximum value
	44	Float_t fPtMin; // pt minimum value
	45	Float_t fPtMax; // pt maximum value
	46	Bool_t fCopyArray; // whether or not the array will be copied to a new one before embedding
	47	Int_t fNEmbClusters; // how many clusters are being embedded
	48	Int_t fNEmbTracks; // how many tracks are being embedded
	49	AliEMCALGeometry *fGeom; //!pointer to EMCal geometry
	50	TClonesArray *fClusters; //!cluster collection
	51	TClonesArray *fOutClusters; //!output cluster collection
	52	TClonesArray *fTracks; //!track collection
	53	TClonesArray *fOutTracks; //!output track collection
	54
	55	private:
	56	AliJetEmbeddingTask(const AliJetEmbeddingTask&); // not implemented
	57	AliJetEmbeddingTask &operator=(const AliJetEmbeddingTask&); // not implemented
	58
	59	ClassDef(AliJetEmbeddingTask, 1) // Jet embedding task
	60	};
	61	#endif