#include "generator/centers_table_builder.hpp"

#include "search/search_trie.hpp"

#include "indexer/centers_table.hpp"
#include "indexer/feature_algo.hpp"
#include "indexer/features_offsets_table.hpp"
#include "indexer/features_vector.hpp"

#include "coding/file_container.hpp"

#include "platform/mwm_traits.hpp"

#include "base/exception.hpp"

#include "defines.hpp"

namespace indexer
{
bool BuildCentersTableFromDataFile(string const & filename, bool forceRebuild)
{
  try
  {
    search::CentersTableBuilder builder;

    {
      FilesContainerR rcont(filename);
      if (!forceRebuild && rcont.IsExist(CENTERS_FILE_TAG))
        return true;

      feature::DataHeader header(rcont);

      version::MwmTraits const traits(header.GetFormat());
      if (!traits.HasOffsetsTable())
      {
        LOG(LERROR, (filename, "does not have an offsets table!"));
        return false;
      }

      auto const table = feature::FeaturesOffsetsTable::Load(rcont);
      if (!table)
      {
        LOG(LERROR, ("Can't load offsets table from:", filename));
        return false;
      }

      FeaturesVector const features(rcont, header, table.get());

      builder.SetCodingParams(header.GetDefCodingParams());
      features.ForEach([&](FeatureType & ft, uint32_t featureId) {
        builder.Put(featureId, feature::GetCenter(ft));
      });
    }

    {
      FilesContainerW writeContainer(filename, FileWriter::OP_WRITE_EXISTING);
      FileWriter writer = writeContainer.GetWriter(CENTERS_FILE_TAG);
      builder.Freeze(writer);
    }
  }
  catch (RootException const & e)
  {
    LOG(LERROR, ("Failed to build centers table:", e.Msg()));
    return false;
  }

  return true;
}
}  // namespace indexer