Track.h

/*
 * Track.h
 *
 *  Created on: 04-05-2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALTRACK_H_
#define HALTRACK_H_
 
#include "V0Track.h"
#include "XiTrack.h"
 
#include "Link.h"
 
#include <TLorentzVector.h>
 
#include <vector>
 
namespace Hal {
  class Event;
  class CompressionMap;
 
  class Track : public TObject {
    friend class Event;
 
  protected:
    Event* fEvent;  
    TLorentzVector fP;
    Int_t fID;
    Int_t fStatus;
    Int_t fHiddenInfo;
    Int_t fMotherID;
    Int_t fThisID;
    UInt_t fType;
    Float_t fCharge;
    enum ETrackBits {
      kPrimary     = 0,
      kV0          = 1,
      kXi          = 2,
      kGlobal      = 3,
      kMother      = 4,  // set if mother is known (and its secondary)
      kV0Daughters = 5,  // set if v0 and daughters are known
      kXiDaughters = 6   // set if xi and daughters are known
    };
    Float_t GetV0IfPossible(Int_t fieldId) const;
 
  public:
    Track();
    Track(const Track& track);
    Track& operator=(const Track& track) = default;
    inline void SetThisID(Int_t id) { fThisID = id; };
    inline Int_t GetThisID() const { return fThisID; };
    inline void SetIDs(Int_t id) { fThisID = fID = id; }
    inline Int_t GetHiddenInfoIndex() const { return fHiddenInfo; };
    inline Double_t GetPx() const { return fP.Px(); };
    inline Double_t GetPy() const { return fP.Py(); };
    inline Double_t GetPz() const { return fP.Pz(); };
    inline Double_t GetE() const { return fP.E(); };
    inline const TLorentzVector& GetMomentum() const { return fP; };
    virtual void Boost(Double_t vx, Double_t vy, Double_t vz);
    virtual void Print(Option_t* option = "") const;
    inline void SetCharge(Double_t charge) { fCharge = charge; };
    inline void SetID(Int_t id) { fID = id; };
    inline void SetStatus(Int_t status) { fStatus = status; };
    inline void SetMomentum(Double_t px, Double_t py, Double_t pz, Double_t e) { fP.SetPxPyPzE(px, py, pz, e); }
    void EnableV0(Bool_t v0, Bool_t daughters = kTRUE);
 
    void EnableXi(Bool_t xi, Bool_t daughters = kTRUE);
    inline void SetMotherIndex(Int_t index) {
      CLRBIT(fType, kPrimary);
      if (index >= 0) SETBIT(fType, kMother);
      fMotherID = index;
    };
    inline Double_t GetMass() const { return fP.M(); };
    inline Double_t GetCharge() const { return fCharge; };
    inline Int_t GetHiddenInfoId() const { return fHiddenInfo; };
    virtual Int_t GetID() const { return fID; };
    inline Int_t GetStatus() const { return fStatus; };
    inline Int_t GetMotherIndex() const { return fMotherID; };
    inline Bool_t IsPrimary() const { return TESTBIT(fType, kPrimary); };
    inline Bool_t IsGoodSecondary() const { return TESTBIT(fType, kMother); };
    inline Bool_t IsGoodV0() const { return TESTBIT(fType, kV0Daughters); };
    inline Bool_t IsGoodXi() const { return TESTBIT(fType, kXiDaughters); }
    inline Bool_t IsGlobal() const { return TESTBIT(fType, kGlobal); };
    V0Track* GetV0Info() const;
    XiTrack* GetXiInfo() const;
    inline void SetGlobal(Bool_t global = kTRUE) {
      if (global)
        SETBIT(fType, kGlobal);
      else
        CLRBIT(fType, kGlobal);
    };
    inline void ClearType() { fType = 0; };
    void SetMass(Double_t mass);
    void SetPrimary();
    void SetSecondary(Bool_t parent);
    void TranslateLinks(const CompressionMap& map);
    inline Bool_t IsV0() const { return TESTBIT(fType, kV0); };
    inline Bool_t IsXi() const { return TESTBIT(fType, kXi); };
    virtual void CopyData(Track* other);
    virtual void CopyAllData(Track* other);
    virtual void Clear(Option_t* opt = "");
    virtual void ResetTrack(Int_t thisID = -1, Event* event = nullptr);
    virtual void RotateZ(Double_t angle);
    Event* GetEvent() const { return fEvent; };
    std::vector<int> GetLinks() const;
    virtual Int_t GetLinksFast(std::vector<int>& vec, Bool_t fast = kTRUE) const;
    virtual void SetLinks(std::vector<int>& vec);
    void SetEvent(Event* event) { fEvent = event; };
    virtual Float_t GetFieldVal(Int_t fieldID) const;
    virtual TString GetFieldName(Int_t fieldID) const;
    virtual ~Track();
    ClassDef(Track, 1)
  };
}  // namespace Hal
 
#endif /* HALTRACK_H_ */