StPicoBTofPidTraits.h

#ifndef StPicoBTofPidTraits_h
#define StPicoBTofPidTraits_h

// C++ headers
#include <limits>

// ROOT headers
#include "TObject.h"
#include "TVector3.h"

//_________________
class StPicoBTofPidTraits : public TObject {

 public:
  StPicoBTofPidTraits();
  StPicoBTofPidTraits(const StPicoBTofPidTraits &traits);
  virtual ~StPicoBTofPidTraits();
  virtual void Print(const Char_t* option = "") const;

  //
  // Getters
  //

  Int_t   trackIndex() const     { return mTrackIndex; }
  Int_t   btofCellId() const     { return (Int_t)mBTofCellId; }
  Int_t   btofMatchFlag() const  { return (Int_t)mBTofMatchFlag; }

  Float_t btof() const           { return mBTof; }
  Float_t btofBeta() const       { return (Float_t)mBTofBeta / 20000.f; }
  Float_t btofYLocal() const     { return (Float_t)mBTofYLocal / 1000.; }
  Float_t btofZLocal() const     { return (Float_t)mBTofZLocal / 1000.; }

  TVector3 btofHitPos() const    { return TVector3( btofHitPosX() , btofHitPosY() , btofHitPosZ() ); }

  Float_t btofHitPosX() const    { return (Float_t)mBTofHitPosX / 100.; }
  Float_t btofHitPosY() const    { return (Float_t)mBTofHitPosY / 100.; }
  Float_t btofHitPosZ() const    { return (Float_t)mBTofHitPosZ / 100.; }

  Float_t nSigmaElectron() const { return (Float_t)mNSigmaElectron / 1000.; }
  Float_t nSigmaPion() const     { return (Float_t)mNSigmaPion / 1000.; }
  Float_t nSigmaKaon() const     { return (Float_t)mNSigmaKaon / 1000.; }
  Float_t nSigmaProton() const   { return (Float_t)mNSigmaProton / 1000.; }

  //
  // Setters
  //

  void setTrackIndex(Int_t idx2PicoTrack)
  { mTrackIndex = (idx2PicoTrack > std::numeric_limits<short>::max()) ? -1 : (Short_t)idx2PicoTrack; }
  void setBTofCellId(Int_t tray, Int_t module, Int_t cell)
  { mBTofCellId  = (Short_t)((tray - 1) * 192 + (module - 1) * 6 + (cell - 1)); }
  void setBTofMatchFlag(UChar_t flag)  { mBTofMatchFlag = flag; }
  void setTOF(Float_t tof)             { mBTof = tof; }
  void setBeta(Float_t beta);
  void setHitPositionXYZ(Float_t x, Float_t y, Float_t z);
  void setHitPositionX(Float_t x);
  void setHitPositionY(Float_t y);
  void setHitPositionZ(Float_t z);
  void setYLocal(Float_t yLocal);
  void setZLocal(Float_t zLocal);

  void setNSigmaElectron(Float_t sigma);
  void setNSigmaPion(Float_t sigma);
  void setNSigmaKaon(Float_t sigma);
  void setNSigmaProton(Float_t sigma);
  
 private:

  Short_t  mTrackIndex;
  Short_t   mBTofCellId;
  UChar_t   mBTofMatchFlag;
  Float_t mBTof;
  UShort_t  mBTofBeta;
  Short_t   mBTofYLocal;
  Short_t   mBTofZLocal;
  Short_t   mBTofHitPosX;
  Short_t   mBTofHitPosY;
  Short_t   mBTofHitPosZ;

  Short_t mNSigmaElectron;
  Short_t mNSigmaPion;
  Short_t mNSigmaKaon;
  Short_t mNSigmaProton;

  ClassDef(StPicoBTofPidTraits, 4);
};

#endif