BmnHypNuclPair.h

Go to the documentation of this file.

#include "TLorentzVector.h"
 
#include <TMath.h>
#include <TNamed.h>
#include <vector>
 
using namespace std;
using namespace TMath;
 
#ifndef BMNHYPNUCLPAIR_H
#define BMNHYPNUCLPAIR_H 1
 
enum class DetectorType : Int_t
{
    kAll = 0,      // FSD + GEM (сумма)
    kGEM = 1,      // только GEM
    kSilicon = 2   // только FSD (Silicon)
};
class BmnHypNuclPair : public TNamed
{
  public:
    /** Default constructor **/
    BmnHypNuclPair();
 
    /** Destructor **/
    virtual ~BmnHypNuclPair();
 
    void SetChi2DCA0(Float_t v) { fChi2DCA0 = v; }
    void SetChi2DCA1(Float_t v) { fChi2DCA1 = v; }
    void SetChi2DCA2(Float_t v) { fChi2DCA2 = v; }
    void SetChi2DCA12(Float_t v) { fChi2DCA12 = v; }
    Float_t GetChi2DCA0() const { return fChi2DCA0; }
    Float_t GetChi2DCA1() const { return fChi2DCA1; }
    Float_t GetChi2DCA2() const { return fChi2DCA2; }
    Float_t GetChi2DCA12() const { return fChi2DCA12; }
 
    void SetPath(Float_t val) { fPath = val; }
 
    void SetInvMass(Float_t val) { fInvMass = val; }
 
    void SetMomXYZ1(Float_t px, Float_t py, Float_t pz)
    {
        fPx1 = px;
        fPy1 = py;
        fPz1 = pz;
    }
 
    void SetMomXYZ2(Float_t px, Float_t py, Float_t pz)
    {
        fPx2 = px;
        fPy2 = py;
        fPz2 = pz;
    }
 
    void SetChi2NDF(Float_t val1, Float_t val2)
    {
        fChi2Ndf1 = val1;
        fChi2Ndf2 = val2;
    }
 
    void SetM2(Float_t val) { fM2 = val; }
 
    void SetNHitsPair(Short_t nFsd1, Short_t nGem1, Short_t nFsd2, Short_t nGem2)
    {
        fNHitsSil1 = nFsd1;
        fNHitsGem1 = nGem1;
        fNHitsSil2 = nFsd2;
        fNHitsGem2 = nGem2;
    }
 
    void SetNTrInEvent(Int_t n) { fNTrInEvent = n; }
 
    Int_t GetNHitsPart1(DetectorType det = DetectorType::kAll) const
    {
        switch (det) {
            case DetectorType::kAll:
                return fNHitsSil1 + fNHitsGem1;
            case DetectorType::kGEM:
                return fNHitsGem1;
            case DetectorType::kSilicon:
                return fNHitsSil1;
        }
        return 0;
    }
 
    Int_t GetNHitsPart2(DetectorType det = DetectorType::kAll) const
    {
        switch (det) {
            case DetectorType::kAll:
                return fNHitsSil2 + fNHitsGem2;
            case DetectorType::kGEM:
                return fNHitsGem2;
            case DetectorType::kSilicon:
                return fNHitsSil2;
        }
        return 0;
    }
 
    Float_t GetInvMass() { return fInvMass; }
 
    Float_t GetPath() { return fPath; }
 
    Float_t GetMom1() { return Sqrt(fPx1 * fPx1 + fPy1 * fPy1 + fPz1 * fPz1); }
 
    Float_t GetPx1() { return fPx1; }
 
    Float_t GetPy1() { return fPy1; }
 
    Float_t GetPz1() { return fPz1; }
 
    Float_t GetPt1() { return Sqrt(fPx1 * fPx1 + fPy1 * fPy1); }
 
    Float_t GetMom2() { return Sqrt(fPx2 * fPx2 + fPy2 * fPy2 + fPz2 * fPz2); }
 
    Float_t GetPx2() { return fPx2; }
 
    Float_t GetPy2() { return fPy2; }
 
    Float_t GetPz2() { return fPz2; }
 
    Float_t GetPt2() { return Sqrt(fPx2 * fPx2 + fPy2 * fPy2); }
 
    Float_t GetTxPart1() { return fPx1 / fPz1; }
 
    Float_t GetTxPart2() { return fPx2 / fPz2; }
 
    Float_t GetTyPart1() { return fPy1 / fPz1; }
 
    Float_t GetTyPart2() { return fPy2 / fPz2; }
 
    Float_t GetChi2Ndf1() { return fChi2Ndf1; }
 
    Float_t GetChi2Ndf2() { return fChi2Ndf2; }
 
    void SetDedx(Float_t val) { fdEdx = val; }
 
    Float_t GetDedx() { return fdEdx; }
 
    Float_t GetM2() { return fM2; }
 
    Int_t GetNTrInEvent() { return fNTrInEvent; }
 
  private:
    Float_t fInvMass;
 
    Float_t fChi2DCA0;
    Float_t fChi2DCA1;
    Float_t fChi2DCA2;
    Float_t fChi2DCA12;
 
    Float_t fPath;
 
    Short_t fNHitsSil1;
    Short_t fNHitsGem1;
    Short_t fNHitsSil2;
    Short_t fNHitsGem2;
 
    Float_t fPx1;
    Float_t fPy1;
    Float_t fPz1;
    Float_t fPx2;
    Float_t fPy2;
    Float_t fPz2;
 
    Float_t fChi2Ndf1;
    Float_t fChi2Ndf2;
 
    Float_t fdEdx;
    Float_t fM2;
 
    Int_t fNTrInEvent;
 
    ClassDef(BmnHypNuclPair, 4);
};
 
#endif