TrackTrajectoryCreationBookKeeper.h

Go to the documentation of this file.

#ifndef TRACKTRAJECTORYCREATIONBOOKKEEPER_H
#define TRACKTRAJECTORYCREATIONBOOKKEEPER_H

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

namespace trkmkr {

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

  class TrackTrajectoryCreationBookKeeper {
  public:
  TrackTrajectoryCreationBookKeeper(std::vector<art::Ptr<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<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<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));
    }
    //
    TrackTrajectory finalizeTrackTrajectory() {
      return TrackTrajectory(std::move(positions),std::move(momenta),std::move(flags),hasMomenta_);
    }
    //
  private:
    bool hasMomenta_;
    std::vector<art::Ptr<Hit> >* hits;
    std::vector<Point_t>         positions;
    std::vector<Vector_t>        momenta;
    std::vector<PointFlags_t>    flags;
    //
  };

}
#endif