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

#ifndef ALI_ANALYSIS_TASK_SE_IMPROVE_ITS_H
#define ALI_ANALYSIS_TASK_SE_IMPROVE_ITS_H

#include "AliAnalysisTaskSE.h"

class TGraph;
class TList;
class AliAODTrack;
class TClonesArray;
class TObjArray;
class AliESDVertex;
class AliVVertex;

class AliAnalysisTaskSEImproveITS:public AliAnalysisTaskSE {
public:
  AliAnalysisTaskSEImproveITS();
  AliAnalysisTaskSEImproveITS(const char *name,
                              const char *resfileCurURI,
                              const char *resfileUpgURI,
                              Int_t ndebug);
  virtual ~AliAnalysisTaskSEImproveITS();

  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
//  virtual void Init();
//  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
//  virtual void Terminate(Option_t *option);
private:
  AliAnalysisTaskSEImproveITS(const AliAnalysisTaskSEImproveITS&);
  AliAnalysisTaskSEImproveITS& operator=(const AliAnalysisTaskSEImproveITS&); 

  // Helper functions
  Double_t EvalGraph(const TGraph *graph,Double_t x) const; 
  void SmearTrack(AliAODTrack *track,const TClonesArray *mcs);
  AliESDVertex* RecalculateVertex(const AliVVertex *old,TObjArray *tracks,Double_t bField);

  TGraph *fD0ZResPCur  ; // old pt dep. d0 res. in z for protons
  TGraph *fD0ZResKCur  ; // old pt dep. d0 res. in z for kaons
  TGraph *fD0ZResPiCur ; // old pt dep. d0 res. in z for pions
  TGraph *fD0RPResPCur ; // old pt dep. d0 res. in rphi for protons
  TGraph *fD0RPResKCur ; // old pt dep. d0 res. in rphi for kaons
  TGraph *fD0RPResPiCur; // old pt dep. d0 res. in rphi for pions
  TGraph *fPt1ResPCur  ; // old pt dep. 1/pt res. for protons
  TGraph *fPt1ResKCur  ; // old pt dep. 1/pt res. for kaons
  TGraph *fPt1ResPiCur ; // old pt dep. 1/pt res. for pions
  TGraph *fD0ZResPUpg  ; // new pt dep. d0 res. in z for protons
  TGraph *fD0ZResKUpg  ; // new pt dep. d0 res. in z for kaons
  TGraph *fD0ZResPiUpg ; // new pt dep. d0 res. in z for pions
  TGraph *fD0RPResPUpg ; // new pt dep. d0 res. in rphi for protons
  TGraph *fD0RPResKUpg ; // new pt dep. d0 res. in rphi for kaons
  TGraph *fD0RPResPiUpg; // new pt dep. d0 res. in rphi for pions
  TGraph *fPt1ResPUpg  ; // new pt dep. 1/pt res. for protons
  TGraph *fPt1ResKUpg  ; // new pt dep. 1/pt res. for kaons
  TGraph *fPt1ResPiUpg ; // new pt dep. 1/pt res. for pions

  TList   *fDebugOutput; //! collection of debug output
  TNtuple *fDebugNtuple; //! debug send on output slot 1
  Float_t *fDebugVars;   //! variables to store as degug info 
  Int_t   fNDebug;       // Max number of debug entries into Ntuple
  ClassDef(AliAnalysisTaskSEImproveITS,1);
};

#endif