trkf::SeedFinderModule Class Reference

Inheritance diagram for trkf::SeedFinderModule:

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

Private Member Functions
void	produce (art::Event &evt) override
art::PtrVector< recob::Hit >	GetHitsFromEvent (std::string HitModuleLabel, art::Event &evt)
void	GetSortedHitsFromClusters (std::string ClusterModuleLabel, art::Event &evt, std::vector< std::vector< art::PtrVector< recob::Hit >>> &SortedHits)

Private Attributes
SeedFinderAlgorithm	fSeedAlg
std::string	fInputModuleLabel
int	fInputSource

Detailed Description

Definition at line 25 of file SeedFinderModule_module.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 56 of file SeedFinderModule_module.cc.

    : EDProducer{pset}, fSeedAlg(pset.get<fhicl::ParameterSet>("SeedAlg"))
  {
    fInputModuleLabel = pset.get<std::string>("InputModuleLabel");
    fInputSource = pset.get<int>("InputSource");

    produces<std::vector<recob::Seed>>();
  }

Member Function Documentation

art::PtrVector< recob::Hit > trkf::SeedFinderModule::GetHitsFromEvent ( std::string  HitModuleLabel, art::Event &  evt  )

private

Definition at line 156 of file SeedFinderModule_module.cc.

  {

    art::PtrVector<recob::Hit> TheHits;
    art::Handle<std::vector<recob::Hit>> hith;
    evt.getByLabel(HitModuleLabel, hith);
    for (unsigned int i = 0; i < hith->size(); ++i) {
      art::Ptr<recob::Hit> hit(hith, i);
      TheHits.push_back(hit);
    }

    return TheHits;
  }

void trkf::SeedFinderModule::GetSortedHitsFromClusters ( std::string  ClusterModuleLabel, art::Event &  evt, std::vector< std::vector< art::PtrVector< recob::Hit >>> &  SortedHits  )

private

Definition at line 121 of file SeedFinderModule_module.cc.

  {

    SortedHits.clear();
    SortedHits.resize(3);
    std::vector<art::Ptr<recob::Cluster>> Clusters;

    art::Handle<std::vector<recob::Cluster>> clusterh;
    evt.getByLabel(ClusterModuleLabel, clusterh);

    if (clusterh.isValid()) { art::fill_ptr_vector(Clusters, clusterh); }

    art::FindManyP<recob::Hit> fm(clusterh, evt, ClusterModuleLabel);

    for (size_t iclus = 0; iclus < Clusters.size(); ++iclus) {
      art::Ptr<recob::Cluster> ThisCluster = Clusters.at(iclus);

      std::vector<art::Ptr<recob::Hit>> ihits = fm.at(iclus);

      art::PtrVector<recob::Hit> HitsThisCluster;
      for (std::vector<art::Ptr<recob::Hit>>::const_iterator i = ihits.begin(); i != ihits.end();
           ++i)
        HitsThisCluster.push_back(*i);

      SortedHits[ThisCluster->View()].push_back(HitsThisCluster);
    }
  }

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

overrideprivate

Definition at line 67 of file SeedFinderModule_module.cc.

  {
    auto seeds = std::make_unique<std::vector<recob::Seed>>();

    std::vector<std::vector<recob::SpacePoint>> SpacePointsWithSeeds;
    std::vector<recob::Seed> SeedVector;

    auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
    auto const detProp =
      art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);

    // Make seeds from clusters
    if (fInputSource == 1) {
      std::vector<std::vector<art::PtrVector<recob::Hit>>> HitsPerSeed;

      std::vector<std::vector<art::PtrVector<recob::Hit>>> SortedHits;
      GetSortedHitsFromClusters(fInputModuleLabel, evt, SortedHits);

      std::vector<std::vector<recob::Seed>> Seeds =
        fSeedAlg.GetSeedsFromSortedHits(clockData, detProp, SortedHits, HitsPerSeed);

      for (size_t i = 0; i != Seeds.size(); ++i)
        for (size_t j = 0; j != Seeds.at(i).size(); ++j)
          SeedVector.push_back(Seeds.at(i).at(j));
    }

    // Make seeds from unsorted hits
    else if (fInputSource == 0) {

      art::PtrVector<recob::Hit> Hits = GetHitsFromEvent(fInputModuleLabel, evt);
      std::vector<art::PtrVector<recob::Hit>> HitCatalogue;
      SeedVector = fSeedAlg.GetSeedsFromUnSortedHits(clockData, detProp, Hits, HitCatalogue);
    }
    else {
      throw cet::exception("SeedFinderModule") << "Unkown source mode " << fInputSource << "\n";
    }

    if (SeedVector.size() > 0) {
      for (size_t i = 0; i != SeedVector.size(); ++i) {
        seeds->push_back(SeedVector.at(i));
      }
    }
    else
      mf::LogInfo("SeedFinder") << "Seed finder made no seeds : no space points in event"
                                << std::endl;

    evt.put(std::move(seeds));
  }

Member Data Documentation

std::string trkf::SeedFinderModule::fInputModuleLabel

private

Definition at line 40 of file SeedFinderModule_module.cc.

int trkf::SeedFinderModule::fInputSource

private

Definition at line 41 of file SeedFinderModule_module.cc.

SeedFinderAlgorithm trkf::SeedFinderModule::fSeedAlg

private

Definition at line 39 of file SeedFinderModule_module.cc.

---

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

• SeedFinderModule_module.cc