#include "testing/testing.hpp"

#include "map/feature_vec_model.hpp"

#include "indexer/scales.hpp"

#include "base/macros.hpp"
#include "base/thread.hpp"

namespace
{
  typedef model::FeaturesFetcher SourceT;

  class FeaturesLoader : public threads::IRoutine
  {
    SourceT const & m_src;
    int m_scale;

    // Get random rect inside m_src.
    m2::RectD GetRandomRect() const
    {
      int const count = max(1, rand() % 50);

      int const x = rand() % count;
      int const y = rand() % count;

      m2::RectD const r = m_src.GetWorldRect();
      double const sizeX = r.SizeX() / count;
      double const sizeY = r.SizeY() / count;

      double const minX = r.minX() + x * sizeX;
      double const minY = r.minY() + y * sizeY;

      return m2::RectD(minX, minY, minX + sizeX, minY + sizeY);
    }

  public:
    FeaturesLoader(SourceT const & src) : m_src(src) {}

    virtual void Do()
    {
      size_t const count = 2000;

      for (size_t i = 0; i < count; ++i)
      {
        m2::RectD const r = GetRandomRect();
        m_scale = scales::GetScaleLevel(r);

        m_src.ForEachFeature_TileDrawing(r, *this, m_scale);
      }
    }

    void operator() (FeatureType const & f)
    {
      // Force load feature.
      // We check asserts here. There is no any other constrains here.
      (void)f.IsEmptyGeometry(m_scale);
    }
  };

  void RunTest(string const & file)
  {
    SourceT src;
    src.InitClassificator();

    UNUSED_VALUE(src.RegisterMap(platform::LocalCountryFile::MakeForTesting(file)));

    // Check that country rect is valid and not infinity.
    m2::RectD const r = src.GetWorldRect();
    TEST ( r.IsValid(), () );

    m2::RectD world(MercatorBounds::FullRect());
    world.Inflate(-10.0, -10.0);

    TEST ( world.IsRectInside(r), () );

    srand(666);

    size_t const count = 20;
    threads::SimpleThreadPool pool(count);

    for (size_t i = 0; i < count; ++i)
      pool.Add(make_unique<FeaturesLoader>(src));

    pool.Join();
  }
}

UNIT_TEST(Threading_ForEachFeature)
{
  RunTest("minsk-pass");
}