// -------------------------------------------------------------------------
// -----              CbmStsFitPerformanceTask header file                -----
// -----               Created 27/01/05  by E. Kryshen                 -----
// -------------------------------------------------------------------------


/**  CbmStsFitPerformanceTask.h
 **@author S.Gorbunov <sgorboun@mail.desy.de>
 **@author I.Kisel    <ikisel@kip.uni-heidelberg.de>
 **@author E.Kryshen  <e.kryshen@gsi.de>
 **
 ** Analysis of the KalmanTrackFitter perfomance
 **/

#ifndef CBMSTSFITPERFORMANCE_H
#define CBMSTSFITPERFORMANCE_H

#include "TClonesArray.h"
#include "CbmTrackParam.h"
#include "CbmStsPoint.h"
#include "CbmStsTrack.h"
#include "CbmTask.h"
#include "TH1D.h"
#include "TH2D.h"
#include "CbmMCTrack.h"
#include "CbmVertex.h"

using namespace std;

class CbmStsFitPerformanceTask : public CbmTask
{

public:

  /** Standard constructor */
  CbmStsFitPerformanceTask(const char *name="CbmStsFitPerformanceTask", Int_t iVerbose = 1);


  /** Destructor */
  ~CbmStsFitPerformanceTask();


  /** Initialization of the task */ 
  InitStatus Init();

  /** ReInitialization of the task */ 
  InitStatus ReInit();


  /** Execute task */ 
  void Exec(Option_t* option);


  /** Finish task */ 
  void Finish();

  /** This function fill state vectors t[], mc[] and covariance 
   ** matrix c[] from the given CbmTrackParam object and 
   ** related CbmStsPoint object. 
   ** @param in  is set to 1 if in values of position and momentum
   ** are considered.
   */
  
  void FillMCStateVectors(CbmStsPoint* point, Double_t mc[], Bool_t out = 0);

  /** This function fills track pulls and residuals at certain z-position */
  void FillTrackHisto(const Double_t mc[6], CbmStsTrack *track, TH1D* hist[10]);
 
  /** This function fills vertex pulls and residuals */
  void FillVertexHisto(TVector3 &mc, CbmVertex *V, TH1D* hist[8]  );

  /** Returns 1 if track has >=8 measurements and stMaxID-stMinID>=4 */
  Bool_t IsLong(CbmStsTrack* track);

  void DoTrackAnalysis(const bool ok=1 ) {fTrackAnalysis=ok;}
  void DoVertexAnalysis(const bool ok=1 ) {fVertexAnalysis=ok;}
  void DoD0Analysis(const bool ok=1 ) {fD0Analysis=ok;}
  
  Double_t GetCharge(CbmMCTrack* mcTrack);
  
 protected:

  void CreateTrackHisto(TH1D* hist[10], const char* name, const char* title );
  void CreateVertexHisto(TH1D* hist[9], const char* name, const char* title );
  void CreateD0Histo(TH1D* hist[14], const char* name, const char* title );

  Int_t fEvent;
  Bool_t fVertexAnalysis, fD0Analysis, fTrackAnalysis;

  TClonesArray* fMCTrackArray;      ///< MCTracks
  TClonesArray* fStsPointArray;     ///< StsPoints
  TClonesArray* fRecStsTrackArray;  ///< Reconstructed StsTracks
  TClonesArray* fHitArray;          ///< Sts hits
  CbmVertex* fPrimaryVertex;     ///< Primary vertex
  TClonesArray* fSTSTrackMatch;  ///< Related MC tracks

  TH1D* fhChi2;      // x=chi2; y=entries for all
  TH1D* fhProb;      // x=Prob function; y=entries for all

  TH1D* fhFrst[10]; 
  TH1D* fhMid[10];
  TH1D* fhLast[10];
  TH1D* fhImp[10];
  TH1D* fhVfit[10];
  TH1D* fhVtx[9];
  TH1D* fhV[13][9];
  TH1D* fhPq[11];
  TH1D *fhD0[4][14];
  TDirectory *histodir;

  ClassDef(CbmStsFitPerformanceTask,1);

};

#endif