]> git.llucax.com Git - z.facultad/75.06/jacu.git/blobdiff - src/jacu.c
Cambios minimos, no se si entraran en la impresion :(
[z.facultad/75.06/jacu.git] / src / jacu.c
index 4b35832d14a3a0035f6e969a7db1a09b887f4e04..d8f6de344f5606fba36779270ed2662336988ac2 100644 (file)
@@ -1,5 +1,24 @@
+/*----------------------------------------------------------------------------
+ *                   jacu - Just Another Compression Utility
+ *----------------------------------------------------------------------------
+ * This file is part of jacu.
+ *
+ * jacu is free software; you can redistribute it and/or modify it under the
+ * terms of the GNU General Public License as published by the Free Software
+ * Foundation; either version 2 of the License, or (at your option) any later
+ * version.
+ *
+ * jacu is distributed in the hope that it will be useful, but WITHOUT ANY
+ * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+ * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
+ * details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with jacu; if not, write to the Free Software Foundation, Inc., 59 Temple
+ * Place, Suite 330, Boston, MA  02111-1307  USA
+ *----------------------------------------------------------------------------
+ */
 
-/* Jacu Team - GPL */
 #include "blocksorting/bs.h"
 #include "mtf/mtf.h"
 #include "zerogrouping/zerogrouping.h"
 
 long fsize(const char* filename);
 
+/* Flags del archivo comprimido */
+#define FLAGS_ZG 0x1
+#define FLAGS_WS 0x2
+#define FLAGS_RESERVED_1 0x4
+#define FLAGS_RESERVED_2 0x8
+#define FLAGS_RESERVED_3 0x16
+#define FLAGS_RESERVED_4 0x64
+#define FLAGS_RESERVED_5 0x128
+#define FLAGS_RESERVED_6 0x255
+
 typedef struct _flags_ {
        int cflag;
        int dflag;
@@ -20,11 +49,27 @@ typedef struct _flags_ {
        int qflag;
        int sflag;
        int mflag;
+       int rflag; /* Richard Dictionary :-) */
 } t_Flags;
 
-int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumensize, t_Flags *flags, char *staticmodel);
+int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags *flags, char *staticmodel);
 int descomprimir(char *src, char *dst);
 
+char is_flags_on(unsigned char flags, unsigned char flag)
+{
+       return (flags & flag);
+}
+
+char flag_on(unsigned char flags, unsigned char flag)
+{
+       return (flags | flag);
+}
+
+char flag_off(unsigned char flags, unsigned char flag)
+{
+       return (flags & (~flag));
+}
+
 int main(int argc, char* argv[])
 {      
        long int volumesize = 0;
@@ -35,7 +80,7 @@ int main(int argc, char* argv[])
                        
        memset(&flags, 0, sizeof(t_Flags));
 
-       while ((ch = getopt(argc, argv, "scdzm:t:q:")) != -1) { 
+       while ((ch = getopt(argc, argv, "rscdzm:t:q:")) != -1) { 
                 
                switch (ch) { 
                        case 'c': flags.cflag = 1; 
@@ -58,6 +103,8 @@ int main(int argc, char* argv[])
                                volumesize = atol(optarg);
                                break; 
 
+                       case 'r': flags.rflag = 1;
+                               break;
                        case 'q': flags.qflag = 1; 
                                switch (atoi(optarg))
                                {
@@ -85,13 +132,13 @@ int main(int argc, char* argv[])
                                }
                                break; 
 
-                       default: fprintf(stderr, "Usage: %s [-cdzs][-q blksize][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
+                       default: fprintf(stderr, "Usage: %s [-cdzsr][-q blksize][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
                                         return(2);
                }
        }
                
        if ( (argc == 1) || (flags.cflag & flags.dflag) || !(flags.cflag | flags.dflag) || ((argc - optind) < 2) || (flags.mflag & flags.sflag)) {
-               fprintf(stderr, "Usage: %s [-cdzs][-q compressionquality][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
+               fprintf(stderr, "Usage: %s [-cdzsr][-q compressionquality][-t volsize][-m modeldumpfile] source target\n", argv[0]); 
                return (3);
        }
        if ((flags.tflag) && (volumesize <= 0l)) {
@@ -119,7 +166,7 @@ long fsize(const char* filename)
        FILE* file;
        long  file_size;
 
-       if (!(file = fopen(filename, "r"))) return -1;
+       if (!(file = fopen(filename, "ab"))) return -1;
        file_size = ftell(file);
        fclose(file);
        return file_size;
@@ -135,8 +182,12 @@ int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags
        unsigned char *mtf;
        unsigned char *salida, *data;
        unsigned char *z;
+       unsigned char file_flags = 0;
        int z_len;
        
+       /* Abrimos el archivo a comprimir y encodeamos bloques */
+       if ((fp = fopen(src, "rb")) == NULL) return 1;
+       
        /* Preparo el compresor huffman */
        if ((shuff = shuff_init_encoder_bychunk(dst, volumesize*1024)) == NULL) return 1;
        if (flags->mflag == 1) shuff_loadmodel(shuff, staticmodel);
@@ -146,28 +197,23 @@ int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags
        salida = malloc(sizeof(unsigned char)*pagesize+sizeof(Uint32));
        bs = bs_create(pagesize);
 
-       /* Abrimos el archivo a comprimir y encodeamos bloques */
-       fp = fopen(src, "rb");
-
        /* Guardamos el pagesize como header (huffencoded) */
        shuff_scanfreq_chunk(shuff,(char*)&pagesize,sizeof(Uint32));
 
        /* Guardamos cabecera para indicar si usamos ZG (huffencoded) */
        if (flags->zflag)
-               shuff_scanfreq_chunk(shuff, "\001", 1);
-       else
-               shuff_scanfreq_chunk(shuff, "\000", 1);
+               file_flags = flag_on(file_flags, FLAGS_ZG);
+       if (flags->rflag)
+               file_flags = flag_on(file_flags, FLAGS_WS);
+
+       shuff_scanfreq_chunk(shuff, &file_flags, 1);
 
        total = 0;
        while (!feof(fp)) {
                i = 0;
-               while ((!feof(fp)) && (i < pagesize)) {
-                       data[i++] = fgetc(fp);
-                       total++;
-               }
+               i = bs_readblock(fp, data, pagesize, flags->rflag);
+               total += i;
 
-               /* Saco un EOF que lee de mas */
-               if (i<pagesize) i--;
 
                /* Aplico BS guardando su resultado + el K en salida */
                bs_solve(data, salida, bs, &k, i);
@@ -184,8 +230,8 @@ int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags
                        Uint32 len;
                        unsigned char buff[2];
                        Uint32 total_size = i + sizeof(Uint32);
-                       ZG zg;                          
-                       /* Guardo la salida del MTF con ceros agrupados (ZG) */                         
+                       ZG zg;
+                       /* Guardo la salida del MTF con ceros agrupados (ZG) */
                        zg_init(&zg);
                        for (j = 0; j < total_size; ++j)
                                if ((len = zg_group(&zg, buff, mtf[j])))
@@ -193,18 +239,20 @@ int comprimir(char *src, char *dst, Uint32 pagesize, Uint32 volumesize, t_Flags
 
                                /* Flusheo ultimo zgrouping */
                                if ((len = zg_group_finish(&zg,buff)))
-                                       shuff_scanfreq_chunk(shuff, buff, len);                         
+                                       shuff_scanfreq_chunk(shuff, buff, len);
                } else {
-                       /* Comprimo la salida del MTF */                                
-                       shuff_scanfreq_chunk(shuff,mtf,i+sizeof(Uint32));                               
+                       /* Comprimo la salida del MTF */
+                       shuff_scanfreq_chunk(shuff,mtf,i+sizeof(Uint32));
                }
                free(mtf);
                free(z);
        }
 
        /* Limpiando */
-       fclose(fp);             
+       if (fclose(fp)) fprintf(stderr, "Error al cerrar archivo de entrada!\n");
        bs_destroy(bs);
+       free(data);
+       free(salida);
 
        /* Comprimo con Huffman */              
        shuff_encode_file(shuff);
@@ -222,13 +270,12 @@ int descomprimir(char *src, char *dst)
 {
        /* Descomprimo */
        FILE *fp_out;
-       Uint32 block_size = 0,zgungrouped = 0, k;
+       Uint32 block_size = 0, k;
        unsigned char *block, *mtf, *orig;
-       unsigned char *z, *zgbuffer;
-       int zgmoved = 0,z_len=0,moredata = 0,decoded = 0;
-       unsigned char use_zg = 0,zgbyte = 0,retbytes = 0;
+       unsigned char *z;
+       Uint32 z_len=0,moredata = 0,decoded = 0;
+       unsigned char file_flags = 0,retbytes = 0;
        HUFF_STATE *shuff;
-       ZG zg;
 
        /* Inicializo el descompresor */
        if ((shuff = shuff_init_decoder(src, NULL)) == NULL) return 1;
@@ -240,11 +287,9 @@ int descomprimir(char *src, char *dst)
        if (!(moredata = shuff_decode_chunk(shuff,(char*)&block_size,sizeof(Uint32),&decoded))) return 1;
 
        /* Descomprimo byte que indica si se usa ZG */
-       if (!(moredata = shuff_decode_chunk(shuff, &use_zg, 1, &decoded))) return 1;
-       if (use_zg) zg_init(&zg);
+       if (!(moredata = shuff_decode_chunk(shuff, &file_flags, 1, &decoded))) return 1;
 
        /* Creo buffers */
-       zgbuffer = malloc(sizeof(unsigned char)*256);
        block = malloc(block_size*sizeof(unsigned char)+sizeof(Uint32));
        orig = malloc(block_size*sizeof(unsigned char));
 
@@ -257,17 +302,22 @@ int descomprimir(char *src, char *dst)
                        moredata = shuff_decode_chunk(shuff,z,z_len,&decoded);                          
                        
                        /* Veo si se uso Zero Grouping para comprimir */
-                       if (use_zg) {                                                                                                                                                           
+                       if (is_flags_on(file_flags, FLAGS_ZG)) {
+                               ZG zg;
+                               unsigned char zgbuffer[255];
+                               unsigned char zgbyte = 0;
+                               int zgmoved = 0;
+                               Uint32 zgungrouped = 0;
                                /* Desagrupo bytes hasta completar la pagina or End of Source File */
-                               zgungrouped = 0;                                        
-                               do {                                                                                                                                    
+                               zg_init(&zg);
+                               do {
                                        /* Levanto un byte zerogrouped y lo paso por el zg_ungroup */
                                        zgmoved = 0;
-                                       moredata = shuff_decode_chunk(shuff,&zgbyte,1,&decoded);                                                                                                                                                                                                                                                                                                
+                                       moredata = shuff_decode_chunk(shuff,&zgbyte,1,&decoded);
                                        retbytes = zg_ungroup(&zg,zgbuffer,zgbyte);
-                                       /* Muevo del zgbuffer a mi bloque lo que corresponda */                                         
-                                       while ((zgmoved < retbytes) && (zgungrouped < block_size+sizeof(Uint32))) {                                                     
-                                               block[zgungrouped++] = zgbuffer[zgmoved++];                                                     
+                                       /* Muevo del zgbuffer a mi bloque lo que corresponda */
+                                       while ((zgmoved < retbytes) && (zgungrouped < block_size+sizeof(Uint32))) {
+                                               block[zgungrouped++] = zgbuffer[zgmoved++];
                                        }
                                } while ((moredata) && (zgungrouped < block_size+sizeof(Uint32)));
 
@@ -276,17 +326,17 @@ int descomprimir(char *src, char *dst)
                                        /* Leo un byte mas (un 0 seguro) y zg_ungroup cambiara su estado */
                                        moredata = shuff_decode_chunk(shuff,&zgbyte,1,&decoded);
                                        zg_ungroup(&zg,zgbuffer,zgbyte);
-                               }                                               
+                               }
 
                                /* Normalizo variables para continuar en common code */
-                               decoded = zgungrouped;                                  
+                               decoded = zgungrouped;
                        }
                        else {
                                /* Levanto una salida de MTF */
                                moredata = shuff_decode_chunk(shuff,block,block_size+sizeof(Uint32),&decoded);
                        }
                        
-                       /* Le aplico MTF inverso a la salida de MTF levantada previamente */    
+                       /* Le aplico MTF inverso a la salida de MTF levantada previamente */
                        mtf = jacu_mtf_inv(z, block, decoded);
 
                        /* Ya tengo la salida del BS, tonces levanto su K */
@@ -295,22 +345,25 @@ int descomprimir(char *src, char *dst)
                        /* Obtengo el chunk original aplicando BS Inverso */
                        bs_restore(orig, mtf+sizeof(Uint32), k, decoded - sizeof(Uint32));
 
-                       fwrite(orig, decoded - sizeof(Uint32), sizeof(unsigned char), fp_out);
+                       decoded -= sizeof(Uint32);
+                       if (is_flags_on(file_flags, FLAGS_WS)) {
+                               orig = bs_finalblock(orig, decoded, &decoded);
+                       }
+
+                       fwrite(orig, decoded, sizeof(unsigned char), fp_out);
                        free(mtf);
-                       free(z);                                
+                       free(z);
                }
                else return 1;
-       } while (moredata);             
+       } while (moredata);
        
        /* Close up files and free mem */
        fclose(fp_out);
        free(block);
        free(orig);
-       free(zgbuffer);
-       
+
        /* Shutdown Huffman */
        shuff_deinit_decoder(shuff);
        free(shuff);
        return 0;
 }
-