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

#define A 259

struct node {
  wchar_t lettre;
  struct cell* listeFils;
};

struct cell {
  struct node* arbre;
  struct cell* arbreSuivant;
};

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) {
	if(((*pL) == NULL) || ((*pL)->arbre->lettre > elem)) {
		ajout_tete(elem, pL);
		return &(*pL)->arbre->listeFils;
	}
	else if((*pL)->arbre->lettre == elem) {
		return &(*pL)->arbre->listeFils;
	}
	else {
		return insertion(elem, &(*pL)->arbreSuivant);
	}
}

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

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

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) {
  	//wprintf(L"mot lu : %ls\n", motLu);
  	int estUneLettre = 1;
	int i = 0;
    cptmot += 1;
     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 == 0) {
	 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);
     }
   }
	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];
  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')) {
       printf("jsuisla\n");
       indice = indice_lettre(tab_arbre_prcp, motLu[0]);
       localisationArbre = &tab_arbre_prcp[indice].listeFils;
       if(indice == 0) {
		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;
      }
    }
    printf("Le 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"Entrez un mot ou une phrase a corriger : \n(0 pour quitter)\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 == 0) {
		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';
        wprintf(L"lettre lue %d\n", motLu[i]);
        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 desalocationArbre(struct cell** pL) {
  
  if ((*pL) == NULL) {
  	printf("return arbre desaloc");
  	return;  
  }
  
  desalocationArbre(&(*pL)-> arbreSuivant);
  desalocationArbre(&(*pL)-> arbre -> listeFils);
  free((*pL) -> arbre);
  free(*pL);  
}

void desalocationTableauArbre(struct node tab_arbre[]){
  int i = 0;
  for(wchar_t u = 'a'; u < A; u++) {
    desalocationArbre(&tab_arbre[i].listeFils);
    if(u>=A-4) printf("pas segfault : i = %d", i+1);
    i++;
  }
}

int main(int argc, char* argv[]) {

 
  setlocale(LC_ALL, "");
  FILE* dico = NULL;
  FILE* txt = NULL;
  struct node tab_arbre[A];
  
  if(argc>2) {
    dico = fopen(argv[1], "r");
    txt = fopen(argv[2], "r");
  }
  else {
    dico = NULL;
    txt = NULL;
  }
  if ((dico == NULL) || (txt == NULL)) {
        wprintf(L"Erreur : il manque un ou plusieurs fichiers !\n");
        return 1;
    }
  initialisation_tab_arbre(tab_arbre);
  remplir_dico(dico, tab_arbre); //on suppose qu'il n'y a pas d'accents dans le dictionnaire
  for(int i = 0; i < A; i++) wprintf(L"%lc\n", tab_arbre[i].lettre);
  correction_txt(txt, tab_arbre);
  correction_mot(tab_arbre);
  desalocationTableauArbre(tab_arbre); 

  printf("fini\n");
  

  return 0;
}