Merging gsi train and aliroot trunk

[u/mrichter/AliRoot.git] / PWG4 / GammaConv / ConvCorrelations / AliAnalysisTaskGammaJet.h

/* This file is property of and copyright                                 *\r
 * ALICE Experiment at CERN, All rights reserved.                         *\r
 * See cxx source for full Copyright notice                               */\r
\r
/// @file   AliAnalysisTaskGammaJet.h\r
/// @author Svein Lindal\r
/// @brief  Class used to run isolation studies of conversion gamma / pions\r
 \r
\r
\r
#ifndef ALIANALYSISTASKGAMMAJET_H\r
#define ALIANALYSISTASKGAMMAJET_H\r
\r
\r
\r
class TH1F;\r
class AliESDEvent;\r
class AliGammaConversionAODObject;\r
class AliAODConversionPhoton;\r
class AliAODPWG4ParticleCorrelation;\r
class AliAODPWG4Particle;\r
class TObjArray;\r
class TString;\r
class TClonesArray;\r
\r
\r
#include "AliAnaConvIsolation.h"\r
#include "AliAnalysisTaskSE.h"\r
\r
class AliAnalysisTaskGammaJet : public AliAnalysisTaskSE {\r
\r
public:\r
 \r
  AliAnalysisTaskGammaJet(); \r
  AliAnalysisTaskGammaJet(const char *name);\r
  virtual ~AliAnalysisTaskGammaJet();\r
  \r
  virtual void   UserCreateOutputObjects();\r
  virtual void   UserExec(Option_t *option);\r
  virtual void   Terminate(Option_t *);\r
\r
  void SetDeltaAODFileName(TString string) { fDeltaAODFileName = string;}\r
 \r
  AliAnaConvIsolation * GetIsolation() const {return fAnaIsolation;}\r
  void SetIsolation( AliAnaConvIsolation * isolation) { fAnaIsolation = isolation; }\r
\r
  void SetGammaCutId(TString cut) { fGammaCutString = Form("GammaConv_%s", cut.Data());}\r
  void SetPionCutId(TString cut) { fPionCutString = Form("GammaConv_%s", cut.Data());}\r
\r
  void SetMinPt(Float_t minpt) { fMinPt = minpt;}\r
  void SetMinNTracks(Int_t nTracks) { fMinNTracks = nTracks; }\r
\r
  void AddIsolationAna(TObject * isoAna) { fAnaIsolationArray->Add(isoAna);}\r
  void AddPhotonHadronAna(TObject * ana) { fAnaPhotonArray->Add(ana);}\r
  void AddPhotonJetAna(TObject * ana) { fAnaPhotonJetArray->Add(ana);}\r
  void AddPionJetAna(TObject * ana) { fAnaPionJetArray->Add(ana);}\r
  void AddPionHadronAna(TObject * ana) { fAnaPionArray->Add(ana);}\r
\r
  void GetPionGrandChildren(const AliAODConversionPhoton * const pion, const TClonesArray * photons, Int_t* trackLabels);\r
  void SetEtaLimits(Float_t eta) { fEtaLimit = eta; }\r
\r
\r
 private:\r
\r
  void NotifyRun();\r
  Bool_t UserNotify();\r
  Bool_t EventIsSynced(const TClonesArray * const tracks, const TClonesArray * const convGamma, const TClonesArray * const pions);\r
  Bool_t BothTracksPresent(const AliAODConversionPhoton * const photon, const TClonesArray * const tracks) const;\r
  Bool_t BothGammaPresent(const AliAODConversionPhoton * const pion, const TClonesArray * const photons, const TClonesArray * const tracks) const;\r
  AliAODEvent * GetAODEvent();\r
\r
  //Get Conversion gammas branch\r
  TClonesArray * GetConversionGammas(const AliAODEvent * aodEvent);\r
  TClonesArray * GetConversionPions(const AliAODEvent * aodEvent);\r
\r
  //Process conv gamma\r
  void ProcessConvGamma( const TClonesArray * const convGamma, const TClonesArray * const pions, const TClonesArray * const tracks);\r
  void ProcessPions( const TClonesArray * const pions, const TClonesArray * const photons, const TClonesArray * const tracks);\r
\r
  //Process calorimeters\r
  void ProcessCalorimeters( const AliAODEvent * const aodEvent );\r
  \r
\r
  //Does any pions have given photon (iPhot) index listed as daughter\r
  Bool_t IsDecayPion(Int_t iPhot, const TClonesArray * const pions); //see above\r
\r
\r
  TList       *fOutputList;       //! Output list\r
  Float_t     fEtaLimit;\r
  TString     fDeltaAODFileName;  //! File where Gamma Conv AOD is located, if not in default AOD\r
  TString     fGammaCutString;   //! The cut string of the conversion analysis used to produce input AOD\r
  TString     fPionCutString;   //! The cut string of the conversion analysis used to produce input AOD\r
\r
\r
  AliAnaConvIsolation * fAnaIsolation;\r
\r
  TObjArray * fAnaIsolationArray; //!Array of isolation analysis objects\r
  TObjArray * fAnaPionArray;      //!Array of pion - hadron ana objects\r
  TObjArray * fAnaPhotonArray;    //!Array of photon - hadron ana objects\r
  TObjArray * fAnaPhotonJetArray; //!Array of photon - jet ana objects\r
  TObjArray * fAnaPionJetArray; //!Array of photon - jet ana objects\r
\r
  Float_t fMinPt; //Minimum pt for leading particles\r
  Int_t fMinNTracks; //Minimum number of tracks in event\r
  \r
  TH2F * fhTracksMissingPt[2];\r
 \r
\r
\r
  \r
  AliAnalysisTaskGammaJet(const AliAnalysisTaskGammaJet&); // not implemented\r
  AliAnalysisTaskGammaJet& operator=(const AliAnalysisTaskGammaJet&); // not implemented\r
  \r
  ClassDef(AliAnalysisTaskGammaJet, 4); // example of analysis\r
};\r
\r
#endif\r