Produce a reco::Track collection, as a result of the fit of an existing recob::TrackTrajectory collection. More...

Inheritance diagram for TrackProducerFromTrackTrajectory:

Public Member Functions
	TrackProducerFromTrackTrajectory (fhicl::ParameterSet const &p)

	TrackProducerFromTrackTrajectory (TrackProducerFromTrackTrajectory const &)=delete

	TrackProducerFromTrackTrajectory (TrackProducerFromTrackTrajectory &&)=delete

TrackProducerFromTrackTrajectory &	operator= (TrackProducerFromTrackTrajectory const &)=delete

TrackProducerFromTrackTrajectory &	operator= (TrackProducerFromTrackTrajectory &&)=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
std::unique_ptr< trkmkr::TrackMaker >	trackMaker_

art::InputTag	trajInputTag

bool	doTrackFitHitInfo_

bool	doSpacePoints_

bool	spacePointsFromTrajP_

Additional Inherited Members
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 >

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

Produce a reco::Track collection, as a result of the fit of an existing recob::TrackTrajectory collection.

This producer takes an input an existing recob::TrackTrajectory collection (and the associated hits) and fits it. The mandatory output are: the resulting recob::Track collection, the associated hits, and the association between the input TrackTrajectory and the output Track. Optional outputs are recob::TrackFitHitInfo and recob::SpacePoint collections, plus the Assns of SpacePoints to Hits. An option is provided to create SpacePoints from the TrajectoryPoints in the Track. The fit is performed by an user-defined tool, which must inherit from larreco/TrackFinder/TrackMaker.

Parameters: trackMaker (fhicl::ParameterSet for the trkmkr::TrackMaker tool used to do the fit), inputCollection (art::InputTag of the input recob::TrackTrajectory collection), doTrackFitHitInfo (bool to decide whether to produce recob::TrackFitHitInfo's), doSpacePoints (bool to decide whether to produce recob::SpacePoint's), and spacePointsFromTrajP (bool to decide whether the produced recob::SpacePoint's are taken from the recob::tracking::TrajectoryPoint_t's of the fitted recob::Track).

Author: G. Cerati (FNAL, MicroBooNE)

Date: 2017

Version: 1.0

Definition at line 40 of file TrackProducerFromTrackTrajectory_module.cc.

Constructor & Destructor Documentation

TrackProducerFromTrackTrajectory::TrackProducerFromTrackTrajectory ( fhicl::ParameterSet const & p )

explicit

Definition at line 62 of file TrackProducerFromTrackTrajectory_module.cc.

   : EDProducer{p}
   , trackMaker_{art::make_tool<trkmkr::TrackMaker>(p.get<fhicl::ParameterSet>("trackMaker"))}
   , trajInputTag{p.get<art::InputTag>("inputCollection")}
   , doTrackFitHitInfo_{p.get<bool>("doTrackFitHitInfo")}
   , doSpacePoints_{p.get<bool>("doSpacePoints")}
   , spacePointsFromTrajP_{p.get<bool>("spacePointsFromTrajP")}
 {
   produces<std::vector<recob::Track>>();
   produces<art::Assns<recob::Track, recob::Hit>>();
   produces<art::Assns<recob::TrackTrajectory, recob::Track>>();
   if (doTrackFitHitInfo_) produces<std::vector<std::vector<recob::TrackFitHitInfo>>>();
   if (doSpacePoints_) {
     produces<std::vector<recob::SpacePoint>>();
     produces<art::Assns<recob::Hit, recob::SpacePoint>>();
   }
 }

TrackProducerFromTrackTrajectory::TrackProducerFromTrackTrajectory ( TrackProducerFromTrackTrajectory const & )

delete

TrackProducerFromTrackTrajectory::TrackProducerFromTrackTrajectory ( TrackProducerFromTrackTrajectory && )

delete

Member Function Documentation

TrackProducerFromTrackTrajectory& TrackProducerFromTrackTrajectory::operator= ( TrackProducerFromTrackTrajectory const & )

delete

TrackProducerFromTrackTrajectory& TrackProducerFromTrackTrajectory::operator= ( TrackProducerFromTrackTrajectory && )

delete

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

overrideprivatevirtual

Implements art::EDProducer.

Definition at line 81 of file TrackProducerFromTrackTrajectory_module.cc.

 {
   // Output collections
   auto outputTracks = std::make_unique<std::vector<recob::Track>>();
   auto outputHits = std::make_unique<art::Assns<recob::Track, recob::Hit>>();
   auto outputTTjTAssn = std::make_unique<art::Assns<recob::TrackTrajectory, recob::Track>>();
   auto outputHitInfo = std::make_unique<std::vector<std::vector<recob::TrackFitHitInfo>>>();
   auto outputSpacePoints = std::make_unique<std::vector<recob::SpacePoint>>();
   auto outputHitSpacePointAssn = std::make_unique<art::Assns<recob::Hit, recob::SpacePoint>>();
 
   // PtrMakers for Assns
   art::PtrMaker<recob::Track> trackPtrMaker(e);
   art::PtrMaker<recob::SpacePoint> spacePointPtrMaker(e);
 
   // Input from event
   art::ValidHandle<std::vector<recob::TrackTrajectory>> inputTrajs =
     e.getValidHandle<std::vector<recob::TrackTrajectory>>(trajInputTag);
   auto const& tjHitsAssn =
     *e.getValidHandle<art::Assns<recob::TrackTrajectory, recob::Hit>>(trajInputTag);
   const auto& trajectoriesWithHits = util::associated_groups(tjHitsAssn);
 
   auto const detProp = art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(e);
 
   // Initialize tool for this event
   trackMaker_->initEvent(e);
 
   // Loop over trajectories to fit
   unsigned int iTraj = 0;
   for (auto hitsRange : trajectoriesWithHits) {
 
     // Get track and its hits
     art::Ptr<recob::TrackTrajectory> traj(inputTrajs, iTraj++);
     std::vector<art::Ptr<recob::Hit>> inHits;
     for (art::Ptr<recob::Hit> const& hit : hitsRange)
       inHits.push_back(hit);
 
     // Declare output objects
     recob::Track outTrack;
     std::vector<art::Ptr<recob::Hit>> outHits;
     trkmkr::OptionalOutputs optionals;
     if (doTrackFitHitInfo_) optionals.initTrackFitInfos();
     if (doSpacePoints_ && !spacePointsFromTrajP_) optionals.initSpacePoints();
 
     // Invoke tool to fit track and fill output objects
     bool fitok = trackMaker_->makeTrack(detProp, traj, inHits, outTrack, outHits, optionals);
     if (!fitok) continue;
 
     // Check that the requirement Nhits == Npoints is satisfied
     // We also require the hits to the in the same order as the points (this cannot be enforced, can it?)
     if (outTrack.NumberTrajectoryPoints() != outHits.size()) {
       throw cet::exception("TrackProducerFromTrackTrajectory")
         << "Produced recob::Track required to have 1-1 correspondance between hits and points.\n";
     }
 
     // Fill output collections, including Assns
     outputTracks->emplace_back(std::move(outTrack));
     const art::Ptr<recob::Track> aptr = trackPtrMaker(outputTracks->size() - 1);
     outputTTjTAssn->addSingle(traj, aptr);
     unsigned int ip = 0;
     for (auto const& trhit : outHits) {
       outputHits->addSingle(aptr, trhit);
 
       if (doSpacePoints_ && spacePointsFromTrajP_ && outputTracks->back().HasValidPoint(ip)) {
         auto& tp = outputTracks->back().Trajectory().LocationAtPoint(ip);
         const double fXYZ[3] = {tp.X(), tp.Y(), tp.Z()};
         const double fErrXYZ[6] = {0};
         recob::SpacePoint sp(fXYZ, fErrXYZ, -1.);
         outputSpacePoints->emplace_back(std::move(sp));
         const art::Ptr<recob::SpacePoint> apsp = spacePointPtrMaker(outputSpacePoints->size() - 1);
         outputHitSpacePointAssn->addSingle(trhit, apsp);
       }
       ip++;
     }
     if (doSpacePoints_ && !spacePointsFromTrajP_) {
       auto osp = optionals.spacePointHitPairs();
       for (auto it = osp.begin(); it != osp.end(); ++it) {
         outputSpacePoints->emplace_back(std::move(it->first));
         const art::Ptr<recob::SpacePoint> apsp = spacePointPtrMaker(outputSpacePoints->size() - 1);
         outputHitSpacePointAssn->addSingle(it->second, apsp);
       }
     }
     if (doTrackFitHitInfo_) { outputHitInfo->emplace_back(optionals.trackFitHitInfos()); }
   }
 
   // Put collections in the event
   e.put(std::move(outputTracks));
   e.put(std::move(outputHits));
   e.put(std::move(outputTTjTAssn));
   if (doTrackFitHitInfo_) { e.put(std::move(outputHitInfo)); }
   if (doSpacePoints_) {
     e.put(std::move(outputSpacePoints));
     e.put(std::move(outputHitSpacePointAssn));
   }
 }