PndFTSCAPerformance.h

Go to the documentation of this file.

//-*- Mode: C++ -*-
// $Id: PndFTSCAPerformance.h,v 1.9 2010/09/01 10:38:27 ikulakov Exp $
// ************************************************************************
// 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 PNDFTSCAPERFORMANCE_H
#define PNDFTSCAPERFORMANCE_H

#include "PndFTSPerformanceBase.h"

#include "PndFTSCADef.h"
#include "PndFTSArray.h"
#include "PndFTSCAMCTrack.h"
#include "PndFTSCAMCPoint.h"

#include <fstream>
#include <cstdio>
#include <map>
#include <string>
using std::string;

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

class PndFTSCAPerformance {
 public:
  typedef PndFTSPerformanceBase::PndFTSCAHitLabel PndFTSCAHitLabel;

  PndFTSCAPerformance();
  virtual ~PndFTSCAPerformance();

  bool SetNewEvent(PndFTSCAGBTracker *const Tracker, string mcTracksFile, string mcPointsFile); // set info for new event
  void InitSubPerformances();

  static PndFTSCAPerformance &Instance();

  void ExecPerformance();

  const PndFTSCAMCTrack &MCTrack(int i) const { return fMCTracks[i]; }
  const PndFTSCAHitLabel &HitLabel(int i) const { return fHitLabels[i]; }
  const PndFTSCAGBTracker *GetTracker() { return fTracker; };

  void SetTracker(PndFTSCAGBTracker *const tracker) { fTracker = tracker; };
  void SetTopoReconstructor(PndFTSTopoReconstructor *const tr) { fTopoReconstructor = tr; };
  void SetMCTracks(vector<PndFTSCAMCTrack> &mcTracks);
  void SetMCPoints(vector<PndFTSCALocalMCPoint> &mcPoints);
  void SetHitLabels(vector<PndFTSCAHitLabel> &hitLabels);

  PndFTSResizableArray<PndFTSCAHitLabel> *GetHitLabels() { return &fHitLabels; }        // array of hit MC labels
  PndFTSResizableArray<PndFTSCAMCTrack> *GetMCTracks() { return &fMCTracks; }           // array of MC tracks
  PndFTSResizableArray<PndFTSCALocalMCPoint> *GetMCPoints() { return &fLocalMCPoints; } // array of MC points in slices CS

  PndFTSPerformanceBase *GetSubPerformance(string name);

  const double *PV() const { return fPV; }

  bool CreateHistos(string name);
  void WriteHistos();

  void SetOutputFile(TFile *oF) { fOutputFile = oF; }

  void SaveDataInFiles(string prefix) const; // Save all MC Data in txt files. @prefix - prefix for file name. Ex: "./data/ev1"
  bool ReadData(vector<int> &labels, vector<PndFTSCALocalMCPoint> &mcpoints, vector<PndFTSCAMCTrack> &mctracks);

  int GetMCPoint(const PndFTSCAGBHit &hit) const;

  void ShiftHitsToMC(float errX = -1, float errY = -1);  // Use spreaded MCposition instead of hits
  void ResimulateHits(float errX = -1, float errY = -1); // Create new hits spreading MCPositions
  void RematchHits();                                    // Match hits with closest MCPoint
#ifdef DRIFT_TUBES
  void CombineHits(); // Rid of hits with same wires positions
  void DivideHitsOnLR();
#endif

  void CreateHistos();

 protected:
  void ReadMCEvent(FILE *in);
  void ReadLocalMCPoints(FILE *in);

  void WriteMCEvent(FILE *out) const;

  struct TSubPerformance {
    PndFTSPerformanceBase *perf;
    string name;
    bool IsGlobalPerf;

    TSubPerformance(){};
    TSubPerformance(PndFTSPerformanceBase *perf_, string name_, bool IsGlobalPerf_ = 1)
    {
      perf = perf_;
      name = name_;
      IsGlobalPerf = IsGlobalPerf_;
      perf->SetHistoCreated(0);
    };
    //       ~TSubPerformance(){if (perf) delete perf;};

    PndFTSPerformanceBase &operator*() { return *perf; }
    PndFTSPerformanceBase *operator->() { return perf; }
  };
  vector<TSubPerformance> subPerformances;
  //     vector<PndFTSCAPerformanceBase*> subPerformances;

  const PndFTSCAGBTracker *fTracker; // pointer to the tracker
  const PndFTSTopoReconstructor *fTopoReconstructor;
  PndFTSResizableArray<PndFTSCAHitLabel> fHitLabels;         // array of hit MC labels
  PndFTSResizableArray<PndFTSCAMCTrack> fMCTracks;           // array of MC tracks
  PndFTSResizableArray<PndFTSCALocalMCPoint> fLocalMCPoints; // array of MC points in slices CS

  double fPV[3]; // primary vertex position (x,y,z)

  int fStatNEvents; // n of events proceed

  TFile *fOutputFile;
  TDirectory *fHistoDir; // ROOT directory with histogramms

 private:
  void WriteDir2Current(TObject *obj);

  PndFTSCAPerformance(const PndFTSCAPerformance &);
  PndFTSCAPerformance &operator=(const PndFTSCAPerformance &);
};

#endif
#endif // DO_TPCCATRACKER_EFF_PERFORMANCE