//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Track on Riemann Sphere
//      Circle parameters can be calculated from plane parameters
//      plane(c,nx,ny,nz);
//
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Sebastian Neubert    TUM            (original author)
//
//
//-----------------------------------------------------------

#ifndef PNDRIEMANNTRACK_HH
#define PNDRIEMANNTRACK_HH

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

// Collaborating Class Headers -------
#include <vector>
#include "TVectorD.h"
#include "TVector3.h"
#include "TMatrixD.h"
#include "TMath.h"

// Collaborating Class Declarations --
#include "PndRiemannHit.h"


class PndRiemannTrack : public TObject{
public:

  // Constructors/Destructors ---------
  PndRiemannTrack();
  //PndRiemannTrack(const PndRiemannTrack& rtrack);
  ~PndRiemannTrack();


  // Accessors -----------------------
  const TVectorD& n() const {return fn;}
  double c() const {return fc;}
  const TVectorD& av() const {return fav;}

  TVectorD orig() const;
  double dX(){return TMath::Sqrt(fcovRXY[1][1]);}
  double dY(){return TMath::Sqrt(fcovRXY[2][2]);}
  double r() const;
  double dR();
  double dip();
  double dipangle();	//< dipangle theta
  double dDip();
  double Pt(double B); //< transvers momentum Pt calculated by the magnetic field B [tesla]
  double P(double B);
  double sign() const;
  double getSZm() const {return fm;}
  double getSZt() const {return ft;}
  unsigned int getNumHits() {return fHits.size();}
  PndRiemannHit* getHit(unsigned int i) {PndRiemannHit* myHit = &(fHits[i]); return myHit;}
  PndRiemannHit* getLastHit() {return getHit(getNumHits()-1);}
  std::vector<PndRiemannHit> getHits(){return fHits;};
  TVector3 getPforHit(int i, double B);

  double calcZPosByS(double s);
  int calcIntersection(PndRiemannTrack& track, TVector3& p1, TVector3& p2);

  double weight() const {return fweight;};
  TMatrixD covPlane() const {return fcovPlane;};
  TMatrixD jacRXY() const {return fjacRXY;};
  TMatrixD covRXY() const {return fcovRXY;};
  double m() const {return fm;}
  double mError() const {return fmError;}
  double t() const {return ft;}
  double tError() const {return ftError;}

  // Modifiers -----------------------
  void addHit(PndRiemannHit& hit);
  void init(double x0, double y0, double R,
	    double dip, double z0);

  // Operations ----------------------
  void refit();
  double dist(PndRiemannHit* hit);
  void szFit();
  double calcSZChi2(PndRiemannHit* hit); //calculates the chi2 of the track plus the additional hit
  double szDist(PndRiemannHit* hit);
  double szError(PndRiemannHit* hit);
  double szChi2(){return fChi2;};

  void SetVerbose(int i){fVerbose = i;}
  void SetVertexCut(double cut){fVertexCut = cut;}
  TVector3 calcErrorPosByS(Double_t s, Double_t dS);

  void PrintHits();

private:

  // Private Data Members ------------
  TVectorD fn;  		///< normal vector to plane;
  double fc;     		///< distance of plane to origin

  double fm; 			///< parameters of sz-fit
  double ft;
  double fmError;		///< Error of fit
  double ftError;		///< Error of fit
  double fChi2;			///< Chisquare of sz fit
  bool fFitDone;
  bool fSZFitDone;

  double fVertexCut;

  std::vector<PndRiemannHit> fHits;
  TVectorD fav;  		///< average over all hits
  double fweight; 		///< sum over all weights (1/(sigmaXY*sigmaXY))
  TMatrixD fcovPlane; 	///< full covarince matrix of the plane;
  TMatrixD fjacRXY;  	///< jacobian matrix to transform from c,n1,n2,n2 to r,x,y
  TMatrixD fcovRXY;
  int fVerbose;
  /////////////////
    bool ftrefit;
  ////////////////


  void calcJacRXY(); 	///< calcualtes fjacRXY

  TVector3 calcErrorLineNorm(PndRiemannTrack& track);
  TVector3 calcErrorLineOffset(PndRiemannTrack& track);
  TVectorD calcErrorXY1XY2(TVector3& line, TVector3& dLine, TVector3& offset, TVector3& dOffset);
  Double_t calcErrorS(TVector2& XY, TVector2& dXY, PndRiemannTrack* track);




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


public:
  ClassDef(PndRiemannTrack,2)

};

#endif

//--------------------------------------------------------------
// $Log$
//--------------------------------------------------------------