KFTopoPerformance.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file KFTopoPerformance.h

/*
 * This file is part of KFParticle package
 * Copyright (C) 2007-2019 FIAS Frankfurt Institute for Advanced Studies
 *               2007-2019 Goethe University of Frankfurt
 *               2007-2019 Ivan Kisel <I.Kisel@compeng.uni-frankfurt.de>
 *               2007-2019 Maksym Zyzak
 *
 * KFParticle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * KFParticle is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifdef DO_TPCCATRACKER_EFF_PERFORMANCE

#ifndef KFTOPOPERFORMANCE_H
#define KFTOPOPERFORMANCE_H


#include "KFParticlePerformanceBase.h"

#include "KFMCVertex.h"
#include "KFMCTrack.h"
#include <cstdio>
#include <map>

#include "KFPartMatch.h"
#include "KFMCParticle.h"

class AliHLTTPCCAGBTracker;

class KFParticleTopoReconstructor;
class KFPHistogram;

class KFTopoPerformance: public KFParticlePerformanceBase
{
 public:
  
  KFTopoPerformance();
  virtual ~KFTopoPerformance();
#ifdef KFPWITHTRACKER
  virtual void SetNewEvent(
    const AliHLTTPCCAGBTracker * const Tracker,
    AliHLTResizableArray<AliHLTTPCCAHitLabel> *hitLabels,
    AliHLTResizableArray<AliHLTTPCCAMCTrack> *mcTracks,
    AliHLTResizableArray<AliHLTTPCCALocalMCPoint> *localMCPoints);
#endif  
  void SetTopoReconstructor( const KFParticleTopoReconstructor * const TopoReconstructor );
  const KFParticleTopoReconstructor * GetTopoReconstructor() const { return fTopoReconstructor; } 
    
    // Check if MC track is reconstructable. Calculate set of MC track. Etc.
  virtual void CheckMCTracks(); // fill mcData.
    // Find reco-MCTracks correspondence
  virtual void MatchTracks();   // fill recoData.
    // Calculate efficiencies
  
  void FillHistos();
  void FillHistos(const KFPHistogram* histograms);
  void FillMCHistos();

  void AddV0Histos();
  
  void SetTrackMatch(const std::vector<int>& trackMatch) { fTrackMatch = trackMatch;} 
  void SetMCTracks(const std::vector<KFMCTrack>& mcTracks) { vMCTracks = mcTracks; } 
  
  const KFPartEfficiencies GetEfficiency() const { return fParteff; } 
  void SetPrintEffFrequency(int n) { fPrintEffFrequency = n;} 

  const std::vector<KFMCVertex> GetPrimVertices() { return fPrimVertices; } 
  const std::vector<KFMCParticle>& MCParticles() { return vMCParticles; } 
  const std::vector<KFPartMatch>& ParticlesMatch() { return RtoMCParticleId; } 
  const std::vector<KFPartMatch>& GetMCtoRPVId() { return MCtoRPVId; } 
  const std::vector<KFPartMatch>& GetRtoMCPVId() { return RtoMCPVId; } 
  const KFMCTrack& GetMCTrack(const int iRecoTrack)
  { 
    int iMCTrack = 0;
    if(RtoMCParticleId[iRecoTrack].IsMatched())
      iMCTrack = RtoMCParticleId[iRecoTrack].GetBestMatch();
    return vMCTracks[iMCTrack]; 
  }
  
  void SetCentralityBin(const int iBin) { fCentralityBin = iBin; } 
  void SetCentralityWeight(const float weight) { fCentralityWeight = weight; } 
  
 private:

  const KFTopoPerformance& operator = (const KFTopoPerformance&); 
  KFTopoPerformance(const KFTopoPerformance&); 
   
  void GetMCParticles();
  void MatchParticles();
  void MatchPV();
  void CalculateEfficiency();
  void CalculatePVEfficiency();
  void FindReconstructableMCParticles();
  void CheckMCParticleIsReconstructable(KFMCParticle &part);
  void FindReconstructableMCVertices();
  void FillParticleParameters(KFParticle& TempPart,
                              int iParticle,
                              int iP,
                              int iPV,
                              TH1F* histoParameters[4][KFPartEfficiencies::nParticles][nHistoPartParam],
                              TH2F* histoParameters2D[4][KFPartEfficiencies::nParticles][nHistoPartParam2D],
                              TH3F* histoParameters3D[1][KFPartEfficiencies::nParticles][nHistoPartParam3D],
                              TH1F* histoFit[KFPartEfficiencies::nParticles][nFitQA] = 0,
                              TH1F* histoFitDaughtersQA[KFPartEfficiencies::nParticles][nFitQA] = 0,
                              TH1F* histoDSToParticleQA[KFPartEfficiencies::nParticles][nDSToParticleQA] = 0,
                              std::vector<int>* multiplicities = 0);
  
  const KFParticleTopoReconstructor *fTopoReconstructor; 

  std::vector<KFMCVertex> fPrimVertices; 
  std::vector<int> fMCTrackToMCPVMatch; 
  std::vector<double> fPVPurity; 
  std::vector<double> fPVTracksRate[4]; 
  std::vector<int> fNCorrectPVTracks; 

  std::vector<int> fTrackMatch; 
  std::vector<KFMCTrack> vMCTracks;  
  std::vector<KFMCParticle> vMCParticles;  
  std::vector<int> fNeutralIndex; 
  
  std::vector<KFPartMatch> MCtoRParticleId;
  std::vector<KFPartMatch> RtoMCParticleId;

  std::vector<KFPartMatch> MCtoRPVId;
  std::vector<KFPartMatch> RtoMCPVId;
  
  int fPrintEffFrequency; 
  
  int fCentralityBin; 
  float fCentralityWeight; 
};

#endif
#endif //DO_TPCCATRACKER_EFF_PERFORMANCE