X-Git-Url: https://git.llucax.com/z.facultad/75.06/jacu.git/blobdiff_plain/86a61940d1d5da3a699fbf12c87e91a7228deda2..d53cc94464bba53e0be1297ca0cc746901ec113b:/src/statichuff/main.c

diff --git a/src/statichuff/main.c b/src/statichuff/main.c
index aa7cde8..ebb7115 100644
--- a/src/statichuff/main.c
+++ b/src/statichuff/main.c
@@ -7,10 +7,14 @@ int main(int argc, char* argv[])
 	int cflag = 0;
 	int dflag = 0;
 	int tflag = 0;
+	int mflag = 0;
+	int sflag = 0;
 	long int volumesize = 0;
 	int ch;
+	HUFF_STATE *shuff;
+	char *staticmodel = NULL;
 			
-	while ((ch = getopt(argc, argv, "cdt:")) != -1) { 
+	while ((ch = getopt(argc, argv, "scdm:t:")) != -1) { 
 		 
 		switch (ch) { 
 			case 'c': cflag = 1; 
@@ -23,25 +27,40 @@ int main(int argc, char* argv[])
 					  volumesize = atoi(optarg);					  
 					  break; 
 			
-			default: fprintf(stderr, "Usage: %s [-cdt] sourcefile targetfile\n", argv[0]); 
+			case 'm': mflag = 1;
+					  staticmodel = optarg;
+					  break; 
+			
+			case 's': sflag = 1; 					  
+					  break;
+			
+			default: fprintf(stderr, "Usage: %s [-cds] [-t volsizekb] sourcefile targetfile [-m modeldumpfile]\n", argv[0]); 
 					 return(2);
 		}
 	}
 		
 	if ( (argc == 1) || (cflag & dflag) || !(cflag | dflag) || ((argc - optind) < 2) ) {
-		fprintf(stderr, "Usage: %s [-cdt] sourcefile targetfile\n", argv[0]); 
+		fprintf(stderr, "Usage: %s [-cds] [-t volsizekb] sourcefile targetfile [-m modeldumpfile]\n", argv[0]); 
 		if ((tflag == 1) && (volumesize < 0)) fprintf(stderr,"Error: The volume size must be a non-zero value\n");
 		return (2);		
 	}
 		
 	if (cflag == 1) {
 		/* Comprimo */
-	    return shuff_encode_file(argv[optind],argv[optind+1],volumesize*1024);
+		shuff = shuff_init_encoder_byfile(argv[optind],argv[optind+1],volumesize*1024);
+		if (mflag == 1) shuff_loadmodel(shuff,staticmodel);
+	    shuff_encode_file(shuff);
+		if (sflag == 1) shuff_savemodel(shuff);
+		shuff_deinit_encoder(shuff);
+		free(shuff);
 	}
 	
 	if (dflag == 1) { 
 		/* Descomprimo */
-		return shuff_decode_file(argv[optind],argv[optind+1]);
+		shuff = shuff_init_decoder(argv[optind],argv[optind+1]);
+		shuff_decode_file(shuff);
+		shuff_deinit_decoder(shuff);
+		free(shuff);
 	}
 		
 	return 0;