AliCaloTrackMatcher.h

Go to the documentation of this file.

#ifndef ALICALOTRACKMATCHER_H
#define ALICALOTRACKMATCHER_H

#include "AliAnalysisTaskSE.h"
#include "AliEMCALGeometry.h"
#include "AliPHOSGeometry.h"
#include <vector>
#include <map>
#include <utility>

class TF1;

using namespace std;

class AliCaloTrackMatcher : public AliAnalysisTaskSE {

  public:
    AliCaloTrackMatcher(const char *name="CaloTrackMatcher_0", Int_t clusterType=0);
    //Uncopyable & operator=(const Uncopyable&);

    virtual ~AliCaloTrackMatcher();                            //virtual destructor
    void UserCreateOutputObjects();

    virtual void UserExec(Option_t *option);
    virtual void Terminate(Option_t *);

    TList* GetCaloTrackMatcherHistograms() {return fListHistos;}

    void SetV0ReaderName(TString name)    {fV0ReaderName = name; return;}
    void SetCorrectionTaskSetting(TString setting) {fCorrTaskSetting = setting;}
    void SetAnalysisTrainMode(TString mode){fAnalysisTrainMode = mode; return;}
    void SetMatchingResidual(Float_t res) {fMatchingResidual = res; return;}
    void SetMatchingWindow(Float_t win) {fMatchingWindow = win; return;}

    // for cluster <-> primary matching
    Bool_t GetTrackClusterMatchingResidual(Int_t trackID, Int_t clusterID, Float_t &dEta, Float_t &dPhi);

    Int_t GetNMatchedTrackIDsForCluster(AliVEvent *event, Int_t clusterID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    Int_t GetNMatchedTrackIDsForCluster(AliVEvent *event, Int_t clusterID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    Int_t GetNMatchedTrackIDsForCluster(AliVEvent *event, Int_t clusterID, Float_t dR);

    Int_t GetNMatchedClusterIDsForTrack(AliVEvent *event, Int_t trackID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    Int_t GetNMatchedClusterIDsForTrack(AliVEvent *event, Int_t trackID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    Int_t GetNMatchedClusterIDsForTrack(AliVEvent *event, Int_t trackID, Float_t dR);

    vector<Int_t> GetMatchedTrackIDsForCluster(AliVEvent *event, Int_t clusterID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    vector<Int_t> GetMatchedTrackIDsForCluster(AliVEvent *event, Int_t clusterID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    vector<Int_t> GetMatchedTrackIDsForCluster(AliVEvent *event, Int_t clusterID, Float_t dR);

    vector<Int_t> GetMatchedClusterIDsForTrack(AliVEvent *event, Int_t trackID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    vector<Int_t> GetMatchedClusterIDsForTrack(AliVEvent *event, Int_t trackID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    vector<Int_t> GetMatchedClusterIDsForTrack(AliVEvent *event, Int_t trackID, Float_t dR);
    
    // for cluster <-> V0-track matching
    Bool_t PropagateV0TrackToClusterAndGetMatchingResidual(AliVTrack* inSecTrack, AliVCluster* cluster, AliVEvent* event, Float_t &dEta, Float_t &dPhi);
    Bool_t IsSecTrackClusterAlreadyTried(Int_t trackID, Int_t clusterID);
    Bool_t GetSecTrackClusterMatchingResidual(Int_t trackID, Int_t clusterID, Float_t &dEta, Float_t &dPhi);

    Int_t GetNMatchedClusterIDsForSecTrack(AliVEvent *event, Int_t clusterID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    Int_t GetNMatchedClusterIDsForSecTrack(AliVEvent *event, Int_t clusterID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    Int_t GetNMatchedClusterIDsForSecTrack(AliVEvent *event, Int_t clusterID, Float_t dR);

    Int_t GetNMatchedSecTrackIDsForCluster(AliVEvent *event, Int_t trackID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    Int_t GetNMatchedSecTrackIDsForCluster(AliVEvent *event, Int_t trackID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    Int_t GetNMatchedSecTrackIDsForCluster(AliVEvent *event, Int_t trackID, Float_t dR);

    vector<Int_t> GetMatchedSecTrackIDsForCluster(AliVEvent *event, Int_t clusterID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    vector<Int_t> GetMatchedSecTrackIDsForCluster(AliVEvent *event, Int_t clusterID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    vector<Int_t> GetMatchedSecTrackIDsForCluster(AliVEvent *event, Int_t clusterID, Float_t dR);

    vector<Int_t> GetMatchedClusterIDsForSecTrack(AliVEvent *event, Int_t trackID, Float_t dEtaMax, Float_t dEtaMin, Float_t dPhiMax, Float_t dPhiMin);
    vector<Int_t> GetMatchedClusterIDsForSecTrack(AliVEvent *event, Int_t trackID, TF1* fFuncPtDepEta, TF1* fFuncPtDepPhi);
    vector<Int_t> GetMatchedClusterIDsForSecTrack(AliVEvent *event, Int_t trackID, Float_t dR);

    //general methods
    Float_t SumTrackEtAroundCluster(AliVEvent* event, Int_t clusterID, Float_t dR);

  private:
    //typedefs
    typedef pair<Int_t, Int_t> pairInt;
    typedef pair<Float_t, Float_t> pairFloat;
    typedef map<pairInt, Int_t> mapT;

    AliCaloTrackMatcher (const AliCaloTrackMatcher&); // not implemented
    AliCaloTrackMatcher & operator=(const AliCaloTrackMatcher&); // not implemented

    // private methods
    void Initialize(Int_t runNumber);
    void ProcessEvent(AliVEvent *event);
    void SetLogBinningYTH2(TH2* histoRebin);

    // debug methods
    void DebugMatching();
    void DebugV0Matching();

    // basic variables/objects
    Int_t                 fClusterType;            // EMCal(1), PHOS(2) or not running (0)
    TString               fV0ReaderName;           // Name of V0Reader
    TString               fCorrTaskSetting;        // Name of Corr Task Setting
    TString               fAnalysisTrainMode;      // AnalysisTrainMode: Grid or GSI //Grid by default
    Double_t              fMatchingWindow;         // matching window to prevent unnecessary propagations
    Float_t               fMatchingResidual;       // matching residual below which track <-> cluster associations should be stored
    Int_t                 fRunNumber;              // current run number

    AliEMCALGeometry*     fGeomEMCAL;              // pointer to EMCAL geometry
    AliPHOSGeometry*      fGeomPHOS;               // pointer to PHOS geometry

    multimap<Int_t,Int_t> fMapTrackToCluster;      // connects a given track ID with all associated cluster IDs
    multimap<Int_t,Int_t> fMapClusterToTrack;      // connects a given cluster ID with all associated track IDs

    Int_t                 fNEntries;               // number of current TrackID/ClusterID -> Eta/Phi connections
    vector<pairFloat>     fVectorDeltaEtaDeltaPhi; // vector of all matching residuals for a specific TrackID/ClusterID
    mapT                  fMap_TrID_ClID_ToIndex;  // map tuple of (trackID,clusterID) to index in vector fVectorDeltaEtaDeltaPhi

    // for cluster <-> V0-track matching (running with different mass hypthesis)
    multimap<Int_t,Int_t> fSecMapTrackToCluster;      // connects a given secondary track ID with all associated cluster IDs
    multimap<Int_t,Int_t> fSecMapClusterToTrack;      // connects a given cluster ID with all associated secondary track IDs

    Int_t                 fSecNEntries;               // number of current V0-trackIDs/clusterID -> Eta/Phi connections
    vector<pairFloat>     fSecVectorDeltaEtaDeltaPhi; // vector of all matching residuals for a specific V0-trackIDs/clusterID
    mapT                  fSecMap_TrID_ClID_ToIndex;  // map tuple of (V0-trackID,clusterID) to index in vector fSecVectorDeltaEtaDeltaPhi
    mapT                  fSecMap_TrID_ClID_AlreadyTried;  // map tuple of (V0-trackID,clusterID) to matching outcome, successful or not

    //histos
    TList*                fListHistos;             // list with histogram(s)
    TH2F*                 fHistControlMatches;     // bookkeeping for processed tracks/clusters and succesful matches
    TH2F*                 fSecHistControlMatches;  // bookkeeping for processed V0-tracks/clusters and succesful matches

    ClassDef(AliCaloTrackMatcher,4)
};

#endif