TrackTrajectoryCreationBookKeeper.h

Go to the documentation of this file.

#ifndef TRACKTRAJECTORYCREATIONBOOKKEEPER_H
#define TRACKTRAJECTORYCREATIONBOOKKEEPER_H

#include "canvas/Persistency/Common/Ptr.h"
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/TrackTrajectory.h"
#include "lardataobj/RecoBase/TrackingTypes.h"

#include <utility>
#include <vector>

namespace trkmkr {

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

  class TrackTrajectoryCreationBookKeeper {
  public:
    TrackTrajectoryCreationBookKeeper(std::vector<art::Ptr<recob::Hit>>& outhits, bool hasMomenta)
      : hasMomenta_(hasMomenta), hits(&outhits)
    {
      hits->clear();
    }
    //
    TrackTrajectoryCreationBookKeeper(const TrackTrajectoryCreationBookKeeper&) = delete;
    TrackTrajectoryCreationBookKeeper(TrackTrajectoryCreationBookKeeper&&) = delete;
    TrackTrajectoryCreationBookKeeper& operator=(const TrackTrajectoryCreationBookKeeper&) = delete;
    TrackTrajectoryCreationBookKeeper& operator=(TrackTrajectoryCreationBookKeeper&&) = delete;
    //
    void addPoint(const Point_t& point,
                  const Vector_t& vect,
                  art::Ptr<recob::Hit> hit,
                  const PointFlags_t& flag)
    {
      positions.push_back(point);
      momenta.push_back(vect);
      hits->push_back(hit);
      flags.push_back(flag);
    }
    void addPoint(Point_t&& point, Vector_t&& vect, art::Ptr<recob::Hit> hit, PointFlags_t&& flag)
    {
      positions.push_back(std::move(point));
      momenta.push_back(std::move(vect));
      hits->push_back(hit);
      flags.push_back(std::move(flag));
    }
    //
    recob::TrackTrajectory finalizeTrackTrajectory()
    {
      return recob::TrackTrajectory(
        std::move(positions), std::move(momenta), std::move(flags), hasMomenta_);
    }
    //
  private:
    bool hasMomenta_;
    std::vector<art::Ptr<recob::Hit>>* hits;
    std::vector<Point_t> positions;
    std::vector<Vector_t> momenta;
    std::vector<PointFlags_t> flags;
    //
  };

}
#endif