TrackCalorimetry_module.cc

Go to the documentation of this file.

// Class:       TrackCalorimetry
// Module Type: producer
// File:        TrackCalorimetry_module.cc
//
// Title:   Track Calorimetry Algorithim Class
// Author:  Wes Ketchum (wketchum@lanl.gov), based on code the Calorimetry_module
//
// Description: Algorithm that produces a calorimetry object given a track
// Input:       recob::Track, Assn<recob::Spacepoint,recob::Track>, Assn<recob::Hit,recob::Track>
// Output:      anab::Calorimetry, (and Assn<anab::Calorimetry,recob::Track>) 
//
// Generated at Tue Oct 21 15:54:15 2014 by Wesley Ketchum using artmod
// from cetpkgsupport v1_07_01.

#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 "art/Framework/Principal/Run.h"
#include "art/Framework/Principal/SubRun.h"
#include "canvas/Utilities/InputTag.h"
#include "fhiclcpp/ParameterSet.h"
#include "messagefacility/MessageLogger/MessageLogger.h"

#include <memory>

#include "larcore/Geometry/Geometry.h"
#include "lardata/Utilities/AssociationUtil.h"
#include "lardata/DetectorInfoServices/DetectorPropertiesService.h"
#include "lardata/DetectorInfoServices/LArPropertiesService.h"
#include "lardata/DetectorInfoServices/ServicePack.h" // lar::extractProviders()
#include "TrackCalorimetryAlg.h"

namespace calo{
  class TrackCalorimetry;
}

class calo::TrackCalorimetry : public art::EDProducer {
public:
  explicit TrackCalorimetry(fhicl::ParameterSet const & p);
  // The destructor generated by the compiler is fine for classes
  // without bare pointers or other resource use.

  // Plugins should not be copied or assigned.
  TrackCalorimetry(TrackCalorimetry const &) = delete;
  TrackCalorimetry(TrackCalorimetry &&) = delete;
  TrackCalorimetry & operator = (TrackCalorimetry const &) = delete;
  TrackCalorimetry & operator = (TrackCalorimetry &&) = delete;

  // Required functions.
  void produce(art::Event & e) override;

  // Selected optional functions.
  void reconfigure(fhicl::ParameterSet const & p) ;

private:

  std::string    fTrackModuleLabel;
  std::string    fHitModuleLabel;

  TrackCalorimetryAlg fTrackCaloAlg;

};

calo::TrackCalorimetry::TrackCalorimetry(fhicl::ParameterSet const & p):
  fTrackModuleLabel(p.get<std::string>("TrackModuleLabel")),
  fHitModuleLabel(p.get<std::string>("HitModuleLabel")),
  fTrackCaloAlg(p.get<fhicl::ParameterSet>("TrackCalorimetryAlg"))
{
  this->reconfigure(p);

  produces< std::vector<anab::Calorimetry>              >();
  produces< art::Assns<recob::Track, anab::Calorimetry> >();
}

void calo::TrackCalorimetry::reconfigure(fhicl::ParameterSet const & p)
{
  fTrackModuleLabel = p.get<std::string>("TrackModuleLabel");
  fHitModuleLabel = p.get<std::string>("HitModuleLabel");

  fTrackCaloAlg.reconfigure(p.get<fhicl::ParameterSet>("TrackCalorimetryAlg"));
}

void calo::TrackCalorimetry::produce(art::Event & e)
{

  art::Handle< std::vector<recob::Track> > trackHandle;
  e.getByLabel(fTrackModuleLabel,trackHandle);
  std::vector<recob::Track> const& trackVector(*trackHandle);
  
  //Get Hits from event.
  art::Handle< std::vector<recob::Hit> > hitHandle;
  e.getByLabel(fHitModuleLabel, hitHandle);
  std::vector<recob::Hit> const& hitVector(*hitHandle);
  
  //Get track<-->hit associations
  art::Handle< art::Assns<recob::Track,recob::Hit> > assnTrackHitHandle;
  e.getByLabel(fTrackModuleLabel,assnTrackHitHandle);
  std::vector< std::vector<size_t> > 
    hit_indices_per_track = util::GetAssociatedVectorManyI(assnTrackHitHandle,
                                                           trackHandle);

  //Make the container for the calo product to put onto the event.
  std::unique_ptr< std::vector<anab::Calorimetry> > caloPtr(new std::vector<anab::Calorimetry>);
  std::vector<anab::Calorimetry> & caloVector(*caloPtr);
  
  //Make a container for the track<-->calo associations. 
  //One entry per track, with entry equal to index in calorimetry collection of associated object.
  std::vector<size_t> assnTrackCaloVector;
  std::unique_ptr< art::Assns<recob::Track,anab::Calorimetry> > assnTrackCaloPtr(new art::Assns<recob::Track,anab::Calorimetry>);

  fTrackCaloAlg.ExtractCalorimetry(trackVector,
                                   hitVector,hit_indices_per_track,
                                   caloVector,assnTrackCaloVector,
                                   lar::extractProviders
                                     <geo::Geometry, detinfo::LArPropertiesService, detinfo::DetectorPropertiesService>()
                                   );

  //Make the associations for ART
  for(size_t calo_iter=0; calo_iter<assnTrackCaloVector.size(); calo_iter++){
    if(assnTrackCaloVector[calo_iter]==std::numeric_limits<size_t>::max()) continue;
    art::Ptr<recob::Track> trk_ptr(trackHandle,assnTrackCaloVector[calo_iter]);
    util::CreateAssn(*this, e, caloVector, trk_ptr, *assnTrackCaloPtr, calo_iter); 
  }

  e.put(std::move(caloPtr));
  e.put(std::move(assnTrackCaloPtr));

}

DEFINE_ART_MODULE(calo::TrackCalorimetry)