Repo created

generator/translator.cpp

@@ -0,0 +1,74 @@
+#include "generator/translator.hpp"
+
+#include "generator/collector_collection.hpp"
+#include "generator/filter_collection.hpp"
+#include "generator/osm_element.hpp"
+
+namespace generator
+{
+Translator::Translator(std::shared_ptr<FeatureProcessorInterface> const & processor,
+                       std::shared_ptr<cache::IntermediateData> const & cache,
+                       std::shared_ptr<FeatureMakerBase> const & maker, std::shared_ptr<FilterInterface> const & filter,
+                       std::shared_ptr<CollectorInterface> const & collector)
+  : m_filter(filter)
+  , m_collector(collector)
+  , m_tagsEnricher(cache->GetCache())
+  , m_featureMaker(maker)
+  , m_processor(processor)
+  , m_cache(cache)
+{
+  m_featureMaker->SetCache(cache->GetCache());
+}
+
+Translator::Translator(std::shared_ptr<FeatureProcessorInterface> const & processor,
+                       std::shared_ptr<cache::IntermediateData> const & cache,
+                       std::shared_ptr<FeatureMakerBase> const & maker)
+  : Translator(processor, cache, maker, std::make_shared<FilterCollection>(), std::make_shared<CollectorCollection>())
+{}
+
+void Translator::SetCollector(std::shared_ptr<CollectorInterface> const & collector)
+{
+  m_collector = collector;
+}
+
+void Translator::SetFilter(std::shared_ptr<FilterInterface> const & filter)
+{
+  m_filter = filter;
+}
+
+void Translator::Emit(OsmElement const & src)
+{
+  // Make a copy because it will be modified below.
+  OsmElement element(src);
+
+  Preprocess(element);  // Might use replaced_tags.txt via a TagReplacer.
+  if (!m_filter->IsAccepted(element))
+    return;
+
+  m_tagsEnricher(element);
+  m_collector->Collect(element);
+  m_featureMaker->Add(element);  // A feature is created from OSM tags.
+  feature::FeatureBuilder feature;
+  while (m_featureMaker->GetNextFeature(feature))
+  {
+    if (!m_filter->IsAccepted(feature))
+      continue;
+
+    // Pass non-modified source element.
+    m_collector->CollectFeature(feature, src);
+    m_processor->Process(feature);
+  }
+}
+
+void Translator::Finish()
+{
+  m_collector->Finish();
+  m_processor->Finish();
+}
+
+bool Translator::Save()
+{
+  m_collector->Finalize(true /* isStable */);
+  return true;
+}
+}  // namespace generator