#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <stdint.h>

#ifdef IS32BIT
#define ALGS 7
#else
#define ALGS 8
#endif

#define RANGES 4

#ifndef MAX_SIZE
#define MAX_SIZE 1024
#endif

uint8_t ties[MAX_SIZE];

void SET_ONE(const uint16_t pos, const uint8_t bit){
        uint8_t cur = ties[pos];
        uint8_t b0 = (bit == 0) ? 1   : (((cur<<7) & 128) >= 128);
        uint8_t b1 = (bit == 1) ? 2   : (((cur<<6) & 128) >= 128)<<1;
        uint8_t b2 = (bit == 2) ? 4   : (((cur<<5) & 128) >= 128)<<2;
        uint8_t b3 = (bit == 3) ? 8   : (((cur<<4) & 128) >= 128)<<3;
        uint8_t b4 = (bit == 4) ? 16  : (((cur<<3) & 128) >= 128)<<4;
        uint8_t b5 = (bit == 5) ? 32  : (((cur<<2) & 128) >= 128)<<5;
        uint8_t b6 = (bit == 6) ? 64  : (((cur<<1) & 128) >= 128)<<6;
        uint8_t b7 = (bit == 7) ? 128 : (((cur) & 128)  >= 128)<<7;

	ties[pos] = b0+b1+b2+b3+b4+b5+b6+b7;
}

uint8_t IS_SET(const uint16_t pos, const uint8_t bit){
        uint8_t cur = ties[pos];
        cur <<= (7-bit);
        cur &= 128;

        if(cur >= 128)
                return 1;
        else
                return 0;
}

void Increment_Winners(uint16_t start, const uint16_t end, uint16_t *output){
	uint8_t a;
	(void)memset((void *)output, '\0', sizeof(uint16_t)*ALGS);
	while(start<end){
		for(a=0; a<ALGS; a++){
			if(IS_SET(start, a))
				output[a]++;
		}
		start++;
	}
}

void Print_with_spaces(const char *alg, const uint16_t length){
	uint16_t a = strlen(alg);
	printf("%s", alg);
	while(a<length){
		printf(" ");
		a++;
	}
}


/*
	for(a=b=0; b<ALGS; b++){
		if(wins_D[b] > a)
			a = wins_D[b];
		if(IS_SET(start_d, b)){
			wins_D[b]--;
			wins_C[b]++;
		}
	}
*/


int main(int argc, char **argv){
	FILE *in;
	uint16_t a,b,c=0; //,d=0;
	uint16_t wt_d, wt_c;
	uint16_t max_wins = 0;
	uint16_t strongest_ab;
	//uint16_t bs=0, cs=0;
	uint16_t winning_starts[RANGES] = {0,0,0,0};
	uint16_t **data;
	uint16_t read_buffer_size = 16;
	char *read_buffer;
	uint16_t start_b = 0;
	uint16_t start_c = 0;
	uint16_t start_d = 0;
	const uint16_t min_start_a = 0;
	const uint16_t min_start_b = 1;
	const uint16_t min_start_c = 2;
	const uint16_t min_start_d = 3;
	const uint16_t max_end_d = MAX_SIZE;
	uint16_t wins_B[ALGS];
	uint16_t wins_C[ALGS];
	uint16_t wins_D[ALGS];
	uint16_t wins_TB[ALGS];
	uint16_t wins_TC[ALGS];
	uint16_t wins_BS[ALGS];
	uint16_t A_Wins[MAX_SIZE-2];

#ifdef IS32BIT
	const char *algs[] = {"byte_loop", "loop", "unrolled_loop", "rep_byte", "rep_4byte", "vector_loop", "libcall"};
#else
	const char *algs[] = {"byte_loop", "loop", "unrolled_loop", "rep_byte", "rep_4byte", "rep_8byte", "vector_loop", "libcall"};
#endif

	//if(argc != 1)
	//	max_end_d = atoi(argv[1]);
	printf("Testing to %u\n", max_end_d);

	data = (uint16_t **)malloc(sizeof(uint16_t *)*max_end_d);
	for(a=min_start_a; a<max_end_d; a++){
		data[a] = (uint16_t *)malloc(sizeof(uint16_t)*ALGS);
	}
	read_buffer = (char *)malloc(read_buffer_size);

	for(a=0; a<8; a++){
#ifdef IS32BIT
		if(a == 5)
			continue;
#endif
		(void)sprintf(read_buffer, "results/r%u.txt", a+1);
		in = fopen(read_buffer, "r");
		b = 0;
		while( fgets(read_buffer, read_buffer_size, in) != NULL){
			if(b == max_end_d)
				break;
			if(atoi(read_buffer) < UINT16_MAX)
				data[b][c] = atoi(read_buffer);
			else
				data[b][c] = UINT16_MAX;
			b++;
		}
		fclose(in);
		c++;
	}
	free(read_buffer);

	for(a = min_start_a; a<max_end_d; a++){
		c = UINT16_MAX;
		for(b=0; b<ALGS; b++){
			if(data[a][b] < c)
				c = data[a][b];
		}
		for(b=0; b<ALGS; b++){
			if(data[a][b] == c)
				SET_ONE(a, b);
		}
		free(data[a]);
	}
	free(data);

	Increment_Winners(min_start_a, min_start_b, wins_B);
	for(start_b = min_start_b; start_b<MAX_SIZE-2; start_b++){
		for(c=b=0; b<ALGS; b++){
			if(wins_B[b] > c){
				c = wins_B[b];
				a = b;
			}
			if(IS_SET(start_b, b))
				wins_B[b]++;
		}
		A_Wins[start_b] = c;
	}


	//calculate wins for each alg of each range on start (NOTE: will be 1 position checked unless min_start_* changed)
	//except range d where max_end_d - min_start_d are checked
	Increment_Winners(min_start_b, min_start_c, wins_BS);
	Increment_Winners(min_start_c, min_start_d, wins_C);
	Increment_Winners(min_start_d, max_end_d, wins_D);

	//tmp2 = min_start_d - min_start_a;
	//pow(tmp2, 3)/6 - pow(tmp2, 2)/2 + tmp2/3;
	//pow(X[n], 2)/2 - 1.5*X[n] + 1;

	//start range d
	for(start_d = min_start_d; start_d<max_end_d; start_d++){
		//printf("%u/MAX_SIZE\n", start_d);
		(void)memcpy((void *)wins_TC, (const void *)wins_C, sizeof(uint16_t)*ALGS);
		(void)memcpy((void *)wins_B, (const void *)wins_BS, sizeof(uint16_t)*ALGS);
		//pick d winning alg
		for(wt_d=b=0; b<ALGS; b++){
			if(wins_D[b] > wt_d)
				wt_d = wins_D[b];
			if(IS_SET(start_d, b)){
				wins_D[b]--;
				wins_C[b]++;
			}
		}
		//start range c
		for(start_c = min_start_c; start_c<start_d; start_c++){
			(void)memcpy((void *)wins_TB, (const void *)wins_B, sizeof(uint16_t)*ALGS);
			for(wt_c=b=0; b<ALGS; b++){
				if(wins_TC[b] > wt_c)
					wt_c = wins_TC[b];
				if(IS_SET(start_c, b)){
					wins_TC[b]--;
					wins_B[b]++;
				}
			}
			wt_c += wt_d;
			//a = 0;
			strongest_ab = 0;
			//start range b
			for(start_b = min_start_b; start_b<start_c; start_b++){
				//pick b and a winning algs
				for(c=b=0; b<ALGS; b++){
					if(wins_TB[b] > c)
						c = wins_TB[b];
					if(IS_SET(start_b, b))
						wins_TB[b]--;
				}
				b = c + A_Wins[start_b];
				if(b > strongest_ab){
					strongest_ab = b;
					a = start_b;
				}
			}
			b = strongest_ab + wt_c;
			if(b > max_wins){
				max_wins = b;
				winning_starts[min_start_b] = a;
				winning_starts[min_start_c] = start_c;
				winning_starts[min_start_d] = start_d;
			}
		}
	}

	printf("\nTotal wins for all data (ties count as win)\n");
	Increment_Winners(min_start_a, max_end_d, wins_B);
	for(a=c=0; a<ALGS; a++){
		printf("\t");
		Print_with_spaces(algs[a], 15);
		printf("%u\n", wins_B[a]);
		c += wins_B[a];
	}
	printf("\ntotal: %u\n\n", c);
	printf("Config with most wins/ties:\n");
	printf("\tstart\twins\talg/s\n");

	for(wt_d=a=0; a<RANGES; a++){
		if(a != min_start_d)
			c = winning_starts[a+1];
		else
			c = max_end_d;
		Increment_Winners(winning_starts[a], c, wins_B);

                max_wins = 0;
                for(b=0; b<ALGS; b++){
                        if(wins_B[b] > max_wins)
                                max_wins = wins_B[b];
                }
		printf("\t%u\t%u\t", winning_starts[a], max_wins);
		wt_d+= max_wins;
		for(b=0; b<ALGS; b++){
			if(wins_B[b] == max_wins)
			printf("%s(%u) ", algs[b], b);
		}
		printf("\n");
	}
	printf("\ntotal: %u\n", wt_d);

	return 0;
}