CachingSeedCleanerBySharedInput Class Reference

#include <CachingSeedCleanerBySharedInput.h>

Inheritance diagram for CachingSeedCleanerBySharedInput:

Public Member Functions
void	add (const Trajectory *traj) override
	Informs the cleaner that a new trajectory has been made, in case the cleaner keeps a local collection of those tracks (i.e. in a map) More...
	CachingSeedCleanerBySharedInput (unsigned int numHitsForSeedCleaner=4, bool onlyPixelHits=false)
void	done () override
	Tells the cleaner that the seeds are finished, and so it can clear any cache it has. More...
bool	good (const TrajectorySeed *seed) override
	Returns true if the seed is not overlapping with another trajectory. More...
void	init (const std::vector< Trajectory > *vect) override
	Provides the cleaner a pointer to the vector where trajectories are stored, in case it does not want to keep a local collection of trajectories. More...
Public Member Functions inherited from RedundantSeedCleaner
void	clean (const std::vector< TrajectorySeed > &, std::vector< TrajectorySeed > &)
	clean More...
void	define (std::vector< TrajectorySeed > &)
	collection definition More...
	RedundantSeedCleaner ()
	constructor More...
virtual	~RedundantSeedCleaner ()
	~RedundantSeedCleaner ()
	destructor More...

Private Attributes
std::unordered_multimap< unsigned int, unsigned int >	theCache
int	theNumHitsForSeedCleaner
bool	theOnlyPixelHits
std::vector< Trajectory::RecHitContainer >	theVault

Detailed Description

Merge of SeedCleanerBySharedInput and CachingSeedCleanerByHitPosition

Definition at line 8 of file CachingSeedCleanerBySharedInput.h.

Constructor & Destructor Documentation

CachingSeedCleanerBySharedInput::CachingSeedCleanerBySharedInput ( unsigned int  numHitsForSeedCleaner = 4, bool  onlyPixelHits = false  )

inline

Definition at line 21 of file CachingSeedCleanerBySharedInput.h.

      : theNumHitsForSeedCleaner(numHitsForSeedCleaner), theOnlyPixelHits(onlyPixelHits) {}

Member Function Documentation

void CachingSeedCleanerBySharedInput::add ( const Trajectory *  traj )

overridevirtual

Informs the cleaner that a new trajectory has been made, in case the cleaner keeps a local collection of those tracks (i.e. in a map)

Implements RedundantSeedCleaner.

Definition at line 23 of file CachingSeedCleanerBySharedInput.cc.

References h, hfClusterShapes_cfi::hits, training_settings::idx, PixelSubdetector::PixelBarrel, PixelSubdetector::PixelEndcap, theCache, theOnlyPixelHits, and theVault.

Referenced by counter.Counter::register().

                                                               {
  unsigned int idx = theVault.size();
  theVault.resize(idx + 1);
  // a vector of shared pointers....
  auto &hits = theVault.back();
  (*trj).validRecHits(hits);

  for (auto const &h : hits) {
    auto detid = h->geographicalId().rawId();

    //For seeds that are made only of pixel hits, it is pointless to store the
    //information about hits on other sub-detector of the trajectory.
    if (theOnlyPixelHits && h->geographicalId().subdetId() != PixelSubdetector::PixelBarrel &&
        h->geographicalId().subdetId() != PixelSubdetector::PixelEndcap)
      continue;
    if (detid)
      theCache.insert(std::pair<uint32_t, unsigned int>(detid, idx));
  }
}

void CachingSeedCleanerBySharedInput::done ( )

overridevirtual

Tells the cleaner that the seeds are finished, and so it can clear any cache it has.

Implements RedundantSeedCleaner.

Definition at line 16 of file CachingSeedCleanerBySharedInput.cc.

References theCache, and theVault.

                                           {
  //edm::LogInfo("CachingSeedCleanerBySharedInput") << " Calls: " << calls_ << ", Tracks: " << tracks_ <<", Comps: " << comps_ << " Vault: " << theVault.size() << ".";
  //calls_ = comps_ = tracks_ = 0;
  theVault.clear();
  theCache.clear();
}

bool CachingSeedCleanerBySharedInput::good ( const TrajectorySeed *  seed )

overridevirtual

Returns true if the seed is not overlapping with another trajectory.

Implements RedundantSeedCleaner.

Definition at line 43 of file CachingSeedCleanerBySharedInput.cc.

References TrackingRecHit::all, dqmdumpme::first, newFWLiteAna::found, dqmdumpme::last, min(), TrajectorySeed::nHits(), FastTimerService_cff::range, TrajectorySeed::recHits(), OrderedSet::t, theCache, theNumHitsForSeedCleaner, and theVault.

                                                                     {
  if (seed->nHits() == 0) {
    return true;
  }

  auto range = seed->recHits();

  auto first = range.first;
  auto last = range.second;
  auto detid = first->geographicalId().rawId();

  //calls_++;
  auto itrange = theCache.equal_range(detid);
  for (auto it = itrange.first; it != itrange.second; ++it) {
    assert(it->first == detid);
    //tracks_++;

    // seeds are limited to the first "theNumHitsForSeedCleaner" hits in trajectory...
    int ext = std::min(theNumHitsForSeedCleaner, int(theVault[it->second].size()));
    auto ts = theVault[it->second].begin();
    auto te = ts + ext;
    auto t = ts;
    auto curr = first;
    for (; curr != last; ++curr) {
      bool found = false;
      for (; t != te; ++t) {
        //comps_++;
        if (curr->sharesInput((**t).hit(), TrackingRecHit::all)) {
          found = true;
          ++t;
          break;
        }
      }
      if (!found)
        break;
    }
    if (curr == last)
      return false;
  }
  return true;
}

void CachingSeedCleanerBySharedInput::init ( const std::vector< Trajectory > *  vect )

overridevirtual

Provides the cleaner a pointer to the vector where trajectories are stored, in case it does not want to keep a local collection of trajectories.

Implements RedundantSeedCleaner.

Definition at line 11 of file CachingSeedCleanerBySharedInput.cc.

References theCache, and theVault.

                                                                            {
  theVault.clear();
  theCache.clear();
}

Member Data Documentation

std::unordered_multimap<unsigned int, unsigned int> CachingSeedCleanerBySharedInput::theCache

private

Definition at line 26 of file CachingSeedCleanerBySharedInput.h.

Referenced by add(), done(), good(), and init().

int CachingSeedCleanerBySharedInput::theNumHitsForSeedCleaner

private

Definition at line 28 of file CachingSeedCleanerBySharedInput.h.

Referenced by good().

bool CachingSeedCleanerBySharedInput::theOnlyPixelHits

private

Definition at line 29 of file CachingSeedCleanerBySharedInput.h.

Referenced by add().

std::vector<Trajectory::RecHitContainer> CachingSeedCleanerBySharedInput::theVault

private

Definition at line 25 of file CachingSeedCleanerBySharedInput.h.

Referenced by add(), done(), good(), and init().