Final Version (ouf)

1 files changed, 235 insertions, 17 deletions

diff --git a/src/test.cc b/src/test.cc
index cc84d6a..05f5414 100644
--- a/src/test.cc
+++ b/src/test.cc
@@ -1,5 +1,8 @@
 #include <stdio.h>
 #include <stdlib.h>
+#include <unistd.h>
+#include <termios.h>
+#include <string.h>
 #include "config.h"
 #include "BHeap.h"
 #include "FHeap.h"
@@ -12,9 +15,33 @@ char N[] = {
 	30, 21, 26, 4, 7, 28, 12, 27, 19, 22,
 };
 
+int method = -1;
+FILE *input;
+struct termios initial_settings, new_settings;
+
+char *method_names[] = {
+	N_("0 - Binary Heap"),
+	N_("1 - Binomial Heap"),
+	N_("2 - Fibonacci Heap"),
+	N_("3 - Sorted chained list")
+};
+
+char *menu[] = {
+	N_("a - Add a key into the priority list"),
+	N_("c - Change priority list type"),
+	N_("d - Delete a Key from the priority list"),
+	N_("e - Extract Min onto the priority list"),
+	N_("l - Lower Key onto a key of the priority list"),
+	N_("p - Print the current priority list on the screen"),
+	N_("r - Remove the whole priority list"),
+	N_("t - Test the priority list algorithms"),
+	N_("q - Quit")
+};
+
 void exception(int e, char *msg)
 {
 	fprintf(stderr, "%s\n", msg);
+	tcsetattr(fileno(input), TCSANOW, &initial_settings);
 	exit(-1);
 }
 
@@ -35,7 +62,8 @@ PriorityList *newlist(int method)
 		break;
 	default:
 		cerr << _("Unknow priority list type: ") << method << endl;
-		exit(-1);
+		exception(1, "Exitting");
+		return NULL;
 	}
 }
 
@@ -49,9 +77,7 @@ void DoCombTest(int method, int number)
 	cerr << _("Creation of a priority list and adding ") << number << _(" random entrie(s)...");
 	T = newlist(method);
 	for (i = 1; i <= number; i++) {
-//              T->Insert(rand() % P_INFINITY, NULL);
-//              T->Insert(rand() % 100, NULL);
-		T->Insert(1, NULL);
+		T->Insert(rand() % 100, NULL);
 	}
 	cerr << _("Ok.\nDeleting the list...");
 	oK = P_INFINITY;
@@ -94,13 +120,6 @@ void FullTest(int method)
 	DoCombTest(method, 30);
 	DoCombTest(method, 70);
 	DoCombTest(method, 1000);
-//        DoCombTest(method, 10000);
-#ifdef BT
-	DoCombTest(method, 100000);
-#ifdef VBT
-	DoCombTest(method, 1000000);
-#endif
-#endif
 
 	cerr << _("Creating a priority list and adding keys:\n");
 	T = newlist(method);
@@ -111,9 +130,6 @@ void FullTest(int method)
 			C3 = C1;
 	}
 
-	if (!C3)
-		exit(-1);
-
 	fprintf(stderr, "59 54 -10\n");
 	C1 = T->Insert(59, NULL);
 	C2 = T->Insert(54, NULL);
@@ -156,9 +172,105 @@ void Usage(void)
 	exit(-1);
 }
 
+int getchoice(char *greet, char *choices[], FILE * in)
+{
+	int chosen = 0;
+	int selected;
+	char **option;
+
+	do {
+		cerr << _("Choice: ") << greet << endl;
+		option = choices;
+		while (*option) {
+			cerr << *(option++) << endl;
+		}
+		do {
+			selected = fgetc(in);
+		} while (selected == '\n' || selected == '\r');
+		option = choices;
+		while (*option) {
+			if (selected == *option[0]) {
+				chosen = 1;
+				break;
+			}
+			option++;
+		}
+		if (!chosen) {
+			cerr << _("Incorrect choice, select again\n");
+		}
+	} while (!chosen);
+	return selected;
+}
+
+char **buildslots(PriorityList * T, Cell * Cells, char option)
+{
+	char **slots, **r, temp[BUFSIZ];
+	int i;
+
+	r = slots = (char **) malloc(12 * sizeof(char *));
+
+	for (i = 0; i < 10; i++) {
+		if (Cells[i] || option) {
+			if (Cells[i]) {
+				sprintf(temp, _("%i - Key: %i"), i, T->ReadKey(Cells[i]));
+			} else {
+				sprintf(temp, _("%i - Empty slot"), i);
+			}
+			*(slots++) = strdup(temp);
+		}
+	}
+
+	if (option)
+		*(slots++) = strdup(_("n - Don't store"));
+	*(slots++) = strdup(_("c - Cancel"));
+
+	*slots = NULL;
+
+	return r;
+
+}
+
+void freeslots(char **slots)
+{
+	char **t;
+
+	for (t = slots; *t; t++)
+		free(*t);
+	free(slots);
+}
+
+void initterm(FILE * in) {
+	tcgetattr(fileno(input), &initial_settings);
+	new_settings = initial_settings;
+	new_settings.c_lflag &= ~ICANON;
+	new_settings.c_lflag &= ~ECHO;
+	new_settings.c_cc[VMIN] = 1;
+	new_settings.c_cc[VTIME] = 0;
+	new_settings.c_lflag &= ~ISIG;
+
+	if (tcsetattr(fileno(input), TCSANOW, &new_settings) != 0) {
+		cerr << _("could not set attributes\n");
+		exit(-1);
+	}
+
+}
+
+void clearterm(FILE * in) {
+	tcsetattr(fileno(input), TCSANOW, &initial_settings);
+}
+
 int main(int argc, char **argv)
 {
-	int method = -1;
+	char choice = 0, **slots;
+	PriorityList *T, * Temp;
+	Datas_t Datas;
+	Cell Cells[10];
+	int i;
+	Key_t Key;
+
+	/*
+	   Lecture des arguments de la ligne de commande.
+	 */
 
 	while (--argc) {
 		argv++;
@@ -175,6 +287,112 @@ int main(int argc, char **argv)
 
 	if (method == -1)
 		method = 0;
-	FullTest(method);
-	return 0;
+
+	/*
+	   Initialisation du terminal.
+	 */
+
+	if (!isatty(fileno(stdout))) {
+		cerr << _("You are not a terminal, Ok.\n");
+	}
+
+	input = fopen("/dev/tty", "r");
+	if (!input) {
+		cerr << _("Unable to open /dev/tty\n");
+		exit(1);
+	}
+	
+	initterm(input);
+
+	/*
+	   Debut du programme de test.
+	 */
+
+	for (i = 0; i < 10; i++) {
+		Cells[i] = NULL;
+	}
+
+	T = newlist(method);
+
+	do {
+		cerr << "\n\n\n\n";
+		cerr << _("Priority list type: ") << method_names[method] << endl;
+		choice = getchoice(_("Please select an action"), menu, input);
+		cerr << _("You have chosen: ") << choice << "\n\n";
+		switch (choice) {
+		case 'a':	// Ajout d'une clef.
+			slots = buildslots(T, Cells, 1);
+			choice = getchoice(_("Please select a slot to save the cell"), slots, input);
+			if (choice == 'c') {
+				freeslots(slots);
+				break;
+			}
+			if (choice == 'n') {
+				choice = -1;
+			} else {
+				choice -= '0';
+			}
+			cerr << _("Please type in a key to add") << endl;
+			clearterm(input);
+			cin >> Key;
+			initterm(input);
+			if (choice == -1) {
+				T->Insert(Key, NULL);
+			} else {
+				Cells[choice] = T->Insert(Key, NULL);
+			}
+			freeslots(slots);
+			break;
+		case 'c':	// Changement du type.
+			Temp = T;
+			cerr << "\n\n\n\n";
+			method = getchoice(_("Please select a priority list type"), method_names, input) - '0';
+			T = newlist(method);
+			T->Union(Temp);
+			delete Temp;
+			break;
+		case 'd':	// Effacement d'une clef.
+			slots = buildslots(T, Cells, 0);
+			choice = getchoice(_("Please select a key to delete"), slots, input);
+			if (choice == 'c') {
+				freeslots(slots);
+				break;
+			}
+			choice -= '0';
+			cerr << _("Delete result: ") << T->Delete(Datas, Cells[choice]);
+			freeslots(slots);
+			break;
+		case 'e':	// Extract Min.
+			cerr << _("Extract Min result: ") << T->Extract_Min(Datas);
+			break;
+		case 'l':	// Lower Key.
+			slots = buildslots(T, Cells, 0);
+			choice = getchoice(_("Please select a key to lower"), slots, input);
+			if (choice == 'c') {
+				freeslots(slots);
+				break;
+			}
+			choice -= '0';
+			cerr << _("Please type in the new key") << endl;
+			clearterm(input);
+			cin >> Key;
+			initterm(input);
+			cerr << _("Lower key result: ") << T->Lower_Key(Cells[choice], Key) ? _("True") : _("False");
+			freeslots(slots);
+			break;
+		case 'p':	// Print
+			T->Dump(cerr);
+			break;
+		case 'r':	// Delete
+			delete T;
+			T = newlist(method);
+			break;
+		case 't':	// Test
+			FullTest(method);
+			break;
+		}
+	} while (choice != 'q');
+
+	clearterm(input);
+	exit(0);
 }