BmnMatchRecoToMC.h

Go to the documentation of this file.

#ifndef BMNMATCHRECOTOMC_H_
#define BMNMATCHRECOTOMC_H_
 
#include "BmnEnums.h"
#include "BmnGemStripHit.h"
#include "BmnGemTrack.h"
#include "BmnLink.h"
#include "BmnMatch.h"
#include "BmnTrackMatch.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "FairLogger.h"
#include "FairMCPoint.h"
#include "FairTask.h"
#include "TClonesArray.h"
 
class TClonesArray;
 
class BmnMatchRecoToMC : public FairTask
{
  public:
    BmnMatchRecoToMC();
    virtual ~BmnMatchRecoToMC();
 
    virtual InitStatus Init();
    virtual void Exec(Option_t* opt);
    virtual void Finish();
 
    void SetInnerTracksBranchName(TString name) { fInnerTrackBranchName = name; }
 
  private:
    void ReadAndCreateDataBranches();
 
    BmnStatus LinkToMC(const TClonesArray* hitMatchArr,
                       const TClonesArray* points,
                       Int_t id,
                       BmnTrackMatch* trackMatch);
    BmnStatus LinkToMC(Int_t id, BmnTrackMatch* trackMatch);
 
    void MatchGlobalTracks();
    void MatchGlobalTracks(const TClonesArray* gemHitMatches,
                           const TClonesArray* silHitMatches,
                           const TClonesArray* ssdHitMatches,
                           const TClonesArray* gemPoints,
                           const TClonesArray* silPoints,
                           const TClonesArray* ssdPoints,
                           const TClonesArray* gemTracks,
                           const TClonesArray* silTracks,
                           const TClonesArray* ssdTracks,
                           const TClonesArray* globTracks,
                           TClonesArray* trackMatches);
 
    // STS
    TClonesArray* fStsHits;     // BmnStsHit array
    TClonesArray* fStsTracks;   // BmnStsTrack array
 
    // GEM
    TClonesArray* fGemPoints;   // BmnStsPoint array
    TClonesArray* fGemTracks;   // BmnGemTrack array
 
    // SILICON
    TClonesArray* fSilPoints;
    TClonesArray* fSilTracks;
 
    // GLOBAL
    TClonesArray* fGlobalTracks;
    TClonesArray* fGlobalTrackMatches;
 
    TString fInnerTrackBranchName;
 
    BmnMatchRecoToMC(const BmnMatchRecoToMC&) = delete;
    BmnMatchRecoToMC& operator=(const BmnMatchRecoToMC&) = delete;
 
    void CalculateTrackQuality(BmnGemTrack* locTr,
                               BmnTrackMatch* trMatch,
                               const TClonesArray* hitMatches,
                               const TClonesArray* points,
                               Int_t& trueCntr,
                               Int_t& falseCntr);
    void CalculateTrackQuality(CbmStsTrack* locTr, BmnTrackMatch* trMatch);
 
    ClassDef(BmnMatchRecoToMC, 1);
};
 
#endif /* BMNMATCHRECOTOMC_H_ */