CollectionSplitting_module.cc

Go to the documentation of this file.

#include "art/Framework/Core/EDProducer.h"
#include "art/Framework/Core/ModuleMacros.h"
#include "art/Framework/Principal/Event.h"

#include "fhiclcpp/ParameterSet.h"

#include <memory>

namespace lar_pandora
{

class CollectionSplitting;


class CollectionSplitting : public art::EDProducer
{
public:
    explicit CollectionSplitting(fhicl::ParameterSet const & pset);

    CollectionSplitting(CollectionSplitting const &) = delete;
    CollectionSplitting(CollectionSplitting &&) = delete;
    CollectionSplitting & operator = (CollectionSplitting const &) = delete;
    CollectionSplitting & operator = (CollectionSplitting &&) = delete;

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

private:
    std::string     m_InputProducerLabel;           
    std::string     m_TrackProducerLabel;           
    std::string     m_ShowerProducerLabel;          
    std::string     m_HitProducerLabel;             
    bool            m_ShouldProduceNeutrinos;       
    bool            m_ShouldProduceCosmics;         
    bool            m_ShouldProduceT0s;             
};

DEFINE_ART_MODULE(CollectionSplitting)

} // namespace lar_pandora

//------------------------------------------------------------------------------------------------------------------------------------------
// implementation follows

#include "lardataobj/RecoBase/SpacePoint.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "lardataobj/RecoBase/Vertex.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardataobj/RecoBase/Shower.h"
#include "lardataobj/RecoBase/PCAxis.h"
#include "lardataobj/RecoBase/Hit.h"

#include "larpandora/LArPandoraEventBuilding/LArPandoraEvent.h"

namespace lar_pandora
{

CollectionSplitting::CollectionSplitting(fhicl::ParameterSet const &pset) : 
    m_InputProducerLabel(pset.get<std::string>("InputProducerLabel")),
    m_TrackProducerLabel(pset.get<std::string>("TrackProducerLabel")),
    m_ShowerProducerLabel(pset.get<std::string>("ShowerProducerLabel")),
    m_HitProducerLabel(pset.get<std::string>("HitProducerLabel")),
    m_ShouldProduceNeutrinos(pset.get<bool>("ShouldProduceNeutrinos", true)),
    m_ShouldProduceCosmics(pset.get<bool>("ShouldProduceCosmics", true)),
    m_ShouldProduceT0s(pset.get<bool>("ShouldProduceT0s", false))
{
    produces< std::vector<recob::PFParticle> >();
    produces< std::vector<recob::SpacePoint> >();
    produces< std::vector<recob::Cluster> >();
    produces< std::vector<recob::Vertex> >();
    produces< std::vector<recob::Track> >(); 
    produces< std::vector<recob::Shower> >();
    produces< std::vector<recob::PCAxis> >();
    produces< std::vector<larpandoraobj::PFParticleMetadata> >();

    produces< art::Assns<recob::PFParticle, recob::SpacePoint> >();
    produces< art::Assns<recob::PFParticle, recob::Cluster> >();
    produces< art::Assns<recob::PFParticle, recob::Vertex> >();
    produces< art::Assns<recob::PFParticle, recob::Track> >();
    produces< art::Assns<recob::PFParticle, recob::Shower> >();
    produces< art::Assns<recob::PFParticle, recob::PCAxis> >();
    produces< art::Assns<recob::PFParticle, larpandoraobj::PFParticleMetadata> >();
    produces< art::Assns<recob::Track, recob::Hit> >();
    produces< art::Assns<recob::Shower, recob::Hit> >();
    produces< art::Assns<recob::Shower, recob::PCAxis> >();
    produces< art::Assns<recob::SpacePoint, recob::Hit> >();
    produces< art::Assns<recob::Cluster, recob::Hit> >();

    if (m_ShouldProduceT0s)
    {
        produces< std::vector<anab::T0> >();
        produces< art::Assns<recob::PFParticle, anab::T0> >();
    }
}

//------------------------------------------------------------------------------------------------------------------------------------------

void CollectionSplitting::produce(art::Event &evt)
{
    if (!m_ShouldProduceNeutrinos && !m_ShouldProduceCosmics) 
        throw cet::exception("LArPandora") << " CollectionSplitting -- Must be configured to produce neutrinos or cosmics or both.";

    const lar_pandora::LArPandoraEvent::Labels labels(m_InputProducerLabel, m_TrackProducerLabel, m_ShowerProducerLabel, m_HitProducerLabel); 
    const lar_pandora::LArPandoraEvent fullEvent(this, &evt, labels, m_ShouldProduceT0s);

    if (m_ShouldProduceNeutrinos && m_ShouldProduceCosmics)
    {
        fullEvent.WriteToEvent();
    }
    else
    {
        const lar_pandora::LArPandoraEvent filteredEvent(fullEvent.FilterByPdgCode(m_ShouldProduceNeutrinos));
        filteredEvent.WriteToEvent();
    }
}

} // namespace lar_pandora