KGTrack.h

Go to the documentation of this file.

#ifndef KGTRACK_H
#define KGTRACK_H

#include <map>

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

#include "lardata/RecoObjects/KHitTrack.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardataobj/RecoBase/Hit.h"

namespace trkf {

  class KGTrack
  {
  public:

    KGTrack(int prefplane);

    ~KGTrack();

    // Accessors.

    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(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:

    // Attributes.

    int fPrefPlane;

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

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

}

#endif