Inheritance diagram for trkf::KalmanFilterTrajectoryFitter:

Classes
struct	Config

struct	Inputs

struct	Options

Public Types
using	Parameters = art::EDProducer::Table< Config >

using	ModuleType = EDProducer

template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >

Public Member Functions
	KalmanFilterTrajectoryFitter (Parameters const &p)

	KalmanFilterTrajectoryFitter (KalmanFilterTrajectoryFitter const &)=delete

	KalmanFilterTrajectoryFitter (KalmanFilterTrajectoryFitter &&)=delete

KalmanFilterTrajectoryFitter &	operator= (KalmanFilterTrajectoryFitter const &)=delete

KalmanFilterTrajectoryFitter &	operator= (KalmanFilterTrajectoryFitter &&)=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)

void	fillProductDescriptions ()

void	registerProducts (ProductDescriptions &productsToRegister)

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)

std::unique_ptr< Worker >	makeWorker (WorkerParams const &wp)

template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)

template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Protected Member Functions
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 ()

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

double	setMomValue (const recob::TrackTrajectory *ptraj, const double pMC, const int pId) const

int	setPId () const

bool	setDirFlip (const recob::TrackTrajectory *ptraj, TVector3 &mcdir) const

void	restoreInputPoints (const recob::TrackTrajectory &track, const std::vector< art::Ptr< recob::Hit >> &inHits, recob::Track &outTrack, std::vector< art::Ptr< recob::Hit >> &outHits) const

Private Attributes
Parameters	p_

TrackStatePropagator	prop

trkf::TrackKalmanFitter	kalmanFitter

trkf::TrackMomentumCalculator	tmc {}

art::InputTag	trajectoryInputTag

art::InputTag	simTrackInputTag

bool	isTT

Detailed Description

Definition at line 34 of file KalmanFilterTrajectoryFitter_module.cc.

Member Typedef Documentation

using art::EDProducer::ModuleType = EDProducer

inherited

Definition at line 17 of file EDProducer.h.

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

Definition at line 104 of file KalmanFilterTrajectoryFitter_module.cc.

template<typename UserConfig , typename KeysToIgnore = void>

using art::detail::Producer::Table = Modifier::Table<UserConfig, KeysToIgnore>

inherited

Definition at line 26 of file Producer.h.

Constructor & Destructor Documentation

KalmanFilterTrajectoryFitter::KalmanFilterTrajectoryFitter ( Parameters const & p )

explicit

Definition at line 138 of file KalmanFilterTrajectoryFitter_module.cc.

References isTT, kalmanFitter, p_, prop, simTrackInputTag, and trajectoryInputTag.

   : EDProducer{p}
   , p_(p)
   , prop{p_().propagator}
   , kalmanFitter{&prop, p_().fitter}
   , trajectoryInputTag{p_().inputs().inputTrajectoryLabel()}
 {
   if (p_().options().pFromMC() || p_().options().dirFromMC())
     simTrackInputTag = art::InputTag{p_().inputs().inputMCLabel()};
 
   isTT = p_().inputs().isTrackTrajectory();
 
   produces<std::vector<recob::Track>>();
   produces<art::Assns<recob::Track, recob::Hit>>();
   produces<art::Assns<recob::Track, recob::Hit, recob::TrackHitMeta>>();
   if (isTT) { produces<art::Assns<recob::TrackTrajectory, recob::Track>>(); }
   else {
     produces<art::Assns<recob::Trajectory, recob::Track>>();
   }
   if (p_().options().produceTrackFitHitInfo()) {
     produces<std::vector<std::vector<recob::TrackFitHitInfo>>>();
   }
   if (p_().options().produceSpacePoints()) {
     produces<std::vector<recob::SpacePoint>>();
     produces<art::Assns<recob::Hit, recob::SpacePoint>>();
   }
 
   //throw expections to avoid possible silent failures due to incompatible configuration options
 
   unsigned int nDirs = 0;
   if (p_().options().dirFromMC()) nDirs++;
   if (p_().options().dirFromVec()) nDirs++;
   if (p_().options().alwaysInvertDir()) nDirs++;
   if (nDirs > 1) {
     throw cet::exception("KalmanFilterTrajectoryFitter")
       << "Incompatible configuration parameters: only at most one can be set to true among "
          "dirFromMC, dirFromVec, and alwaysInvertDir."
       << "\n";
   }
 
   unsigned int nPFroms = 0;
   if (p_().options().pFromLength()) nPFroms++;
   if (p_().options().pFromMC()) nPFroms++;
   if (nPFroms > 1) {
     throw cet::exception("KalmanFilterTrajectoryFitter")
       << "Incompatible configuration parameters: only at most one can be set to true among "
          "pFromLength, and pFromMC."
       << "\n"; //pFromMSChi2,
   }
 
   if (p_().options().keepInputTrajectoryPoints()) {
     if (p_().fitter().sortHitsByPlane() || p_().fitter().sortOutputHitsMinLength() ||
         p_().fitter().skipNegProp()) {
       throw cet::exception("KalmanFilterTrajectoryFitter")
         << "Incompatible configuration parameters: keepInputTrajectoryPoints needs the following "
            "fitter options all set to false: sortHitsByPlane, sortOutputHitsMinLength, skipNegProp."
         << "\n";
     }
   }
 }

trkf::KalmanFilterTrajectoryFitter::KalmanFilterTrajectoryFitter ( KalmanFilterTrajectoryFitter const & )

delete

trkf::KalmanFilterTrajectoryFitter::KalmanFilterTrajectoryFitter ( KalmanFilterTrajectoryFitter && )

delete

Member Function Documentation

template<typename T , BranchType BT>

ProductToken< T > art::ModuleBase::consumes ( InputTag const & tag )

protectedinherited

Definition at line 61 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::consumes().

   {
     return collector_.consumes<T, BT>(tag);
   }

ConsumesCollector & art::ModuleBase::consumesCollector ( )

protectedinherited

Definition at line 57 of file ModuleBase.cc.

References art::ModuleBase::collector_.

   {
     return collector_;
   }

template<typename T , BranchType BT>

void art::ModuleBase::consumesMany ( )

protectedinherited

Definition at line 75 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::consumesMany().

   {
     collector_.consumesMany<T, BT>();
   }

template<typename Element , BranchType = InEvent>

ViewToken<Element> art::ModuleBase::consumesView ( InputTag const & )

protectedinherited

template<typename T , BranchType BT>

ViewToken<T> art::ModuleBase::consumesView ( InputTag const & tag )

inherited

Definition at line 68 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::consumesView().

   {
     return collector_.consumesView<T, BT>(tag);
   }

void art::detail::Producer::doBeginJob ( SharedResources const & resources )

inherited

Definition at line 22 of file Producer.cc.

References art::detail::Producer::beginJobWithFrame(), and art::detail::Producer::setupQueues().

   {
     setupQueues(resources);
     ProcessingFrame const frame{ScheduleID{}};
     beginJobWithFrame(frame);
   }

bool art::detail::Producer::doBeginRun	(	RunPrincipal &	rp,
		ModuleContext const &	mc
	)

inherited

Definition at line 65 of file Producer.cc.

References art::detail::Producer::beginRunWithFrame(), art::RangeSet::forRun(), art::RunPrincipal::makeRun(), r, art::RunPrincipal::runID(), and art::ModuleContext::scheduleID().

   {
     auto r = rp.makeRun(mc, RangeSet::forRun(rp.runID()));
     ProcessingFrame const frame{mc.scheduleID()};
     beginRunWithFrame(r, frame);
     r.commitProducts();
     return true;
   }

bool art::detail::Producer::doBeginSubRun	(	SubRunPrincipal &	srp,
		ModuleContext const &	mc
	)

inherited

Definition at line 85 of file Producer.cc.

References art::detail::Producer::beginSubRunWithFrame(), art::RangeSet::forSubRun(), art::SubRunPrincipal::makeSubRun(), art::ModuleContext::scheduleID(), and art::SubRunPrincipal::subRunID().

   {
     auto sr = srp.makeSubRun(mc, RangeSet::forSubRun(srp.subRunID()));
     ProcessingFrame const frame{mc.scheduleID()};
     beginSubRunWithFrame(sr, frame);
     sr.commitProducts();
     return true;
   }

void art::detail::Producer::doEndJob ( )

inherited

Definition at line 30 of file Producer.cc.

References art::detail::Producer::endJobWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     endJobWithFrame(frame);
   }

bool art::detail::Producer::doEndRun	(	RunPrincipal &	rp,
		ModuleContext const &	mc
	)

inherited

Definition at line 75 of file Producer.cc.

References art::detail::Producer::endRunWithFrame(), art::RunPrincipal::makeRun(), r, art::ModuleContext::scheduleID(), and art::Principal::seenRanges().

   {
     auto r = rp.makeRun(mc, rp.seenRanges());
     ProcessingFrame const frame{mc.scheduleID()};
     endRunWithFrame(r, frame);
     r.commitProducts();
     return true;
   }

bool art::detail::Producer::doEndSubRun	(	SubRunPrincipal &	srp,
		ModuleContext const &	mc
	)

inherited

Definition at line 95 of file Producer.cc.

References art::detail::Producer::endSubRunWithFrame(), art::SubRunPrincipal::makeSubRun(), art::ModuleContext::scheduleID(), and art::Principal::seenRanges().

   {
     auto sr = srp.makeSubRun(mc, srp.seenRanges());
     ProcessingFrame const frame{mc.scheduleID()};
     endSubRunWithFrame(sr, frame);
     sr.commitProducts();
     return true;
   }

bool art::detail::Producer::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
	)

inherited

Definition at line 105 of file Producer.cc.

References art::detail::Producer::checkPutProducts_, e, art::EventPrincipal::makeEvent(), art::detail::Producer::produceWithFrame(), and art::ModuleContext::scheduleID().

   {
     auto e = ep.makeEvent(mc);
     ++counts_run;
     ProcessingFrame const frame{mc.scheduleID()};
     produceWithFrame(e, frame);
     e.commitProducts(checkPutProducts_, &expectedProducts<InEvent>());
     ++counts_passed;
     return true;
   }

void art::detail::Producer::doRespondToCloseInputFile ( FileBlock const & fb )

inherited

Definition at line 44 of file Producer.cc.

References art::detail::Producer::respondToCloseInputFileWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToCloseInputFileWithFrame(fb, frame);
   }

void art::detail::Producer::doRespondToCloseOutputFiles ( FileBlock const & fb )

inherited

Definition at line 58 of file Producer.cc.

References art::detail::Producer::respondToCloseOutputFilesWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToCloseOutputFilesWithFrame(fb, frame);
   }

void art::detail::Producer::doRespondToOpenInputFile ( FileBlock const & fb )

inherited

Definition at line 37 of file Producer.cc.

References art::detail::Producer::respondToOpenInputFileWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToOpenInputFileWithFrame(fb, frame);
   }

void art::detail::Producer::doRespondToOpenOutputFiles ( FileBlock const & fb )

inherited

Definition at line 51 of file Producer.cc.

References art::detail::Producer::respondToOpenOutputFilesWithFrame().

   {
     ProcessingFrame const frame{ScheduleID{}};
     respondToOpenOutputFilesWithFrame(fb, frame);
   }

void art::Modifier::fillProductDescriptions ( )

inherited

Definition at line 10 of file Modifier.cc.

References art::ProductRegistryHelper::fillDescriptions(), and art::ModuleBase::moduleDescription().

   {
     ProductRegistryHelper::fillDescriptions(moduleDescription());
   }

std::array< std::vector< ProductInfo >, NumBranchTypes > const & art::ModuleBase::getConsumables ( ) const

inherited

Definition at line 43 of file ModuleBase.cc.

References art::ModuleBase::collector_, and art::ConsumesCollector::getConsumables().

   {
     return collector_.getConsumables();
   }

std::unique_ptr< Worker > art::ModuleBase::makeWorker ( WorkerParams const & wp )

inherited

Definition at line 37 of file ModuleBase.cc.

References art::ModuleBase::doMakeWorker(), and art::NumBranchTypes.

   {
     return doMakeWorker(wp);
   }

template<typename T , BranchType BT>

ProductToken< T > art::ModuleBase::mayConsume ( InputTag const & tag )

protectedinherited

Definition at line 82 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::mayConsume().

   {
     return collector_.mayConsume<T, BT>(tag);
   }

template<typename T , BranchType BT>

void art::ModuleBase::mayConsumeMany ( )

protectedinherited

Definition at line 96 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::mayConsumeMany().

   {
     collector_.mayConsumeMany<T, BT>();
   }

template<typename Element , BranchType = InEvent>

ViewToken<Element> art::ModuleBase::mayConsumeView ( InputTag const & )

protectedinherited

template<typename T , BranchType BT>

ViewToken<T> art::ModuleBase::mayConsumeView ( InputTag const & tag )

inherited

Definition at line 89 of file ModuleBase.h.

References art::ModuleBase::collector_, and art::ConsumesCollector::mayConsumeView().

   {
     return collector_.mayConsumeView<T, BT>(tag);
   }

ModuleDescription const & art::ModuleBase::moduleDescription ( ) const

inherited

Definition at line 13 of file ModuleBase.cc.

References art::errors::LogicError.

Referenced by art::OutputModule::doRespondToOpenInputFile(), art::OutputModule::doWriteEvent(), art::Modifier::fillProductDescriptions(), art::OutputModule::makePlugins_(), art::OutputWorker::OutputWorker(), reco::shower::LArPandoraModularShowerCreation::produce(), art::Modifier::registerProducts(), and art::OutputModule::registerProducts().

   {
     if (md_.has_value()) {
       return *md_;
     }
 
     throw Exception{errors::LogicError,
                     "There was an error while calling moduleDescription().\n"}
       << "The moduleDescription() base-class member function cannot be called\n"
          "during module construction.  To determine which module is "
          "responsible\n"
          "for calling it, find the '<module type>:<module "
          "label>@Construction'\n"
          "tag in the message prefix above.  Please contact artists@fnal.gov\n"
          "for guidance.\n";
   }

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

delete

KalmanFilterTrajectoryFitter& trkf::KalmanFilterTrajectoryFitter::operator= ( KalmanFilterTrajectoryFitter && )

delete

void KalmanFilterTrajectoryFitter::produce ( art::Event & e )

overrideprivatevirtual

Implements art::EDProducer.

Definition at line 200 of file KalmanFilterTrajectoryFitter_module.cc.

References art::Assns< L, R, D >::begin(), art::Assns< L, R, D >::end(), trkf::TrackKalmanFitter::fitTrack(), art::ProductRetriever::getByLabel(), art::ProductRetriever::getProductID(), art::ProductRetriever::getValidHandle(), trkmkr::OptionalOutputs::initTrackFitInfos(), isTT, kalmanFitter, sim::MCStep::Momentum(), p_, sim::MCTrack::PdgCode(), sim::MCTrack::Process(), art::Handle< T >::product(), art::ProductRetriever::productGetter(), art::Event::put(), restoreInputPoints(), setDirFlip(), setMomValue(), setPId(), simTrackInputTag, sim::MCTrack::Start(), trkmkr::OptionalOutputs::trackFitHitInfos(), and trajectoryInputTag.

 {
 
   auto outputTracks = std::make_unique<std::vector<recob::Track>>();
   auto outputHitsMeta =
     std::make_unique<art::Assns<recob::Track, recob::Hit, recob::TrackHitMeta>>();
   auto outputHits = std::make_unique<art::Assns<recob::Track, recob::Hit>>();
   auto outputHitInfo = std::make_unique<std::vector<std::vector<recob::TrackFitHitInfo>>>();
 
   //only one will be filled and pushed into the event:
   auto outputTTjTAssn = std::make_unique<art::Assns<recob::TrackTrajectory, recob::Track>>();
   auto outputTjTAssn = std::make_unique<art::Assns<recob::Trajectory, recob::Track>>();
 
   auto const tid = e.getProductID<std::vector<recob::Track>>();
   auto const tidgetter = e.productGetter(tid);
 
   auto outputSpacePoints = std::make_unique<std::vector<recob::SpacePoint>>();
   auto outputHitSpacePointAssn = std::make_unique<art::Assns<recob::Hit, recob::SpacePoint>>();
   auto const spid = e.getProductID<std::vector<recob::SpacePoint>>();
   auto const spidgetter = e.productGetter(spid);
 
   //FIXME, eventually remove this (ok only for single particle MC)
   double pMC = -1.;
   TVector3 mcdir;
   if (p_().options().pFromMC() || p_().options().dirFromMC()) {
     art::ValidHandle<std::vector<sim::MCTrack>> simTracks =
       e.getValidHandle<std::vector<sim::MCTrack>>(simTrackInputTag);
     for (unsigned int iMC = 0; iMC < simTracks->size(); ++iMC) {
       const sim::MCTrack& mctrack = simTracks->at(iMC);
       //fiducial cuts on MC tracks
       if (mctrack.PdgCode() != 13) continue;
       if (mctrack.Process() != "primary") continue;
       pMC = mctrack.Start().Momentum().P() * 0.001;
       mcdir = TVector3(mctrack.Start().Momentum().X() * 0.001 / pMC,
                        mctrack.Start().Momentum().Y() * 0.001 / pMC,
                        mctrack.Start().Momentum().Z() * 0.001 / pMC);
       break;
     }
     //std::cout << "mc momentum value = " << pval << " GeV" << std::endl;
   }
 
   unsigned int nTrajs = 0;
 
   art::Handle<std::vector<recob::TrackTrajectory>> inputTrackTrajectoryH;
   art::Handle<std::vector<recob::Trajectory>> inputTrajectoryH;
   const std::vector<recob::TrackTrajectory>* trackTrajectoryVec = nullptr;
   const std::vector<recob::Trajectory>* trajectoryVec = nullptr;
   const art::Assns<recob::TrackTrajectory, recob::Hit>* trackTrajectoryHitsAssn = nullptr;
   const art::Assns<recob::Trajectory, recob::Hit>* trajectoryHitsAssn = nullptr;
   if (isTT) {
     bool ok = e.getByLabel(trajectoryInputTag, inputTrackTrajectoryH);
     if (!ok)
       throw cet::exception("KalmanFilterTrajectoryFitter")
         << "Cannot find recob::TrackTrajectory art::Handle with inputTag " << trajectoryInputTag;
     trackTrajectoryVec = inputTrackTrajectoryH.product();
     trackTrajectoryHitsAssn =
       e.getValidHandle<art::Assns<recob::TrackTrajectory, recob::Hit>>(trajectoryInputTag)
         .product();
     nTrajs = trackTrajectoryVec->size();
   }
   else {
     bool ok = e.getByLabel(trajectoryInputTag, inputTrajectoryH);
     if (!ok)
       throw cet::exception("KalmanFilterTrajectoryFitter")
         << "Cannot find recob::Trajectory art::Handle with inputTag " << trajectoryInputTag;
     trajectoryVec = inputTrajectoryH.product();
     trajectoryHitsAssn =
       e.getValidHandle<art::Assns<recob::Trajectory, recob::Hit>>(trajectoryInputTag).product();
     nTrajs = trajectoryVec->size();
   }
 
   auto const detProp = art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(e);
 
   for (unsigned int iTraj = 0; iTraj < nTrajs; ++iTraj) {
 
     const recob::TrackTrajectory& inTraj =
       (isTT ? trackTrajectoryVec->at(iTraj) :
               recob::TrackTrajectory(trajectoryVec->at(iTraj),
                                      std::vector<recob::TrajectoryPointFlags>()));
     // this is not computationally optimal, but at least preserves the order
     // unlike FindManyP
     std::vector<art::Ptr<recob::Hit>> inHits;
     if (isTT) {
       for (auto it = trackTrajectoryHitsAssn->begin(); it != trackTrajectoryHitsAssn->end(); ++it) {
         if (it->first.key() == iTraj)
           inHits.push_back(it->second);
         else if (inHits.size() > 0)
           break;
       }
     }
     else {
       for (auto it = trajectoryHitsAssn->begin(); it != trajectoryHitsAssn->end(); ++it) {
         if (it->first.key() == iTraj)
           inHits.push_back(it->second);
         else if (inHits.size() > 0)
           break;
       }
     }
     const int pId = setPId();
     const double mom = setMomValue(&inTraj, pMC, pId);
     const bool flipDir = setDirFlip(&inTraj, mcdir);
 
     recob::Track outTrack;
     std::vector<art::Ptr<recob::Hit>> outHits;
     trkmkr::OptionalOutputs optionals;
     if (p_().options().produceTrackFitHitInfo()) optionals.initTrackFitInfos();
     bool fitok = kalmanFitter.fitTrack(detProp,
                                        inTraj,
                                        iTraj,
                                        SMatrixSym55(),
                                        SMatrixSym55(),
                                        inHits, // inFlags,
                                        mom,
                                        pId,
                                        flipDir,
                                        outTrack,
                                        outHits,
                                        optionals);
     if (!fitok) continue;
 
     if (p_().options().keepInputTrajectoryPoints()) {
       restoreInputPoints(inTraj, inHits, outTrack, outHits);
     }
 
     outputTracks->emplace_back(std::move(outTrack));
     art::Ptr<recob::Track> aptr(tid, outputTracks->size() - 1, tidgetter);
     unsigned int ip = 0;
     for (auto const& trhit : outHits) {
       //the fitter produces collections with 1-1 match between hits and point
       recob::TrackHitMeta metadata(ip, -1);
       outputHitsMeta->addSingle(aptr, trhit, metadata);
       outputHits->addSingle(aptr, trhit);
       if (p_().options().produceSpacePoints() && outputTracks->back().HasValidPoint(ip)) {
         auto& tp = outputTracks->back().Trajectory().LocationAtPoint(ip);
         double fXYZ[3] = {tp.X(), tp.Y(), tp.Z()};
         double fErrXYZ[6] = {0};
         recob::SpacePoint sp(fXYZ, fErrXYZ, -1.);
         outputSpacePoints->emplace_back(std::move(sp));
         art::Ptr<recob::SpacePoint> apsp(spid, outputSpacePoints->size() - 1, spidgetter);
         outputHitSpacePointAssn->addSingle(trhit, apsp);
       }
       ip++;
     }
     outputHitInfo->emplace_back(optionals.trackFitHitInfos());
     if (isTT) {
       outputTTjTAssn->addSingle(art::Ptr<recob::TrackTrajectory>(inputTrackTrajectoryH, iTraj),
                                 aptr);
     }
     else {
       outputTjTAssn->addSingle(art::Ptr<recob::Trajectory>(inputTrajectoryH, iTraj), aptr);
     }
   }
   e.put(std::move(outputTracks));
   e.put(std::move(outputHitsMeta));
   e.put(std::move(outputHits));
   if (p_().options().produceTrackFitHitInfo()) { e.put(std::move(outputHitInfo)); }
   if (p_().options().produceSpacePoints()) {
     e.put(std::move(outputSpacePoints));
     e.put(std::move(outputHitSpacePointAssn));
   }
   if (isTT)
     e.put(std::move(outputTTjTAssn));
   else
     e.put(std::move(outputTjTAssn));
 }

void art::Modifier::registerProducts ( ProductDescriptions & productsToRegister )

inherited

Definition at line 16 of file Modifier.cc.

References art::ModuleBase::moduleDescription(), and art::ProductRegistryHelper::registerProducts().

   {
     ProductRegistryHelper::registerProducts(productsToRegister,
                                             moduleDescription());
   }

void KalmanFilterTrajectoryFitter::restoreInputPoints	(	const recob::TrackTrajectory &	track,
		const std::vector< art::Ptr< recob::Hit >> &	inHits,
		recob::Track &	outTrack,
		std::vector< art::Ptr< recob::Hit >> &	outHits
	)		const

private

Definition at line 366 of file KalmanFilterTrajectoryFitter_module.cc.

References recob::Track::Chi2(), recob::Track::Covariances(), recob::TrackTrajectory::DirectionAtPoint(), recob::Track::FlagsAtPoint(), recob::Track::ID(), recob::TrackTrajectory::LocationAtPoint(), recob::Track::Ndof(), recob::TrajectoryPointFlagTraits::NoPoint, recob::Track::NumberTrajectoryPoints(), recob::Track::ParticleId(), and recob::Track::VertexMomentum().

Referenced by produce().

 {
   const auto np = outTrack.NumberTrajectoryPoints();
   std::vector<Point_t> positions(np);
   std::vector<Vector_t> momenta(np);
   std::vector<recob::TrajectoryPointFlags> outFlags(np);
   //
   for (unsigned int p = 0; p < np; ++p) {
     auto flag = outTrack.FlagsAtPoint(p);
     auto mom = outTrack.VertexMomentum();
     auto op = flag.fromHit();
     positions[op] = track.LocationAtPoint(op);
     momenta[op] = mom * track.DirectionAtPoint(op);
     auto mask = flag.mask();
     if (mask.isSet(recob::TrajectoryPointFlagTraits::NoPoint))
       mask.unset(recob::TrajectoryPointFlagTraits::NoPoint);
     outFlags[op] = recob::TrajectoryPointFlags(op, mask);
   }
   auto covs = outTrack.Covariances();
   outTrack = recob::Track(
     recob::TrackTrajectory(std::move(positions), std::move(momenta), std::move(outFlags), true),
     outTrack.ParticleId(),
     outTrack.Chi2(),
     outTrack.Ndof(),
     std::move(covs.first),
     std::move(covs.second),
     outTrack.ID());
   //
   outHits.clear();
   for (auto h : inHits)
     outHits.push_back(h);
 }

bool KalmanFilterTrajectoryFitter::setDirFlip	(	const recob::TrackTrajectory *	ptraj,
		TVector3 &	mcdir
	)		const

private

Definition at line 421 of file KalmanFilterTrajectoryFitter_module.cc.

References DEFINE_ART_MODULE, dir, p_, and recob::TrackTrajectory::VertexDirection().

Referenced by produce().

 {
   bool result = false;
   if (p_().options().alwaysInvertDir()) { return true; }
   else if (p_().options().dirFromMC()) {
     auto tdir = ptraj->VertexDirection();
     if ((mcdir.X() * tdir.X() + mcdir.Y() * tdir.Y() + mcdir.Z() * tdir.Z()) < 0.) result = true;
   }
   else if (p_().options().dirFromVec()) {
     std::array<float, 3> dir = p_().options().dirVec();
     auto tdir = ptraj->VertexDirection();
     if ((dir[0] * tdir.X() + dir[1] * tdir.Y() + dir[2] * tdir.Z()) < 0.) result = true;
   }
   return result;
 }

void art::ModuleBase::setModuleDescription ( ModuleDescription const & md )

inherited

Definition at line 31 of file ModuleBase.cc.

References art::ModuleBase::md_.

   {
     md_ = md;
   }

double KalmanFilterTrajectoryFitter::setMomValue	(	const recob::TrackTrajectory *	ptraj,
		const double	pMC,
		const int	pId
	)		const

private

Definition at line 403 of file KalmanFilterTrajectoryFitter_module.cc.

References trkf::TrackMomentumCalculator::GetTrackMomentum(), recob::TrackTrajectory::Length(), p_, and tmc.

Referenced by produce().

 {
   double result = p_().options().pval();
   if (p_().options().pFromLength()) { result = tmc.GetTrackMomentum(ptraj->Length(), pId); }
   else if (p_().options().pFromMC() && pMC > 0.) {
     result = pMC;
   }
   return result;
 }

int KalmanFilterTrajectoryFitter::setPId ( ) const

private

Definition at line 415 of file KalmanFilterTrajectoryFitter_module.cc.

References p_.

Referenced by produce().

 {
   int result = p_().options().pdgId();
   return result;
 }

void art::ModuleBase::sortConsumables ( std::string const & current_process_name )

inherited

Definition at line 49 of file ModuleBase.cc.

References art::ModuleBase::collector_, and art::ConsumesCollector::sortConsumables().

   {
     // Now that we know we have seen all the consumes declarations,
     // sort the results for fast lookup later.
     collector_.sortConsumables(current_process_name);
   }