[u/mrichter/AliRoot.git] / PWG4 / CaloCalib / AliAnalysisTaskPHOSPi0CalibSelection.h

#ifndef ALIANALYSISTASKPHOSPI0CALIBSELECTION_H
#define ALIANALYSISTASKPHOSPI0CALIBSELECTION_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//---------------------------------------------------------------------------// 
// Fill histograms with two-cluster invariant mass                           //
// using calibration coefficients of the previous iteration.                 //
//---------------------------------------------------------------------------//

// Root includes
#include "TH1.h"

// AliRoot includes
#include "AliAnalysisTaskSE.h"
#include "AliPHOSGeometry.h"
#include "AliAODCaloCluster.h"
#include "AliAODCaloCells.h"
#include "AliPHOSCalibData.h"

class AliAnalysisTaskPHOSPi0CalibSelection : public AliAnalysisTaskSE
{
public:

  AliAnalysisTaskPHOSPi0CalibSelection();
  AliAnalysisTaskPHOSPi0CalibSelection(const char* name);
  AliAnalysisTaskPHOSPi0CalibSelection(const AliAnalysisTaskPHOSPi0CalibSelection&); 
  AliAnalysisTaskPHOSPi0CalibSelection& operator=(const AliAnalysisTaskPHOSPi0CalibSelection&); 
  virtual ~AliAnalysisTaskPHOSPi0CalibSelection();

  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t * opt);
  
  void SetClusterMinEnergy(Float_t emin) {fEmin=emin;}
  void SetLogWeight(Float_t weight) {fLogWeight=weight;}
  void SetCalibCorrections(AliPHOSCalibData* cdata);
  void CreateAODFromESD();
  void CreateAODFromAOD();	

  void CopyAOD(Bool_t copy)   { fCopyAOD = copy ; }
  Bool_t IsAODCopied() const { return fCopyAOD ; }

private:

  void MaxEnergyCellPos(AliAODCaloCells *cells, AliAODCaloCluster* clu, Int_t& maxId);

private:

  TList* fOutputContainer;
  TH1F*  fHmpi0[5][64][56];// two-cluster inv. mass assigned to each cell.

  AliPHOSGeometry * fPhosGeo;   // PHOS geometry
  AliPHOSCalibData* fCalibData; // corrections to CC from the previous iteration
	
  TH1F*   fHmgg; //two-cluster inv.mass
  Float_t fEmin; // min. cluster energy
  Float_t fLogWeight; // log weight used in cluster recalibration
  Bool_t  fCopyAOD; // Copy calo information only to AOD?
	
  ClassDef(AliAnalysisTaskPHOSPi0CalibSelection,3);

};

#endif //ALIANALYSISTASKPHOSPI0CALIBSELECTION_H
Commit	Line	Data
e727d2f7	1	#ifndef ALIANALYSISTASKPHOSPI0CALIBSELECTION_H
	2	#define ALIANALYSISTASKPHOSPI0CALIBSELECTION_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//---------------------------------------------------------------------------//
	8	// Fill histograms with two-cluster invariant mass //
	9	// using calibration coefficients of the previous iteration. //
	10	//---------------------------------------------------------------------------//
	11
4b4efabc	12	// Root includes
4b4efabc	13	#include "TH1.h"
e727d2f7	14
4b4efabc	15	// AliRoot includes
e727d2f7	16	#include "AliAnalysisTaskSE.h"
e727d2f7	17	#include "AliPHOSGeometry.h"
	18	#include "AliAODCaloCluster.h"
	19	#include "AliAODCaloCells.h"
4b4efabc	20	#include "AliPHOSCalibData.h"
e727d2f7	21
	22	class AliAnalysisTaskPHOSPi0CalibSelection : public AliAnalysisTaskSE
	23	{
	24	public:
	25
	26	AliAnalysisTaskPHOSPi0CalibSelection();
	27	AliAnalysisTaskPHOSPi0CalibSelection(const char* name);
29a41d05	28	AliAnalysisTaskPHOSPi0CalibSelection(const AliAnalysisTaskPHOSPi0CalibSelection&);
29a41d05	29	AliAnalysisTaskPHOSPi0CalibSelection& operator=(const AliAnalysisTaskPHOSPi0CalibSelection&);
e727d2f7	30	virtual ~AliAnalysisTaskPHOSPi0CalibSelection();
	31
	32	// Implementation of interface methods
	33	virtual void UserCreateOutputObjects();
	34	virtual void UserExec(Option_t * opt);
	35
	36	void SetClusterMinEnergy(Float_t emin) {fEmin=emin;}
4b4efabc	37	void SetLogWeight(Float_t weight) {fLogWeight=weight;}
4b4efabc	38	void SetCalibCorrections(AliPHOSCalibData* cdata);
7a4cf423	39	void CreateAODFromESD();
7a4cf423	40	void CreateAODFromAOD();
e727d2f7	41
7a4cf423	42	void CopyAOD(Bool_t copy) { fCopyAOD = copy ; }
	43	Bool_t IsAODCopied() const { return fCopyAOD ; }
	44
	45	private:
e727d2f7	46
	47	void MaxEnergyCellPos(AliAODCaloCells cells, AliAODCaloCluster clu, Int_t& maxId);
	48
	49	private:
	50
	51	TList* fOutputContainer;
	52	TH1F* fHmpi0[5][64][56];// two-cluster inv. mass assigned to each cell.
	53
e727d2f7	54	AliPHOSGeometry * fPhosGeo; // PHOS geometry
4b4efabc	55	AliPHOSCalibData* fCalibData; // corrections to CC from the previous iteration
4b4efabc	56
7a4cf423	57	TH1F* fHmgg; //two-cluster inv.mass
e727d2f7	58	Float_t fEmin; // min. cluster energy
4b4efabc	59	Float_t fLogWeight; // log weight used in cluster recalibration
7a4cf423	60	Bool_t fCopyAOD; // Copy calo information only to AOD?
4b4efabc	61
7a4cf423	62	ClassDef(AliAnalysisTaskPHOSPi0CalibSelection,3);
e727d2f7	63
	64	};
	65
	66	#endif //ALIANALYSISTASKPHOSPI0CALIBSELECTION_H