trkf::VertexFitter Class Reference

Module for fitting a vertex using the Geometric3DVertexFitter. More...

Inheritance diagram for trkf::VertexFitter:

Classes
struct	Config
struct	Inputs

Public Types
using	Parameters = art::EDProducer::Table< Config >
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer
using	WorkerType = WorkerT< EDProducer >
Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >
Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >

Public Member Functions
	VertexFitter (Parameters const &p)
	VertexFitter (VertexFitter const &)=delete
	VertexFitter (VertexFitter &&)=delete
VertexFitter &	operator= (VertexFitter const &)=delete
VertexFitter &	operator= (VertexFitter &&)=delete
Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDProducer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept
	Producer (fhicl::ParameterSet const &)
	Producer (Producer const &)=delete
	Producer (Producer &&)=delete
Producer &	operator= (Producer const &)=delete
Producer &	operator= (Producer &&)=delete
void	doBeginJob (SharedResources const &resources)
void	doEndJob ()
void	doRespondToOpenInputFile (FileBlock const &fb)
void	doRespondToCloseInputFile (FileBlock const &fb)
void	doRespondToOpenOutputFiles (FileBlock const &fb)
void	doRespondToCloseOutputFiles (FileBlock const &fb)
bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)
Public Member Functions inherited from art::Modifier
	~Modifier () noexcept
	Modifier ()
	Modifier (Modifier const &)=delete
	Modifier (Modifier &&)=delete
Modifier &	operator= (Modifier const &)=delete
Modifier &	operator= (Modifier &&)=delete
Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept
	ModuleBase ()
ModuleDescription const &	moduleDescription () const
void	setModuleDescription (ModuleDescription const &)
std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const
void	sortConsumables (std::string const &current_process_name)
template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)
template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Member Functions
void	produce (art::Event &e) override

Private Attributes
art::InputTag	pfParticleInputTag
art::InputTag	trackInputTag
Geometric3DVertexFitter	fitter

Additional Inherited Members
Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)
Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()
template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)
template<typename T , BranchType = InEvent>
void	consumesMany ()
template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)
template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Module for fitting a vertex using the Geometric3DVertexFitter.

It selects primary PFParticles, and then collects all tracks associated to its daughters; if at least 2 tracks are found, they are passed to the vertex fitter.

Inputs are: a PFParticle collection, and the associated tracks.

Outputs are: vector of recob::Vertex, Assns of (neutrino) recob::PFParticle to recob::Vertex, Assns of recob::Vertex and recob::Track with recob::VertexAssnMeta.

For configuration options see Geometric3DVertexFitter::Parameters

Author

G. Cerati (FNAL, MicroBooNE)

Date

2017

Version

1.0

Definition at line 41 of file VertexFitter_module.cc.

Member Typedef Documentation

using trkf::VertexFitter::Parameters = art::EDProducer::Table<Config>

Definition at line 62 of file VertexFitter_module.cc.

Constructor & Destructor Documentation

trkf::VertexFitter::VertexFitter ( Parameters const &  p )

explicit

Definition at line 81 of file VertexFitter_module.cc.

  : EDProducer{p}
  , pfParticleInputTag(p().inputs().inputPFParticleLabel())
  , trackInputTag(p().inputs().inputTracksLabel())
  , fitter(p().geom3dvtxfit, p().propagator)
{
  produces<std::vector<recob::Vertex>>();
  produces<art::Assns<recob::PFParticle, recob::Vertex>>();
  produces<art::Assns<recob::Vertex, recob::Track, recob::VertexAssnMeta>>();
}

trkf::VertexFitter::VertexFitter ( VertexFitter const &  )

delete

trkf::VertexFitter::VertexFitter ( VertexFitter &&  )

delete

Member Function Documentation

VertexFitter& trkf::VertexFitter::operator= ( VertexFitter const &  )

delete

VertexFitter& trkf::VertexFitter::operator= ( VertexFitter &&  )

delete

void trkf::VertexFitter::produce ( art::Event &  e )

overrideprivatevirtual

Implements art::EDProducer.

Definition at line 93 of file VertexFitter_module.cc.

{
  using namespace std;

  auto outputVertices = make_unique<vector<recob::Vertex>>();
  auto outputPFVxAssn = make_unique<art::Assns<recob::PFParticle, recob::Vertex>>();
  auto outputVxTkMtAssn =
    make_unique<art::Assns<recob::Vertex, recob::Track, recob::VertexAssnMeta>>();

  const auto& inputPFParticle = e.getValidHandle<vector<recob::PFParticle>>(pfParticleInputTag);
  art::FindManyP<recob::Track> const assocTracks{inputPFParticle, e, trackInputTag};

  auto const detProp = art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(e);

  // PtrMakers for Assns
  art::PtrMaker<recob::Vertex> vtxPtrMaker(e);

  for (size_t iPF = 0; iPF < inputPFParticle->size(); ++iPF) {

    art::Ptr<recob::PFParticle> pfp(inputPFParticle, iPF);
    if (pfp->IsPrimary() == false || pfp->NumDaughters() < 2) continue;
    vector<art::Ptr<recob::Track>> tracks;
    auto& pfd = pfp->Daughters();
    for (auto ipfd : pfd) {
      // Daugthers returns the id as in pfp->Self() not the key
      // so need to find the key for the pfp with Self==ipfd
      for (size_t jPF = 0; jPF < inputPFParticle->size(); ++jPF) {
        art::Ptr<recob::PFParticle> pfpd(inputPFParticle, jPF);
        if (pfpd->Self() != ipfd) continue;
        vector<art::Ptr<recob::Track>> pftracks = assocTracks.at(jPF);
        for (auto t : pftracks) {
          tracks.push_back(t);
        }
        break;
      }
    }
    if (tracks.size() < 2) continue;

    VertexWrapper vtx = fitter.fitTracks(detProp, tracks);
    if (vtx.isValid() == false) continue;
    vtx.setVertexId(outputVertices->size());

    auto meta = fitter.computeMeta(detProp, vtx, tracks);

    // Fill the output collections

    outputVertices->emplace_back(vtx.vertex());
    const art::Ptr<recob::Vertex> aptr = vtxPtrMaker(outputVertices->size() - 1);
    outputPFVxAssn->addSingle(art::Ptr<recob::PFParticle>(inputPFParticle, iPF), aptr);

    size_t itt = 0;
    for (auto t : tracks) {
      outputVxTkMtAssn->addSingle(aptr, t, meta[itt]);
      itt++;
    }
  }

  e.put(std::move(outputVertices));
  e.put(std::move(outputPFVxAssn));
  e.put(std::move(outputVxTkMtAssn));
}

Member Data Documentation

Geometric3DVertexFitter trkf::VertexFitter::fitter

private

Definition at line 77 of file VertexFitter_module.cc.

art::InputTag trkf::VertexFitter::pfParticleInputTag

private

Definition at line 75 of file VertexFitter_module.cc.

art::InputTag trkf::VertexFitter::trackInputTag

private

Definition at line 76 of file VertexFitter_module.cc.

---

The documentation for this class was generated from the following file:

• larreco/larreco/VertexFinder/VertexFitter_module.cc