BmnVertex.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                       BmnVertex source file                   -----
// -----                  Created 13.12.2018  by D. Baranov            -----
// -------------------------------------------------------------------------
#include "BmnVertex.h"
 
#include <iomanip>
#include <sstream>
#include "FairLogger.h"
 
using namespace std;
 
 
// -----   Default constructor   -------------------------------------------
BmnVertex::BmnVertex()
  : TNamed("Vertex", "Global"),
    fX(0.),
    fY(0.),
    fZ(0.),
    fChi2(0.),
    fNDF(0),
    fNTracks(0),
    fCovMatrix()
{
  for(Int_t i=0; i<6; i++) fCovMatrix[i] = 0;
}
// -------------------------------------------------------------------------
 
// -----   Constructor with name and title   -------------------------------
BmnVertex::BmnVertex(const char* name, const char* title)
  : TNamed(name, title),
    fX(0.),
    fY(0.),
    fZ(0.),
    fChi2(0.),
    fNDF(0),
    fNTracks(0),
    fCovMatrix()
{
  for(Int_t i=0; i<6; i++) fCovMatrix[i] = 0;
}
// -------------------------------------------------------------------------
 
 
 
// -----   Constructor with all parameters   -------------------------------
BmnVertex::BmnVertex(const char* name, const char* title,
         Double_t x, Double_t y, Double_t z, Double_t chi2,
         Int_t ndf, Int_t nTracks,
         const TMatrixFSym& covMat)
  : TNamed(name, title),
    fX(x),
    fY(y),
    fZ(z),
    fChi2(chi2),
    fNDF(ndf),
    fNTracks(nTracks),
    fCovMatrix()
{
  if ( (covMat.GetNrows() != 3) && (covMat.GetNcols() != 3) ) {
    LOG(error) << "Wrong dimension of passed covariance matrix. Clear the covariance matrix.";
    for(Int_t i=0; i<6; i++) fCovMatrix[i] = 0;
  } else {
    Int_t index = 0;
    for (Int_t i=0; i<3; i++) {
      for (Int_t j=i; j<3; j++) fCovMatrix[index++] = covMat[i][j];
    }
  }
}
// -------------------------------------------------------------------------
 
 
 
// -----   Destructor   ----------------------------------------------------
BmnVertex::~BmnVertex() { }
// -------------------------------------------------------------------------
 
 
 
// -----   Public method Print   -------------------------------------------
void BmnVertex::Print(Option_t*) const {
  Double_t chi2ndf;
  if (fNDF) chi2ndf = fChi2 / Double_t(fNDF);
  else chi2ndf = 0.;
  LOG(info) << "Vertex coord. (" << fX << "," << fY << "," << fZ << ") cm, "
            << "chi2/ndf = " << chi2ndf << ", " << fNTracks
            << " tracks used";
}
// -------------------------------------------------------------------------
 
 
 
// -----   Accessor to covariance matrix    --------------------------------
void BmnVertex::CovMatrix(TMatrixFSym& covMat) const {
  Int_t index = 0;
  for (int i=0; i<3; i++) {
    for (int j=i; j<3; j++) {
      covMat[i][j] = fCovMatrix[index];
      covMat[j][i] = fCovMatrix[index];
      index++;
    }
  }
}
// -------------------------------------------------------------------------
 
 
 
// -----   Accessor to covariance matrix elements   ------------------------
Double_t BmnVertex::GetCovariance(Int_t i, Int_t j) const {
  TMatrixFSym* mat = new TMatrixFSym(3);
  CovMatrix(*mat);
  Double_t element = (*mat)[i][j];
  delete mat;
  return element;
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method SetVertex   ---------------------------------------
void BmnVertex::SetVertex(Double_t x, Double_t y, Double_t z, Double_t chi2,
        Int_t ndf, Int_t nTracks,
        const TMatrixFSym& covMat)
{
  fX       = x;
  fY       = y;
  fZ       = z;
  fChi2    = chi2;
  fNDF     = ndf;
  fNTracks = nTracks;
  if ( (covMat.GetNrows() != 3) && (covMat.GetNcols() != 3) ) {
    LOG(error) << "Wrong dimension of passed covariance matrix. Clear the covariance matrix.";
    for(Int_t i=0; i<6; i++) fCovMatrix[i] = 0;
  } else {
    Int_t index = 0;
    for (Int_t i=0; i<3; i++) {
      for (Int_t j=i; j<3; j++) fCovMatrix[index++] = covMat[i][j];
    }
  }
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method Reset   -------------------------------------------
void BmnVertex::Reset() {
  fX = fY = fZ = fChi2 = 0.;
  fNDF = fNTracks = 0;
  for(Int_t i=0; i<6; i++) fCovMatrix[i] = 0;
}
// -------------------------------------------------------------------------
 
 
 
// --- String output  ------------------------------------------------------
string BmnVertex::ToString() const {
 
  Double_t chi2ndf = ( fNDF ? fChi2 / Double_t(fNDF) : 0.);
  stringstream ss;
  ss << "Vertex: position (" << fixed << setprecision(4) << fX << ", " << fY << ", " << fZ
   << ") cm, chi2/ndf = " << chi2ndf << ", tracks used: " << fNTracks;
   return ss.str();
}
// -------------------------------------------------------------------------