#include <numeric>
#include <vector>
#include <algorithm>

#include "cppunit/cppunit_proxy.h"

#if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)
using namespace std;
#endif

//
// TestCase class
//
class PartialTest : public CPPUNIT_NS::TestCase
{
  CPPUNIT_TEST_SUITE(PartialTest);
  CPPUNIT_TEST(parsrt0);
  CPPUNIT_TEST(parsrt1);
  CPPUNIT_TEST(parsrt2);
  CPPUNIT_TEST(parsrtc0);
  CPPUNIT_TEST(parsrtc1);
  CPPUNIT_TEST(parsrtc2);
  CPPUNIT_TEST(partsum0);
  CPPUNIT_TEST(partsum1);
  CPPUNIT_TEST(partsum2);
  CPPUNIT_TEST_SUITE_END();

protected:
  void parsrt0();
  void parsrt1();
  void parsrt2();
  void parsrtc0();
  void parsrtc1();
  void parsrtc2();
  void partsum0();
  void partsum1();
  void partsum2();

  static bool str_compare(const char* a_, const char* b_)
  {
    return strcmp(a_, b_) < 0 ? 1 : 0;
  }
};

CPPUNIT_TEST_SUITE_REGISTRATION(PartialTest);

//
// tests implementation
//
void PartialTest::parsrt0()
{
  int numbers[6] = { 5, 2, 4, 3, 1, 6 };

  partial_sort((int*)numbers, (int*)numbers + 3, (int*)numbers + 6);

  // 1 2 3 5 4 6 
  CPPUNIT_ASSERT(numbers[0]==1);
  CPPUNIT_ASSERT(numbers[1]==2);
  CPPUNIT_ASSERT(numbers[2]==3);
  CPPUNIT_ASSERT(numbers[3]==5);
  CPPUNIT_ASSERT(numbers[4]==4);
  CPPUNIT_ASSERT(numbers[5]==6);
}

void PartialTest::parsrt1()
{
  // 8 8 5 3 7 6 5 3 2 4 
  // 2 3 3 4 5 8 8 7 6 5 
  int numbers[10] ={ 8, 8, 5, 3, 7, 6, 5, 3, 2, 4 };

  vector <int> v1(numbers, numbers+10);
  partial_sort(v1.begin(), v1.begin() + v1.size() / 2, v1.end());

  CPPUNIT_ASSERT(v1[0]==2);
  CPPUNIT_ASSERT(v1[1]==3);
  CPPUNIT_ASSERT(v1[2]==3);
  CPPUNIT_ASSERT(v1[3]==4);
  CPPUNIT_ASSERT(v1[4]==5);
  CPPUNIT_ASSERT(v1[5]==8);
  CPPUNIT_ASSERT(v1[6]==8);
  CPPUNIT_ASSERT(v1[7]==7);
  CPPUNIT_ASSERT(v1[8]==6);
  CPPUNIT_ASSERT(v1[9]==5);
}

void PartialTest::parsrt2()
{
  char* names[] = { "aa", "ff", "dd", "ee", "cc", "bb" };

  const unsigned nameSize = sizeof(names) / sizeof(names[0]);
  vector <char*> v1(nameSize);
  for(size_t i = 0; i < v1.size(); i++)
    v1[i] = names[i];
  
  partial_sort(v1.begin(), v1.begin() + nameSize / 2, v1.end(), str_compare);

  // aa bb cc ff ee dd 
  CPPUNIT_ASSERT( strcmp(v1[0], "aa") == 0 );
  CPPUNIT_ASSERT( v1[0] == names[0] );
  CPPUNIT_ASSERT( strcmp(v1[1], "bb") == 0 );
  CPPUNIT_ASSERT( v1[1] == names[5] );
  CPPUNIT_ASSERT( strcmp(v1[2], "cc") == 0 );
  CPPUNIT_ASSERT( v1[2] == names[4] );
  CPPUNIT_ASSERT( strcmp(v1[3], "ff") == 0 );
  CPPUNIT_ASSERT( v1[3] == names[1] );
  CPPUNIT_ASSERT( strcmp(v1[4], "ee") == 0 );
  CPPUNIT_ASSERT( v1[4] == names[3] );
  CPPUNIT_ASSERT( strcmp(v1[5], "dd") == 0 );
  CPPUNIT_ASSERT( v1[5] == names[2] );
}

void PartialTest::parsrtc0()
{
  int numbers[6] = { 5, 2, 4, 3, 1, 6 };

  int result[3];
  partial_sort_copy((int*)numbers, (int*)numbers + 6, (int*)result, (int*)result + 3);
  //1 2 3 
  CPPUNIT_ASSERT(result[0]==1);
  CPPUNIT_ASSERT(result[1]==2);
  CPPUNIT_ASSERT(result[2]==3);
}

void PartialTest::parsrtc1()
{
  int numbers[10] ={ 3, 0, 4, 3, 2, 8, 2, 7, 7, 5 };

  //3 0 4 3 2 8 2 7 7 5 
  //0 2 2 3 3 

  vector <int> v1(numbers, numbers+10);
  vector <int> result(5);

  partial_sort_copy(v1.begin(), v1.end(), result.begin(), result.end());
  CPPUNIT_ASSERT(result[0]==0);
  CPPUNIT_ASSERT(result[1]==2);
  CPPUNIT_ASSERT(result[2]==2);
  CPPUNIT_ASSERT(result[3]==3);
  CPPUNIT_ASSERT(result[4]==3);
}

void PartialTest::parsrtc2()
{
  char* names[] = { "aa", "ff", "dd", "ee", "cc", "bb" };

  const unsigned nameSize = sizeof(names) / sizeof(names[0]);
  vector <char*> v1(nameSize);
  for(size_t i = 0; i < v1.size(); i++)
    v1[i] = names[i];
  vector <char*> result(3);
  partial_sort_copy(v1.begin(), v1.end(), result.begin(), result.end(), str_compare);
  
  // aa bb cc
  CPPUNIT_ASSERT( strcmp( result[0], "aa" ) == 0 );
  CPPUNIT_ASSERT( result[0] == names[0] );
  CPPUNIT_ASSERT( strcmp( result[1], "bb" ) == 0 );
  CPPUNIT_ASSERT( result[1] == names[5] );
  CPPUNIT_ASSERT( strcmp( result[2], "cc" ) == 0 );
  CPPUNIT_ASSERT( result[2] == names[4] );
}

void PartialTest::partsum0()
{
  int numbers[6] = { 1, 2, 3, 4, 5, 6 };

  int result[6];
  partial_sum((int*)numbers, (int*)numbers + 6, (int*)result);

  // 1 3 6 10 15 21
  CPPUNIT_ASSERT(result[0]==1);
  CPPUNIT_ASSERT(result[1]==3);
  CPPUNIT_ASSERT(result[2]==6);
  CPPUNIT_ASSERT(result[3]==10);
  CPPUNIT_ASSERT(result[4]==15);
  CPPUNIT_ASSERT(result[5]==21);
}

void PartialTest::partsum1()
{
  vector <int> v1(10);
  iota(v1.begin(), v1.end(), 0);
  vector <int> v2(v1.size());
  partial_sum(v1.begin(), v1.end(), v2.begin());

  // 0 1 3 6 10 15 21 28 36 45 
  CPPUNIT_ASSERT(v2[0]==0);
  CPPUNIT_ASSERT(v2[1]==1);
  CPPUNIT_ASSERT(v2[2]==3);
  CPPUNIT_ASSERT(v2[3]==6);
  CPPUNIT_ASSERT(v2[4]==10);
  CPPUNIT_ASSERT(v2[5]==15);
  CPPUNIT_ASSERT(v2[6]==21);
  CPPUNIT_ASSERT(v2[7]==28);
  CPPUNIT_ASSERT(v2[8]==36);
  CPPUNIT_ASSERT(v2[9]==45);
}

void PartialTest::partsum2()
{
  vector <int> v1(5);
  iota(v1.begin(), v1.end(), 1);
  vector <int> v2(v1.size());
  partial_sum(v1.begin(), v1.end(), v2.begin(), multiplies<int>());
  // 1 2 6 24 120 
  CPPUNIT_ASSERT(v2[0]==1);
  CPPUNIT_ASSERT(v2[1]==2);
  CPPUNIT_ASSERT(v2[2]==6);
  CPPUNIT_ASSERT(v2[3]==24);
  CPPUNIT_ASSERT(v2[4]==120);
}
