CbmStsFindTracksQa.h

Go to the documentation of this file.

/* Copyright (C) 2006-2021 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Denis Bertini [committer], Sergey Gorbunov, Volker Friese */
 
// -------------------------------------------------------------------------
// -----                  CbmStsFindTracksQa header file               -----
// -----                  Created 11/01/06  by V. Friese               -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMSTSTRACKFINDERQA_H
#define CBMSTSTRACKFINDERQA_H 1
 
#include "CbmLink.h"
 
#include <FairTask.h>
 
#include <TFolder.h>
#include <TStopwatch.h>
#include <TVector3.h>
 
class TClonesArray;
class TH1;
class TH1F;
class CbmMCDataArray;
class CbmStsSetup;
class CbmMCDataManager;
class CbmTimeSlice;
class FairRootManager;
 
class CbmStsFindTracksQa : public FairTask {
 
 public:
  CbmStsFindTracksQa(Int_t iVerbose = 1);
 
  CbmStsFindTracksQa(Int_t minHits, Double_t quota, Int_t iVerbose);
 
  virtual ~CbmStsFindTracksQa();
 
  virtual void SetParContainers();
 
  virtual InitStatus Init();
 
  virtual InitStatus ReInit();
 
  virtual void Exec(Option_t* opt);
 
 private:
  virtual void Finish();
 
  InitStatus GetGeometry();
 
  void GetTargetPosition();
 
  void CreateHistos();
 
  void Reset();
 
  // collect id's of MC events participating to the data
  void CollectMcEvents(TClonesArray* Matches);
 
  void FillHitMap();
 
  void FillMatchMap(Int_t& nRec, Int_t& nGhosts, Int_t& nClones);
 
  void DivideHistos(TH1* histo1, TH1* histo2, TH1* histo3);
 
  struct McTrackInfo {
    std::map<Int_t, Int_t> fHitMap = {};  
    Int_t fStsTrackMatch           = -1;  
    Double_t fQuali                = 0.;  
    Int_t fMatchedNHitsAll         = 0;   
    Int_t fMatchedNHitsTrue        = 0;   
  };
 
  McTrackInfo& getMcTrackInfo(const CbmLink& link)
  {
    assert(fMcTrackInfoMap.find(link) != fMcTrackInfoMap.end());
    return fMcTrackInfoMap[link];
  }
 
  std::map<CbmLink, McTrackInfo> fMcTrackInfoMap = {};  
 
  FairRootManager* fManager    = nullptr;  
  CbmMCDataManager* fMcManager = nullptr;  
  CbmTimeSlice* fTimeSlice     = nullptr;  
 
  CbmMCDataArray* fMCTracks = nullptr;  
 
  Bool_t fIsMvdActive        = kTRUE;  // is the Mvd module active
  Int_t fMvdNstations        = 0;
  CbmMCDataArray* fMvdPoints = nullptr;  
  TClonesArray* fMvdCluster  = nullptr;  
  TClonesArray* fMvdHits     = nullptr;  
  TClonesArray* fMvdHitMatch = nullptr;  
 
  CbmStsSetup* fStsSetup         = nullptr;  // STS setup interface
  Int_t fStsNstations            = 0;        // Number of STS stations
  CbmMCDataArray* fStsPoints     = nullptr;  
  TClonesArray* fStsHits         = nullptr;  
  TClonesArray* fStsHitMatch     = nullptr;  
  TClonesArray* fStsClusterMatch = nullptr;  
  TClonesArray* fStsTracks       = nullptr;  
  TClonesArray* fStsTrackMatches = nullptr;  
 
  TVector3 fTargetPos = {0., 0., 0.};  // Target centre position
 
  Int_t fMinStations = 4;  // Minimal number of stations with hits for considered MCTrack
 
  Double_t fQuota = 0.7;  // True/all hits for track to be considered reconstructed
 
  TFolder fOutFolder = {"StsFindTracksQA", "StsFindTracksQA"};  
 
  // eff. vs. p, all
  TH1F* fhMomAccAll = nullptr;
  TH1F* fhMomRecAll = nullptr;
  TH1F* fhMomEffAll = nullptr;
 
  // eff. vs. p, vertex
  TH1F* fhMomAccPrim = nullptr;
  TH1F* fhMomRecPrim = nullptr;
  TH1F* fhMomEffPrim = nullptr;
 
  // eff. vs. p, non-vertex
  TH1F* fhMomAccSec = nullptr;
  TH1F* fhMomRecSec = nullptr;
  TH1F* fhMomEffSec = nullptr;
 
  // eff. vs. np, all
  TH1F* fhNpAccAll = nullptr;
  TH1F* fhNpRecAll = nullptr;
  TH1F* fhNpEffAll = nullptr;
 
  // eff. vs. np, vertex
  TH1F* fhNpAccPrim = nullptr;
  TH1F* fhNpRecPrim = nullptr;
  TH1F* fhNpEffPrim = nullptr;
 
  // eff. vs. np, non-vertex
  TH1F* fhNpAccSec = nullptr;
  TH1F* fhNpRecSec = nullptr;
  TH1F* fhNpEffSec = nullptr;
 
  // eff. vs. z, non-vertex
  TH1F* fhZAccSec = nullptr;
  TH1F* fhZRecSec = nullptr;
  TH1F* fhZEffSec = nullptr;
 
  // # hits of clones and ghosts
  TH1F* fhNhClones = nullptr;
  TH1F* fhNhGhosts = nullptr;
 
 
  TList* fHistoList = nullptr;
 
 
  Int_t fNAll          = 0;
  Int_t fNAccAll       = 0;
  Int_t fNAccPrim      = 0;
  Int_t fNAccRef       = 0;
  Int_t fNAccRefLong   = 0;
  Int_t fNAccSec       = 0;
  Int_t fNRecAll       = 0;
  Int_t fNRecPrim      = 0;
  Int_t fNRecRef       = 0;
  Int_t fNRecRefLong   = 0;
  Int_t fNRecSec       = 0;
  Int_t fNGhosts       = 0;
  Int_t fNClones       = 0;
  Int_t fNEvents       = 0;  
  Int_t fNEventsFailed = 0;  
  Double_t fTime       = 0.; 
  TStopwatch fTimer = {};
 
  CbmStsFindTracksQa(const CbmStsFindTracksQa&);
  CbmStsFindTracksQa operator=(const CbmStsFindTracksQa&);
 
  ClassDef(CbmStsFindTracksQa, 0);
};
 
 
#endif