update tp-huffman

tp-huffman/hufftree.c

@@ -1,4 +1,6 @@
 #include "hufftree.h"
+#include "error.h"
+#include <stdlib.h>
 
 /************************************/
 /* Primitives des arbres de Huffman */
@@ -6,43 +8,60 @@
 
 /* Création d'une feuille */
 huffnode *create_huffleaf(int byte, int freq) {
-  // À écrire
-  return NULL;
+  huffnode *h = malloc(sizeof(huffnode));
+  h -> freq = 0;
+  h -> byte = 0;
+  h -> leftchild = NULL;
+  h -> rightchild = NULL;
+  return h;
 }
 
 /* Fusion de deux arbres avec un nouveau noeud racine */
 huffnode *merge_hufftree(huffnode *pl, huffnode *pr) {
-  // À écrire
-  return NULL;
+    huffnode *h = create_huffleaf(0, pl->freq + pr->freq);
+    h->leftchild = pl;
+    h->rightchild = pr;
+    return h;
 }
 
 /* Teste si un noeud est une feuille */
 bool isleaf_huffnode(huffnode *p) {
-  // À écrire
-  return true;
+    return p->leftchild == NULL && p->rightchild == NULL;
 }
 
 /* Retourne la valeur de l'octet correspondant à un noeud */
 int getbyte_huffnode(huffnode *p) {
-  // À écrire
-  return 0;
+  if (isleaf_huffnode(p)) {
+      ERROR("getbyte_huffnode: noeud feuille attendu\n");
+      exit(1);
+  }
+  return p->byte;
 }
 
 /* Retournent les fils d'un noeud */
 
 huffnode *getleft_huffnode(huffnode *p) {
-  // À écrire
-  return NULL;
+  if (isleaf_huffnode(p)) {
+      ERROR("getleft_huffnode: noeud interne attendu\n");
+      exit(1);
+  }
+  return p->leftchild;
 }
 huffnode *getright_huffnode(huffnode *p) {
-  // À écrire
-  return NULL;
+    if (isleaf_huffnode(p)) {
+          ERROR("getright_huffnode: noeud interne attendu\n");
+          exit(1);
+      }
+  return p->rightchild;
 }
 
 /* Libération d'un arbre */
 
 void free_hufftree(huffnode *p) {
-  // À écrire
+  if (p == NULL) return;
+  free_hufftree(p -> leftchild);
+  free_hufftree(p -> rightchild);
+  free(p);
 }
 
 /**********************************************/
@@ -51,15 +70,20 @@ void free_hufftree(huffnode *p) {
 
 /* Comparaison de deux arbres */
 bool compare_hufftree(void *p1, void *p2) {
-  // À écrire
-  return true;
+    huffnode *v1 = (huffnode*)p1;
+    huffnode *v2 = (huffnode*)p2;
+    if (v1 == NULL || v2 == NULL) {
+        ERROR("compare_hufftree: NULL pointeur\n");
+        exit(1);
+    }
+    return v1->freq < v2->freq;
 }
 
 /* Création de l'arbre de Huffman à partir du fichier à compresser */
 huffnode *datafile_to_hufftree(FILE *input) {
   /* Phase 1: création du tableau de fréquences */
 
-  // À écrire
+  int *tab = malloc(256 * sizeof(int));
 
   /* Phase 2: intialisation de la file de priorité à partir du tableau de
    * fréquences */