AliAnalysisTaskSEImproveITS.h

Go to the documentation of this file.

/* 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 *period,
                              const char *systematic,
                              Bool_t isRunInVertexing,
                              Int_t ndebug);

  virtual ~AliAnalysisTaskSEImproveITS();

  virtual void UserCreateOutputObjects();
  //  virtual void Init();
  //  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  //  virtual void Terminate(Option_t *option);
  void SetImproveTracks(Bool_t flag=kTRUE) { fImproveTracks=flag; return; }
  void SetUpdateSecVertCovMat(Bool_t flag=kTRUE) { fUpdateSecVertCovMat=flag; return; }
  void SetUpdateSTCovMatrix(Bool_t opt=kTRUE){fUpdateSTCovMatrix=opt;}
  void SetUpdatePulls(Bool_t opt=kTRUE){fUpdatePulls=opt;}
  void SetMimicData(Bool_t opt=kFALSE){fMimicData=opt;}
  void SetAOD(Bool_t flag=kTRUE) { fIsAOD=flag; return; }

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

  Double_t EvalGraph(Double_t x,const TGraph *graph,const TGraph *graphSA=0) const; 
  void SmearTrack(AliVTrack *track, Double_t bz);
  AliESDVertex* RecalculateVertex(const AliVVertex *old,TObjArray *tracks,Double_t bField);
  Int_t PhiBin(Double_t phi) const;

  TGraph *fD0ZResPCur  ; 
  TGraph *fD0ZResKCur  ; 
  TGraph *fD0ZResPiCur ; 
  TGraph *fD0ZResECur ; 
  TGraph *fD0RPResPCur; 
  TGraph *fD0RPResKCur; 
  TGraph *fD0RPResPiCur; 
  TGraph *fD0RPResECur; 
  TGraph *fD0RPSigmaPullRatioP; 
  TGraph *fD0RPSigmaPullRatioK; 
  TGraph *fD0RPSigmaPullRatioPi; 
  TGraph *fD0RPSigmaPullRatioE; 
  TGraph *fD0RPMeanPCur[2][4]; 
  TGraph *fD0RPMeanKCur[2][4]; 
  TGraph *fD0RPMeanPiCur[2][4]; 
  TGraph *fD0RPMeanECur[2][4]; 
  TGraph *fPt1ResPCur  ; 
  TGraph *fPt1ResKCur  ; 
  TGraph *fPt1ResPiCur ; 
  TGraph *fPt1ResECur ; 
  TGraph *fD0ZResPUpg  ; 
  TGraph *fD0ZResKUpg  ; 
  TGraph *fD0ZResPiUpg ; 
  TGraph *fD0ZResEUpg ; 
  TGraph *fD0RPResPUpg; 
  TGraph *fD0RPResKUpg; 
  TGraph *fD0RPResPiUpg; 
  TGraph *fD0RPResEUpg; 
  TGraph *fD0RPMeanPUpg[2][4]; 
  TGraph *fD0RPMeanKUpg[2][4]; 
  TGraph *fD0RPMeanPiUpg[2][4]; 
  TGraph *fD0RPMeanEUpg[2][4]; 
  TGraph *fPt1ResPUpg  ; 
  TGraph *fPt1ResKUpg  ; 
  TGraph *fPt1ResPiUpg ; 
  TGraph *fPt1ResEUpg ; 
  TGraph *fD0ZResPCurSA  ; 
  TGraph *fD0ZResKCurSA  ; 
  TGraph *fD0ZResPiCurSA ; 
  TGraph *fD0ZResECurSA ; 
  TGraph *fD0RPResPCurSA ; 
  TGraph *fD0RPResKCurSA ; 
  TGraph *fD0RPResPiCurSA; 
  TGraph *fD0RPResECurSA; 
  TGraph *fPt1ResPCurSA  ; 
  TGraph *fPt1ResKCurSA  ; 
  TGraph *fPt1ResPiCurSA ; 
  TGraph *fPt1ResECurSA ; 
  TGraph *fD0ZResPUpgSA  ; 
  TGraph *fD0ZResKUpgSA  ; 
  TGraph *fD0ZResPiUpgSA ; 
  TGraph *fD0ZResEUpgSA ; 
  TGraph *fD0RPResPUpgSA ; 
  TGraph *fD0RPResKUpgSA ; 
  TGraph *fD0RPResPiUpgSA; 
  TGraph *fD0RPResEUpgSA; 
  TGraph *fPt1ResPUpgSA  ; 
  TGraph *fPt1ResKUpgSA  ; 
  TGraph *fPt1ResPiUpgSA ; 
  TGraph *fPt1ResEUpgSA ; 

  Bool_t fRunInVertexing; 
  Bool_t fImproveTracks; 
  Bool_t fUpdateSecVertCovMat; 
  Bool_t fUpdateSTCovMatrix; 
  Bool_t fUpdatePulls; 
  Bool_t fMimicData;
  Bool_t fIsAOD;          // flag to run on AOD 
  TClonesArray *fMCs;      // pointer to AOD MC info
  TList   *fDebugOutput; 
  TNtuple *fDebugNtuple; 
  Float_t *fDebugVars;   
  Int_t   fNDebug;       

  ClassDef(AliAnalysisTaskSEImproveITS,10);
};

#endif