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

    //////////////////////////////////////////////////////////////////////////
    //                                                                      //
    //  Task for Isolated Gamma in p-p,p-Pb and eventually g-h Correlation  //
    //                                                                      //
    //  Author: Davide Francesco Lodato (Utrecht University)                //
    //          Lucile Ronflette (Subatech, Nantes)                         //
    //          Marco Marquard   (University Frankfurt am Main)             //
    //////////////////////////////////////////////////////////////////////////

    //ROOT System

class TH1;
class TH2;
class TH3;
class THnSparse;
class TList;
class TObjArray;
class AliEMCALGeometry;
class AliESDCaloCells;
class AliESDEvent;
class AliESDtrack;
class TClonesArray;
class TList;
class TString;
class AliVCluster;
class AliVParticle;
class AliESDtrackCuts;
class AliAODEvent;
class AliAODCaloCells;
class AliVCluster;
class AliMCEvent;
class AliStack;
class TParticle;
class AliClusterContainer;
class AliParticleContainer;
class AliEmcalParticle;
    //AliRoot System
class AliEMCALTrack;
    //class AliMagF;
class AliEMCALRecoUtils;
    //class AliAnalysisFilter;
class AliAODTrack;
class AliAODCaloCluster;
class AliESDCaloCluster;
class AliVCaloCells;
    //class AliEventPoolManager;

#include "AliAnalysisTaskEmcal.h"

class AliAnalysisTaskEMCALPhotonIsolation : public AliAnalysisTaskEmcal {
public:
    AliAnalysisTaskEMCALPhotonIsolation();
    AliAnalysisTaskEMCALPhotonIsolation(const char *name, Bool_t histo=kFALSE);
    virtual ~AliAnalysisTaskEMCALPhotonIsolation();
    
    void                     UserCreateOutputObjects();
                    
    void                     SetIsoConeRadius(Float_t r)                                     { fIsoConeRadius = r ;}
    void                     SetCTMdeltaEta (Float_t r)                                      { fdetacut = r ;}
    void                     SetCTMdeltaPhi (Float_t r)                                      { fdphicut = r ;}
    void                     SetIsoMethod (Int_t r )                                         { fIsoMethod = r ;}
    void                     SetUEMethod (Int_t UE)                                          { fUEMethod = UE;}
    void                     SetOutputFormat (Int_t iOut)                                    { fWho = iOut;}
    void                     SetQA (Bool_t QA)                                               { fQA = QA;}
    void                     SetMC (Bool_t MC)                                               { fIsMC = MC;}
    void                     SetUSEofTPC (Bool_t TPC)                                        { fTPC4Iso = TPC;}
    void                     SetLCAnalysis (Bool_t LC)                                       { fisLCAnalysis = LC;}

protected:
    
    void                     EtIsoCellPhiBand(TLorentzVector c, Float_t &etIso, Float_t &phiBand);    //EIsoCone via Cells UE via PhiBand EMCAL
    void                     EtIsoCellEtaBand(TLorentzVector c, Float_t &etIso, Float_t &etaBand);    //EIsoCone via Cells UE via EtaBand EMCAL
    void                     EtIsoClusPhiBand(TLorentzVector c, Float_t &etIso, Float_t &etaBand, Int_t index);    //EIsoCone via Clusters UE via EtaBand EMCAL
    void                     EtIsoClusEtaBand(TLorentzVector c, Float_t &etIso, Float_t &etaBand, Int_t index);    //EIsoCone via Clusters UE via EtaBand EMCAL
    void                     PtIsoTrackPhiBand(TLorentzVector c, Float_t &ptIso, Float_t &phiBand);   //PIsoCone via Track UE via PhiBand TPC
    void                     PtIsoTrackEtaBand(TLorentzVector c, Float_t &ptIso, Float_t &etaBand);   //PIsoCone via Track UE via EtaBand TPC
  //  void                     PtIsoTraClusPhiBand(TLorentzVector c, Float_t &ptIso, Float_t &phiBand); //(P+E)IsoCone via Track/Clus UE via PhiBand TPC+EMCAL
  //  void                     PtIsoTraClusEtaBand(TLorentzVector c, Float_t &ptIso, Float_t &etaBand); //(P+E)IsoCone via Track/Clus UE via EtaBand TPC+EMCAL
    void                     PtIsoTrackOrthCones(TLorentzVector c, Float_t &ptIso, Float_t &cones);   //PIsoCone via Tracks UE via Orthogonal Cones in Phi
    void                     PtIsoTrackFullTPC(TLorentzVector c, Float_t &ptIso, Float_t &full);      //PIsoCone via Tracks UE via FullTPC - IsoCone - B2BEtaBand
    Bool_t                   ClustTrackMatching(AliVCluster *cluster);
    Bool_t                   CheckBoundaries(TLorentzVector vecCOI);
   // void                     FillNCOutput(AliVCluster *COI, TLorentzVector vecCOI, Int_t index);
    
    Float_t*                 GenerateFixedBinArray(Int_t n, Float_t min, Float_t max) const;
    void                     ExecOnce();   
    Bool_t                   Run();
     
    using AliAnalysisTaskEmcal::FillGeneralHistograms;
    Bool_t FillGeneralHistograms(AliVCluster *COI, TLorentzVector VecCOI, Int_t index);
    //Bool_t                   FillGeneralHistograms(AliVCluster *COI, TLorentzVector VecCOI, Int_t index);
    
  
  //    TObjArray                   fParticleCollArray;              // Neutral Clusters collection array
    TClonesArray               *fNCluster;                       // Neutral clusters
    
    Int_t       fWho;           // MODE for the Output Object (TTree or THnSparse)
  
  
		//IMPLEMENT ALL THE HISTOGRAMS AND ALL THE OUTPUT OBJECTS WE WANT!!!
   //    TList       *fOutputList; //! Output list
//    TGeoHMatrix *fGeomMatrix[12];//! Geometry misalignment matrices for EMCal
    
    
    TH1        *fTrackMult;                      //!Track Multiplicity ---QA
    TH1        *fTrackMultEMCAL;                 //!Track Multiplicity EMCAL ---QA
    TH1        *fClustMult;                      //!Cluster Multiplicity EMCAL ---QA
    TH1        *fPVZBefore;                      //!Z Vertex distribution before cuts. ---QA
    TH2        *fEtaPhiCell;                     //!EMCAL Active Cells Distribution EtaPhi ---QA
    TH2        *fEtaPhiClus;                     //!EMCAL Cluster Distribution EtaPhi ---QA
    TH2        *fClusEvsClusT;                   //!Cluster Energy vs Cluster Time ---QA
    TH1        *fGoodEventsOnPVZ;                //!Number of selected events After the Cut on Primary Vertex
    TH1        *fPT;                             //!Pt distribution
    TH2        *fM02;                            //!Squared_Lambda0 distribution
    TH1        *fNLM;                            //!NLM distribution
    TH2        *fDeltaETAClusTrackVSpT;          //!dEta Cluster-Track VS pT!
    TH2        *fDeltaPHIClusTrackVSpT;          //!dPhi Cluster-Track VS pT!
    TH1        *fEtIsoCells;                     //!Isolation Energy with EMCAL Cells
    TH1        *fEtIsoClust;                     //!Isolation Energy with EMCAL Clusters
    TH1        *fPtIsoTrack;                     //!Isolation Pt with Tracks
    TH1        *fPtEtIsoTC;                      //!Isolation with Pt from Tracks and Et from NON-Matched Clusters
    TH2        *fPhiBandUEClust;                 //!UE with Phi Band (Clusters)
    TH2        *fEtaBandUEClust;                 //!UE with Eta Band (Clusters)
    TH2        *fPhiBandUECells;                 //!UE with Phi Band (Cells)
    TH2        *fEtaBandUECells;                 //!UE with Eta Band (Cells)
    TH2        *fPhiBandUETracks;                //!UE with Phi Band (Tracks)
    TH2        *fEtaBandUETracks;                //!UE with Eta Band (Tracks)
    TH2        *fPerpConesUETracks;              //!UE with Cones (Tracks ONLY)
    TH2        *fTPCWithoutIsoConeB2BbandUE;     //!UE with Full TPC except IsoCone and EtaBand in Back2Back
    TH1        *fNTotClus10GeV;                  //!number of TOTAL clusters with Energy bigger than 10 GeV 
    TH1        *fRecoPV;                         //! primary vertex reconstruction
    TH1        *fEtIsolatedCells;                //! Isolated photons, isolation with cells
    TH1        *fEtIsolatedClust;                //! Isolated photons, isolation with clusters
    TH1        *fEtIsolatedTracks;                //! Isolated photons, isolation with tracks
    TH1        *fTest;					                 //! Test
    
    THnSparse   *fOutputTHnS;                    //! 1st Method 4 Output
    THnSparse   *fOutPTMAX;                      //! 1st Method 4 Isolation on pTMax
    
    TTree       *fOutputTree;                    //! 2nd Method 4 Output
    
    Float_t     fIsoConeRadius;                  // Radius for the Isolation Cont
    Int_t       fEtIsoMethod;                    // Isolation definition 0=SumEt<EtThr, 1=SumEt<%Ephoton, 2=Etmax<EtThr
    Double_t    fEtIsoThreshold;                 // Et isolation threshold, supposed to be % if method one is choosed
    Double_t    fdetacut;                        // cut on deta between track and cluster
    Double_t    fdphicut;                        // cut on dphi between track and cluster
    Double_t    fM02mincut;                      // lambda0^2 minimum cut
    Double_t    fM02maxcut;                      // lambda0^2 maximum cut
    Bool_t      fQA;                             // Flag for few further QA plots wrt the ones already done in the EMCALTask
    Bool_t      fIsMC;                           // Flag for MC Truth Analysis
    Bool_t      fTPC4Iso;                        //0=EMCAL_ONLY; 1=EMCAL+TPC
    Int_t       fIsoMethod;                      //0=Cells, 1=Clusters (EMCAL_ONLY),  2=Tracks (EMCAL w/o TPC)
    Int_t       fUEMethod;                       //0=PhiBand, 1=EtaBand, (EMCAL or TPC) 2= Ort Cones, 3=FullTPC (only with TPC)
    Double_t    *fVertex;                        //!event vertex
    Int_t       fNDimensions;                    //!number of Dimensions for the THnSPARSE
    Bool_t      fisLCAnalysis;                   // Flag to pass from Leading Clusters Analysis to a NC One

// Initialization for TTree variables
    Int_t       fevents;                         // N events
    Double_t    flambda0T;                       // M02
    Double_t    fEtT;                            // Et
    Double_t    fPtT;                            // Pt
    Double_t    fEtisoT;                         // Iso Et
    Double_t    fPtisoT;                         // Iso Pt
    Double_t    fetaT;                           // Eta
    Double_t    fphiT;                           // Phi
    Double_t    fsumEtisoconeT;                  // Iso sum cone
    Double_t    fsumEtUE;                        // sum UE
  
    //  AliParticleContainer       *fTracksCont;     //!Tracks
    //  AliParticleContainer       *fclusters;                       //!Container for Particle container 4 clusters

private:
   AliAnalysisTaskEMCALPhotonIsolation(const AliAnalysisTaskEMCALPhotonIsolation&);            // not implemented
   AliAnalysisTaskEMCALPhotonIsolation &operator=(const AliAnalysisTaskEMCALPhotonIsolation&); // not implemented
    
  
    ClassDef(AliAnalysisTaskEMCALPhotonIsolation, 1)    //EMCAL Neutrals base analysis task
};
#endif