StPicoTrackCovMatrix.h

#ifndef StPicoTrackCovMatrix_h
#define StPicoTrackCovMatrix_h

// ROOT headers
#include "TObject.h"

#if defined (__TFG__VERSION__)
#include "StEvent/StDcaGeometry.h"
#endif /* __TFG__VERSION__ */

//_________________
class StPicoTrackCovMatrix : public TObject {
  
 public:
  
  StPicoTrackCovMatrix();
  StPicoTrackCovMatrix(const StPicoTrackCovMatrix &matrix);
  virtual ~StPicoTrackCovMatrix();
  virtual void Print(Char_t const* option = "") const;

  //
  // Getters
  //

  Float_t* params()                   { return &mImp; }
  const Float_t* params() const       { return &mImp; }
  const Float_t* sigmas() const       { return mSigma; }
  const Float_t* correlations() const { return mCorr; }
  Float_t imp() const                 { return mImp; }
  Float_t z() const                   { return mZ; }
  Float_t psi() const                 { return mPsi; }
  Float_t pti() const                 { return mPti; }
  Float_t tan() const                 { return mTan; }
  Float_t curv() const                { return mCurv; }

  Bool_t isBadCovMatrix();

#if defined (__TFG__VERSION__)
  StDcaGeometry &dcaGeometry() const;
#endif /* __TFG__VERSION__ */

  //
  // Setters
  //

  void setParams(Float_t params[6])
  { mImp = params[0]; mZ = params[1]; mPsi = params[2]; mPti = params[3]; mTan = params[4]; mCurv = params[5]; }
  void setSigmas(Float_t sigmas[5]) { for(Int_t iIter=0; iIter<5; iIter++) { mSigma[iIter] = sigmas[iIter]; } }
  void setCorrelations(Float_t corr[10]) { for(Int_t iIter=0; iIter<10; iIter++) { mCorr[iIter] = corr[iIter]; } }

  void setImp(Float_t imp)       { mImp = (Float16_t)imp; }
  void setZ(Float_t z)           { mZ = (Float16_t)z; }
  void setPsi(Float_t psi)       { mPsi = (Float16_t)psi; }
  void setPti(Float_t pti)       { mPti = (Float16_t)pti; }
  void setTan(Float_t tan)       { mTan = (Float16_t)tan; }
  void setCurv(Float_t curv)     { mCurv = (Float16_t)curv; }
  
 private:

    /*                                                          j    0     1     2     3     4
    Float_t  mImpImp;                                       i 0  0(0) 
    Float_t  mZImp,   mZZ;                                    1  1(0)  2(1)
    Float_t  mPsiImp, mPsiZ, mPsiPsi;                         2  3(1)  4(2)  5(2)
    Float_t  mPtiImp, mPtiZ, mPtiPsi, mPtiPti;                3  6(3)  7(4)  8(5)  9(3)
    Float_t  mTanImp, mTanZ, mTanPsi, mTanPti, mTanTan;       4 10(6) 11(7) 12(8) 13(9) 14(4)
   */

  Float16_t mImp;
  Float16_t mZ;
  Float16_t mPsi;        //[-pi,pi,20]
  Float16_t mPti;
  Float16_t mTan;        //[-10,10,20]
  Float16_t mCurv;
  Float16_t mSigma[5];
  Float16_t mCorr[10];   //[-1,1,20] 

  ClassDef(StPicoTrackCovMatrix, 1)
};

#endif