PndCAPerformance.h

Go to the documentation of this file.

//-*- Mode: C++ -*-
// $Id: PndCAPerformance.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 PNDCAPERFORMANCE_H
#define PNDCAPERFORMANCE_H

#include "PndCAPerformanceBase.h"

#include "PndCADef.h"
#include "PndCAMCTrack.h"
#include "PndCAMCPoint.h"

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

class TObject;
class TParticle;
class PndCAHit;
class PndCAMCPoint;
class PndCAGBTracker;
class PndCATopoReconstructor;
class TDirectory;
class TH1D;
class TH2D;
class TProfile;

class PndCAPerformance
{
  public:

    typedef PndCAPerformanceBase::PndCAHitLabel PndCAHitLabel;

    PndCAPerformance();
    virtual ~PndCAPerformance();

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

    static PndCAPerformance &Instance();
    
    void ExecPerformance();

  const PndCAMCTrack &MCTrack(int i) const { return fMCTracks[i]; }
  const PndCAHitLabel &HitLabel(int i) const { return fHitLabels[i]; }
  const PndCAGBTracker *GetTracker(){ return fTracker; };
    

  void SetTracker( PndCAGBTracker* const tracker ){ fTracker = tracker; };
  void SetTopoReconstructor( PndCATopoReconstructor* const tr ){ fTopoReconstructor = tr; };
  void SetMCTracks(vector<PndCAMCTrack>& mcTracks);
  void SetMCPoints(vector<PndCALocalMCPoint>& mcPoints);
  void SetHitLabels(vector<PndCAHitLabel>& hitLabels);

  vector<PndCAHitLabel>     * GetHitLabels() { return &fHitLabels; }      // array of hit MC labels
  vector<PndCAMCTrack>      * GetMCTracks()  { return &fMCTracks;  }      // array of MC tracks
  vector<PndCALocalMCPoint> * GetMCPoints()  { return &fLocalMCPoints;}   // array of MC points in slices CS

  PndCAPerformanceBase* 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 ReadDataFromFiles( string prefix ); // @prefix - prefix for file name. Ex: "./data/ev1"


  void CombineHits(); // Rid of hits with same wires positions
  
  protected:

    void CreateHistos();
    
    void ReadMCEvent( FILE *in );
    void ReadLocalMCPoints( FILE *in );

    void WriteMCEvent( FILE *out ) const;

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

      TSubPerformance(){};
      TSubPerformance(PndCAPerformanceBase* perf_, string name_, bool IsGlobalPerf_ = 1){
        perf = perf_;
        name = name_;
        IsGlobalPerf = IsGlobalPerf_;
        perf->SetHistoCreated(0);
      };
//       ~TSubPerformance(){if (perf) delete perf;};
      
      PndCAPerformanceBase& operator*(){return *perf;}
      PndCAPerformanceBase* operator->(){return perf;}
    };
    vector<TSubPerformance> subPerformances;
//     vector<PndCAPerformanceBase*> subPerformances;
    
    const PndCAGBTracker *fTracker; // pointer to the tracker
    const PndCATopoReconstructor* fTopoReconstructor;
    vector<PndCAHitLabel> fHitLabels; // array of hit MC labels
    vector<PndCAMCTrack> fMCTracks;   // array of MC tracks
    vector<PndCALocalMCPoint> 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 );
  
    PndCAPerformance( const PndCAPerformance& );
    PndCAPerformance &operator=( const PndCAPerformance& );

};

#endif
#endif //DO_TPCCATRACKER_EFF_PERFORMANCE