MCParticleHitMatching_module.cc

Go to the documentation of this file.

// Framework includes
#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Principal/Event.h"
#include "art/Utilities/make_tool.h"
#include "fhiclcpp/ParameterSet.h"

#include <memory>

// LArSoft
#include "larana/T0Finder/AssociationsTools/IHitParticleAssociations.h"

#include "lardataobj/RecoBase/Hit.h"

#include "nusimdata/SimulationBase/MCParticle.h"

namespace t0 {
  class MCParticleHitMatching;

  class MCParticleHitMatching : public art::EDProducer {
  public:
    explicit MCParticleHitMatching(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.
    MCParticleHitMatching(MCParticleHitMatching const&) = delete;
    MCParticleHitMatching(MCParticleHitMatching&&) = delete;
    MCParticleHitMatching& operator=(MCParticleHitMatching const&) = delete;
    MCParticleHitMatching& operator=(MCParticleHitMatching&&) = delete;

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

    // For keeping track of the replacement backtracker
    std::unique_ptr<IHitParticleAssociations> fHitParticleAssociations;
    bool fOverrideRealData; 
  };

  MCParticleHitMatching::MCParticleHitMatching(fhicl::ParameterSet const& pset) : EDProducer{pset}
  {
    // Get the tool for MC Truth matching
    //const fhicl::ParameterSet& hitPartAssnsParams = pset.get<fhicl::ParameterSet>("HitParticleAssociations"); // unused

    fOverrideRealData = pset.get<bool>("OverrideRealData", false);

    // Get the tool for MC Truth matching
    fHitParticleAssociations = art::make_tool<IHitParticleAssociations>(
      pset.get<fhicl::ParameterSet>("HitParticleAssociations"));

    produces<HitParticleAssociations>();
  }

  void MCParticleHitMatching::produce(art::Event& evt)
  {
    if (evt.isRealData() && !fOverrideRealData) return;

    std::unique_ptr<HitParticleAssociations> MCPartHitassn(new HitParticleAssociations);

    fHitParticleAssociations->CreateHitParticleAssociations(evt, MCPartHitassn.get());

    evt.put(std::move(MCPartHitassn));
  } // Produce

  DEFINE_ART_MODULE(MCParticleHitMatching)
} // end namespace