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

#define A 26

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

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

void lien_listeFils(struct cell** pL) {
  struct cell* p;
  p = (struct cell*)malloc(sizeof(struct cell));
 
  (*pL)->arbre->listeFils = p;
}

void initialisation_tab_arbre(struct node tab[]) {
  for(int i = 0; i < A; i++) {
    tab[i].lettre = 97+i; //ajout lettres minuscules
    tab[i].listeFils = NULL;
  }
}

void ajout_tete(wchar_t elem, struct cell** pL) {
  struct cell* p;
  p = (struct cell*)malloc(sizeof(struct cell));
  p->arbre = (struct node*)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 remplir_dico(FILE* fd, struct node tab_arbre_prcp[]) {

  struct cell** localisationArbre = NULL; 
  int cptmot = 0;
  wchar_t motLu[50];
  while(fwscanf(fd, L"%ls", motLu)==1) {
  	wprintf(L"mot lu : %ls\n", motLu);
  	int wchar_tEstUneLettre = 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 {
       wprintf(L"Erreur remplissage dico : L'un des caracteres n'est pas une lettre\n");
       wprintf(L"Mot : %ls incorrect\n", motLu);
       wchar_tEstUneLettre = 0;
     }
	 while((motLu[i] != '\0') && (wchar_tEstUneLettre == 1)) {
  		i += 1;
  		wprintf(L"lettre lue %ld\n", motLu[i]);
		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;
  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 {
        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;
        localisationArbre = test_mot(motLu[i], localisationArbre, &verif);
      }
      if(verif == 0) wprintf(L"Mot %ls non present dans le dicitonnaire.\n", motLu);
    }
    fclose(fd);
} 

void correction_mot(struct node tab_arbre_prcp[]) {
  struct cell** localisationArbre = NULL;
  int verif;
  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 {
        wprintf(L"Erreur correction txt : L'un des caract%lcres n'est pas une lettre\n", 130);
        verif = 0;
      }
      while((verif == 1) && (motLu[i] != '\0')) {
        i += 1;
        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);
    }
} 


int main(int argc, char* argv[]) {
  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
  correction_txt(txt, tab_arbre);
  correction_mot(tab_arbre);

  return 0;
}