#include "testing/testing.hpp"

#include "map/feature_vec_model.hpp"

#include "indexer/data_header.hpp"
#include "indexer/interval_index.hpp"

#include "platform/platform.hpp"

#include "base/logging.hpp"


UNIT_TEST(CheckMWM_LoadAll)
{
  Platform & pl = GetPlatform();

  Platform::FilesList maps;
  pl.GetFilesByExt(pl.WritableDir(), DATA_FILE_EXTENSION, maps);

  model::FeaturesFetcher m;
  m.InitClassificator();

  for (string const & s : maps)
  {
    try
    {
      pair<MwmSet::MwmLock, bool> const p = m.RegisterMap(s);
      TEST(p.first.IsLocked(), ());
      TEST(p.second, ());
    }
    catch (RootException const & ex)
    {
      TEST(false, ("Bad mwm file:", s));
    }
  }
}

UNIT_TEST(CheckMWM_GeomIndex)
{
  // Open mwm file from data path.
  FilesContainerR cont(GetPlatform().GetReader("minsk-pass.mwm"));

  // Initialize index reader section inside mwm.
  typedef ModelReaderPtr ReaderT;
  ReaderT reader = cont.GetReader(INDEX_FILE_TAG);
  ReaderSource<ReaderT> source(reader);
  VarSerialVectorReader<ReaderT> treesReader(source);

  // Make interval index objects for each scale bucket.
  vector<unique_ptr<IntervalIndex<ReaderT>>> scale2Index;
  for (size_t i = 0; i < treesReader.Size(); ++i)
    scale2Index.emplace_back(new IntervalIndex<ReaderT>(treesReader.SubReader(i)));

  // Pass full coverage as input for test.
  uint64_t beg = 0;
  uint64_t end = static_cast<uint64_t>((1ULL << 63) - 1);

  // Count objects for each scale bucket.
  map<size_t, uint64_t> resCount;
  for (size_t i = 0; i < scale2Index.size(); ++i)
    scale2Index[i]->ForEach([i, &resCount](uint32_t)
    {
      ++resCount[i];
    }, beg, end);

  // Print results.
  LOG(LINFO, (resCount));
}