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 "TSystem.h"

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


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

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, AliAODMCParticle *mcpart, AliAODMCParticle *mcdau1, AliAODMCParticle *mcdau2, Int_t mcnused);
  void MakeAnalysis(AliAODEvent *aod, TClonesArray *mcArray);


  void SetMC(Bool_t theMCon) {fUseMCInfo = theMCon;}
  Bool_t GetMC() const {return fUseMCInfo;}

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

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


 private:

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

  void DefineTreeVariables();
  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;                
  AliRDHFCutsLctopK0sfromAODtracks *fAnalCuts; 
  Bool_t fIsEventSelected;          
  Bool_t    fWriteVariableTree;     
  TTree    *fVariablesTree;         
  Bool_t fReconstructPrimVert;       
  Bool_t fIsMB;       
  Bool_t fIsSemi;     
  Bool_t fIsCent;     
  Bool_t fIsINT7;     
  Bool_t fIsEMC7;     
  Float_t *fCandidateVariables;   
  AliAODVertex *fVtx1;            
  AliESDVertex *fV1;              
  Double_t fBzkG;                 
  Float_t  fCentrality;           
  Float_t  fTriggerCheck;         

  //--------------------- My histograms ------------------
  THnSparse* fHistoLcK0SpMass;         

  TH1F* fHistoBachPt;      
  TH1F* fHistod0Bach;      
  TH1F* fHistod0V0;        
  TH1F* fHistod0d0;        
  TH1F* fHistoV0CosPA;     
  TH1F* fHistoProbProton;  
  TH1F* fHistoDecayLength; 
  TH1F* fHistoK0SMass;     


  ClassDef(AliAnalysisTaskSELc2pK0sfromAODtracks,2); 
};
#endif