BmnParticlePair.h

Go to the documentation of this file.

// @(#)bmnroot/physics/particles:$Id$
// Author: Pavel Batyuk <pavel.batyuk@jinr.ru> 2017-06-20
 
////////////////////////////////////////////////////////////////////////////////
//                                                                            //
// BmnParticlePair                                                            //
//                                                                            //
//  A class to consider selected pair of particles                            //
//                                                                            //
////////////////////////////////////////////////////////////////////////////////
 
#include <TNamed.h>
#include <TVector3.h>
#include <vector>
 
using namespace std;
 
#ifndef BMNPARTICLEPAIR_H
#define BMNPARTICLEPAIR_H 1
 
class BmnParticlePair : public TNamed
{
  public:
    /** Default constructor **/
    BmnParticlePair();
 
    /** Destructor **/
    virtual ~BmnParticlePair();
 
    void SetV0X(Double_t val) { fV0X = val; }
 
    void SetV0Y(Double_t val) { fV0Y = val; }
 
    void SetV0Z(Double_t val) { fV0Z = val; }
 
    void SetDCA0(Double_t val) { fDCA0 = val; }
 
    void SetDCA1(Double_t val) { fDCA1 = val; }
 
    void SetDCA2(Double_t val) { fDCA2 = val; }
 
    void SetDCA12(Double_t val) { fDCA12 = val; }
 
    void SetPath(Double_t val) { fPath = val; }
 
    void SetInvMass(Double_t val) { fInvMass = val; }
 
    void SetAngleDecayProducts(Double_t val) { fAngleDecayProducts = val; }
 
    void SetMomPair(Double_t val1, Double_t val2)
    {
        fMomPart1 = val1;
        fMomPart2 = val2;
    }
 
    void SetEtaPair(Double_t val1, Double_t val2)
    {
        fEtaPart1 = val1;
        fEtaPart2 = val2;
    }
 
    void SetTxPair(Double_t val1, Double_t val2)
    {
        fTxPart1 = val1;
        fTxPart2 = val2;
    }
 
    void SetTyPair(Double_t val1, Double_t val2)
    {
        fTyPart1 = val1;
        fTyPart2 = val2;
    }
 
    void SetChi2Pair(Double_t val1, Double_t val2)
    {
        fChi2Part1 = val1;
        fChi2Part2 = val2;
    }
 
    void SetNDFPair(Int_t val1, Int_t val2)
    {
        fNDFPart1 = val1;
        fNDFPart2 = val2;
    }
 
    void SetBeta400Pair(Double_t val1, Double_t val2)
    {
        fBeta400Part1 = val1;
        fBeta400Part2 = val2;
    }
 
    void SetBeta700Pair(Double_t val1, Double_t val2)
    {
        fBeta700Part1 = val1;
        fBeta700Part2 = val2;
    }
 
    void SetMCTrackIdPart1(Int_t id);
    void SetMCTrackIdPart2(Int_t id);
    void SetRecoTrackIdPart1(Int_t id);   // actually it is GEM track Id
    void SetRecoTrackIdPart2(Int_t id);   // actually it is GEM track Id
 
    void SetMCMomPart1(Double_t px, Double_t py, Double_t pz);
    void SetMCMomPart2(Double_t px, Double_t py, Double_t pz);
 
    void SetNHitsPair(vector<Int_t> part1, vector<Int_t> part2)
    {
        fNHitsSilPart1 = part1[0];   // SILICON
        fNHitsGemPart1 = part1[1];   // GEM
 
        fNHitsSilPart2 = part2[0];   // SILICON
        fNHitsGemPart2 = part2[1];   // GEM
    }
 
    void SetNHitsPair(Int_t nHitsPart1, Int_t nHitsPart2)
    {
        fNHitsGemPart1 = nHitsPart1;
        // fNHitsSilPart1 = 0;
 
        fNHitsGemPart2 = nHitsPart2;
        // fNHitsSilPart2 = 0;
    }
 
    Int_t GetNHitsPart1(TString det = "")
    {
        if (det.IsNull())
            return fNHitsSilPart1 + fNHitsGemPart1;   // nHits per glob. track
 
        else if (det.Contains("GEM"))
            return fNHitsGemPart1;
 
        else if (det.Contains("SILICON"))
            return fNHitsSilPart1;
 
        else
            return 0;
    }
 
    Int_t GetNHitsPart2(TString det = "")
    {
        if (det.IsNull())
            return fNHitsSilPart2 + fNHitsGemPart2;   // nHits per glob. track
 
        else if (det.Contains("GEM"))
            return fNHitsGemPart2;
 
        else if (det.Contains("SILICON"))
            return fNHitsSilPart2;
 
        else
            return 0;
    }
 
    Double_t GetInvMass() { return fInvMass; }
 
    Double_t GetAngleDecayProducts() { return fAngleDecayProducts; }
 
    Double_t GetDCA12() { return fDCA12; }
 
    Double_t GetPath() { return fPath; }
 
    Double_t GetDCA1() { return fDCA1; }
 
    Double_t GetDCA2() { return fDCA2; }
 
    Double_t GetDCA0() { return fDCA0; }
 
    Double_t GetMomPart1() { return fMomPart1; }
 
    Double_t GetMomPart2() { return fMomPart2; }
 
    Double_t GetEtaPart1() { return fEtaPart1; }
 
    Double_t GetEtaPart2() { return fEtaPart2; }
 
    Double_t GetV0X() { return fV0X; }
 
    Double_t GetV0Y() { return fV0Y; }
 
    Double_t GetV0Z() { return fV0Z; }
 
    Double_t GetTxPart1() { return fTxPart1; }
 
    Double_t GetTxPart2() { return fTxPart2; }
 
    Double_t GetTyPart1() { return fTyPart1; }
 
    Double_t GetTyPart2() { return fTyPart2; }
 
    Double_t GetChi2Part1() { return fChi2Part1; }
 
    Double_t GetChi2Part2() { return fChi2Part2; }
 
    Int_t GetNDFPart1() { return fNDFPart1; }
 
    Int_t GetNDFPart2() { return fNDFPart2; }
 
    void SetAlpha(Double_t val) { fAlpha = val; }
 
    void SetPtPodol(Double_t val) { fPtPodol = val; }
 
    Double_t GetAlpha() { return fAlpha; }
 
    Double_t GetPtPodol() { return fPtPodol; }
 
    Double_t GetBeta400Part1() { return fBeta400Part1; }
 
    Double_t GetBeta400Part2() { return fBeta400Part2; }
 
    Double_t GetBeta700Part1() { return fBeta700Part1; }
 
    Double_t GetBeta700Part2() { return fBeta700Part2; }
 
    TVector3 GetMCMomPart1();
    TVector3 GetMCMomPart2();
 
    Int_t GetMCTrackIdPart1();
    Int_t GetMCTrackIdPart2();
 
    Int_t GetRecoTrackIdPart1();   // actually it is GEM track Id
    Int_t GetRecoTrackIdPart2();   // actually it is GEM track Id
 
    // Omega cut from M. Zavertyaev
 
    Double_t Omega() { return (fDCA12 * fDCA12) / (fDCA0 * fDCA0 + 4 * fDCA1 * fDCA2); }
 
  private:
    Double_t fInvMass;   // Invariant mass of a considering pair
 
    Double_t fAngleDecayProducts;   // Angle between the two products of decay
 
    Double_t fBeta400Part1;
    Double_t fBeta400Part2;
 
    Double_t fBeta700Part1;
    Double_t fBeta700Part2;
 
    Double_t fV0X;
    Double_t fV0Y;
    Double_t fV0Z;
 
    Double_t fDCA0;
    Double_t fDCA1;
    Double_t fDCA2;
 
    Double_t fDCA12;
 
    Double_t fPath;
 
    Double_t fMomPart1;
    Double_t fMomPart2;
 
    Double_t fEtaPart1;
    Double_t fEtaPart2;
 
    Int_t fNHitsSilPart1;
    Int_t fNHitsGemPart1;
 
    Int_t fNHitsSilPart2;
    Int_t fNHitsGemPart2;
 
    Double_t fTxPart1;
    Double_t fTxPart2;
    Double_t fTyPart1;
    Double_t fTyPart2;
 
    Double_t fChi2Part1;
    Double_t fChi2Part2;
    Int_t fNDFPart1;
    Int_t fNDFPart2;
 
    // Armenteros-Podolyansky ...
    Double_t fAlpha;
    Double_t fPtPodol;
 
    TVector3 fMCMomPart1;
    TVector3 fMCMomPart2;
 
    TVector3 fRecoMomPart1;
    TVector3 fRecoMomPart2;
 
    Int_t fMCTrackIdPart1;
    Int_t fMCTrackIdPart2;
    Int_t fRecoTrackIdPart1;
    Int_t fRecoTrackIdPart2;
 
    ClassDef(BmnParticlePair, 1);
};
 
#endif