AliAnalysisTaskSEHFQA.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSEHFQUALITYASSURANCE_H
#define ALIANALYSISTASKSEHFQUALITYASSURANCE_H

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

/* $Id$ */ 

//*************************************************************************
//*************************************************************************

#include <TROOT.h>
#include <TSystem.h>

#include "AliAnalysisTaskSE.h"

class AliRDHFCuts;
class TH1F;
class AliAODEvent;
class AliFlowEvent;
class AliFlowTrackCuts;

class AliAnalysisTaskSEHFQA : public AliAnalysisTaskSE
{

 public:

  enum DecChannel {kDplustoKpipi,kD0toKpi,kDstartoKpipi,kDstoKKpi,kD0toKpipipi,kLambdactopKpi,kLambdactoV0};

  AliAnalysisTaskSEHFQA();
  AliAnalysisTaskSEHFQA(const char *name, DecChannel ch, AliRDHFCuts* cuts);
  virtual ~AliAnalysisTaskSEHFQA();

  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);

  void SetReadMC(Bool_t mcflag){fReadMC=mcflag;}
  void SetSimpleMode(Bool_t flag){fSimpleMode=flag;}
  void SetTrackOn(Bool_t trackon=kTRUE){fOnOff[0]=trackon;}
  void SetPIDOn(Bool_t pidon=kTRUE){fOnOff[1]=pidon;}
  void SetCentralityOn(Bool_t centron=kTRUE){fOnOff[2]=centron;}
  void SetEvSelectionOn(Bool_t evselon=kTRUE){fOnOff[3]=evselon;}
  void SetFlowObsOn(Bool_t flowobson=kTRUE){fOnOff[4]=flowobson;}
  void SetUseSelectionBit(Bool_t selectionbiton=kTRUE){fUseSelectionBit=selectionbiton;}
  void SetSecondCentralityEstimator(AliRDHFCuts::ECentrality est){fEstimator = est;}
  void SetFillDistributionsForTrackEffChecks(Bool_t filldistrtrackeffcheckson=kFALSE){fFillDistrTrackEffChecks = filldistrtrackeffcheckson;}

  AliRDHFCuts* GetCutObject() const {return fCuts;}
  DecChannel GetDecayChannel()const {return fDecayChannel;}
  Bool_t GetTrackStatus() const {return fOnOff[0];}
  Bool_t GetPIDStatus() const {return fOnOff[1];}
  Bool_t GetCentralityStatus() const {return fOnOff[2];}
  Bool_t GetEvSelStatus() const {return fOnOff[3];}
  Bool_t GetFlowObsStatus() const {return fOnOff[4];}
  Bool_t GetUseSelectionBit() const {return fUseSelectionBit;}
  AliRDHFCuts::ECentrality GetSecondCentralityEstimator()const {return fEstimator;}
  Bool_t GetFillDistributionsForTrackEffChecks()const {return fFillDistrTrackEffChecks;}

 private:
  AliAnalysisTaskSEHFQA(const AliAnalysisTaskSEHFQA &source);
  AliAnalysisTaskSEHFQA operator=(const AliAnalysisTaskSEHFQA &source);
  void FillFlowObs(AliAODEvent *aod);

 TList* fOutputEntries;    
 TList* fOutputPID;        
 TList* fOutputTrack;      
 TList* fOutputCounters;   
 TList* fOutputCheckCentrality;   
 TList* fOutputEvSelection; 
 TList* fOutputFlowObs;    
 DecChannel fDecayChannel; //identify the decay channel
 AliRDHFCuts* fCuts;       // object containing cuts 
 AliFlowEvent *fFlowEvent; 
 AliFlowTrackCuts *fRFPcuts; 
 AliRDHFCuts::ECentrality fEstimator; //2nd estimator for centrality
 Bool_t fReadMC;           
 Bool_t fSimpleMode;       
 Bool_t fUseSelectionBit;  
 Bool_t fOnOff[5];         
 Bool_t fFillDistrTrackEffChecks;

 ClassDef(AliAnalysisTaskSEHFQA,10); 
};

#endif