CornerFinder_module.cc

Go to the documentation of this file.

#ifndef CORNERFINDER_H
#define CORNERFINDER_H

//Basic includes
#include <vector>
#include <string>

//Framework includes
#include "art/Framework/Core/ModuleMacros.h" 
#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Principal/Event.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"

//LArSoft includes
#include "lardataobj/RecoBase/Wire.h"
#include "lardataobj/RecoBase/EndPoint2D.h"
#include "larreco/RecoAlg/CornerFinderAlg.h"
#include "larcore/Geometry/Geometry.h"

//header info for CornerFinder class

namespace vertex {
   
  class CornerFinder :  public art::EDProducer {
    
  public:
    
    explicit CornerFinder(fhicl::ParameterSet const& pset); 
    virtual ~CornerFinder();        
    void reconfigure(fhicl::ParameterSet const& p);

    
    void produce(art::Event& evt);
    
  private:
    corner::CornerFinderAlg  fCornerAlg;
    art::ServiceHandle<geo::Geometry> fGeometryHandle;

    std::string               fCalDataModuleLabel;

    void printEndpoints(std::vector<recob::EndPoint2D> const& corner_vector);

  }; //class CornerFinder
  
  

  //-----------------------------------------------------------------------------
  CornerFinder::CornerFinder(fhicl::ParameterSet const& pset):
    fCornerAlg(pset.get<fhicl::ParameterSet>("CornerAlgParamSet"))
  {  
    this->reconfigure(pset);    
    produces< std::vector<recob::EndPoint2D> >();
  }
  
  //-----------------------------------------------------------------------------
  CornerFinder::~CornerFinder(){}

  //---------------------------------------------------------------------------
  void CornerFinder::reconfigure(fhicl::ParameterSet const& pset) {
    fCalDataModuleLabel = pset.get<std::string>("CalDataModuleLabel");
    fCornerAlg.reconfigure(pset.get<fhicl::ParameterSet>("CornerAlgParamSet"));
    return;
  }

  //-----------------------------------------------------------------------------
  void CornerFinder::produce(art::Event& evt){
  
    //We need do very little here, as it's all handled by the corner finder.
    const bool DEBUG_TEST = false; //turn on/off some messages

    //We need to grab out the wires.
    art::Handle< std::vector<recob::Wire> > wireHandle;
    evt.getByLabel(fCalDataModuleLabel,wireHandle);
    std::vector<recob::Wire> const& wireVec(*wireHandle);

    //First, have it process the wires.
    fCornerAlg.GrabWires(wireVec,*fGeometryHandle);

    //now, make a vector of recob::EndPoint2Ds, and hand that to CornerAlg to fill out
    std::unique_ptr< std::vector<recob::EndPoint2D> > corner_vector(new std::vector<recob::EndPoint2D>);
    fCornerAlg.get_feature_points_fast(*corner_vector,*fGeometryHandle);

    mf::LogInfo("CornerFinderModule") << "CornerFinderAlg finished, and returned " 
                                          << corner_vector->size() << " endpoints.";

    if(DEBUG_TEST) printEndpoints(*corner_vector);

    //and now, put this on the event.
    evt.put(std::move(corner_vector));

    //Done!

  } // end of produce


  //-----------------------------------------------------------------------------
  void CornerFinder::printEndpoints(std::vector<recob::EndPoint2D> const& corner_vector){

    for(auto iter=corner_vector.begin(); iter!=corner_vector.end(); iter++){
      geo::WireID wid = iter->WireID();
      mf::LogVerbatim("CornerFinderModule") << "Endpoint found: (plane,wire,time,strength)=(" 
                                            << wid.Plane << "," 
                                            << wid.Wire << "," 
                                            << iter->DriftTime() << ","
                                            << iter->Strength() << ")";
    }

  }


  DEFINE_ART_MODULE(CornerFinder)

}
#endif // CORNERFINDER_H