#include <iostream>
#include <iomanip>
#include "ColorT.h"
#include <map>
#include <fstream>
#include <bitset>
#include <cmath>

using namespace std;

const int TRIALS = 100'000;

void RandomTest();
void NameTest();

int main() {
    RandomTest();
    NameTest();
    return 0;
}

void TestPermutations(ColorT color) {
    string name{ColorTToString(color)};
    int i;
    size_t j;

    for( i = 0; i <  pow(2,static_cast<int>(name.size())) ; ++i) {
         bitset<10> state{static_cast<unsigned long long>(i)};
         for (j =0; j < name.size(); ++j) {
             if (state[j] == 0) {
                 name[j] = static_cast<char>(tolower(name[j]));
             } else {
                 name[j] = static_cast<char>(toupper(name[j]));
             }
         }
         if (color != StringToColorT(name)) {
            cout << "Name conversion of " << ColorTToString(color) 
                 << " failed for " << name << endl;
         }
    }
}

void NameTest() {
    ColorT color;

    cout << "Performing Color Name Test " << endl;

    for(color = FIRST_COLOR; color != ColorT::NONE; color = NextColor(color)) {
       cout << "Testing " << ColorTToString(color) << endl;
       TestPermutations(color);
    }

    cout << "Color Name Test Complete" << endl;
    cout << endl;
}

void RandomTest() {
    map <ColorT,int> count;
    int trial;
    ColorT color;
    double percent;

    cout << "Performing Random Color Test " << endl;

    srand (static_cast<int>(time(nullptr)));

    for( trial = 1; trial <= TRIALS; trial++) {
       color = RandomColor();
       ++(count[color]);
    }

    cout << "Frequency Analysis " << endl;

    cout << fixed << setprecision(2);

    for(color = FIRST_COLOR; color != ColorT::NONE; color = NextColor(color)) {
       percent = static_cast<double>(count[color])/ TRIALS * 100.0 ;

       cout <<setw(10) << ColorTToString(color)
            << setw(7) << count[color]
            << setw(10) << percent << "%" << endl;
    }
    cout << "Random Color Test Complete" << endl;
    cout << endl;

    return;
}