geosrc_creator.h

#pragma once

#include "address_constructor.h"
#include "config.h"
#include "toponym_relations_provider.h"

#include <maps/search/libs/geosrc/include/geosrc_paths.h>

#include <mapreduce/yt/interface/client.h>

#include <util/generic/size_literals.h>
#include <util/memory/blob.h>

namespace maps::search::geoindexer {

class GeosrcCreator {
public:
using RecognitionPovsMap = std::unordered_map<TString, std::unordered_map<TString, TString>>;

GeosrcCreator(const Config& config,
const fbs::wrappers::RecognitionMap& recognitionMap,
const ToponymRelationsProvider& toponymRelationsProvider);

void createGeosrc();

private:

/*
If Russia (country) from Russia's point of view (001) contains Crimea
and from Ukraine's point of view (UA) doesn't contain Crimea, we build 2 variants of Russia.
It means, that countryToRecognitionSubjects contains ```Russia -> [001, UA]```
*/
THashMap<Region, SubjectSet> buildCountryToRecognitionSubjects() const;

std::pair<std::unordered_set<TString>, RecognitionPovsMap> buildAllPovsAndRecognitionPovsMap(
const THashMap<Region, SubjectSet>& countryToRecognitionSubjects) const;

void createPovsRecognitionTable(const NYT::ITransactionPtr& tx,
const TString& recognizingPovPath,
const RecognitionPovsMap& recognitionPovsMap) const;

void generateTables(const THashMap<Region, SubjectSet>& countryToRecognitionSubjects) const;


const std::string& ytServer_;
const std::string& geostoragePath_;

IndexStorageConfig indexStorage_;
geosrc::GeosrcPaths geosrcPaths_;

const fbs::wrappers::RecognitionMap& recognitionMap_;
const ToponymRelationsProvider& toponymRelationsProvider_;

static constexpr uint64_t JOB_MEMORY_LIMIT = 4_GB;
};

} // namespace maps::search::geoindexer