//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Data object carrying infomration about a fitted track
//      (residuals, chi2,...)
//      This class is meant to serve as a container which can be
//      used witht the ROOT TreeViewer to produce plots
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//
//
//-----------------------------------------------------------

#ifndef TRACKFITSTAT_HH
#define TRACKFITSTAT_HH

// Base Class Headers ----------------
#include "TObject.h"
#include "TVector3.h"

// Collaborating Class Headers -------
#include <ostream> // remove if you do not need streaming op
#include <vector>
// Collaborating Class Declarations --



class TrackFitStat : public TObject{
public:

  // Constructors/Destructors ---------
  TrackFitStat();
  virtual ~TrackFitStat(){}

  // Operators
  
  // Accessors -----------------------
  Int_t GetPdgCode()	{ return _nPDG; }
  Int_t GetMotherID()	{ return _motherID; }
  Double_t GetMCP()		{ return _pmc;	}
  Double_t GetP()       {return _p;}
  TVector3 GetMom()     {return _mom;}

  const std::vector<double>* GetResX() {return &_TpcResX;}
  const std::vector<double>* GetResY() {return &_TpcResY;}
  const std::vector<double>* GetResZ() {return &_TpcResZ;}
  const std::vector<double>* GetResU() {return &_TpcResU;}
  const std::vector<double>* GetResV() {return &_TpcResV;}
  const std::vector<double>* GetSigX() {return &_TpcSigX;}
  const std::vector<double>* GetSigY() {return &_TpcSigY;}
  const std::vector<double>* GetSigZ() {return &_TpcSigZ;}
  const std::vector<double>* GetChi2X() {return &_TpcChi2X;}
  const std::vector<double>* GetChi2Y() {return &_TpcChi2Y;}
  const std::vector<double>* GetChi2Z() {return &_TpcChi2Z;}
  const std::vector<unsigned int>* GetClSizes() {return &_TpcClusterSize;}
  const std::vector<unsigned int>* Get2DClSizes() {return &_Tpc2DClusterSize;}
  const std::vector<double>* GetAmps() {return &_TpcClusterAmp;}
  const std::vector<TVector3>* GetProjectionPoints() {return &_TpcProj;}
  const std::vector<double>* GetHitPositionsX() {return &_TpcPosX;}
  const std::vector<double>* GetHitPositionsY() {return &_TpcPosY;}
  const std::vector<double>* GetHitPositionsZ() {return &_TpcPosZ;}
  const std::vector<unsigned int>* GetHitIDs() {return &_TpcHitID;}


  // Modifiers -----------------------

  void fillTpcResX(const std::vector<double>& r){_TpcResX=r;}
  void fillTpcResY(const std::vector<double>& r){_TpcResY=r;}
  void fillTpcResZ(const std::vector<double>& r){_TpcResZ=r;}
  void fillTpcResU(const std::vector<double>& r){_TpcResU=r;}
  void fillTpcResV(const std::vector<double>& r){_TpcResV=r;}
  void fillTpcSigX(const std::vector<double>& r){_TpcSigX=r;}
  void fillTpcSigY(const std::vector<double>& r){_TpcSigY=r;}
  void fillTpcSigZ(const std::vector<double>& r){_TpcSigZ=r;}
  void fillTpcChi2X(const std::vector<double>& r){_TpcChi2X=r;}
  void fillTpcChi2Y(const std::vector<double>& r){_TpcChi2Y=r;}
  void fillTpcChi2Z(const std::vector<double>& r){_TpcChi2Z=r;}
  void fillTpcClusterSize(const std::vector<unsigned int>& s){_TpcClusterSize=s;}
  void fillTpc2DClusterSize(const std::vector<unsigned int>& s){_Tpc2DClusterSize=s;}
  void fillTpcClusterAmp(const std::vector<double>& s){_TpcClusterAmp=s;}
  void fillTpcProjPoints(const std::vector<TVector3>& s){_TpcProj=s;}
  void fillTpcHitPositionsX(const std::vector<double>& s){_TpcPosX=s;}
  void fillTpcHitPositionsY(const std::vector<double>& s){_TpcPosY=s;}
  void fillTpcHitPositionsZ(const std::vector<double>& s){_TpcPosZ=s;}
  void fillTpcHitIDs(const std::vector<unsigned int>& s){_TpcHitID=s;}
  
  void addFailedHits(int NFH) {_failedHits+=NFH;}
  void setNDF(unsigned int ndf) {_NDF=ndf;}
  void setChi2(double chi2) {_ChiSqu=chi2;}

  // Operations ----------------------

  void setp(double p){_p=p;}
  void setmom(TVector3 p){_mom=p;}
  void setpmc(double p){_pmc=p;_hasMc=true;}
  void setpstart(double pstart){_pstart=pstart;}
  void setsigp(double s){_sigp=s;}
  void setcharge(double q){_charge=q;}
  void setmccharge(double q){_mccharge=q;}
  void setpdg(Int_t pdg){_nPDG=pdg;}
  void setmotherid(Int_t motherID)	{_motherID=motherID;}
  unsigned int nhits() const {return _TpcResX.size();}

  unsigned int getFailedHits() {return _failedHits;}
  unsigned int getNDF() const {return _NDF;}    //NDF from fit
  double getChi2() const {return _ChiSqu;}      //ChiSqu from fit
  

private:

  // Private Data Members ------------
  std::vector<double> _TpcResX;
  std::vector<double> _TpcResY;
  std::vector<double> _TpcResZ;
  std::vector<double> _TpcResU;
  std::vector<double> _TpcResV;
  std::vector<double> _TpcSigX;
  std::vector<double> _TpcSigY;
  std::vector<double> _TpcSigZ;

  std::vector<double> _TpcChi2X;
  std::vector<double> _TpcChi2Y;
  std::vector<double> _TpcChi2Z;
  std::vector<unsigned int> _TpcClusterSize;
  std::vector<unsigned int> _Tpc2DClusterSize;
  std::vector<double> _TpcClusterAmp;
  std::vector<TVector3> _TpcProj;
  std::vector<double> _TpcPosX;
  std::vector<double> _TpcPosY;
  std::vector<double> _TpcPosZ;
  std::vector<unsigned int> _TpcHitID;

  double _p;
  double _pmc;
  double _pstart;
  TVector3 _mom;
  double _charge;
  double _mccharge;
  double _sigp;
  Int_t _nPDG;
  Int_t _motherID;

  unsigned int _failedHits;   //stores the number of failed hits (cov was zero)
  
  bool _hasMc;
  double _ChiSqu;
  unsigned int _NDF;

  // Private Methods -----------------

public:
  ClassDef(TrackFitStat,12)

};

#endif

//--------------------------------------------------------------
// $Log$
//--------------------------------------------------------------
//Version 8 : Chi2 info added