PndFTSTopoPerformance.h

Go to the documentation of this file.

//-*- Mode: C++ -*-
// ************************************************************************
// This file is property of and copyright by the ALICE HLT Project        *
// ALICE Experiment at CERN, All rights reserved.                         *
// See cxx source for full Copyright notice                               *
//                                                                        *
//*************************************************************************
#ifdef DO_TPCCATRACKER_EFF_PERFORMANCE

#ifndef PNDFTSTOPOPERFORMANCE_H
#define PNDFTSTOPOPERFORMANCE_H

#include "PndFTSParticlePerformanceBase.h"

#include "PndFTSCADef.h"
#include "PndFTSArray.h"

#include "PndFTSCAMCTrack.h"
#include "PndFTSCAMCPoint.h"
#include "PndFTSCAMCVertex.h"
#include <fstream>
#include <cstdio>
#include <map>

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

class TObject;
class TParticle;
class PndFTSCAMCPoint;
class PndFTSCAGBTracker;
class PndFTSTopoReconstructor;
class TDirectory;
class TH1D;
class TH2D;
class TProfile;

class TFile;
class PndFTSCATracker;

class PndFTSTopoPerformance : public PndFTSParticlePerformanceBase {
 public:
  PndFTSTopoPerformance();
  virtual ~PndFTSTopoPerformance(){};

  virtual void SetNewEvent(const PndFTSCAGBTracker *const Tracker, PndFTSResizableArray<PndFTSCAHitLabel> *hitLabels, PndFTSResizableArray<PndFTSCAMCTrack> *mcTracks,
                           PndFTSResizableArray<PndFTSCALocalMCPoint> *localMCPoints);

  void SetNewEvent2(const PndFTSTopoReconstructor *const TopoReconstructor); // use together with SetNewEvent !!!

  // 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
  virtual void EfficiencyPerformance(){}; // current don't use eff

  virtual void PrintEfficiencyStatistic(){}; // current don't use eff
  virtual void PrintEfficiency(){};

  //     virtual void CreateHistos(string histoDir);
  virtual void FillHistos();

 private:
  void GetMCParticles();
  void MatchParticles();
  void CalculateEfficiency();
  void FindReconstructableMCParticles();
  void CheckMCParticleIsReconstructable(KFMCParticle &part);

  const PndFTSTopoReconstructor *fTopoReconstructor;

  vector<PndFTSCAMCVertex> fPrimVertices; // primary vertex positions (currently only one vertex is implemented)

  vector<KFMCParticle> vMCParticles; // MC particles

  vector<KFPartMatch> MCtoRParticleId; // array for match MC and reco particles
  vector<KFPartMatch> RtoMCParticleId;

  // defines the mode of an efficiency calculation
  // 1 - efficiency is normalized on the all MCParticles
  // 2 - efficiency is normalized on the MCParticles, which has all daughters reconstructable
  // 3 - efficiency is normalized on the MCParticles, which has all daughters reconstructed by the track finder
  const int fFindParticlesMode;
};

#endif
#endif // DO_TPCCATRACKER_EFF_PERFORMANCE