Add new compiler flag to turn on debugging features

Makefile

@@ -5,6 +5,8 @@ SRC_FILES := \
 	huffman.c \
 	huffman.c \
 	print_helper.c
 	print_helper.c
 
 
+DEBUG := 1
+
 default:
 default:
-	gcc $(SRC_FILES) -o huffman
+	gcc $(SRC_FILES) -o huffman -DDEBUG=$(DEBUG)
 
 

huffman.c

@@ -57,6 +57,9 @@ int encode_file(FILE *in, FILE *out)
 	if (!tree)
 	if (!tree)
 		return -1;
 		return -1;
 
 
+	if (DEBUG)
+		print_tree(tree);
+
 	encode_header(&bitbuf, tree);
 	encode_header(&bitbuf, tree);
 
 
 	fseek(in, 0, SEEK_SET);
 	fseek(in, 0, SEEK_SET);
@@ -81,6 +84,9 @@ int decode_file(FILE *in, FILE *out)
 
 
 	decode_tree(&bitbuf, out, decoded);
 	decode_tree(&bitbuf, out, decoded);
 
 
+	if (DEBUG)
+		print_tree(decoded);
+
 	free(decoded);
 	free(decoded);
 
 
 	return 0;
 	return 0;