TrackMCH.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_MCH_TRACKMCH_H_
#define ALICEO2_MCH_TRACKMCH_H_
 
#include <TMatrixD.h>
#include <iosfwd>
 
#include "CommonDataFormat/RangeReference.h"
#include "CommonDataFormat/InteractionRecord.h"
#include "CommonDataFormat/TimeStamp.h"
 
namespace o2
{
namespace mch
{
 
class TrackMCH
{
  using ClusRef = o2::dataformats::RangeRefComp<5>;
 
 public:
  using Time = o2::dataformats::TimeStampWithError<float, float>;
 
  TrackMCH() = default;
  TrackMCH(double z, const TMatrixD& param, const TMatrixD& cov, double chi2, int firstClIdx, int nClusters,
           double zAtMID, const TMatrixD& paramAtMID, const TMatrixD& covAtMID, const Time& time);
  ~TrackMCH() = default;
 
  TrackMCH(const TrackMCH& track) = default;
  TrackMCH& operator=(const TrackMCH& track) = default;
  TrackMCH(TrackMCH&&) = default;
  TrackMCH& operator=(TrackMCH&&) = default;
 
  double getX() const { return mParam[0]; }
  double getY() const { return mParam[2]; }
  double getZ() const { return mZ; }
  void setZ(double z) { mZ = z; }
 
  double getPx() const;
  double getPy() const;
  double getPz() const;
  double getP() const;
 
  short getSign() const { return (mParam[4] < 0) ? -1 : 1; }
 
  const double* getParameters() const { return mParam; }
  void setParameters(const TMatrixD& param) { param.GetMatrix2Array(mParam); }
 
  const double* getCovariances() const { return mCov; }
  double getCovariance(int i, int j) const { return mCov[SCovIdx[i][j]]; }
  void setCovariances(const TMatrixD& cov) { setCovariances(cov, mCov); }
 
  double getZAtMID() const { return mZAtMID; }
  void setZAtMID(double z) { mZAtMID = z; }
 
  const double* getParametersAtMID() const { return mParamAtMID; }
  void setParametersAtMID(const TMatrixD& param) { param.GetMatrix2Array(mParamAtMID); }
 
  const double* getCovariancesAtMID() const { return mCovAtMID; }
  double getCovarianceAtMID(int i, int j) const { return mCovAtMID[SCovIdx[i][j]]; }
  void setCovariancesAtMID(const TMatrixD& cov) { setCovariances(cov, mCovAtMID); }
 
  double getChi2() const { return mChi2; }
  void setChi2(double chi2) { mChi2 = chi2; }
  int getNDF() const { return 2 * mClusRef.getEntries() - 5; }
  double getChi2OverNDF() const { return mChi2 / getNDF(); }
 
  int getNClusters() const { return mClusRef.getEntries(); }
  int getFirstClusterIdx() const { return mClusRef.getFirstEntry(); }
  int getLastClusterIdx() const { return mClusRef.getFirstEntry() + mClusRef.getEntries() - 1; }
  void setClusterRef(int firstClusterIdx, int nClusters) { mClusRef.set(firstClusterIdx, nClusters); }
 
  const Time& getTimeMUS() const { return mTimeMUS; }
  Time& getTimeMUS() { return mTimeMUS; }
  void setTimeMUS(const Time& t) { mTimeMUS = t; }
  void setTimeMUS(float t, float te)
  {
    mTimeMUS.setTimeStamp(t);
    mTimeMUS.setTimeStampError(te);
  }
 
  InteractionRecord getMeanIR(uint32_t refOrbit) const;
 
 private:
  static constexpr int SNParams = 5;  
  static constexpr int SCovSize = 15; 
  static constexpr int SCovIdx[SNParams][SNParams] = {{0, 1, 3, 6, 10},
                                                      {1, 2, 4, 7, 11},
                                                      {3, 4, 5, 8, 12},
                                                      {6, 7, 8, 9, 13},
                                                      {10, 11, 12, 13, 14}};
 
  void setCovariances(const TMatrixD& src, double (&dest)[SCovSize]);
 
  double mZ = 0.;                 
  double mParam[SNParams] = {0.}; 
  double mCov[SCovSize] = {0.};
 
  double mChi2 = 0.; 
 
  ClusRef mClusRef{}; 
 
  double mZAtMID = 0.;                 
  double mParamAtMID[SNParams] = {0.}; 
  double mCovAtMID[SCovSize] = {0.};   
  Time mTimeMUS{};                     
 
  ClassDefNV(TrackMCH, 3);
};
 
std::ostream& operator<<(std::ostream& os, const TrackMCH& t);
 
std::string asString(const o2::mch::TrackMCH& t);
 
} // namespace mch
} // namespace o2
 
#endif // ALICEO2_MCH_TRACKMCH_H_