KGTrack.h

Go to the documentation of this file.

#ifndef KGTRACK_H
#define KGTRACK_H

#include <iosfwd>
#include <map>
#include <vector>

#include "canvas/Persistency/Common/PtrVector.h"

#include "lardata/RecoObjects/KHitTrack.h"

namespace detinfo {
  class DetectorPropertiesData;
}

namespace recob {
  class Hit;
  class Track;
}

namespace trkf {

  class KGTrack {
  public:
    KGTrack(int prefplane);

    int getPrefPlane() const { return fPrefPlane; }

    const std::multimap<double, KHitTrack>& getTrackMap() const { return fTrackMap; }

    size_t numHits() const { return fTrackMap.size(); }

    const KHitTrack& startTrack() const;

    const KHitTrack& endTrack() const;

    bool isValid() const { return fTrackMap.size() > 0; }

    // Modifiers.

    std::multimap<double, KHitTrack>& getTrackMap() { return fTrackMap; }

    KHitTrack& startTrack();

    KHitTrack& endTrack();

    void addTrack(const KHitTrack& trh);

    void recalibrate();

    void clear() { fTrackMap.clear(); }

    // Methods.

    void fillTrack(detinfo::DetectorPropertiesData const& detProp,
                   recob::Track& track,
                   int id) const;

    void fillHits(art::PtrVector<recob::Hit>& hits, std::vector<unsigned int>& hittpindex) const;

    const std::multimap<double, KHitTrack> TrackMap() const { return fTrackMap; }

    std::ostream& Print(std::ostream& out) const;

  private:
    int fPrefPlane;

    std::multimap<double, KHitTrack> fTrackMap;
  };

  std::ostream& operator<<(std::ostream& out, const KGTrack& trg);

}

#endif