AliAnalysisTaskSEXic2eleXifromAODtracks.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSEXIC2ELEXIFROMAODTRACKS_H
#define ALIANALYSISTASKSEXIC2ELEXIFROMAODTRACKS_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 "AliRDHFCutsXictoeleXifromAODtracks.h"


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

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

  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);
  void FillMixROOTObjects(TLorentzVector *et, TLorentzVector *ev, Int_t charge);
  void FillElectronROOTObjects(AliAODTrack *trk, TClonesArray *mcArray);
  void FillCascROOTObjects(AliAODcascade *casc, TClonesArray *mcArray);
  void FillMCROOTObjects(AliAODMCParticle *part, AliAODMCParticle *mcepart, AliAODMCParticle *mcv0part, Int_t decaytype);
  void FillMCEleROOTObjects(AliAODMCParticle *mcepart, TClonesArray *mcArray);
  void FillMCCascROOTObjects(AliAODMCParticle *mccpart, TClonesArray *mcArray);
  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);
  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);
  void ResetPool(Int_t poolIndex);
  Int_t GetPoolIndex(Double_t zvert, Double_t mult);


 private:

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

  void DefineTreeVariables();
  void DefineEleTreeVariables();
  void DefineCascTreeVariables();
  void DefineMCTreeVariables();
  void DefineMCEleTreeVariables();
  void DefineMCCascTreeVariables();
  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;                
  AliRDHFCutsXictoeleXifromAODtracks *fAnalCuts;// Cuts - sent to output slot 2
  Bool_t fIsEventSelected;          
  Bool_t    fWriteVariableTree;     
  TTree    *fVariablesTree;         
  Bool_t    fWriteEachVariableTree;     
  Bool_t    fWriteMCVariableTree;     
  TTree    *fEleVariablesTree;         
  TTree    *fCascVariablesTree;         
  TTree    *fMCVariablesTree;         
  TTree    *fMCEleVariablesTree;         
  TTree    *fMCCascVariablesTree;         
  Bool_t fReconstructPrimVert;       
  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;   
  Float_t *fCandidateMCEleVariables;   
  Float_t *fCandidateMCCascVariables;   
  AliAODVertex *fVtx1;            
  AliESDVertex *fV1;              
  Float_t  fVtxZ;         
  Double_t fBzkG;                 
  Float_t  fCentrality;           
  Int_t  fRunNumber;           
  Float_t  fTriggerCheck;         
  Bool_t  fUseCentralityV0M;         
  Int_t  fEvNumberCounter;         

  //--------------------- My histograms ------------------
  THnSparse* fHistoEleXiMass;         
  THnSparse* fHistoEleXiMassRS;         
  THnSparse* fHistoEleXiMassWS;         
  THnSparse* fHistoEleXiMassRSMix;         
  THnSparse* fHistoEleXiMassWSMix;         
  THnSparse* fHistoEleXiMassRSSide;         
  THnSparse* fHistoEleXiMassWSSide;         
  THnSparse* fHistoEleXiMassvsElePtRS;         
  THnSparse* fHistoEleXiMassvsElePtWS;         
  THnSparse* fHistoEleXiMassvsElePtRSMix;         
  THnSparse* fHistoEleXiMassvsElePtWSMix;         
  THnSparse* fHistoEleXiMassvsElePtRSSide;         
  THnSparse* fHistoEleXiMassvsElePtWSSide;         
  THnSparse* fHistoEleXiMassvsElePtRS1;         
  THnSparse* fHistoEleXiMassvsElePtWS1;         
  THnSparse* fHistoEleXiMassvsElePtRSMix1;         
  THnSparse* fHistoEleXiMassvsElePtWSMix1;         
  THnSparse* fHistoEleXiMassvsElePtRSSide1;         
  THnSparse* fHistoEleXiMassvsElePtWSSide1;         
  THnSparse* fHistoEleXiMassvsElePtRS2;         
  THnSparse* fHistoEleXiMassvsElePtWS2;         
  THnSparse* fHistoEleXiMassvsElePtRSMix2;         
  THnSparse* fHistoEleXiMassvsElePtWSMix2;         
  THnSparse* fHistoEleXiMassvsElePtRSSide2;         
  THnSparse* fHistoEleXiMassvsElePtWSSide2;         
  TH2F* fHistoElePtRS;         
  TH2F* fHistoElePtWS;         
  TH2F* fHistoElePtRSMix;         
  TH2F* fHistoElePtWSMix;         
  THnSparse* fHistoEleXiMassMCS;         
  THnSparse* fHistoEleXiMassMCGen;         
  THnSparse* fHistoEleXiMassvsElePtMCS;         
  THnSparse* fHistoEleXiMassvsElePtMCGen;         
  THnSparse* fHistoEleXiMassvsElePtMCS1;         
  THnSparse* fHistoEleXiMassvsElePtMCGen1;         
  THnSparse* fHistoEleXiMassvsElePtMCS2;         
  THnSparse* fHistoEleXiMassvsElePtMCGen2;         
  TH2F* fHistoElePtMCS;         
  TH2F* fHistoElePtMCGen;         

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

  THnSparse* fHistoElePtvsXiPtRS;         
  THnSparse* fHistoElePtvsXiPtWS;         
  THnSparse* fHistoElePtvsXiPtRSMix;         
  THnSparse* fHistoElePtvsXiPtWSMix;         
  THnSparse* fHistoElePtvsXiPtMCS;         
  THnSparse* fHistoElePtvsXiPtvsXicPtMCS;         
  THnSparse* fHistoElePtvsXiPtMCGen;         
  THnSparse* fHistoElePtvsXiPtvsXicPtMCGen;         
  THnSparse* fHistoElePtvsXiPtMCXicGen;         

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

  TH1F* fHistoBachPt;      
  TH1F* fHistoBachPtMCS;      
  TH1F* fHistoBachPtMCGen;      
  TH1F* fHistod0Bach;      
  TH2F* fHistoXiMassvsPt;     
  TH2F* fHistoXiMassvsPtMCS;     
  TH2F* fHistoXiMassvsPtMCGen;     
  TH2F* fHistoOmegaMassvsPt;     
  TH2F* fHistoElectronTPCPID;     
  TH2F* fHistoElectronTOFPID;     
  TH2F* fHistoElectronTPCSelPID;     
  TH2F* fHistoElectronTOFSelPID;     
  TH2F* fHistoElectronTPCPIDSelTOF;     
  TH2F* fHistoElectronTOFPIDSelTPC;     
  TH2F* fHistoElectronTPCPIDSelTOFSmallEta;     
  TH2F* fHistoElectronTPCPIDSelTOFLargeEta;     
  TH2F* fHistoElectronTPCPIDSelTOFEtaDep[8];     
  THnSparse* fHistoElePtvsCutVarsRS[23];         
  THnSparse* fHistoElePtvsCutVarsWS[23];         
  THnSparse* fHistoElePtvsCutVarsMCS[23];         
  TH2F* fHistoElectronQovPtvsPhi;     
  TH2F* fHistoXiQovPtvsPhi;     
  THnSparse* fHistoXicMCGen;         
  THnSparse* fHistoXicMCGen1;         
  THnSparse* fHistoXicMCGen2;         
  THnSparse* fHistoXicElectronMCGen;         
  THnSparse* fHistoXicElectronMCGen1;         
  THnSparse* fHistoXicElectronMCGen2;         
  THnSparse* fHistoElectronMCGen;         
  THnSparse* fHistoXiMCGen;         

  AliNormalizationCounter *fCounter;
  TH1F *fHistonEvtvsRunNumber;
  TH1F *fHistonElevsRunNumber;
  TH1F *fHistonXivsRunNumber;

  //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; 
  TObjArray* fCascadeTracks1; 
  TObjArray* fCascadeTracks2; 

  ClassDef(AliAnalysisTaskSEXic2eleXifromAODtracks,10); 
};
#endif