AliAnalysisTaskSEOmegac2eleOmegafromAODtracks.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSEOMEGAC2ELEOMEGAFROMAODTRACKS_H
#define ALIANALYSISTASKSEOMEGAC2ELEOMEGAFROMAODTRACKS_H

/**************************************************************************
 * Copyright(c) 1998-2009, 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.                  *
 **************************************************************************/

/* $Id$ */ 

#include "TROOT.h"
#include "TSystem.h"

#include "AliAnalysisTaskSE.h"
#include "AliAODEvent.h"
#include "AliPID.h"
#include "AliRDHFCutsOmegactoeleOmegafromAODtracks.h"


class THnSparse;
class TH1F;
class TClonesArray;
class AliAODRecoCascadeHF;
class AliESDVertex;
class AliAODMCParticle;
class AliEventPoolManager;
class AliNormalizationCounter;

class AliAnalysisTaskSEOmegac2eleOmegafromAODtracks : public AliAnalysisTaskSE 
{
 public:
  AliAnalysisTaskSEOmegac2eleOmegafromAODtracks();
  AliAnalysisTaskSEOmegac2eleOmegafromAODtracks(const Char_t* name, AliRDHFCutsOmegactoeleOmegafromAODtracks* cuts, Bool_t writeVariableTree=kTRUE);
  virtual ~AliAnalysisTaskSEOmegac2eleOmegafromAODtracks();

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

  void FillROOTObjects(AliAODRecoCascadeHF *elobj, AliAODcascade *casc, AliAODTrack *trk, TClonesArray *mcArray, Bool_t mixing);
  void FillElectronROOTObjects(AliAODTrack *trk, TClonesArray *mcArray);
  void FillCascROOTObjects(AliAODcascade *casc, TClonesArray *mcArray);
  void FillMCROOTObjects(AliAODMCParticle *part, AliAODMCParticle *mcepart, AliAODMCParticle *mcv0part, Int_t decaytype);
  void MakeMCAnalysis(TClonesArray *mcArray);
  void MakeAnalysis(AliAODEvent *aod, TClonesArray *mcArray);

  void SelectCascade( const AliVEvent *event,Int_t nCasc,Int_t &nSeleCasc, Bool_t *seleCascFlags, TClonesArray *mcArray);
  void SelectTrack( const AliVEvent *event, Int_t trkEntries, Int_t &nSeleTrks,Bool_t *seleFlags, TClonesArray *mcArray);

  void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
  Bool_t GetMC() const {return fUseMCInfo;}
  void SetUseCentralityV0M(Bool_t centon) {fUseCentralityV0M = centon;}
  Bool_t GetUseCentralityV0M() const {return fUseCentralityV0M;}
  void SetWriteEachVariableTree(Bool_t a) {fWriteEachVariableTree = a;}
  Bool_t GetWriteEachVariableTree() const {return fWriteEachVariableTree;}
  void SetWriteMCVariableTree(Bool_t a) {fWriteMCVariableTree = a;}
  Bool_t GetWriteMCVariableTree() const {return fWriteMCVariableTree;}

  void SetReconstructPrimVert(Bool_t a) { fReconstructPrimVert=a; }

  AliAODRecoCascadeHF* MakeCascadeHF(AliAODcascade *casc, AliAODTrack *trk, AliAODEvent *aod, AliAODVertex *vert, Bool_t mixing);
  AliAODVertex* ReconstructSecondaryVertex(AliAODcascade *casc, AliAODTrack *trk, AliAODEvent *aod);
  Int_t MatchToMC(AliAODRecoCascadeHF *elobj, TClonesArray *mcArray, Int_t *pdgarray_ele, Int_t *pdgarray_casc, Int_t *labelarray_ele, Int_t *labelarray_casc,  Int_t &ngen_ele, Int_t &ngen_casc);
  Int_t MatchToMCCascade(AliAODcascade *theCascade, Int_t pdgabscasc, Int_t *pdgDgcasc, Int_t *pdgDgv0, TClonesArray *mcArray) const;


  void SetEventMixingWithPools(){fDoEventMixing=1;}
  void SetEventMixingOff(){fDoEventMixing=0;}
  void SetNumberOfEventsForMixing(Int_t events){fNumberOfEventsForMixing=events;}
  void SetPoolPVzBinLimits(Int_t Nzvtxbins,const Double_t *ZvtxBins){
    fNzVtxBins = Nzvtxbins;
    for(int ix = 0;ix<fNzVtxBins+1;ix++){fZvtxBins[ix] = ZvtxBins[ix];}
  }
  void SetPoolCentBinLimits(Int_t Ncentbins,const Double_t *CentBins){
    fNCentBins = Ncentbins;
    for(int ix = 0;ix<fNCentBins+1;ix++){fCentBins[ix] = CentBins[ix];}
  }
  void DoEventMixingWithPools(Int_t index,AliAODEvent *aodEvent, Bool_t *seleFlags);
  void ResetPool(Int_t poolIndex);
  Int_t GetPoolIndex(Double_t zvert, Double_t mult);


 private:

  AliAnalysisTaskSEOmegac2eleOmegafromAODtracks(const AliAnalysisTaskSEOmegac2eleOmegafromAODtracks &source);
  AliAnalysisTaskSEOmegac2eleOmegafromAODtracks& operator=(const AliAnalysisTaskSEOmegac2eleOmegafromAODtracks& source); 

  void DefineTreeVariables();
  void DefineEleTreeVariables();
  void DefineCascTreeVariables();
  void DefineMCTreeVariables();
  void DefineGeneralHistograms();
  void DefineAnalysisHistograms();

  AliAODVertex *CallPrimaryVertex(AliAODcascade *casc, AliAODTrack *trk, AliAODEvent *evt);
  AliAODVertex* PrimaryVertex(const TObjArray *trkArray,AliVEvent *event);

  Bool_t fUseMCInfo;                 
  TList *fOutput;                    
  TList *fOutputAll;                 
  TList *fListCuts;                  
  TH1F *fCEvents;                    
  TH1F *fHTrigger;                   
  TH1F *fHCentrality;                
  AliRDHFCutsOmegactoeleOmegafromAODtracks *fAnalCuts;// Cuts - sent to output slot 2
  Bool_t fIsEventSelected;          // flag for event selected
  Bool_t    fWriteVariableTree;     // flag to decide whether to write the candidate variables on a tree variables
  TTree    *fVariablesTree;         
  Bool_t    fWriteEachVariableTree;     
  Bool_t    fWriteMCVariableTree;     
  TTree    *fEleVariablesTree;         
  TTree    *fCascVariablesTree;         
  TTree    *fMCVariablesTree;         
  Bool_t fReconstructPrimVert;       // /Reconstruct primary vertex excluding candidate tracks
  Bool_t fIsMB;       
  Bool_t fIsSemi;     
  Bool_t fIsCent;     
  Bool_t fIsINT7;     
  Bool_t fIsEMC7;     
  Float_t *fCandidateVariables;   
  Float_t *fCandidateEleVariables;   
  Float_t *fCandidateCascVariables;   
  Float_t *fCandidateMCVariables;   
  AliAODVertex *fVtx1;            
  AliESDVertex *fV1;              
  Float_t  fVtxZ;         
  Double_t fBzkG;                 
  Float_t  fCentrality;           
  Float_t  fTriggerCheck;         
  Bool_t  fUseCentralityV0M;         
  Int_t  fEvNumberCounter;         

  //--------------------- My histograms ------------------
  THnSparse* fHistoEleOmegaMass;         
  THnSparse* fHistoEleOmegaMassRS;         
  THnSparse* fHistoEleOmegaMassWS;         
  THnSparse* fHistoEleOmegaMassRSMix;         
  THnSparse* fHistoEleOmegaMassWSMix;         
  THnSparse* fHistoEleOmegaMassvsElePtRS;         
  THnSparse* fHistoEleOmegaMassvsElePtWS;         
  THnSparse* fHistoEleOmegaMassvsElePtRSMix;         
  THnSparse* fHistoEleOmegaMassvsElePtWSMix;         
  TH2F* fHistoElePtRS;         
  TH2F* fHistoElePtWS;         
  TH2F* fHistoElePtRSMix;         
  TH2F* fHistoElePtWSMix;         
  THnSparse* fHistoEleOmegaMassMCS;         
  THnSparse* fHistoEleOmegaMassMCGen;         
  THnSparse* fHistoEleOmegaMassvsElePtMCS;         
  THnSparse* fHistoEleOmegaMassvsElePtMCGen;         
  TH2F* fHistoElePtMCS;         
  TH2F* fHistoElePtMCGen;         

  THnSparse* fHistoElePtvsEtaRS;         
  THnSparse* fHistoElePtvsEtaWS;         
  THnSparse* fHistoElePtvsEtaRSMix;         
  THnSparse* fHistoElePtvsEtaWSMix;         
  THnSparse* fHistoElePtvsEtaMCS;         
  THnSparse* fHistoElePtvsEtaMCGen;         

  THnSparse* fHistoElePtvsOmegaPtRS;         
  THnSparse* fHistoElePtvsOmegaPtWS;         
  THnSparse* fHistoElePtvsOmegaPtRSMix;         
  THnSparse* fHistoElePtvsOmegaPtWSMix;         
  THnSparse* fHistoElePtvsOmegaPtMCS;         
  THnSparse* fHistoElePtvsOmegaPtMCGen;         

  THnSparse* fHistoElePtvsd0RS;         
  THnSparse* fHistoElePtvsd0WS;         
  THnSparse* fHistoElePtvsd0RSMix;         
  THnSparse* fHistoElePtvsd0WSMix;         
  THnSparse* fHistoElePtvsd0MCS;         

  TH1F* fHistoBachPt;      
  TH1F* fHistoBachPtMCS;      
  TH1F* fHistoBachPtMCGen;      
  TH1F* fHistod0Bach;      
  TH2F* fHistoXiMassvsPt;     
  TH2F* fHistoOmegaMassvsPt;     
  TH2F* fHistoOmegaMassvsPtMCS;     
  TH2F* fHistoOmegaMassvsPtMCGen;     
  TH2F* fHistoElectronTPCPID;     
  TH2F* fHistoElectronTOFPID;     
  TH2F* fHistoElectronTPCSelPID;     
  TH2F* fHistoElectronTOFSelPID;     
  TH2F* fHistoElectronTPCPIDSelTOF;     
  TH2F* fHistoElectronTPCPIDSelTOFSmallEta;     
  TH2F* fHistoElectronTPCPIDSelTOFLargeEta;     

  AliNormalizationCounter *fCounter;
  TH1F *fHistonEvtvsRunNumber;
  TH1F *fHistonElevsRunNumber;
  TH1F *fHistonOmegavsRunNumber;

  //Mixing
  Int_t fDoEventMixing; 
  Int_t  fNumberOfEventsForMixing; 
  Int_t fNzVtxBins;               
  Double_t fZvtxBins[100];            // [fNzVtxBinsDim]
  Int_t fNCentBins;               
  Double_t fCentBins[100];            // [fNCentBinsDim]
  Int_t  fNOfPools; 
  TTree** fEventBuffer;   
  TObjString *fEventInfo; 
  TObjArray* fElectronTracks; 


  ClassDef(AliAnalysisTaskSEOmegac2eleOmegafromAODtracks,4); 
};
#endif