MCReco_module.cc

Go to the documentation of this file.

// Class:       MCReco
// Module Type: producer
// File:        MCReco_module.cc
//
// Generated at Mon Aug 11 05:40:00 2014 by Kazuhiro Terao using artmod
// from cetpkgsupport v1_05_04.

#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "canvas/Persistency/Common/FindOneP.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 "MCShowerRecoAlg.h"
#include "MCTrackRecoAlg.h"

#include <memory>

class MCReco;

class MCReco : public art::EDProducer {
public:
  explicit MCReco(fhicl::ParameterSet const & p);
//  virtual ~MCReco();

  void produce(art::Event & e) override;

private:

  // Declare member data here.
  art::InputTag fMCParticleLabel;
  art::InputTag fSimChannelLabel;

  ::sim::MCRecoPart fPart;
  ::sim::MCRecoEdep fEdep;
  ::sim::MCShowerRecoAlg fMCSAlg;
  ::sim::MCTrackRecoAlg  fMCTAlg;
};

MCReco::MCReco(fhicl::ParameterSet const & pset)
  : fPart   (pset.get< fhicl::ParameterSet >("MCRecoPart"))
  , fEdep   (pset.get< fhicl::ParameterSet >("MCRecoEdep"))
  , fMCSAlg (pset.get< fhicl::ParameterSet >("MCShowerRecoAlg"))
  , fMCTAlg (pset.get< fhicl::ParameterSet >("MCTrackRecoAlg"))
{

  //for backwards compatibility to using the "G4ModName" label...
  if(!(pset.get_if_present<art::InputTag>("MCParticleLabel",fMCParticleLabel) &&
       pset.get_if_present<art::InputTag>("SimChannelLabel",fSimChannelLabel)) ){

    mf::LogWarning("MCReco_module") << "USING DEPRECATED G4ModName CONFIG IN MCRECO_MODULE"
                                    << "\nUse 'MCParticleLabel' and 'SimChannelLabel' instead.";

    fMCParticleLabel = pset.get<art::InputTag>("G4ModName","largeant");
    fSimChannelLabel = pset.get<art::InputTag>("G4ModName","largeant");
  }

  produces< std::vector< sim::MCShower> >();
  produces< std::vector< sim::MCTrack>  >();
  // Call appropriate produces<>() functions here.

//MCReco::~MCReco()
//{
  // Clean up dynamic memory and other resources here.
//}
}

void MCReco::produce(art::Event & evt)
{
//  std::unique_ptr< std::vector<sim::MCTrack> > outTrackArray(new std::vector<sim::MCTrack>);

  // Retrieve mcparticles
  art::Handle<std::vector<simb::MCParticle> > mcpHandle;
  evt.getByLabel(fMCParticleLabel,mcpHandle);
  if(!mcpHandle.isValid()) throw cet::exception(__FUNCTION__) << "Failed to retrieve simb::MCParticle";;

  // Find associations
  art::FindOneP<simb::MCTruth> ass(mcpHandle, evt, fMCParticleLabel);
  std::vector<simb::Origin_t> orig_array;
  orig_array.reserve(mcpHandle->size());
  for(size_t i=0; i<mcpHandle->size(); ++i) {
    const art::Ptr<simb::MCTruth> &mct = ass.at(i);
    orig_array.push_back(mct->Origin());
  }

  // Retrieve SimChannel
  art::Handle<std::vector<sim::SimChannel> > schHandle;
  evt.getByLabel(fSimChannelLabel,schHandle);
  if(!schHandle.isValid()) throw cet::exception(__FUNCTION__) << "Failed to retrieve sim::SimChannel";

  const std::vector<simb::MCParticle>& mcp_array(*mcpHandle);
  fPart.AddParticles(mcp_array,orig_array);

  const std::vector<sim::SimChannel>&  sch_array(*schHandle);
  fEdep.MakeMCEdep(sch_array);

  //Add MCShowers and MCTracks to the event 
  evt.put(fMCSAlg.Reconstruct(fPart,fEdep));
  evt.put(fMCTAlg.Reconstruct(fPart,fEdep));
  
  fEdep.Clear();
  fPart.clear();
}

DEFINE_ART_MODULE(MCReco)