TrackCreationBookKeeper.h

Go to the documentation of this file.

#ifndef TRACKCREATIONBOOKKEEPER_H
#define TRACKCREATIONBOOKKEEPER_H

#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardataobj/RecoBase/TrackFitHitInfo.h"
#include "lardataobj/RecoBase/TrackingTypes.h"
#include "larreco/RecoAlg/TrackTrajectoryCreationBookKeeper.h"
#include "larreco/TrackFinder/TrackMaker.h"

namespace trkmkr {

  using Point_t = recob::tracking::Point_t;
  using Vector_t = recob::tracking::Vector_t;
  using PointFlags_t = recob::TrackTrajectory::PointFlags_t;

  class TrackCreationBookKeeper {
  public:
    TrackCreationBookKeeper(std::vector<art::Ptr<recob::Hit>>& outhits,
                            OptionalOutputs& optionals,
                            int tkID,
                            int pdgHyp,
                            bool hasMomenta,
                            int nfitpars = 4)
      : ttcbk_(outhits, hasMomenta)
      , tkID_(tkID)
      , pdgHyp_(pdgHyp)
      , totChi2_(0)
      , opts(&optionals)
      , nfittedpars(nfitpars)
    {
      opts->reset();
    }
    //
    TrackCreationBookKeeper(const TrackCreationBookKeeper&) = delete;
    TrackCreationBookKeeper(TrackCreationBookKeeper&&) = delete;
    TrackCreationBookKeeper& operator=(const TrackCreationBookKeeper&) = delete;
    TrackCreationBookKeeper& operator=(TrackCreationBookKeeper&&) = delete;
    //
    void addPoint(const Point_t& point,
                  const Vector_t& vect,
                  art::Ptr<recob::Hit> hit,
                  const PointFlags_t& flag,
                  double chi2)
    {
      ttcbk_.addPoint(point, vect, hit, flag);
      if (chi2 >= 0) {
        chi2v.push_back(chi2);
        totChi2_ += chi2;
      }
    }
    void addPoint(const Point_t& point,
                  const Vector_t& vect,
                  art::Ptr<recob::Hit> hit,
                  const PointFlags_t& flag,
                  double chi2,
                  OptionalPointElement& ope)
    {
      addPoint(point, vect, hit, flag, chi2);
      opts->addPoint(ope);
    }
    void addPoint(Point_t&& point,
                  Vector_t&& vect,
                  art::Ptr<recob::Hit> hit,
                  PointFlags_t&& flag,
                  double chi2)
    {
      ttcbk_.addPoint(std::move(point), std::move(vect), hit, std::move(flag));
      if (chi2 >= 0) {
        chi2v.push_back(chi2);
        totChi2_ += chi2;
      }
    }
    void addPoint(Point_t&& point,
                  Vector_t&& vect,
                  art::Ptr<recob::Hit> hit,
                  PointFlags_t&& flag,
                  double chi2,
                  OptionalPointElement& ope)
    {
      addPoint(std::move(point), std::move(vect), hit, std::move(flag), chi2);
      opts->addPoint(ope);
    }
    //
    void setTotChi2(double totChi2) { totChi2_ = totChi2; }
    //
    recob::Track finalizeTrack(const recob::tracking::SMatrixSym55& covStart,
                               const recob::tracking::SMatrixSym55& covEnd)
    {
      return recob::Track(ttcbk_.finalizeTrackTrajectory(),
                          pdgHyp_,
                          totChi2_,
                          int(chi2v.size()) - nfittedpars,
                          recob::tracking::SMatrixSym55(covStart),
                          recob::tracking::SMatrixSym55(covEnd),
                          tkID_);
    }
    recob::Track finalizeTrack(recob::tracking::SMatrixSym55&& covStart,
                               recob::tracking::SMatrixSym55&& covEnd)
    {
      return recob::Track(ttcbk_.finalizeTrackTrajectory(),
                          pdgHyp_,
                          totChi2_,
                          int(chi2v.size()) - nfittedpars,
                          std::move(covStart),
                          std::move(covEnd),
                          tkID_);
    }
  private:
    trkmkr::TrackTrajectoryCreationBookKeeper ttcbk_;
    int tkID_;
    int pdgHyp_;
    double totChi2_;
    OptionalOutputs* opts;
    std::vector<double> chi2v;
    int
      nfittedpars; // hits are 1D measurement, i.e. each hit is one d.o.f.; no B field: 4 fitted parameters by default
    //
  };

}
#endif