#ifndef __FdRecGeometry_h__
#define __FdRecGeometry_h__

#include <FdGeometry.h>

#include <FdRecLevel.h>


class FdRecGeometry : public FdGeometry {

public:
  FdRecGeometry();

  // -------- low level geometry variables ---------

  /// get error of theta of SDP fit [rad]
  Double_t GetSDPThetaError() const { return fSDPThetaError; }
  /// get error of phi of SDP fit [rad]
  Double_t GetSDPPhiError() const { return fSDPPhiError; }
  /// get the correlation coefficient between Theta and Phi
  Double_t GetSDPThetaPhiCorrelation() const { return fRhoThetaPhi; }
  /// get chi2 of SDP fit [rad]
  Double_t GetSDPChi2() const { return fSDPChi2; }
  /// get NdF of SDP fit [rad]
  UInt_t GetSDPNdF() const { return fSDPNdF; }

  /// get error of T0  of time fit [ns]
  Double_t GetT0Error() const { return fT0Error; }
  /// get error of Rp of time fit [m]
  Double_t GetRpError() const { return fRpError; }
  /// get error of Chi0 of time fit [rad]
  Double_t GetChi0Error() const { return fChi0Error; }
  /// get the correlation coefficient between Chi0 and T0
  Double_t GetChi0T0Correlation() const { return fRhoChi0T0; }
  /// get the correlation coefficient between Chi0 and Rp
  Double_t GetChi0RpCorrelation() const { return fRhoChi0Rp; }
  /// get the correlation coefficient between  Rp and T0
  Double_t GetRpT0Correlation() const { return fRhoRpT0; }
  /// get chi2 of hybrid time fit
  Double_t GetTimeFitChi2() const { return fTimeFitChi2; }
  /// get chi2 of FD only time fit
  Double_t GetTimeFitFDChi2() const { return fTimeFitFDChi2; }
  /// get chi2 of linear fit to FD times
  Double_t GetLinearTimeFitChi2() const { return fLinearTimeFitChi2; }
  /// get NdF of time fit
  UInt_t GetTimeFitNdof() const { return fTimeFitNdF; }

  /// get the correlation coefficient between northing and easting
  Double_t GetNorthEastCorrelation() const { return fNorthEastCorrelation; }
  /// get the correlation coefficient between northing and theta
  Double_t GetNorthThetaCorrelation() const { return fNorthThetaCorrelation; }
  /// get the correlation coefficient between northing and phi
  Double_t GetNorthPhiCorrelation() const { return fNorthPhiCorrelation; }
  /// get the correlation coefficient between northing and TCore
  Double_t GetNorthTCoreCorrelation() const { return fNorthTCoreCorrelation; }
  /// get the correlation coefficient between easting and theta
  Double_t GetEastThetaCorrelation() const { return fEastThetaCorrelation; }
  /// get the correlation coefficient between easting and phi
  Double_t GetEastPhiCorrelation() const { return fEastPhiCorrelation; }
  /// get the correlation coefficient between easting and TCore
  Double_t GetEastTCoreCorrelation() const { return fEastTCoreCorrelation; }
  /// get the correlation coefficient between theta and phi
  Double_t GetThetaPhiCorrelation() const { return fThetaPhiCorrelation; }
  /// get the correlation coefficient between theta and TCore
  Double_t GetThetaTCoreCorrelation() const { return fThetaTCoreCorrelation; }
  /// get the correlation coefficient between phi and TCore
  Double_t GetPhiTCoreCorrelation() const { return fPhiTCoreCorrelation; }
  /// get chi2 of axis/core fit
  Double_t GetAxisFitChi2() const { return fAxisFitChi2; }
  /// get Ndf of axis/core fit
  UInt_t GetAxisFitNdof() const { return fAxisFitNdF; }

  /// get the number of pixels in the time fit
  UInt_t GetNTimeFitPixels() const { return fNTimeFitPixels; }

  /// get the rec level of the geometry reconstruction
  EFdGeomRecLevel GetGeomRecLevel() const { return fFdGeomRecLevel; }

  /// calculate uncertainty of distance from eye to core
  Double_t GetCoreEyeDistanceError() const;

  // -------- SD related variables ---------

  /// get number of stations used in hybrid fit
  UInt_t GetNHybridStations() const { return fNTank; }
  /// get the hottest station of hybrid fit
  UInt_t GetHottestStation() const { return fHottestTank; }
  /// get the distance of the hottest station to the axis in SP coords. [m]
  Double_t GetStationAxisDistance() const { return fNTank > 0 ? fAxisDist : 1e99; }
  /// get the FD/SD time residual after fit [ns]
  Double_t GetSDFDTimeOffset() const { return fNTank > 0 ? fSDFDdT : 1e99; }
  /// get the distance of the hottest station to the SDP [m]
  Double_t GetStationSDPDistance() const { return fNTank > 0 ? fSDPDist : 1e99; }

  /// set error of theta of SDP fit [rad]
  void SetSDPThetaError(const Double_t arg) { fSDPThetaError = arg; }
  /// set error of phi of SDP fit [rad]
  void SetSDPPhiError(const Double_t arg) { fSDPPhiError = arg; }
  /// set the correlation coefficient between Theta and Phi
  void SetSDPThetaPhiCorrelation(const Double_t arg) { fRhoThetaPhi = arg; }
  /// set chi2 of SDP fit [rad]
  void SetSDPChi2(const Double_t arg) { fSDPChi2 = arg; }
  /// set NdF of SDP fit [rad]
  void SetSDPNdF(const UInt_t arg) { fSDPNdF = arg; }

  /// set error of T0  of time fit [ns]
  void SetT0Error(const Double_t arg) { fT0Error = arg; }
  /// set error of Rp of time fit [m]
  void SetRpError(const Double_t arg) { fRpError = arg; }
  /// set error of Chi0 of time fit [rad]
  void SetChi0Error(const Double_t arg) { fChi0Error = arg; }
  /// set the correlation coefficient between Chi0 and T0
  void SetChi0T0Correlation(const Double_t arg) { fRhoChi0T0 = arg; }
  /// set the correlation coefficient between Chi0 and Rp
  void SetChi0RpCorrelation(const Double_t arg) { fRhoChi0Rp = arg; }
  /// set the correlation coefficient between  Rp and T0
  void SetRpT0Correlation(const Double_t arg) { fRhoRpT0 = arg; }
  /// set chi2 of hybrid time fit
  void SetTimeFitChi2(const Double_t arg) { fTimeFitChi2 = arg; }
  /// set chi2 of FD only time fit
  void SetTimeFitFDChi2(const Double_t arg) { fTimeFitFDChi2 = arg; }
  /// get chi2 of linear fit to FD times
  void SetLinearTimeFitChi2(const Double_t chi2) { fLinearTimeFitChi2 = chi2; }
  /// set NdF of time fit
  void SetTimeFitNdof(const UInt_t arg) { fTimeFitNdF = arg; }

  /// set the correlation coefficient between northing and easting
  void SetNorthEastCorrelation(const Double_t arg) { fNorthEastCorrelation = arg; }
  /// set the correlation coefficient between northing and theta
  void SetNorthThetaCorrelation(const Double_t arg) { fNorthThetaCorrelation = arg; }
  /// set the correlation coefficient between northing and phi
  void SetNorthPhiCorrelation(const Double_t arg) { fNorthPhiCorrelation = arg; }
  /// set the correlation coefficient between northing and TCore
  void SetNorthTCoreCorrelation(const Double_t arg) { fNorthTCoreCorrelation = arg; }
  /// set the correlation coefficient between easting and theta
  void SetEastThetaCorrelation(const Double_t arg) { fEastThetaCorrelation = arg; }
  /// set the correlation coefficient between easting and phi
  void SetEastPhiCorrelation(const Double_t arg) { fEastPhiCorrelation = arg; }
  /// set the correlation coefficient between easting and TCore
  void SetEastTCoreCorrelation(const Double_t arg) { fEastTCoreCorrelation = arg; }
  /// set the correlation coefficient between theta and phi
  void SetThetaPhiCorrelation(const Double_t arg) { fThetaPhiCorrelation = arg; }
  /// set the correlation coefficient between theta and TCore
  void SetThetaTCoreCorrelation(const Double_t arg) { fThetaTCoreCorrelation = arg; }
  /// set the correlation coefficient between phi and TCore
  void SetPhiTCoreCorrelation(const Double_t arg) { fPhiTCoreCorrelation = arg; }
  /// set chi2 of axis/core fit
  void SetAxisFitChi2(const Double_t arg) { fAxisFitChi2 = arg; }
  /// set NdF of axis/core fit
  void SetAxisFitNdof(const UInt_t arg) { fAxisFitNdF = arg; }

  /// set the rec level of the geometry reconstruction
  void SetGeomRecLevel(const EFdGeomRecLevel arg) { fFdGeomRecLevel = arg; }

  /// set the number of pixels in the time fit
  void SetNTimeFitPixels(const UInt_t arg) { fNTimeFitPixels = arg; }

  // -------- SD related variables ---------

  /// set number of stations used in hybrid fit
  void SetNHybridStations(const UInt_t arg) { fNTank = arg; }
  /// set the hottest station of hybrid fit
  void SetHottestStation(const UInt_t arg) { fHottestTank = arg; }
  /// set the distance of the hottest station to the axis in SP coords. [m]
  void SetStationAxisDistance(const Double_t arg) { fAxisDist = arg; }
  /// set the FD/SD time residual after fit [ns]
  void SetSDFDTimeOffset(const Double_t arg) { fSDFDdT = arg; }
  /// set the distance of the hottest station to the SDP [m]
  void SetStationSDPDistance(const Double_t arg) { fSDPDist = arg; }

private:
  // SDP
  Double_t fSDPThetaError;
  Double_t fSDPPhiError;
  Double_t fRhoThetaPhi;
  Double_t fSDPChi2;
  UInt_t fSDPNdF;

  // axis fit
  Double_t fT0Error;
  Double_t fChi0Error;
  Double_t fRpError;
  Double_t fRhoChi0T0;
  Double_t fRhoChi0Rp;
  Double_t fRhoRpT0;
  Double_t fTimeFitChi2;
  Double_t fTimeFitFDChi2;
  Double_t fLinearTimeFitChi2;
  UInt_t fTimeFitNdF;

  // axis/core fit
  Double_t fNorthEastCorrelation;
  Double_t fNorthThetaCorrelation;
  Double_t fNorthPhiCorrelation;
  Double_t fNorthTCoreCorrelation;
  Double_t fEastThetaCorrelation;
  Double_t fEastPhiCorrelation;
  Double_t fEastTCoreCorrelation;
  Double_t fThetaPhiCorrelation;
  Double_t fThetaTCoreCorrelation;
  Double_t fPhiTCoreCorrelation;
  Double_t fAxisFitChi2;
  UInt_t fAxisFitNdF;

  EFdGeomRecLevel fFdGeomRecLevel;

  UInt_t fNTimeFitPixels;

  // SD related variables
  UInt_t fNTank;
  UInt_t fHottestTank;
  Double_t fAxisDist;
  Double_t fSDFDdT;
  Double_t fSDPDist;

  ClassDef(FdRecGeometry, 4);

};


#endif