// Copyright (c) 2014 CIYAM Developers
//
// Distributed under the MIT/X11 software license, please refer to the file license.txt
// in the root project directory or http://www.opensource.org/licenses/mit-license.php.

#include <ctime>
#include <cmath>
#include <cstdlib>

#include <string>
#include <vector>
#include <sstream>
#include <iostream>

#define NUM_DAYS 1
#define NUM_YEARS 10

//#define USE_PENALISING

//#define PREVENT_IMMEDIATE_REPEAT

//#define SHOW_WINNERS
//#define SHOW_WINNERS_WEIGHT

using namespace std;

#ifndef NUM_YEARS
const size_t c_num_blocks = 1440 * NUM_DAYS;
#else
const size_t c_num_blocks = 1440 * 365 * NUM_YEARS;
#endif

int main( )
{
#ifdef SHOW_WINNERS
   string winners;
#endif
   vector< int > wins;
   vector< int > streaks;
   vector< int > balances;
   vector< int > best_streak;

   vector< int > combined;
   vector< long > weights;

   int combined_streak = 0;
   int best_combined_streak = 0;

   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 5 );
   balances.push_back( 50 );

   srand( ( unsigned int )time( 0 ) );

   for( size_t i = 0; i < balances.size( ); i++ )
   {
      wins.push_back( 0 );
      weights.push_back( 0 );
      streaks.push_back( 0 );
      combined.push_back( 0 );
      best_streak.push_back( 0 );
   }

   size_t penalising1 = 0;
   size_t penalising2 = 0;
   size_t penalising3 = 0;
   size_t last_winner = 0;
   for( size_t blocks = 0; blocks < c_num_blocks; blocks++ )
   {
#ifdef USE_PENALISING
      if( blocks % 1440 == 0 )
      {
         penalising1 = rand( ) % balances.size( );
         penalising2 = rand( ) % balances.size( );
         penalising3 = rand( ) % balances.size( );
         penalising4 = rand( ) % balances.size( );
      }
#endif

      long total_weight = 0;
      for( size_t i = 0; i < weights.size( ); i++ )
      {
         int divisor = rand( ) % 10;

         if( divisor == 0 )
            ++divisor;

         weights[ i ] = ( rand( ) % 10000 ) * ( balances[ i ] divisor );

#ifdef USE_PENALISING
         if( i == penalising1 || i == penalising2 || i == penalising3 || i == penalising4 )
            weights[ i ] = 0;
#endif

         total_weight += weights[ i ];
      }

      size_t winner = 0;
      size_t runner_up = 0;
      long best_target = 0;
#ifdef PREVENT_IMMEDIATE_REPEAT
      long second_best_target = 0;
#endif

      for( size_t i = 0; i < balances.size( ); i++ )
      {
         long adjusted_weight = weights[ i ] * 1000 total_weight;

         if( adjusted_weight > best_target )
         {
            winner = i;
            best_target = adjusted_weight;
         }
#ifdef PREVENT_IMMEDIATE_REPEAT
         else if( adjusted_weight > second_best_target )
         {
            runner_up = i;
            second_best_target = adjusted_weight;
         }
#endif
      }

#ifdef PREVENT_IMMEDIATE_REPEAT
      if( winner == last_winner )
         winner = runner_up;
#endif

#ifdef SHOW_WINNERS
      winners += ( char )( 'a' + winner );
#  ifdef SHOW_WINNERS_WEIGHT
      ostringstream osstr;
      osstr << best_target;
      winners += "(" + osstr.str( ) + ")";
#  endif
#endif
      ++wins[ winner ];

      if( winner != balances.size( ) - 1 )
      {
         ++combined_streak;
         if( combined_streak > best_combined_streak )
            best_combined_streak = combined_streak;
      }
      else
         combined_streak = 0;

      if( winner == last_winner )
      {
         ++streaks[ winner ];
         if( streaks[ winner ] > best_streak[ winner ] )
            best_streak[ winner ] = streaks[ winner ];
      }
      else
         streaks[ winner ] = 0;

      last_winner = winner;
   }

   cout << "blocks = " << c_num_blocks << endl;

   for( size_t i = 0; i < balances.size( ); i++ )
      cout << ( char )( 'a' + i ) << ": " << balances[ i ] << endl;

#ifdef SHOW_WINNERS
   cout << winners << endl;
#endif
   for( size_t i = 0; i < wins.size( ); i++ )
      cout << "wins( " << ( char )( 'a' + i ) << " ) = " << wins[ i ] << endl;

   for( size_t i = 0; i < best_streak.size( ); i++ )
      cout << "best_streak( " << ( char )( 'a' + i ) << " ) = " << ( best_streak[ i ] + 1 ) << endl;

   cout << "best_combined_streak = " << best_combined_streak << endl;
}