Intento de poner a andar el ZG.

[z.facultad/75.06/jacu.git] / src / statichuff / main.c

diff --git a/src/statichuff/main.c b/src/statichuff/main.c
index aa7cde86840a441dab46a74e36719e14bc6c120b..ebb711588504e5ce89ad8347f7ef7183be557552 100644 (file)
--- 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;