Rewritten spectrum2 task, old running is optional, various minor updates to AddTask...

[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskCorrectionsUE.h

#ifndef ALIANALYSISTASKCORRECTIONSUE_H
#define ALIANALYSISTASKCORRECTIONSUE_H

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

////////////////////////////////////////////////////////////////////////
//
// Analysis class to Correct Underlying Event studies
//
// This class needs as input ESDs.\r
// The output is an analysis-specific container.\r
//
// The class is used to get the contamination from secondaries\r
// from tracks DCA distribution \r
// as function of track pT and pseudo-rapidity.\r
// It provides additional information for the corrections \r
// that can not be retrieved by the AliAnalysisTaskLeadingTackUE\r
// task, which is running on AODs.\r
// 
////////////////////////////////////////////////////////////////////////

#include "AliAnalysisTask.h"

class AliAnalyseLeadingTrackUE;\r
class AliESDtrackCuts;\r
class AliInputEventHandler;
class AliAODEvent;
class AliCFContainer;\r
class AliESDEvent;
class AliESDtrack;\r
class AliAODInputHandler;
class AliESDInputHandler;\r
class AliMCEventHandler;
class AliMCEvent;
class TH1F;
class TH2F;
class TH3F;\r
class TH1I;
class TProfile;
class TTree;
class TVector3;

class  AliAnalysisTaskCorrectionsUE : public AliAnalysisTask
  {
  public:
    // track cuts steps\r
    enum CFSteps {\r
    	kCFStepAll     = 0,\r
	kCFStepCuts    = 1, // standard ITS+TPC 2009 cuts w.o. DCA cut and SPD cluster requirement\r
	kCFStepSPD     = 2, // add SPD cluster requirement\r
	kCFStepDCA     = 3  // add pT dependent DCA cut\r
    	};\r

    AliAnalysisTaskCorrectionsUE(const char* name="AliAnalysisTaskCorrectionsUE");
    virtual           ~AliAnalysisTaskCorrectionsUE() {if ( fListOfHistos ) delete fListOfHistos; }
    AliAnalysisTaskCorrectionsUE(const  AliAnalysisTaskCorrectionsUE &det);
    AliAnalysisTaskCorrectionsUE&   operator=(const  AliAnalysisTaskCorrectionsUE &det);
       
    // return instance of the singleton
    static  AliAnalysisTaskCorrectionsUE* Instance();
      
    // Implementation of interace methods
    virtual     Bool_t Notify();
    virtual     void   ConnectInputData(Option_t *);
    virtual     void   CreateOutputObjects();
    virtual     void   Exec(Option_t *option);
    virtual     void   Terminate(Option_t *);

    //  Setters/Getters
    virtual     void   SetDebugLevel( Int_t level )  { fDebug = level; }\r
    virtual     void   SetMode(Int_t mode)           { fMode  = mode;  }\r

    //Event QA
    void  SetZVertex( Double_t val )          { fZVertex = val; }\r
    void  SetTracksInVertex( Int_t val ){ fnTracksVertex = val; }

    // Track selection cuts\r
    void  SetTrackEtaCut( Double_t val ) { fTrackEtaCut = val; }\r
    void  SetTrackPtCut( Double_t val )  { fTrackPtCut = val; }\r
	
  protected:
    static     AliAnalysisTaskCorrectionsUE*     fgTaskCorrectionsUE;        // Pointer to single instance\r
  private:
    void       AddSettingsTree();                                  // add list of settings to output list\r
    // Analysis methods\r
    void       AnalyseCorrectionMode();                            // main algorithm to get correction maps\r
    void       CreateContainer();  			           // create the output CF container\r
    void       FillContainer(AliESDtrack *track, Int_t step,Bool_t mcvertex, Double_t matchLeading); // fill container \r
    AliAnalyseLeadingTrackUE*       fAnalyseUE;         //! points to AliAnalyseLeadingTrackUE class\r
    Int_t                           fDebug;             //  Debug flag\r
    AliESDEvent*                    fESDEvent;          //! ESD Event\r
    AliESDInputHandler*             fESDHandler;        //! ESD Input Handler\r
    AliInputEventHandler*           fInputHandler;      //  Input event handler\r
    TList*                          fListOfHistos;      //  Output list of histograms\r
    AliMCEvent*                     fMcEvent;           //  pointer to MC event\r
    AliMCEventHandler*              fMcHandler;         //  pointer to MC handler\r
    Int_t 	                    fMode;              //  fMode = 0: data-like analysis \r
    				                        //  fMode = 1: corrections analysis	\r
    AliCFContainer*		    fOutCFcont;         //  output CF container   \r
    TH1F*			    fhEntries;          //  count events	 \r
    TH1F* 			    fhFakes;           	//  counts the amount of fake tracks \r
    TH1F*                           fhPtMCAll;          //  pT distribution of all accepted MC tracks \r
    TH1F*                           fhPtMCPrim;         //  pT distribution MC primaries\r
    TH1F*                           fhPtMCSec;          //  pT distribution MC secondaries\r
    TH1F*                           fhPtMCPrimFake;     //  pT distribution MC fake primaries\r
    TH1F*                           fhPtMCSecFake;      //  pT distribution MC fake secondaries\r
\r
    // Cuts Events QA\r
    Int_t          fnTracksVertex;        // QA tracks pointing to principal vertex (= 3 default) 
    Double_t       fZVertex;              // Position of Vertex in Z direction
    TH1F*          fhVertexContributors;  // Plot number of contributors in vertex  \r
    TH3F*          fhVertexReso;          //  vertex resolution in XY and Z vs. number of contributors\r
    // Track cuts\r
    Double_t       fTrackEtaCut;          // Eta cut on tracks in the regions (fRegionType=1)\r
    Double_t       fTrackPtCut;           // Pt cut of tracks in the regions\r
    AliESDtrackCuts* fEsdTrackCuts;       // ITS+TPC 2009 cuts (no SPD requirement, no DCA cut) \r
    AliESDtrackCuts* fEsdTrackCutsSPD;     // Require 1 cluser in SPD\r
    AliESDtrackCuts* fEsdTrackCutsSDD;     // Require 1 cluser in 1st layer SDD\r
    AliESDtrackCuts* fEsdTrackCutsDCA;     // Add pT dependent DCA cut\r
    ClassDef( AliAnalysisTaskCorrectionsUE, 5); // Analysis task to correct Underlying Event analysis\r
  };

#endif