MpdRoot_KFParticle/physics/femto/MpdFemtoMaker/MpdFemtoKink.h

/**
 * \class MpdFemtoKink
 * \brief Main class holding kink information
 *
 * MpdFemtoKink holds all the necessary information about a kink
 *
 * \author Grigory Nigmatkulov (NRNU MEPhI)
 * \date May 18, 2019
 * \email nigmatkulov@gmail.com
 */

#ifndef MpdFemtoKink_h
#define MpdFemtoKink_h

// MpdFemtoMaker headers
#include "MpdFemtoTrack.h"
#include "MpdFemtoTypes.h"

// ROOT headers
#include "TVector3.h"

//_________________
class MpdFemtoKink {
 public:
  /// Default constructor
  MpdFemtoKink();
  /// Copy constructor
  MpdFemtoKink(const MpdFemtoKink&);
  /// Copy constructor
  MpdFemtoKink& operator=(const MpdFemtoKink &k);
  /// Default destructor
  virtual ~MpdFemtoKink();

  //
  // Getters
  //

  float dcaParentDaughter() const {
    return mDcaParentDaughter;
  }

  float dcaDaughterPrimaryVertex() const {
    return mDaughter.gDCA().Mag();
  }

  float dcaParent2PrimaryVertex() const {
    return mParent.gDCA().Mag();
  }

  float hitDistanceParentDaughter() const {
    return mHitDistanceParentDaughter;
  }

  float hitDistanceParentVertex() const {
    return mHitDistanceParentVertex;
  }

  float deltaEnergy(int i = 0) const {
    return mDeltaEnergy[i];
  }

  float decayAngle() const {
    return mDecayAngle;
  }

  float decayAngleCM() const {
    return mDecayAngleCM;
  }

  MpdFemtoTrack daughter() const {
    return mDaughter;
  }

  MpdFemtoTrack parent() const {
    return mParent;
  }

  TVector3 position() const {
    return TVector3(mPositionX, mPositionY, mPositionZ);
  }

  TVector3 kinkPoint() const {
    return position();
  }

  TVector3 decayPoint() const {
    return position();
  }

  TVector3 primaryVertex() const {
    return mParent.primaryVertex();
  }

  //
  // Setters
  //

  void setDcaParentDaughter(const float& dca) {
    mDcaParentDaughter = dca;
  }

  void setHitDistanceParentDaugher(const float& dist) {
    mHitDistanceParentDaughter = dist;
  }

  void setHitDistanceParentVertex(const float& dist) {
    mHitDistanceParentVertex = dist;
  }

  void setDeltaEnergy(const int& i, const float& e) {
    mDeltaEnergy[i] = e;
  }

  void setDecayAngle(const float& angle) {
    mDecayAngle = angle;
  }

  void setDecayAngleCM(const float& angle) {
    mDecayAngleCM = angle;
  }
  void setDaugherHbtTrack(const MpdFemtoTrack& trk);
  void setParetntHbtTrack(const MpdFemtoTrack& trk);

  void setKinkPoint(const TVector3& pos) {
    mPositionX = pos.X();
    mPositionY = pos.Y();
    mPositionZ = pos.Z();
  }

  void setKinkPoint(const float& x, const float& y, const float& z) {
    mPositionX = x;
    mPositionY = y;
    mPositionZ = z;
  }

  void setPoint(const TVector3& pos) {
    setKinkPoint(pos);
  }

  void setPoint(const float& x, const float& y, const float& z) {
    setKinkPoint(x, y, z);
  }

 protected:

  /// DCA between parent and daugher
  float mDcaParentDaughter;
  /// Distance between parent track and daughter track
  float mHitDistanceParentDaughter;
  /// Distance between parent track and primary vertex
  float mHitDistanceParentVertex;
  /// Missed energy
  float mDeltaEnergy[3];
  /// Decay angles in the Lab frame
  float mDecayAngle;
  /// Decay angles in the CMS frame
  float mDecayAngleCM;
  /// Daughter track
  MpdFemtoTrack mDaughter;
  /// Parent track
  MpdFemtoTrack mParent;
  /// Decay point x of the kink
  float mPositionX;
  /// Decay point y of the kink
  float mPositionY;
  /// Decay point z of the kink
  float mPositionZ;

  ClassDef(MpdFemtoKink, 1)
};

#endif // #define MpdFemtoKink_h