mirror of
https://codeberg.org/comaps/comaps
synced 2025-12-22 22:23:44 +00:00
45 lines
1.2 KiB
C++
45 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include "generator/collector_interface.hpp"
|
|
#include "generator/mini_roundabout_info.hpp"
|
|
#include "generator/way_nodes_mapper.hpp"
|
|
|
|
#include "3party/ankerl/unordered_dense.h"
|
|
|
|
namespace generator
|
|
{
|
|
|
|
class MiniRoundaboutCollector : public generator::CollectorInterface
|
|
{
|
|
public:
|
|
MiniRoundaboutCollector(std::string const & filename, IDRInterfacePtr cache);
|
|
|
|
// CollectorInterface overrides:
|
|
std::shared_ptr<CollectorInterface> Clone(IDRInterfacePtr const & = {}) const override;
|
|
|
|
void Collect(OsmElement const & element) override;
|
|
void CollectFeature(feature::FeatureBuilder const & feature, OsmElement const & element) override;
|
|
|
|
IMPLEMENT_COLLECTOR_IFACE(MiniRoundaboutCollector);
|
|
void MergeInto(MiniRoundaboutCollector & collector) const;
|
|
|
|
protected:
|
|
void Save() override;
|
|
|
|
template <typename Fn>
|
|
void ForEachMiniRoundabout(Fn && toDo)
|
|
{
|
|
for (auto & p : m_miniRoundabouts)
|
|
if (m_miniRoundaboutsExceptions.count(p.first) == 0)
|
|
toDo(p.second);
|
|
}
|
|
|
|
private:
|
|
IDRInterfacePtr m_cache;
|
|
|
|
WaysIDHolder m_roads;
|
|
ankerl::unordered_dense::map<uint64_t, MiniRoundaboutInfo> m_miniRoundabouts;
|
|
ankerl::unordered_dense::set<uint64_t> m_miniRoundaboutsExceptions;
|
|
};
|
|
} // namespace generator
|