trkf::TrackStitcher Class Reference

Inheritance diagram for trkf::TrackStitcher:

Public Member Functions
	TrackStitcher (fhicl::ParameterSet const &pset)

Private Member Functions
void	produce (art::Event &evt) override
art::PtrVector< recob::Hit >	GetHitsFromComponentTracks (const art::PtrVector< recob::Track > &, const art::Event &evt)
art::PtrVector< recob::SpacePoint >	GetSpacePointsFromComponentTracks (const art::PtrVector< recob::Track > &, const art::Event &evt)
std::vector< art::Ptr < recob::Hit > >	GetHitsFromAssdSpacePoints (const art::PtrVector< recob::SpacePoint > &, const art::Event &evt, std::vector< std::pair< std::vector< art::Ptr< recob::Hit > >::const_iterator, std::vector< art::Ptr< recob::Hit > >::const_iterator > > &vpi)

Private Attributes
std::string	fTrackModuleLabel
std::string	fSpptModuleLabel
bool	fStizatch
StitchAlg	fStitchAlg

Detailed Description

Definition at line 41 of file TrackStitcher_module.cc.

Constructor & Destructor Documentation

trkf::TrackStitcher::TrackStitcher ( fhicl::ParameterSet const &  pset )

explicit

Definition at line 63 of file TrackStitcher_module.cc.

                                                             :
     EDProducer{pset},
   fStitchAlg(pset.get< fhicl::ParameterSet >("StitchAlg"))
   {
      fTrackModuleLabel    = pset.get< std::string >("TrackModuleLabel");
      fSpptModuleLabel     = pset.get< std::string >("SpptModuleLabel");
      fStizatch            = pset.get< bool >       ("CommonComponentStitch",true);

      produces< std::vector<recob::Track>                  >();
      produces<std::vector<art::PtrVector<recob::Track> >  >();
      produces<art::Assns<recob::Track, recob::Hit>        >();
      produces<art::Assns<recob::Track, recob::SpacePoint> >();
      produces<art::Assns<recob::SpacePoint, recob::Hit>   >();

   }

Member Function Documentation

std::vector< art::Ptr< recob::Hit > > trkf::TrackStitcher::GetHitsFromAssdSpacePoints ( const art::PtrVector< recob::SpacePoint > &  sppts, const art::Event &  evt, std::vector< std::pair< std::vector< art::Ptr< recob::Hit > >::const_iterator, std::vector< art::Ptr< recob::Hit > >::const_iterator > > &  vpi  )

private

Definition at line 247 of file TrackStitcher_module.cc.

   {

      std::vector<art::Ptr<recob::Hit>> hits;
      art::FindManyP<recob::Hit> hitAssns(sppts, evtGHFCT, fSpptModuleLabel);

      size_t start(0), finish(0);
      for (unsigned int ii=0; ii < sppts.size(); ++ii )
      {
         hits.insert(hits.end(),hitAssns.at(ii).begin(), hitAssns.at(ii).end());
         finish = start+(size_t)(hitAssns.at(ii).end() - hitAssns.at(ii).begin());
         std::pair< std::vector<art::Ptr<recob::Hit> >::const_iterator, std::vector<art::Ptr<recob::Hit> >::const_iterator > pithittmp(hitAssns.at(ii).begin(),hitAssns.at(ii).end());
         pithit.push_back(pithittmp);
         start += (finish+1);
      }

      return hits;
   }

art::PtrVector< recob::Hit > trkf::TrackStitcher::GetHitsFromComponentTracks ( const art::PtrVector< recob::Track > &  tcomp, const art::Event &  evt  )

private

Definition at line 217 of file TrackStitcher_module.cc.

   {

      art::PtrVector<recob::Hit> hits;
      art::FindManyP<recob::Hit> hitAssns(tcomp, evtGHFCT, fTrackModuleLabel);

      for (unsigned int ii=0; ii < tcomp.size(); ++ii )
      {
         hits.insert(hits.end(),hitAssns.at(ii).begin(), hitAssns.at(ii).end());
      }


      //    const art::PtrVector<recob::Hit> chits(hits);
      return hits;
   }

art::PtrVector< recob::SpacePoint > trkf::TrackStitcher::GetSpacePointsFromComponentTracks ( const art::PtrVector< recob::Track > &  tcomp, const art::Event &  evt  )

private

Definition at line 233 of file TrackStitcher_module.cc.

   {

      art::PtrVector<recob::SpacePoint> sppts;
      art::FindManyP<recob::SpacePoint> spptAssns(tcomp, evtGHFCT, fTrackModuleLabel);
      for (unsigned int ii=0; ii < tcomp.size(); ++ii )
      {
         sppts.insert(sppts.end(),spptAssns.at(ii).begin(), spptAssns.at(ii).end());
      }

      //    const art::PtrVector<recob::Hit> chits(hits);
      return sppts;
   }

void trkf::TrackStitcher::produce ( art::Event &  evt )

overrideprivate

Definition at line 80 of file TrackStitcher_module.cc.

   {

      // get services
      art::ServiceHandle<geo::Geometry const> geom;

      //////////////////////////////////////////////////////
      // Make a std::unique_ptr<> for the thing you want to put into the event
      //////////////////////////////////////////////////////
      // tcol is the collection of new tracks
      std::unique_ptr<std::vector<recob::Track> > tcol(new std::vector<recob::Track>);
      std::unique_ptr<art::PtrVector<recob::SpacePoint> > scol(new art::PtrVector<recob::SpacePoint>);
      // tvcol is the collection of vectors that comprise each tcol
      std::unique_ptr<std::vector< art::PtrVector<recob::Track> > > tvcol(new std::vector< art::PtrVector<recob::Track> >);
      std::unique_ptr< art::Assns<recob::Track, recob::Hit> > thassn(new art::Assns<recob::Track, recob::Hit>);
      std::unique_ptr< art::Assns<recob::Track, recob::SpacePoint> > tsptassn(new art::Assns<recob::Track, recob::SpacePoint>);
      std::unique_ptr< art::Assns<recob::SpacePoint, recob::Hit > > spthassn(new art::Assns<recob::SpacePoint, recob::Hit>);


      // Get the original Spacepoints. Trackers other than CosmicTracker wrote the
      // SpacePoints as a PtrVec of vecs. If they look like that, flatten into one vec.

      art::Handle< std::vector < recob::SpacePoint > > sppth;
      try
      {
         mf::LogWarning("TrackStitcher") << "Trying to read Track3DKalmanXYZ-style PtrVector of std::vector of SpacePoints" << std::endl;
         art::Handle< std::vector < art::PtrVector <recob::SpacePoint> > > sppth;
         evt.getByLabel(fSpptModuleLabel, sppth);
         for (size_t ii=0; ii<sppth->size() ;ii++)
            for (size_t jj=0; jj<sppth->at(ii).size() ;ii++)
            {
               art::Ptr<recob::SpacePoint> sptmp(sppth->at(ii).at(jj));
               scol->push_back(sptmp );
            }
      }
      catch(...)
      {
         mf::LogWarning("TrackStitcher") << "Trying instead to read CosmicTracker-style already-flattened vector of SpacePoints" << std::endl;
         art::Handle< std::vector < recob::SpacePoint > > sppthf;
         evt.getByLabel(fSpptModuleLabel, sppthf);
         for (size_t ii=0; ii<sppthf->size() ;ii++)
         {
            art::Ptr<recob::SpacePoint> sptmpf(sppthf,ii);
            scol->push_back(sptmpf);
         }

      }


      // Find the best match for each track's Head and Tail.
      fStitchAlg.FindHeadsAndTails(evt,fTrackModuleLabel);
      // walk through each vertex of one track to its match on another, and so on and stitch 'em.
      fStitchAlg.WalkStitch();
      // search composite tracks and stitch further if there are components in common. Do it until all are stitched.
      bool stizatch(fStizatch);
      while (stizatch)
      {
         stizatch = fStitchAlg.CommonComponentStitch();
      }
      mf::LogVerbatim("TrackStitcher.beginning") << "There are " <<  fStitchAlg.ftListHandle->size() << " Tracks in this event before stitching.";

      fStitchAlg.GetTracks(*tcol);
      fStitchAlg.GetTrackComposites(*tvcol);

      if (tcol->size()!=tvcol->size())
         throw cet::exception("TrackStitcher") << "Tracks and TrackComposites do not match: "<<tcol->size()<<" vs "<<tvcol->size()<<"\n";

      std::vector<size_t> spIndices(scol->size());
      // create spIndices, index array for searching into original scol SpacePoints.
      for ( size_t ii=0; ii<scol->size(); ii++ )
      {spIndices[ii]=ii;}

      for (size_t ii=0; ii<tvcol->size(); ii++)
      {
         const art::PtrVector<recob::Hit>& hits(GetHitsFromComponentTracks(tvcol->at(ii), evt));
         // Now make the Assns of relevant Hits to stitched Track
         util::CreateAssn(*this, evt, *tcol, hits, *thassn, ii);
         const art::PtrVector<recob::SpacePoint>& sppts(GetSpacePointsFromComponentTracks(tvcol->at(ii), evt));
         // Now make the Assns of relevant Sppts to stitched Track
         util::CreateAssn(*this, evt, *tcol, sppts, *tsptassn, ii);

         // Now Assns of sppts to hits. For this Sppt
         // I call the function to bring back the vec of associated Hits and the vector of
         // pairs of iterators that allow to pull those Hits needed from each Sppt.
         std::vector<std::pair<std::vector<art::Ptr<recob::Hit> >::const_iterator, std::vector<art::Ptr<recob::Hit> >::const_iterator > >  pits;

         std::vector<art::Ptr<recob::Hit>> hitsFromSppts;
         art::FindManyP<recob::Hit> hitAssns(sppts, evt, fSpptModuleLabel);

         size_t start(0), finish(0);
         for (unsigned int ii=0; ii < sppts.size(); ++ii )
         {
            hitsFromSppts.insert(hitsFromSppts.end(),hitAssns.at(ii).begin(), hitAssns.at(ii).end());
            finish = start+(size_t)(hitAssns.at(ii).end() - hitAssns.at(ii).begin());
            std::pair< std::vector<art::Ptr<recob::Hit> >::const_iterator, std::vector<art::Ptr<recob::Hit> >::const_iterator > pithittmp(hitAssns.at(ii).begin(),hitAssns.at(ii).end());
            pits.push_back(pithittmp);
            start += (finish+1);
         }
         //     std::cout << "TrackStitcher_module: scol->size() is " << scol->size() << std::endl;
         //     std::cout << "TrackStitcher_module: sppts.size() is " << sppts.size() << std::endl;
         for ( size_t jj=0; jj<sppts.size(); jj++ )
         {
            // find jjth sppt in the list of scol. Meaning, find kkth element of sppth.
            size_t ll(scol->size());
            // this gives indices into the vector of original spacepoints in which
            // to look for our sppts.
            size_t off(0);
            for ( auto& kk : spIndices )
            {
               const art::Ptr<recob::SpacePoint> spptnc(scol->at(kk));
               if ( spptnc != sppts.at(jj)) { off++; continue;}
               ll = kk;
               //               std::cout << "TrackStitcher_module: index into spacepoints for which to write out sppt-hit Assns is " << ll << std::endl;
               // drop this one for future searches, since we've used it.

               break;
            }
            if (ll<scol->size())
            {
               std::vector <art::Ptr  <recob::Hit> > hitsThisSppt;
               hitsThisSppt.insert(hitsThisSppt.begin(),pits.at(jj).first,pits.at(jj).second);
               util::CreateAssn(*this, evt, scol->at(ll), hitsThisSppt, *spthassn);
            }
         }
      }


      mf::LogVerbatim("TrackStitcher.end") << "There are " <<  tvcol->size() << " Tracks in this event after stitching.";
      evt.put(std::move(tcol));
      evt.put(std::move(tvcol));
      // Add Hit-to-Track and Sppt-to-Track Assns.
      evt.put(std::move(thassn));
      evt.put(std::move(tsptassn));
      evt.put(std::move(spthassn));

   }

Member Data Documentation

std::string trkf::TrackStitcher::fSpptModuleLabel

private

Definition at line 52 of file TrackStitcher_module.cc.

StitchAlg trkf::TrackStitcher::fStitchAlg

private

Definition at line 54 of file TrackStitcher_module.cc.

bool trkf::TrackStitcher::fStizatch

private

Definition at line 53 of file TrackStitcher_module.cc.

std::string trkf::TrackStitcher::fTrackModuleLabel

private

Definition at line 51 of file TrackStitcher_module.cc.

---

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

• TrackStitcher_module.cc