EndPointModule_module.cc

Go to the documentation of this file.

//
// EndPointModule class
//
// joshua.spitz@yale.edu
//
//  This algorithm is designed to find (weak) vertices from hits after deconvolution and hit finding.
//  A weak vertex is a vertex that has been found using a dedicated vertex finding algorithm only. A
//  strong vertex is a vertex that has been found using a dedicated vertex finding algorithm and matched
//  to a crossing of two or more HoughLineFinder lines. The VertexMatch module finds strong vertices.
//Thanks to B. Morgan of U. of Warwick for comments and suggestions

#include <string>
#include <vector>

// Framework includes
#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Principal/Handle.h"
#include "canvas/Persistency/Common/Assns.h"
#include "canvas/Persistency/Common/Ptr.h"
#include "canvas/Persistency/Common/PtrVector.h"
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

// LArSoft includes
#include "lardata/Utilities/AssociationUtil.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "lardataobj/RecoBase/EndPoint2D.h"
#include "lardataobj/RecoBase/Hit.h"
#include "larreco/RecoAlg/EndPointAlg.h"

namespace cluster {

  class EndPointModule : public art::EDProducer {

  public:
    explicit EndPointModule(fhicl::ParameterSet const& pset);

  private:
    void produce(art::Event& evt);

    std::string fDBScanModuleLabel;

    EndPointAlg fEPAlg; 
  };

}

namespace cluster {

  //-----------------------------------------------------------------------------
  EndPointModule::EndPointModule(fhicl::ParameterSet const& pset)
    : EDProducer{pset}, fEPAlg(pset.get<fhicl::ParameterSet>("EndPointAlg"))
  {
    fDBScanModuleLabel = pset.get<std::string>("DBScanModuleLabel");

    produces<std::vector<recob::EndPoint2D>>();
    produces<art::Assns<recob::EndPoint2D, recob::Hit>>();
  }

  //-----------------------------------------------------------------------------

  void EndPointModule::produce(art::Event& evt)
  {

    art::Handle<std::vector<recob::Cluster>> clusterListHandle;
    evt.getByLabel(fDBScanModuleLabel, clusterListHandle);
    //Point to a collection of vertices to output.

    //.......................................
    art::PtrVector<recob::Cluster> clusIn;
    for (unsigned int ii = 0; ii < clusterListHandle->size(); ++ii) {
      art::Ptr<recob::Cluster> cluster(clusterListHandle, ii);
      clusIn.push_back(cluster);
    }

    // make a std::vector<recob::Cluster> for the output of the
    // Hough Transform
    std::vector<recob::EndPoint2D> vtxOut;
    std::vector<art::PtrVector<recob::Hit>> vtxHitsOut;
    size_t numvtx = fEPAlg.EndPoint(clusIn, vtxOut, vtxHitsOut, evt, fDBScanModuleLabel);

    MF_LOG_DEBUG("Vertex") << "found " << numvtx << "vertices with VertexService";

    //Point to a collection of vertices to output.
    std::unique_ptr<std::vector<recob::EndPoint2D>> vtxcol(
      new std::vector<recob::EndPoint2D>(vtxOut));
    std::unique_ptr<art::Assns<recob::EndPoint2D, recob::Hit>> assn(
      new art::Assns<recob::EndPoint2D, recob::Hit>);

    for (size_t v = 0; v < vtxcol->size(); ++v)
      util::CreateAssn(evt, *(vtxcol.get()), vtxHitsOut[v], *(assn.get()), v);

    evt.put(std::move(vtxcol));
    evt.put(std::move(assn));
  }

} // end namespace

namespace cluster {

  DEFINE_ART_MODULE(EndPointModule)

}