Decay.h

/*
 * Decay.h
 *
 *  Created on: 21 gru 2023
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
#ifndef HAL_FEATURES_PHYS_DECAY_H_
#define HAL_FEATURES_PHYS_DECAY_H_
 
#include <TObject.h>
#include <vector>
 
 
namespace Hal {
  class McTrack;
  class McEvent;
  class DecayChannel : public TObject {
    const Int_t fDaughters;
    std::vector<Int_t> fPdgCodes;
    std::vector<Double_t> fMass;
    Double_t fMassThres = {-1};
    Double_t fBranchRatio;
 
  public:
    DecayChannel(Int_t dau1 = 0, Int_t dau2 = 0, Double_t ratio = 0.0);
    DecayChannel(Int_t dau1, Int_t dau2, Int_t dau3, Double_t ratio);
    Int_t GetDaughterPdg(Int_t dau) const { return fPdgCodes[dau]; }
    Int_t GetDaughtersNo() const { return fPdgCodes.size(); }
    Double_t GetDaughterMass(Int_t dau) const { return fMass[dau]; }
    Double_t GetBranchingRatio() const { return fBranchRatio; }
    Double_t GetMassThreshold() const { return fMassThres; }
    void SetBranchRatio(Double_t x) { fBranchRatio = x; }
    virtual Bool_t Init();
    virtual ~DecayChannel() {};
    ClassDef(DecayChannel, 1)
  };
  class Decay : public TObject {
    std::vector<DecayChannel> fDecayChannels;
    Double_t fGamma       = {0};
    Bool_t fBreightWigher = {kFALSE};
    Double_t fMotherPdg   = {-1};
    Double_t fMotherMass  = {-1};
 
  protected:
    virtual Double_t GetDecayTime(McTrack& mother, Double_t mass) const;
 
    virtual void Decay2Body(McTrack& mother, std::vector<McTrack*>& daughters, const DecayChannel& channel) const;
    virtual void Decay3Body(McTrack& mother, std::vector<McTrack*>& daughters, const DecayChannel& channel) const;
 
  public:
    Decay(Int_t motherPdg = -1);
    Int_t GetMotherPdg() const { return fMotherPdg; }
    virtual Int_t DecayParticle(McTrack& mother, std::vector<McTrack*>& daughters, Bool_t addToEvent = kFALSE) const;
    void EnableBreightWigner(Bool_t bw = kTRUE) { fBreightWigher = bw; }
    void SetGamma(Double_t gamma) { fGamma = gamma; };
    void AddDecayChannel(const DecayChannel& decay) { fDecayChannels.push_back(decay); };
    virtual Bool_t Init();
    virtual ~Decay() {};
    ClassDef(Decay, 1)
  };
}  // namespace Hal
 
#endif /* HAL_FEATURES_PHYS_DECAY_H_ */