new class

[u/mrichter/AliRoot.git] / PWG4 / AliAnaGammaDirect.h

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

/* History of cvs commits:
 *
 * $Log$
 *
 */

//_________________________________________________________________________

// Class for the analysis of gamma  (gamma-jet, 
// gamma-hadron(Arleo, TODO))
// This class only contains 3 methods: one to fill lists of particles (ESDs) comming 
//  from the CTS (ITS+TPC) and the calorimeters: The other search in the 
//  corresponing calorimeter for the highest energy cluster, identify it as 
//  prompt photon(Shower Shape (done) and Isolation Cut (TODO in new class?)).
//
//  Class created from old AliPHOSGammaJet
//  (see AliRoot versions previous Release 4-09)

//*-- Author: Gustavo Conesa (INFN-LNF)

// --- ROOT system ---
#include <TROOT.h>
#include <TChain.h>
#include "TTask.h"
#include "TArrayD.h"
#include "TChain.h"
#include <TH2F.h>
#include <TTree.h> 
#include <TParticle.h> 
#include "AliAnalysisTask.h" 

class AliESD ; 
 
class AliAnaGammaDirect : public AliAnalysisTask {

public: 

  AliAnaGammaDirect(const char *name) ; // default ctor
  AliAnaGammaDirect(const AliAnaGammaDirect & g) ; // cpy ctor
  virtual ~AliAnaGammaDirect() ; //virtual dtor
  virtual void Exec(Option_t * opt = "") ;
  virtual void Init(Option_t * opt = "");
  virtual void Terminate(Option_t * opt = "");
  
  TTree *     GetChain()                const {return fChain ; }
  AliESD *    GetESD()                  const {return fESD ; }
  TObjArray * GetOutputContainer()      const {return fOutputContainer ; }
  Double_t  GetMinGammaPt()    const {return fMinGammaPt ; }
  TString    GetCalorimeter()       const {return fCalorimeter ; }
  Bool_t      GetPrintInfo()           const {return fPrintInfo ; }
  Float_t     GetConeSize()          const {return fConeSize ; }
  Float_t     GetPtThreshold()      const {return fPtThreshold ; }
  Float_t     GetPtSumThres()     const {return fPtSumThreshold ; }
  Int_t        GetICMethod()          const {return fMakeICMethod ; }

  Bool_t   IsEMCALPIDOn() const {return fEMCALPID ; }
  Bool_t   IsPHOSPIDOn() const {return fPHOSPID ; }

  void Print(const Option_t * opt)const;

  void SetMinGammaPt(Double_t ptcut){fMinGammaPt =ptcut;}
  void SetCalorimeter(TString calo){ fCalorimeter= calo ; }
  void SetPrintInfo(Bool_t print){ fPrintInfo = print ; }
  void SetConeSize(Float_t r)              {fConeSize = r ; }
  void SetPtThreshold(Float_t pt)        {fPtThreshold = pt; };
  void SetPtSumThreshold(Float_t pt) {fPtSumThreshold = pt; };
  void SetICMethod(Int_t i )          {fMakeICMethod = i ; }
  
  void SetConeSizes(Int_t i, Float_t r)              {fConeSizes[i] = r ; }
  void SetPtThresholds(Int_t i, Float_t pt)        {fPtThresholds[i] = pt; };

  void SetEMCALPIDOn(Bool_t pid){ fEMCALPID= pid ; }
  void SetPHOSPIDOn(Bool_t pid){ fPHOSPID= pid ; }

  void CreateParticleList(TClonesArray * particleList, 
			  TClonesArray * plCh, TClonesArray * plNe, 
			  TClonesArray * plNePHOS);
  
  
  void GetPromptGamma(TClonesArray * plNe, TClonesArray * plCTS, TParticle * pGamma, Bool_t &Is)  const;
  
  void MakeIsolationCut(TClonesArray * plCTS, TClonesArray * plNe, 
			TParticle *pCandidate, Int_t index, 
			Bool_t &imcpt, Bool_t &icms, Float_t &ptsum) const ;
  
  void MakeHistos() ;
  
  
 private:

  TTree       *fChain ;   //!pointer to the analyzed TTree or TChain
  AliESD       *fESD ;     //! Declaration of leave types
  TObjArray  *fOutputContainer ; //! output data container
  Bool_t        fPrintInfo ;      //Print most interesting information on screen
  Double_t    fMinGammaPt ;  // Min pt in Calorimeter
  TString      fCalorimeter ; //PHOS or EMCAL detects Gamma
  Bool_t       fEMCALPID ;//Fill EMCAL particle lists with particles with corresponding pid
  Bool_t       fPHOSPID;  //Fill PHOS particle lists with particles with corresponding pid
  Float_t      fConeSize ; //Size of the isolation cone 
  Float_t      fPtThreshold ; //Mimium pt of the particles in the cone to set isolation
  Float_t      fPtSumThreshold ; //Mimium pt sum of the particles in the cone to set isolation  
  Int_t         fNCones   ; //Number of cone sizes to test
  Int_t         fNPtThres ; //Number of ptThres to test
  Float_t     fConeSizes[10] ; // Arrat with cones to test
  Float_t     fPtThresholds[10] ; // Array with pt thresholds to test
  Int_t        fMakeICMethod ; //Isolation cut method to be used
                                           // 0: No isolation
                                           // 1: Pt threshold method
                                           // 2: Cone pt sum method
                                           // 3: Study both methods for several cones and pt.
  //Histograms  
  TH1F * fhNGamma    ; 
  TH2F * fhPhiGamma    ; 
  TH2F * fhEtaGamma    ; 
  TH1F * fhPtCandidate ;
  TH1F* fhPtThresIsolated[10][10] ;
  TH2F* fhPtSumIsolated[10] ;

  ClassDef(AliAnaGammaDirect,0)
} ;
 

#endif //ALIANAGAMMADIRECT_H