#include "../include/display.h"
#include "../include/common.h"
#include <stdio.h>
#include <string.h>
#include <locale.h>

/**
 * Détermine la profondeur de l'arbre binaire
 * à partir du sommet donné
 */
static int _maxDepth(HufVertex);

/**
 * Affiche l'arborescence à partir du sommet donné,
 * indenté sur le niveau donné
 */
static void _printVertex(HufVertex, wchar_t*, int);

/**
 * Transforme le caractère donné en espace s'il est
 * non-imprimable, ou le laisse tel quel sinon
 */
static char _safeChar(char input);

void _printVertex(HufVertex vert, wchar_t *buffer, int length) {
	if (vert.child_l != NULL && vert.child_r != NULL) {
        // Affichage d'un sommet avec enfants, on montre
        // la probabilité somme des enfants
		printf("■\n", vert.frequency);

        // Affichage du fils gauche. Augmentation du préfixe
        // pour l'affichage des enfants du fils gauche
        buffer[length] = '\0';
        printf("%ls├── ", buffer);

        buffer[length] = L'│';
        buffer[length + 1] = ' ';
        buffer[length + 2] = ' ';
        buffer[length + 3] = ' ';
        buffer[length + 4] = '\0';
        _printVertex(*vert.child_l, buffer, length + 4);

        // Affichage du fils droite. Augmentation du préfixe
        // pour l'affichage des enfants du fils droit
        buffer[length] = '\0';
        printf("%ls└── ", buffer);

        buffer[length] = ' ';
        _printVertex(*vert.child_r, buffer, length + 4);
	} else {
        // Affichage d'une feuille de l'arbre, correspondant
        // à un caractère du corpus
		printf("%c\n", vert.character, vert.frequency);
	}
}

int _maxDepth(HufVertex vert) {
    if (vert.child_l != NULL && vert.child_r != NULL) {
        int left = _maxDepth(*vert.child_l);
        int right = _maxDepth(*vert.child_r);

        return (left > right ? left : right) + 1;
    } else {
        return 1;
    }
}

void printTree(HufTree tree) {
    // Allocation d'un tampon suffisamment grand pour contenir
    // les indentations des sommets les plus profonds de l'arbre
    wchar_t *buffer = malloc(_maxDepth(*tree.root) * 4 * sizeof(*buffer));
	_printVertex(*tree.root, buffer, 0);
    free(buffer);
}

char _safeChar(char input) {
    if (input < 32 || input == 127) {
        return ' ';
    }

    return input;
}

void printFrequenciesTable(double* table, size_t size) {
    printf("╭─┬───┬─────────╮\n");
    printf("│C│num│fréquence│\n");
    printf("├─┼───┼─────────┤\n");

    double sum = 0;

    for (size_t i = 0; i < size; i++) {
        if (table[i] != 0) {
            printf("│%c│%3d│%9lf│\n", _safeChar(i), (int) i, table[i]);
            sum += table[i];
        }
    }

    printf("├─┴───┼─────────┤\n");
    printf("│Total│%9lf│\n", sum);
    printf("╰─────┴─────────╯\n");
}

void printLabelsTable(char** table, size_t size) {
    printf("╭─┬───┬────────────────╮\n");
    printf("│C│num│       étiquette│\n");
    printf("├─┼───┼────────────────┤\n");

    for (size_t i = 0; i < size; i++) {
        if (table[i] != 0) {
            printf("│%c│%3d│%16s│\n", _safeChar(i), (int) i, table[i]);
        }
    }

    printf("╰─┴───┴────────────────╯\n");
}