75 lines
2.1 KiB
C++
75 lines
2.1 KiB
C++
|
#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
|