PndTrackingQA.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

#pragma once

#include "PndTrackFunctor.h"
#include "PndTrackingQARecoInfo.h"
#include "PndTrackingQAMCInfo.h"
#include "PndTrackingQABranchEnum.h"
#include "PndTrack.h"

#include "TString.h"

class PndTrackingQA {
 public:
  PndTrackingQA(TString trackBranchName, TString idealTrackName, Bool_t pndTrackNotTrackCand = kTRUE);

  PndTrackingQA(TString trackBranchName, TString idealTrackName, PndTrackFunctor *posTrack, Bool_t pndTrackNotTrackCand = kTRUE);

  virtual ~PndTrackingQA();

  virtual void Init();

  void AddHitsBranchName(TString name) { fBranchNames.push_back(name); }
  void SetHitsBranchNames(std::vector<TString> names) { fBranchNames = names; }

  void SetSttInfo(TClonesArray *sttHitArray, TClonesArray *sttTubeArray)
  {
    fSttHitArray = sttHitArray;
    fSttTubeArray = sttTubeArray;
  }

  void SetVerbose(int val) { fVerbose = val; }

  void AnalyseEvent(TClonesArray *recoTrackInfo, TClonesArray *mcTrackInfo);

 protected:
  std::map<FairLink, PndTrackingQAMCInfo *> FillMCTrackInfo(TClonesArray *mcTrackInfo);
  double CalcPCA(PndTrack *track);
  int GetMCInfoQualification(PndTrackingQAMCInfo *mcInfo, PndTrack *idealTrack);
  std::map<TString, FairMultiLinkedData> AnalyseTrackCand(PndTrackCand *trackCand);
  FairMultiLinkedData GetMCInfoForBranch(TString branchName, PndTrackCand *trackCand);
  void PrintTrackInfo(std::map<TString, FairMultiLinkedData> info);
  FairLink GetMostProbableMCTrack(std::map<TString, FairMultiLinkedData> &trackInfo);
  PndTrackingQARecoInfo GetRecoInfoFromRecoTrack(FairLink trackId, PndTrackingQAMCInfo *mcTrackInfo);
  void AssociateRecoTracksToMCTracks(PndTrackingQARecoInfo &recoInfo);

 private:
  std::vector<TString> fBranchNames; 
  FairRootManager *ioman = nullptr;
  TString fTrackBranchName;
  TString fIdealTrackName;
  Bool_t fPndTrackNotTrackCand = kTRUE; // kTRUE if track and kFALSE if track cand
  PndTrackFunctor *fPossibleTrack = nullptr;

  std::map<FairLink, PndTrackingQAMCInfo *> fMCInfoMap; //< MCTrackId, Associated MCInfo>

  Int_t fVerbose = 0;

  TClonesArray *fTrack = nullptr;
  TClonesArray *fMCTrack = nullptr;
  TClonesArray *fIdealTrack = nullptr;
  TClonesArray *fTrackCand = nullptr;
  TClonesArray *fIdealTrackCand = nullptr;

  TClonesArray *fSttTubeArray = nullptr;
  TClonesArray *fSttHitArray = nullptr;

  ClassDef(PndTrackingQA, 1);
};