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

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_

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 )

overrideprivate

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));
   }
 }