#include <stdio.h>
#include <stdlib.h>
#include <wchar.h>
#include <locale.h>

#include "fonctions.h"

#define A 256

void ajout_tete(wchar_t elem, struct cell** pL) {
  struct cell* p;
  p = malloc(sizeof(struct cell));
  p->arbre = malloc(sizeof(struct node));
  p->arbre->listeFils = NULL;
  p->arbre->lettre = elem;
  p->arbreSuivant = *pL;
  *pL = p;
}

struct cell ** insertion(wchar_t elem, struct cell** pL, int* lettreAjoute) {
	if(((*pL) == NULL) || ((*pL)->arbre->lettre > elem)) {
		ajout_tete(elem, pL);
		if(lettreAjoute != NULL) *lettreAjoute = 1;
		return &(*pL)->arbre->listeFils;
	}
	else if((*pL)->arbre->lettre == elem) {
		return &(*pL)->arbre->listeFils;
	}
	else {
	  return insertion(elem, &(*pL)->arbreSuivant, lettreAjoute);
	}
}

void initialisation_tab_arbre(struct node tab[]) {
  int i = 0;
  for(wchar_t u = 'a'; u < A; u++) {
    tab[i].lettre = u; //ajout lettres minuscules
    tab[i].listeFils = NULL;
    i++;
  }
  tab[i].lettre = '?';
  tab[i+1].lettre = '!';
  tab[i+2].lettre = '\'';
  tab[i+3].lettre = '.';	//Ajout des caractères de ponctuation par défau
  tab[i+4].lettre = ':';
  tab[i+5].lettre = ';';
  for(int j = 0; j <= 5; j++) { 
    tab[i+j].listeFils = NULL;
    insertion('\0', &(tab[i+j].listeFils), NULL);
  }
}

int indice_lettre(struct node tab_arbre_prcp[], wchar_t lettre) {
  int i = 0;
  while(i >= 0 && i < A){  
   if(lettre == tab_arbre_prcp[i].lettre ){
     return i;
   }  
   i++;
  }
return -1;
}

void remplir_dico(FILE* fd, struct node tab_arbre_prcp[]) {

  struct cell** localisationArbre = NULL; 
  int cptmot = 0, indice = 0;
  wchar_t motLu[50];
  while(fwscanf(fd, L"%ls", motLu)==1) {
    int estUneLettre = 1, lettreAjoute = 0;
	int i = 0;
     if((motLu[0] >= 'A') && (motLu[0] <= 'Z')) {
       localisationArbre = &tab_arbre_prcp[motLu[0]-65].listeFils;
     }
     else if((motLu[0] >= 'a') && (motLu[0] <= 'z')) {
       localisationArbre = &tab_arbre_prcp[motLu[0]-97].listeFils;
     }
     else if((motLu[0] < 'A') || (motLu[0] > 'z')) {
       indice = indice_lettre(tab_arbre_prcp, motLu[0]);
       localisationArbre = &tab_arbre_prcp[indice].listeFils;
       if(indice == -1) {
	 wprintf(L"Erreur remplissage dico : L'un des caracteres n'est pas une lettre\n");
	 wprintf(L"Mot : '%ls' incorrect\n", motLu);
	 estUneLettre = 0;
       }
     }
     else {
       wprintf(L"Erreur remplissage dico : L'un des caracteres n'est pas une lettre\n");
       wprintf(L"Mot : '%ls' incorrect\n", motLu);
       estUneLettre = 0;
     }
     while((motLu[i] != '\0') && (estUneLettre == 1)) {
  	i += 1;
	localisationArbre = insertion(motLu[i], localisationArbre, &lettreAjoute);
     }
     if(lettreAjoute == 1) cptmot += 1;
   }
	wprintf(L"\n");
	fclose(fd);
  	wprintf(L"%d mots inseres dans le dictionnaire.\n", cptmot);
}

struct cell ** test_mot(wchar_t mot, struct cell** localisation, int* verif) {

  if((*localisation == NULL) || (*localisation)->arbre->lettre > mot) {
    *verif = 0;
    return NULL;
  }
  if((*localisation)->arbre->lettre == mot) {
    return &(*localisation)->arbre->listeFils;
  }
  else {
    return test_mot(mot, &(*localisation)->arbreSuivant, verif);
  }
}

void correction_txt(FILE* fd, struct node tab_arbre_prcp[]) {
  struct cell** localisationArbre = NULL;
  int verif;
  int indice = 0, cptfaute = 0;
  wchar_t motLu[50];
  wprintf(L"\nCorrection du fichier texte : \n\n");
  while(fwscanf(fd, L"%ls", motLu)==1) {
      verif = 1;
      int i = 0;
      if((motLu[0] >= 'A') && (motLu[0] <= 'Z')) {
        localisationArbre = &tab_arbre_prcp[motLu[0]-65].listeFils;
      }
      else if((motLu[0] >= 'a') && (motLu[0] <= 'z')) {
        localisationArbre = &tab_arbre_prcp[motLu[0]-97].listeFils;
      }
      else if((motLu[0] < 'A') || (motLu[0] > 'z')) {
       indice = indice_lettre(tab_arbre_prcp, motLu[0]);
       localisationArbre = &tab_arbre_prcp[indice].listeFils;
       if(indice == -1) {
		wprintf(L"Erreur correction txt : L'un des caracteres n'est pas une lettre\n");
        verif = 0;
       }
      }
      else {
        wprintf(L"Erreur correction txt : L'un des caracteres n'est pas une lettre\n");
        verif = 0;
      }
      while((verif == 1) && (motLu[i] != '\0')) {
        i += 1;
        if((motLu[i] == '.') || (motLu[i] == ',') || (motLu[i] == ';') || (motLu[i] == ':') || (motLu[i] == '!') || (motLu[i] == '?')) motLu[i] = '\0';		//prendre en compte les fins de phrase
        localisationArbre = test_mot(motLu[i], localisationArbre, &verif);
      }
      if(verif == 0) {
      	wprintf(L"Mot '%ls' non present dans le dicitonnaire.\n", motLu);
      	cptfaute += 1;
      }
    }
    wprintf(L"\nLe texte comporte %d faute(s)\n", cptfaute);
    fclose(fd);
} 

void correction_mot(struct node tab_arbre_prcp[]) {
  struct cell** localisationArbre = NULL;
  int verif;
  int indice = 0;
  wchar_t motLu[50];
  wprintf(L"\nEntrez un mot ou une phrase a corriger : \n(0 pour quitter)\n\n");
  while(wscanf(L"%ls", motLu)==1) {
      if(motLu[0]=='0') return;
      verif = 1;
      int i = 0;
      if((motLu[0] >= 'A') && (motLu[0] <= 'Z')) {
        localisationArbre = &tab_arbre_prcp[motLu[0]-65].listeFils;
      }
      else if((motLu[0] >= 'a') && (motLu[0] <= 'z')) {
        localisationArbre = &tab_arbre_prcp[motLu[0]-97].listeFils;
      }
      else if((motLu[0] < 'A') || (motLu[0] > 'z')) {
       indice = indice_lettre(tab_arbre_prcp, motLu[0]);
       localisationArbre = &tab_arbre_prcp[indice].listeFils;
       if(indice == -1) {
		wprintf(L"Erreur correction mot : L'un des caracteres n'est pas une lettre\n");
        verif = 0;
       }
      }
      else {
        wprintf(L"Erreur correction mot : L'un des caracteres n'est pas une lettre\n");
        verif = 0;
      }
      while((verif == 1) && (motLu[i] != '\0')) {
        i += 1;
        if((motLu[i] == '.') || (motLu[i] == ',') || (motLu[i] == ';') || (motLu[i] == ':') || (motLu[i] == '!') || (motLu[i] == '?')) motLu[i] = '\0';
        localisationArbre = test_mot(motLu[i], localisationArbre, &verif);
      }
      if(verif == 0) {
      	wprintf(L"Mot : '%ls' non present dans le dicitonnaire.\n", motLu);
      }
      else wprintf(L"Mot : '%ls' correct\n", motLu);
    }
} 

void desallocationArbre(struct cell** pL) {
  
  if ((*pL) == NULL) return;

  desallocationArbre(&(*pL)-> arbreSuivant);
  desallocationArbre(&(*pL)-> arbre -> listeFils);
  free((*pL) -> arbre);
  free(*pL);  
}

void desallocationTableauArbre(struct node tab_arbre[]) {
  int i = 0;
  for(wchar_t u = 'a'; u <= A+5; u++) { //A+5 car ajout de 5 caractères de ponctuation en plus des lettres
    desallocationArbre(&tab_arbre[i].listeFils);
    i++;
  }
}