/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

//-----------------------------------------------------------------------
// This class stores QA variables as function of pT for different type
// of tracks and track selection criteria
// Author : Marta Verweij - UU
//-----------------------------------------------------------------------

#ifndef ALIPWG4HIGHPTTRACKQA_H
#define ALIPWG4HIGHPTTRACKQA_H

#include "AliAnalysisTaskSE.h"

class TH1F;
class TH2F;
class TH3F;
class TProfile;
class TList;
class TArrayF;

class AliVEvent;
class AliESDEvent;
class AliESDtrackCuts;
class AliESDVertex;
class AliAODVertex;
class AliAODTrack;
class AliESDtrack;

class AliGenPythiaEventHeader;
class AliMCEvent;
//class AliAnalysisHelperJetTasks;

class AliPWG4HighPtTrackQA: public AliAnalysisTaskSE {

 public:
  AliPWG4HighPtTrackQA();
  AliPWG4HighPtTrackQA(const char *name);
  virtual ~AliPWG4HighPtTrackQA() {;}
 
  //  virtual void   ConnectInputData(Option_t *);
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2

  enum DataType {kESD,kAOD};

  Bool_t IsPbPb() {return fIsPbPb;}  //is PbPb data?
  Bool_t SelectEvent();              //decides if event is used for analysis
  Int_t CalculateCentrality(AliVEvent *ev);
  Int_t CalculateCentrality(AliESDEvent *esd);
  Int_t CalculateCentrality(const AliAODEvent *aod);
  Int_t GetCentralityClass(Float_t cent=-1.) const;
  void DoAnalysisESD();
  void DoAnalysisAOD();
  void FillHistograms();

  //Setters
  void SetDataType(DataType d)             {fDataType = d;}
  void SetIsPbPb(Bool_t cs)                {fIsPbPb = cs;}
  void SetCentralityClass(int cent)        {fCentClass=cent;}
  void SetCuts(AliESDtrackCuts* trackCuts)         {fTrackCuts         = trackCuts;}
  void SetCutsITSLoose(AliESDtrackCuts* trackCuts) {fTrackCutsITSLoose = trackCuts;}
  void SetCutsTPConly(AliESDtrackCuts* trackCuts)  {fTrackCutsTPConly  = trackCuts;}
  void SetTrackType(Int_t trackType) {fTrackType = trackType;}
  void SetFilterMask(UInt_t filterMask)    {fFilterMask    = filterMask ;}
  void SetIncludeNoITS(Bool_t f)           {fIncludeNoITS  = f          ; }

  void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
  void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
  void SetPtBinEdges(Int_t region, Double_t ptmax, Double_t ptBinWidth);
  void SetNVariables(Int_t nv) {fNVariables = nv;}

  Float_t GetPtMax()           {return fPtMax;}
  Float_t GetTPCClusterInfo(const AliAODTrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159, Bool_t useFitMap=kFALSE) const;
  Float_t GetTPCClusterInfoFitMap(const AliESDtrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159) const;
  Int_t   GetTrackLengthTPC(const AliESDtrack *track) const;
  Int_t   GetTrackLengthTPC(const AliAODTrack *track) const;
  Float_t GetGoldenChi2(AliESDtrack *origtrack);
  Float_t GetGGCChi2(AliESDtrack *origtrack);

  static AliGenPythiaEventHeader*  GetPythiaEventHeader(const AliMCEvent *mcEvent);
  static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root

 protected:

 private:
  AliPWG4HighPtTrackQA(const AliPWG4HighPtTrackQA&);
  AliPWG4HighPtTrackQA& operator=(const AliPWG4HighPtTrackQA&);

  DataType fDataType;             // kESD or kAOD

  AliVEvent   *fEvent;            //! AliVEvent object
  AliESDEvent *fESD;              //! ESD object
  const AliESDVertex   *fVtx;     //! vertex object
  const AliAODVertex   *fVtxAOD;  //! vertex object AOD

  AliESDtrackCuts *fTrackCuts;         // TrackCuts
  AliESDtrackCuts *fTrackCutsITSLoose; // Loose ITS track cuts
  AliESDtrackCuts *fTrackCutsTPConly;  // TPC track cuts
  Int_t   fTrackType;                  // 0: global track; 1:TPConly track 2: TPConly constrained track 3: global ITSrefit 4: TPConly constrained track with QA selection based on global track
  UInt_t fFilterMask;                  // Select tracks from specific track cuts belonging to certain filter mask for AOD analysis
  Bool_t fIncludeNoITS;                // includes tracks with failed ITS refit

  Double_t fSigmaConstrainedMax;  // max sigma on constrained fit
  Float_t fPtMax;                 // Maximum pT for histograms
  Float_t fPtBinEdges[3][2];      // 3 regions total with different binning for pT axis of histos

  Bool_t   fIsPbPb;               // kTRUE if PbPb
  Int_t fCentClass;               // Select only events from predefined centrality class
  

  /*
    QA variables stored in TArrayF *fVariables
  0: pt
  1: phi
  2: eta
  3: dca2D
  4: dcaZ 
  5: nClustersTPC
  6: nPointITS   
  7: chi2C       
  8: nSigmaToVertex
  9: relUncertainty1Pt
  10: chi2PerClusterTPC
  11: #crossed rows
  12: (#crossed rows)/(#findable clusters)
  13: SigmaY2
  14: SigmaZ2
  15: SigmaSnp2
  16: SigmaTgl2
  17: Sigma1Pt2
  18: NClustersTPCIter1
  19: TPCChi2Iter1
  20: NClustersTPCShared
  21: Chi2Gold (TPC constrained vs global)
  22: Chi2GGC (global constrained vs global)
  23: NCrossed rows from fit map
  24: (#crossed rows)/(#findable clusters) from fit map
  */

  Int_t fNVariables;             // Number of variables
  TArrayF *fVariables;           // QA variables

  UChar_t fITSClusterMap;  // map of clusters, one bit per a layer

  Float_t fAvgTrials;             // Average number of trials
  
  TH1F *fNEventAll;                            //! Event counter
  TH1F *fNEventSel;                            //! Event counter
  TH1F *fNEventReject;                         //! Book keeping of reason of rejecting events
 
  TH1F *fh1Centrality;                         //! Centrality

  TProfile*     fh1Xsec;                       //! pythia cross section and trials
  TH1F*         fh1Trials;                     //! trials which are added
  TH1F*         fh1PtHard;                     //! pt hard of the event
  TH1F*         fh1PtHardTrials;               //! pt hard of the event

  TH1F *fh1NTracksAll;                         //! All tracks
  TH1F *fh1NTracksReject;                      //! Reason why track was rejected
  TH1F *fh1NTracksSel;                         //! Number of accepted tracks

  TH1F *fPtAll;                                //! Pt spectrum all charged particles
  TH1F *fPtSel;                                //! Pt spectrum all selected charged particles by fTrackCuts
  TH2F *fPtPhi;                                //! Pt vs Phi
  TH2F *fPtEta;                                //! Pt vs Eta
  TH3F *fPtEtaPhi;                             //! Pt vs Eta vs Phi
  TH2F *fPtDCA2D;                              //! Pt vs DCA2D
  TH2F *fPtDCAZ;                               //! Pt vs DCAZ
  TH2F *fPtNClustersTPC;                       //! Pt vs nClustersTPC
  TH2F *fPtNClustersTPCPhi;                    //! Phi vs nClustersTPC
  TH2F *fPtNClustersTPCIter1;                  //! Pt vs nClustersTPCIter1
  TH3F *fPtNClustersTPCIter1Phi;               //! Pt vs nClustersTPCIter1 vs Phi
  TH2F *fPtNClustersTPCShared;                 //! Pt vs nClustersTPCShared
  TH2F *fPtNClustersTPCSharedFrac;             //! Pt vs nClustersTPCSharedFrac
  TH2F *fPtNPointITS;                          //! Pt vs nPointITS
  TH3F *fPtNPointITSPhi;                       //! Pt vs nPointITS vs phi
  TH2F *fPtChi2C;                              //! Pt vs Chi2C
  TH2F *fPtNSigmaToVertex;                     //! Pt vs nSigmaToVertex

  TH2F *fPtRelUncertainty1Pt;                  //! Pt vs relUncertainty1Pt
  TH3F *fPtRelUncertainty1PtNClus;             //! Pt vs relUncertainty1Pt vs NClustersTPC
  TH3F *fPtRelUncertainty1PtNClusIter1;        //! Pt vs relUncertainty1Pt vs NClustersTPCIter1
  TH3F *fPtRelUncertainty1PtNPointITS;         //! Pt vs relUncertainty1Pt vs NPointITS
  TH3F *fPtRelUncertainty1PtITSClusterMap;     //! Pt vs relUncertainty1Pt vs byte of ITS clustermap

  TH3F *fPtRelUncertainty1PtChi2;              //! Pt vs relUncertainty1Pt vs Chi2TPC/NClus
  TH3F *fPtRelUncertainty1PtChi2Iter1;         //! Pt vs relUncertainty1Pt vs Chi2TPC/NClusIter1
  TH3F *fPtRelUncertainty1PtPhi;               //! Pt vs relUncertainty1PtPhi

  TH2F *fPtChi2PerClusterTPC;                  //! Pt vs Chi2PerClusterTPC
  TH2F *fPtChi2PerClusterTPCIter1;             //! Pt vs Chi2PerClusterTPCIter1
  TH2F *fPtNCrossedRows;                       //! Pt vs NCrossedRows
  TH3F *fPtNCrossedRowsPhi;                    //! Pt vs NCrossedRows vs Phi
  TH3F *fPtNCrossedRowsNClusFPhi;              //! Pt vs NCrossedRows/NClusF vs Phi
  TH3F *fPtNCrRNCrRNClusF;                     //! Pt vs NCrossedRows vs NCrossedRows/NClusF 
  TH2F *fPtNCrossedRowsFit;                    //! Pt vs NCrossedRows from NClusterFitMap
  TH3F *fPtNCrossedRowsFitPhi;                 //! Pt vs NCrossedRows from FitMap vs Phi
  TH3F *fPtNCrossedRowsNClusFFitPhi;           //! Pt vs NCrossedRows/NClusF from NClusterFitMap
  TH2F *fNCrossedRowsNCrossedRowsFit;          //! NCrossedRows from ClusterMap vs NCrossedRows from NClusterFitMap
  TH2F *fNClustersNCrossedRows;                //! NClusters vs NCrossedRows
  TH2F *fNClustersNCrossedRowsFit;             //! NClusters vs NCrossedRows from fit map
  TH3F *fPtNClustersNClustersFitMap;           //! pT vs ncls vs ncls from fit map
  TH3F *fPtRelUncertainty1PtNCrossedRows;      //! Pt vs relUncertainty1Pt vs NCrossedRows
  TH3F *fPtRelUncertainty1PtNCrossedRowsFit;   //! Pt vs relUncertainty1Pt vs NCrossedRowsFitMap

  TH2F *fPtChi2Gold;                           //! Pt vs Chi2 between global and TPC constrained track
  TH2F *fPtChi2GGC;                            //! Pt vs Chi2 between global and global constrained track
  TH3F *fPtChi2GoldPhi;                        //! Pt vs Chi2 between global and TPC constrained track vs phi
  TH3F *fPtChi2GGCPhi;                         //! Pt vs Chi2 between global and global constrained track vs phi
  TH2F *fChi2GoldChi2GGC;                      //! Correlations between gold chi2 and GGC chi2

  TH3F *fPtChi2ITSPhi;                         //! Pt vs ITS chi2 vs phi

  //histos for covariance matrix elements
  TH2F *fPtSigmaY2;                            //! 1/Pt vs sigma(y) extCov[0]
  TH2F *fPtSigmaZ2;                            //! 1/Pt vs sigma(z) extCov[2]
  TH2F *fPtSigmaSnp2;                          //! 1/Pt vs sigma(Snp) extCov[5]
  TH2F *fPtSigmaTgl2;                          //! 1/Pt vs sigma(Tgl) extCov[9]
  TH2F *fPtSigma1Pt2;                          //! 1/Pt vs sigma(1/pT) extCov[14]

  //profiles for covariance matrix elements
  TProfile *fProfPtSigmaY2;                    //! 1/Pt vs sigma(y) extCov[0]
  TProfile *fProfPtSigmaZ2;                    //! 1/Pt vs sigma(z) extCov[2]
  TProfile *fProfPtSigmaSnp2;                  //! 1/Pt vs sigma(Snp) extCov[5]
  TProfile *fProfPtSigmaTgl2;                  //! 1/Pt vs sigma(Tgl) extCov[9]
  TProfile *fProfPtSigma1Pt2;                  //! 1/Pt vs sigma(1/pT) extCov[14]

  TProfile *fProfPtSigma1Pt;                   //! pT vs sigma(1/Pt)
  TProfile *fProfPtPtSigma1Pt;                 //! pT vs pT*sigma(1/Pt)

  TList *fHistList; //! List of Histograms
 
  ClassDef(AliPWG4HighPtTrackQA,8) 
};
#endif