//-*- Mode: C++ -*-
// $Id: AliHLTTPCCAMergerPerformance.h,v 1.3 2010/08/24 18:52:20 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 AliHLTTPCCAMergerPerformance_H
#define AliHLTTPCCAMergerPerformance_H


#include "AliHLTTPCCASlicePerformance.h"

#include "AliHLTTPCCADef.h"

#include "AliHLTTPCCAMCTrack.h"
#include "AliHLTTPCCAMCPoint.h"
#include <fstream>
#include <cstdio>
#include <map>

#include "TTree.h"
#include "TObject.h"
#include "TClonesArray.h"

class TObject;
class TParticle;
class AliHLTTPCCAMCPoint;
class AliHLTTPCCAGBTracker;
class TDirectory;
class TH1D;
class TH2D;
class TProfile;

class TFile;
class TBranch;
class AliHLTTPCCATracker;

/**
 * @class AliHLTTPCCAMergerPerformance
 */
 
 class MergerPerfDataEvent: public TObject{
 public:
  MergerPerfDataEvent() {};
  virtual ~MergerPerfDataEvent() 
  {
    if(MergerData) delete MergerData;
  }

 friend class AliHLTTPCCAMergerPerformance;
 
 protected:
  TClonesArray *MergerData;
  
  ClassDef(MergerPerfDataEvent,3)
};
 
class MergerPerfData: public TObject{
 public:
  MergerPerfData() {};
  virtual ~MergerPerfData() {};

  bool IsMerged;
  bool IsDzDs;              // = 1 if dz/ds the same
  bool IsSameTrack;         // = 1 if tracks originated from the same MC track
  bool IsCovMatrixPositive;  // = 1 if the diagonal elements > 0
  bool IsCovMatrixFinite;   // = 1 if the elements are not huge ()
  float chi2;
  float delta2; 
  float Si;
  float dy;
  float dz;
  float dsin;
  
  float nClust1;
  float nClust2;
    
  ClassDef(MergerPerfData,4)
};
 
class AliHLTTPCCAMergerPerformance: public AliHLTTPCCATrackPerformanceBase
{
  public:

    AliHLTTPCCAMergerPerformance():first_call(true){};
    virtual ~AliHLTTPCCAMergerPerformance()
    { 
      if(fMPDE) delete fMPDE;
    };

    virtual void SetNewEvent(const AliHLTTPCCAGBTracker * const Tracker,
                             AliHLTResizableArray<AliHLTTPCCAHitLabel> *hitLabels,
                             AliHLTResizableArray<AliHLTTPCCAMCTrack> *mcTracks,
                             AliHLTResizableArray<AliHLTTPCCALocalMCPoint> *localMCPoints);
    
      /// Calculate Efficiencies
    virtual void Exec(bool print = false);
    
      /// Histograms
    virtual void CreateHistos(string histoDir = "", TFile* outFile = 0);
    virtual void FillHistos();
    
    void FillTree();
    
    void AddMergerData(int step, int iSlice1, int iTrack1, int iSlice2, int iTrack2, bool IsDzDs, bool IsCovMatrixPositiv, 
                       bool IsCovMatrixFinite, float &chi2, float &delta2, float &Si, float &dy, float &dz, float &dsin, int nClust1, int nClust2);
    void SetMerged(int step);
    void FillMC();
    
  private:
    int iData;
    int iData2Step;
    vector<AliHLTTPCCASlicePerformance*> slicePerformances;
    bool first_call;
    TTree *MPDTree; // Merger Performance Data tree
    TTree *MPDTree2Step;
    TClonesArray *MergerData;
    MergerPerfDataEvent *fMPDE;
    MergerPerfDataEvent *fMPDE2Step;
};

#endif
#endif //DO_TPCCATRACKER_EFF_PERFORMANCE