V0Track.h

/*
 * V0Track.h
 *
 *  Created on: 04-05-2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HALV0TRACK_H_
#define HALV0TRACK_H_
 
#include <TLorentzVector.h>
#include <TObject.h>
#include <TVector3.h>
 
#include "Const.h"
#include "HiddenInfo.h"
 
namespace Hal {
  class Track;
  class V0Track : public HiddenInfo {
    Int_t fTrackId;
    Int_t fPosId;
    Int_t fNegId;
    Int_t fAssumedPdg;
    Int_t fAssumedPdgPos;
    Int_t fAssumedPdgNeg;
    Double_t fS1, fS2;
    Double_t fAlpha;
    Double_t fPtArm;
    Double_t fDauDist;
    Double_t fDecLenght;
    Double_t fCosAngle;
    TVector3 fDecay;
    TVector3 fMomPos;
    TVector3 fMomNeg;
 
  public:
    V0Track();
    V0Track(const V0Track& other) = default;
    virtual void CopyData(V0Track* v);
    void SetTrackId(Int_t id) { fTrackId = id; }
    inline Int_t GetTrackId() const { return fTrackId; }
    inline Int_t GetPdg() const { return fAssumedPdg; };
    inline Int_t GetPdgPosDau() const { return fAssumedPdgPos; };
    inline Int_t GetPdgNegDau() const { return fAssumedPdgNeg; };
    Int_t GetNegId() const { return fNegId; }
    Int_t GetPosId() const { return fPosId; }
    Double_t GetAlphaArm() const { return fAlpha; }
    Double_t GetPtArm() const { return fPtArm; }
    Double_t GetDauDist() const { return fDauDist; }
    Double_t GetDecLength() const { return fDecLenght; };
    Double_t GetCosAngle() const { return fCosAngle; }
    void SetS(Double_t s1, Double_t s2) {
      fS1 = s1;
      fS2 = s2;
    };
    void SetPdgDaughters(Int_t pos, Int_t neg) {
      fAssumedPdgNeg = neg;
      fAssumedPdgPos = pos;
    }
    void SetPdg(Int_t pid) { fAssumedPdg = pid; };
    void SetNegId(Int_t negId) { fNegId = negId; }
    void SetPosId(Int_t posId) { fPosId = posId; }
    void SetDecLenght(Double_t L) { fDecLenght = L; };
    Double_t GetLambdaMass() const { return GetHypoMass(Const::ProtonMass(), Const::PionPlusMass()); }
    Double_t GetAntiLambdaMass() const { return GetHypoMass(Const::PionPlusMass(), Const::ProtonMass()); }
    Double_t GetK0Mass() const { return GetHypoMass(Const::PionPlusMass(), Const::PionPlusMass()); }
    Double_t GetHypoMass(Double_t m_dau1, Double_t m_dau2) const;
    std::pair<Double_t, Double_t> GetS() const { return std::pair<Double_t, Double_t>(fS1, fS2); }
    const TVector3& GetMomPos() const { return fMomPos; };
    const TVector3& GetMomNeg() const { return fMomNeg; }
    const TVector3& GetDecay() const { return fDecay; }
    void SetMomentumPos(const TVector3& positiveDaughter) { fMomPos = positiveDaughter; }
    void SetMomPos(Double_t px, Double_t py, Double_t pz);
    void SetMomentumNeg(const TVector3& negativeDaughter) { fMomNeg = negativeDaughter; }
    void SetMomNeg(Double_t px, Double_t py, Double_t pz);
    void SetDecayPos(Double_t x, Double_t y, Double_t z);
    void SetDecayPos(const TVector3& pos) { fDecay = pos; }
    void SetAlphaArm(Double_t alpha) { fAlpha = alpha; }
    void SetPtArm(Double_t ptArm) { fPtArm = ptArm; }
    void SetDauDist(Double_t dauDist) { fDauDist = dauDist; }
    virtual void RotateZ(Double_t angle);
    TVector3 Recalc(const Track& track);
    virtual ~V0Track();
    ClassDef(V0Track, 1)
  };
}  // namespace Hal
#endif /* HALV0TRACK_H_ */