AliAnalysisTaskMuonFakes.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKMUONFAKES_H
#define ALIANALYSISTASKMUONFAKES_H

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

/* $Id$ */ 

//Author: Philippe Pillot - SUBATECH Nantes

#include <TString.h>

#include "AliAnalysisTaskSE.h"
#include "AliMuonTrackCuts.h"

class TObjArray;
class AliCounterCollection;
class AliESDMuonTrack;
class AliMUONVTrackStore;
class AliMUONTrack;

class AliAnalysisTaskMuonFakes : public AliAnalysisTaskSE {
public:
  
  AliAnalysisTaskMuonFakes();
  AliAnalysisTaskMuonFakes(const char *name);
  virtual ~AliAnalysisTaskMuonFakes();
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *);
  virtual void   NotifyRun();
  virtual void   Terminate(Option_t *);
  
  void ShowProgressBar(Bool_t flag = kTRUE) { fShowProgressBar = flag; }
  
  void UseMCLabels(Bool_t flag = kTRUE) { fUseLabel = flag; }
  
  void CombineMCId(Bool_t flag = kTRUE) { fCombineMCId = flag; }
  
  void SetExternalSigmaCut(Double_t cut) { fExternalSigmaCut = cut; }
  
  void MatchTrigger(Bool_t flag = kTRUE) { fMatchTrig = flag; }
  
  void ApplyAccCut(Bool_t flag = kTRUE) { fApplyAccCut = flag; }
  
  void Chi2Cut(Double_t cut) { fChi2Cut = cut; }
  
  void PtCut(Double_t cut) { fPtCut = cut; }
  
  // set standard cuts to select tracks to be considered
  void SetMuonTrackCuts(AliMuonTrackCuts &trackCuts);
  
  void RecoParamLocation(const char* ocdbPath) { fRecoParamLocation = ocdbPath; }
  
  void DecayAsFake(Bool_t flag = kTRUE) { fDecayAsFake = flag; }
  
  void PrintDecayChain(Bool_t flag = kTRUE) { fPrintDecayChain = flag; }
  
  void DisableDetailedCounters(Bool_t flag = kTRUE) { fDisableDetailedCounters = flag; }
  
  TObjArray* GetCanvases() {return fCanvases;}
  
private:
  
  AliAnalysisTaskMuonFakes(const AliAnalysisTaskMuonFakes& rhs);
  AliAnalysisTaskMuonFakes& operator = (const AliAnalysisTaskMuonFakes& rhs);
  
  // return kTRUE if the track pass the section criteria
  Bool_t IsSelected(AliESDMuonTrack &esdTrack);
  
  // fill global histograms at track level
  void FillHistoTrack(Int_t histShift, Int_t nClusters, Int_t nChamberHit, Double_t normalizedChi2,
          Double_t p, Double_t pT, Double_t eta, Double_t phi, Double_t dca,
          Double_t thetaTrackAbsEnd, Double_t pdca, Double_t rAbs);
  
  void FillHistoPair(Int_t histShift, Double_t mass, Double_t p, Double_t pt,
         Double_t y, Double_t eta, Double_t phi);
  
  // look for fake tracks still connected to a reconstructible simulated track
  Int_t RemoveConnectedFakes(AliMUONVTrackStore &fakeTrackStore, AliMUONVTrackStore &trackRefStore,
           TString &selected, TString &centrality);
  
  // Check whether this combination of clusters correspond to a decaying particle or not
  Int_t IsDecay(Int_t nClusters, Int_t *chId, Int_t *labels, Bool_t &isReconstructible, Int_t &lastCh) const;
  
  // Try to match clusters between track and trackRef and add the corresponding MC labels to the arrays
  void AddCompatibleClusters(const AliMUONTrack &track, const AliMUONTrack &trackRef,
           TArrayI *labels, Int_t *nLabels) const;
  
  // Check whether this track correspond to a decaying particle by using cluster MC labels
  Int_t IsDecayByLabel(const AliMUONTrack &track, Bool_t &isReconstructible, Int_t &lastCh) const;
  
  // Check whether this track correspond to a decaying particle by comparing clusters position
  Int_t IsDecayByPosition(const AliMUONTrack &track, const AliMUONVTrackStore &trackRefStore,
        const AliMUONVTrackStore &usedTrackRefStore, Bool_t &isReconstructible,
        Int_t &lastCh) const;
  
private:
  
  enum histoIndex {
    kNumberOfClusters        = 0,  
    kNumberOfChamberHit      = 6,  
    kChi2PerDof              = 12, 
    kChi2PerDofVsNClusters   = 18, 
    kChi2PerDofVsNChamberHit = 24, 
    kP                       = 30, 
    kPt                      = 36, 
    kEta                     = 42, 
    kPhi                     = 48, 
    kDCA                     = 54, 
    kPDCA23                  = 60, 
    kPDCA310                 = 66, 
    kRAbs                    = 72, 
    kNhistTrack              = 78  
  };
  
  enum histoIndexAdd {
    kNumberOfTracks,             
    kNumberOfAdditionalTracks,   
    kNumberOfClustersMC,         
    kFractionOfMatchedClusters,  
    kFractionOfConnectedClusters 
  };
  
  enum histo2Index {
    k2Mass     = 0,  
    k2P        = 4,  
    k2Pt       = 8,  
    k2Y        = 12, 
    k2Eta      = 16, 
    k2Phi      = 20, 
    kNhistPair = 24  
  };
  
  TObjArray* fList;     
  TObjArray* fList2;    
  TObjArray* fCanvases; 
  
  AliCounterCollection* fTrackCounters;        
  AliCounterCollection* fFakeTrackCounters;    
  AliCounterCollection* fMatchedTrackCounters; 
  AliCounterCollection* fEventCounters;        
  AliCounterCollection* fPairCounters;         
  
  TString  fCurrentFileName;         
  UInt_t   fRequestedStationMask;    
  Bool_t   fRequest2ChInSameSt45;    
  Double_t fSigmaCut;                
  Int_t    fNEvents;                 
  Bool_t   fShowProgressBar;         
  Bool_t   fUseLabel;                
  Bool_t   fCombineMCId;             
  Double_t fExternalSigmaCut;        
  Bool_t   fMatchTrig;               
  Bool_t   fApplyAccCut;             
  Double_t fChi2Cut;                 
  Double_t fPtCut;                   
  TString  fRecoParamLocation;       
  Bool_t   fDecayAsFake;             
  Bool_t   fPrintDecayChain;         
  Bool_t   fDisableDetailedCounters; 
  
  AliMuonTrackCuts* fMuonTrackCuts; 
  
  ClassDef(AliAnalysisTaskMuonFakes, 3); // fake muon analysis
};


//________________________________________________________________________
inline void AliAnalysisTaskMuonFakes::SetMuonTrackCuts(AliMuonTrackCuts &trackCuts)
{
  delete fMuonTrackCuts;
  fMuonTrackCuts = new AliMuonTrackCuts(trackCuts);
}

#endif