WCVertex_module.cc

Go to the documentation of this file.

///
/// \file    WCVertex_module.cc
/// \brief   Place a vertex on a cosmic ray track.
/// \author  messier@indiana.edu
/// \version $Id:$
///
#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "canvas/Persistency/Common/FindMany.h"
#include "canvas/Persistency/Common/FindManyP.h"
#include "art/Framework/Services/Optional/TFileService.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"

#include "RecoBase/FilterList.h"
#include "RecoBase/Cluster.h"
#include "RecoBase/Vertex.h"
#include "Utilities/AssociationUtil.h"
#include "GeometryObjects/GeometryBase.h"
#include "Geometry/Geometry.h"
#include "GeometryObjects/Geo.h"

namespace wcvtx {
  ///
  /// Module to associate wire chamber vertex with every slice in the event
  ///
  class WCVertex : public art::EDProducer {
  private:
    // folder labels
    std::string fSliceLabel;       ///< Where to get slices?
    std::string fVertexLabel;      ///< Vertex to grab

  public:
    //..................................................................
    explicit WCVertex(fhicl::ParameterSet const &pset):
      fSliceLabel    (pset.get< std::string >("SliceLabel")),
      fVertexLabel   (pset.get< std::string >("VertexLabel"))
    {
      //this->produces< std::vector<rb::Vertex> > ();
      this->produces< art::Assns<rb::Vertex, rb::Cluster> >();
    }
    
    //..................................................................
    virtual ~WCVertex() {  }
   
    //..................................................................
    void beginJob() { }
 
    //..................................................................
    void produce(art::Event& evt) {
      //
      // Associate the WC vertex with every single slice in event.
      // This is a super hacky way to do this, but you don't know a priori 
      // which slice has the particle that caused the trigger.
      //
      std::unique_ptr< art::Assns<rb::Vertex, rb::Cluster> > 
        vtx_to_slice(new art::Assns<rb::Vertex, rb::Cluster>);
      std::vector<art::Ptr<rb::Cluster>> slicevec;
      // 
      // Find the slices
      //
      art::Handle<std::vector<rb::Cluster> > slice;
      evt.getByLabel(fSliceLabel,slice);
      
      for(unsigned int i = 0; i < slice->size(); ++i) {
        art::Ptr<rb::Cluster> sptr(slice, i);
        slicevec.push_back(sptr);
      }
      
      art::Handle<std::vector<rb::Vertex>> vtxHandle;
      evt.getByLabel(fVertexLabel,vtxHandle);
      art::Ptr<rb::Vertex>vtx(vtxHandle, 0);
            
      util::CreateAssn(*this, evt, vtx, slicevec, *vtx_to_slice);
      evt.put(std::move(vtx_to_slice));
    }
    
    //......................................................................
    void endJob() { }
  };
}
////////////////////////////////////////////////////////////////////////
namespace wcvtx
{
  DEFINE_ART_MODULE(WCVertex)
}
////////////////////////////////////////////////////////////////////////