BmnParticle.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      BmnParticle header file                  -----
// -----                  MpdParticle created 21/01/2013  by A. Zinchenko             -----
// -----                  BmnParticle created 02/07/2025  by A. Zinchenko             -----
// -------------------------------------------------------------------------
 
#ifndef BMNPARTICLE_H
#define BMNPARTICLE_H 1
 
// #include "CbmKFTrack.h"
 
// #include "TArrayI.h"
#include "TMatrixD.h"
// #include "TMatrixFSym.h"
#include "TMath.h"
#include "TVector3.h"
 
#include <vector>
// class MpdKalmanTrack;
class CbmStsTrack;
// class MpdVertex;
class CbmVertex;
// class MpdKalmanFilter;
class CbmKF;
class CbmKFTrack;
 
using namespace std;
 
class BmnParticle : public TObject
{
 
  public:
    BmnParticle();                          
    BmnParticle(const BmnParticle& part);   
    BmnParticle(CbmKFTrack& track,
                Int_t indx = -1,
                Double_t mass = 0.1396,
                Double_t* orig = NULL);   
    BmnParticle(CbmStsTrack& track,
                Int_t indx = -1,
                Double_t mass = 0.1396,
                Double_t* orig = NULL);   
 
    BmnParticle& operator=(const BmnParticle& part);   
 
    virtual ~BmnParticle();
 
    void Print();
 
    int GetNDF() const { return fNDF; }
    Int_t GetIndx() const { return fIndx; }
    Int_t GetPdg() const { return fPdg; }
    Int_t GetCharge() const { return fCharge; }
    Double_t GetMass() const { return fMass; }
    Double_t GetMeas(Int_t i) const { return fMeas(i, 0); }
    Double_t GetXY(Int_t i) const { return fXY0[i]; }
    Int_t GetFlag() const { return fFlag; }
    Double_t GetZ0() const { return fZ; }
    // Double_t Phi() const { return GetMeas(2); }
    // Double_t Pt() const { return TMath::Min (TMath::Abs(1./GetMeas(4)*fieldConst), 100.); }
    // Double_t Theta() const { return GetMeas(3); }
    // Double_t Momentum() const { return Pt() / TMath::Sin(Theta()); }
    // TVector3 Momentum3() const { return TVector3(Pt()*TMath::Cos(Phi()), Pt()*TMath::Sin(Phi()),
    //               Momentum()*TMath::Cos(Theta())); }
    Double_t Phi() const { return TMath::ATan2(fq(1, 0), fq(0, 0)); }   // smoothed value
    Double_t Pt() const { return TMath::Min(Momentum() * TMath::Sin(Theta()), 100.); }
    Double_t GetPt() const { return Pt(); }
    // Double_t Theta() const { return fq(1,0); }
    Double_t Theta() const;
    Double_t GetTheta() const { return Theta(); }
    // Double_t Momentum() const { return fCharge == 0 ? fq(2, 0) : Pt() / TMath::Sin(Theta()); }
    Double_t Momentum() const { return fCharge == 0 ? fq(2, 0) : TMath::Abs(fCharge / fq(2, 0)); }
    Double_t GetMomentum() const { return Momentum(); }
    TVector3 Momentum3() const
    {
        return TVector3(Pt() * TMath::Cos(Phi()), Pt() * TMath::Sin(Phi()), Momentum() * TMath::Cos(Theta()));
    }
    Double_t Energy() const;
    Double_t Rapidity() const;
    Double_t Dca() const { return fMeas(0, 0); }   
    Int_t Ndaughters() const { return fDaughtersInds.size(); }
    const vector<Int_t>& DaughterInds() const { return fDaughtersInds; }
    void Track2Part(CbmKFTrack& track, Bool_t setWeight, Double_t* orig);   // conversion from track to particle
    const Double_t Chi2Vertex() { return fChi2ver; }                        
    Double_t Chi2Vertex(CbmVertex* vtx);                                    
    Double_t Chi2() const { return fChi2; }                                 
    Bool_t Point00() const { return fPoint00; }                             
    void FillJ();                                                           // fill Jacobian matrix fJ
    void FillJinv(TVector3& mom3);                                          // fill Jacobian matrix fJinv
 
    TMatrixD& GetMeas() { return fMeas; }
    TMatrixD& GetJ() { return fJ; }
    TMatrixD& GetJinv() { return fJinv; }
    TMatrixD& GetD() { return fD; }
    TMatrixD& GetE() { return fE; }
    TMatrixD& GetA() { return fA; }
    TMatrixD& GetB() { return fB; }
    TMatrixD& GetC() { return fC; }
    TMatrixD& GetG() { return fG; }
    TMatrixD& GetW() { return fW; }
    TMatrixD& Getq() { return fq; }
    TMatrixD& Getx() { return fx; }
    // TMatrixD& GetW() const { return fW; }
    Double_t GetX() const { return fx(0, 0); }
    Double_t GetY() const { return fx(1, 0); }
    Double_t GetZ() const { return fx(2, 0); }
    Double_t GetPx() const { return Pt() * TMath::Cos(Phi()); }
    Double_t GetPy() const { return Pt() * TMath::Sin(Phi()); }
    Double_t GetPz() const { return Momentum() * TMath::Cos(Theta()); }
 
    void SetIndx(Int_t indx) { fIndx = indx; }
    void SetPdg(Int_t pdg)
    {
        fPdg = pdg;
        SetMass();
    }
    void SetCharge(Int_t charge) { fCharge = charge; }
    void SetMass(Double_t mass = -2.0);
    void AddDaughter(Int_t indx) { fDaughtersInds.push_back(indx); }
    Double_t BuildMother(vector<BmnParticle*>& vDaught);
    Double_t BuildMother(vector<CbmStsTrack*>& vTracks, vector<BmnParticle*>& vDaught);
    void SetChi2(Double_t chi2) { fChi2 = chi2; }
    void SetMeas(TMatrixD& matr) { fMeas = matr; }
    void SetCovD(TMatrixD& matr) { fD = matr; }
    void SetCovE(TMatrixD& matr) { fE = matr; }
    // void SetCovQ(TMatrixD &matr) { fQ = matr; }
    void SetA(TMatrixD& matr) { fA = matr; }
    void SetB(TMatrixD& matr) { fB = matr; }
    void SetC(TMatrixD& matr) { fC = matr; }
    void SetG(TMatrixD& matr) { fG = matr; }
    void SetW(TMatrixD& matr) { fW = matr; }
    void Setq(TMatrixD& matr) { fq = matr; }
    void Setx(TMatrixD& matr) { fx = matr; }
    void SetXY(Double_t x, Double_t y)
    {
        fXY0[0] = x;
        fXY0[1] = y;
    }
    void SetFlag(Int_t flag) { fFlag = flag; }
    void SetNDF(int ndf) { fNDF = ndf; }
    void SetZ(Double_t z) { fZ = z; }
    void ParamsAtDca();
    CbmKFTrack GetKFTrack() const;
 
  private:
    // FindPca(CbmKFTrackInterface& tr, Double_t* vert);
    void FindPca(CbmKFTrack& tr, Double_t* vert);
    TVector3 Parabola(TVector3& pos0, TVector3& pos1, TVector3& pos2);
    // void WeightAtDca(CbmStsTrack &track, Double_t *vert); // obtain BmnParticle weight at DCA
    void WeightAtDca(CbmKFTrack& track, Double_t* vert);   // obtain BmnParticle weight at DCA
 
    Int_t fIndx;                    // index of particle
    Int_t fPdg;                     // PDG hypothesis
    Int_t fCharge;                  // charge
    Double_t fMass;                 // particle mass (GeV)
    Double_t fieldConst;            
    vector<Int_t> fDaughtersInds;   // indices of particles it is created from
    TMatrixD fMeas;                 // vector of measurements (params)
    TMatrixD fq;                    // geometrical momentum
    TMatrixD fx;                    // particle origin (production vertex)
    Double_t fXY0[2];               // X and Y at DCA
    TMatrixD fJ;                    
    TMatrixD fJinv;                 
    TMatrixD fD;                    
    TMatrixD fE;                    
    // TMatrixD fQ;                  //! covariance cov(qk,qj)
    TMatrixD fA;         
    TMatrixD fB;         
    TMatrixD fC;         
    TMatrixD fG;         // covariance of params
    TMatrixD fW;         
    Double_t fChi2;      // Chi2 of mother particle
    Double_t fChi2ver;   // Chi2 of particle w.r.t. vertex
    Int_t fFlag;         // status flag
    Bool_t fPoint00;     // flag for tracks extrapolated to (0,0)
    Double_t fZ;         // Z-position
    int fNDF;
 
    // MpdKalmanFilter *fKF;
    CbmKF* fKF;
 
    ClassDef(BmnParticle, 1);
};
 
//__________________________________________________________________________
 
inline Double_t BmnParticle::Theta() const
{
    // Theta angle
 
    return TMath::ATan(TMath::Sqrt(fq(1, 0) * fq(1, 0) + fq(0, 0) * fq(0, 0)));
}
 
//__________________________________________________________________________
 
#endif