KHitContainerWireX.cxx

Go to the documentation of this file.

#include <map>

#include "lardata/RecoObjects/KHitContainerWireX.h"

#include "cetlib_except/exception.h"

#include "lardata/RecoObjects/KHitWireX.h"
#include "larcore/Geometry/Geometry.h"

namespace trkf {

  KHitContainerWireX::KHitContainerWireX()
  {}

  KHitContainerWireX::~KHitContainerWireX()
  {}

  void KHitContainerWireX::fill(const art::PtrVector<recob::Hit>& hits,
                                int only_plane)
  {
    // Get services.

    art::ServiceHandle<geo::Geometry> geom;

    // Make a temporary map from channel number to KHitGroup objects.
    // The KHitGroup pointers are borrowed references to KHitGroup
    // objects stored by value in the base class.

    std::map<unsigned int, KHitGroup*> group_map;

    // Loop over hits.

    for(art::PtrVector<recob::Hit>::const_iterator ihit = hits.begin();
        ihit != hits.end(); ++ihit) {
      const recob::Hit& hit = **ihit;

      // Extract the wire id from the Hit.
      geo::WireID hitWireID = hit.WireID();
                
      uint32_t channel = hit.Channel();

      // Choose plane.
      if(only_plane >= 0 && hitWireID.Plane != (unsigned int)(only_plane))
        continue;

      // See if we need to make a new KHitGroup.

      KHitGroup* pgr = 0;
      if(group_map.count(channel) == 0) {
        getUnsorted().push_back(KHitGroup());
        pgr = &(getUnsorted().back());
        group_map[channel] = pgr;
      }
      else
        pgr = group_map[channel];
      if (!pgr) {
        throw cet::exception("KHitContainerWireX")
          << __func__ << ": no group map for channel " << channel << "\n";
      }

      // Get surface from KHitGroup (might be null pointer).

      const std::shared_ptr<const Surface>& psurf = pgr->getSurface();

      // Construct KHitWireX object.

      std::shared_ptr<const KHitBase> phit(new KHitWireX(*ihit, psurf));

      // Insert hit into KHitGroup.

      pgr->addHit(phit);
    }
  }

} // end namespace trkf