AliAnalysisTaskSELc2pK0sfromAODtracks.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKSELC2PK0SFROMAODTRACKS_H
#define ALIANALYSISTASKSELC2PK0SFROMAODTRACKS_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 "TVector.h"
#include "TSystem.h"

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


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

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

  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 *lcobj, AliAODv0 *v0, AliAODTrack *trk, AliAODTrack *trkpid, AliAODEvent *aod, TClonesArray *mcarray);
  void FillMixROOTObjects(TLorentzVector *pt, TLorentzVector *ev, TVector *tinfo, TVector *vinfo);
  Bool_t MakeMCAnalysis(TClonesArray *mcArray);
  void MakeAnalysis(AliAODEvent *aod, TClonesArray *mcArray);
  void FillProtonROOTObjects(AliAODTrack *trk, TClonesArray *mcArray);
  void FillMCROOTObjects(AliAODMCParticle *part, AliAODMCParticle *mcepart, AliAODMCParticle *mcv0part, Int_t decaytype);
  void FillV0ROOTObjects(AliAODv0 *v0, TClonesArray *mcArray);
  void FillMCProtonROOTObjects(AliAODMCParticle *mcepart, TClonesArray *mcArray);
  void FillMCV0ROOTObjects(AliAODMCParticle *mcv0part, TClonesArray *mcArray);
  void SelectV0( const AliVEvent *event,Int_t nV0,Int_t &nSeleV0, Bool_t *seleV0Flags, TClonesArray *mcArray);
  void SelectTrack( const AliVEvent *event, Int_t trkEntries, Int_t &nSeleTrks,Int_t *seleFlags, TClonesArray *mcArray);
  Int_t MatchToMC(AliAODRecoCascadeHF *elobj, TClonesArray *mcArray, Int_t *pdgarray_pr, Int_t *pdgarray_v0, Int_t *labelarray_pr, Int_t *labelarray_v0,  Int_t &ngen_pr, Int_t &ngen_v0);


  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 SetReconstructPrimVert(Bool_t a) { fReconstructPrimVert=a; }

  AliAODRecoCascadeHF* MakeCascadeHF(AliAODv0 *casc, AliAODTrack *trk, AliAODTrack *trkpid, AliAODEvent *aod, AliAODVertex *vert);
  AliAODVertex* ReconstructSecondaryVertex(AliAODv0 *casc, AliAODTrack *trk, AliAODEvent *aod);

  void StoreGlobalTrackReference(AliAODTrack *track, Int_t);
  void ResetGlobalTrackReference();

  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 SetPoolRPBinLimits(Int_t Nrpbins,const Double_t *rpbins){
    fNRPBins = Nrpbins;
    for(int ix = 0;ix<fNRPBins+1;ix++){fRPBins[ix] = rpbins[ix];}
  }
  void DoEventMixingWithPools(Int_t index);
  void ResetPool(Int_t poolIndex);
  Int_t GetPoolIndex(Double_t zvert, Double_t mult, Double_t rp);

 private:

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

  void DefineTreeVariables();
  void DefineProtonTreeVariables();
  void DefineV0TreeVariables();
  void DefineMCTreeVariables();
  void DefineMCProtonTreeVariables();
  void DefineMCV0TreeVariables();
  void DefineGeneralHistograms();
  void DefineAnalysisHistograms();

  AliAODVertex *CallPrimaryVertex(AliAODv0 *v0, 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;                
  TH1F *fHReactionPlane;                
  AliRDHFCutsLctopK0sfromAODtracks *fAnalCuts; 
  Bool_t fIsEventSelected;          
  Bool_t    fWriteVariableTree;     
  Bool_t    fWriteEachVariableTree;     
  Bool_t    fWriteMCVariableTree;     
  TTree    *fVariablesTree;         
  TTree    *fProtonVariablesTree;         
  TTree    *fV0VariablesTree;         
  TTree    *fMCVariablesTree;         
  TTree    *fMCProtonVariablesTree;         
  TTree    *fMCV0VariablesTree;         
  Bool_t fReconstructPrimVert;       
  Bool_t fIsMB;       
  Bool_t fIsSemi;     
  Bool_t fIsCent;     
  Bool_t fIsINT7;     
  Bool_t fIsEMC7;     
  Float_t *fCandidateVariables;   
  Float_t *fCandidateProtonVariables;   
  Float_t *fCandidateV0Variables;   
  Float_t *fCandidateMCVariables;   
  Float_t *fCandidateMCProtonVariables;   
  Float_t *fCandidateMCV0Variables;   
  AliAODVertex *fVtx1;            
  AliESDVertex *fV1;              
  Float_t  fVtxZ;         
  Double_t fBzkG;                 
  Float_t  fCentrality;           
  Float_t  fReactionPlane;           
  Int_t  fRunNumber;           
  Float_t  fTriggerCheck;         
  Bool_t  fUseCentralityV0M;         
  Int_t  fEvNumberCounter;         
  AliNormalizationCounter *fCounter;

  //--------------------- My histograms ------------------
  TH1F *fHistonEvtvsRunNumber;     
  TH1F *fHistonProtonvsRunNumber;     
  TH1F *fHistonK0svsRunNumber;     

  THnSparse* fHistoLcMCGen;         
  THnSparse* fHistoLcK0SpMass;         
  THnSparse* fHistoLcK0SpMassMix;         
  THnSparse* fHistoLcK0SpMassCoarse;         
  THnSparse* fHistoLcK0SpMassMixCoarse;         
  THnSparse* fHistoK0spCorrelation;         
  THnSparse* fHistoK0spCorrelationMix;         
  THnSparse* fHistoK0spCorrelationMCS;         
  THnSparse* fHistoLcK0SpMassMCS;         
  THnSparse* fHistoLcK0SpPi0MassMCS;         
  THnSparse* fHistoLcKPluspMass;         
  THnSparse* fHistoLcKMinuspMass;         
  THnSparse* fHistoLcKPluspMassMix;         
  THnSparse* fHistoLcKMinuspMassMix;         

  TH2F* fHistoBachPt;      
  TH2F* fHistoBachPtMCS;      
  TH2F* fHistoBachPtMCGen;      
  TH2F* fHistoKaonPt;      
  TH2F* fHistoKaonPtMCS;      
  TH2F* fHistoKaonPtMCGen;      
  TH3F* fHistoK0sMassvsPt;     
  TH3F* fHistoK0sMassvsPtMCS;     
  TH3F* fHistoK0sMassvsPtMCGen;     
  TH1F* fHistod0Bach;      
  TH1F* fHistod0V0;        
  TH1F* fHistod0d0;        
  TH1F* fHistoV0CosPA;     
  TH1F* fHistoProbProton;  
  TH1F* fHistoDecayLength; 
  TH1F* fHistoK0SMass;     
  TH1F* fHistoMassTagV0Min; 
  TH1F* fHistoMassTagV0SameSignMin; 

  TH2D *fHistoResponseLcPt; 
  TH2D *fHistoResponseLcPt1; 
  TH2D *fHistoResponseLcPt2; 

  // Store pointers to global tracks for pid and dca
  AliAODTrack **fGTI;                
  Int_t *fGTIndex;                
  const UShort_t  fTrackBuffSize;          

  //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 fNRPBins;               
  Double_t fRPBins[100];            // [fNRPBinsDim]
  Int_t  fNOfPools; 
  TTree** fEventBuffer;   
  TObjString *fEventInfo; 
  TObjArray* fProtonTracks; 
  TObjArray* fV0Tracks; 
  TObjArray* fProtonCutVarsArray; 
  TObjArray* fV0CutVarsArray; 


  ClassDef(AliAnalysisTaskSELc2pK0sfromAODtracks,5); 
};
#endif