src/jacu.c

   1
   2 #include "statichuff/statichuff.h"
   3 #include "blocksorting/bs.h"
   4 #include "mtf/mtf.h"
   5 #include "vfile/vfile.h"
   6 #include <stdlib.h>
   7 #include <stdio.h>
   8 #include <unistd.h>
   9
  10 int main(int argc, char* argv[])
  11 {
  12         int cflag = 0;
  13         int dflag = 0;
  14         int tflag = 0;
  15         int pflag = 0;
  16         long int volumesize = 0;
  17         size_t pagesize = 32768; /* 32KB */
  18         int ch;
  19         t_BlockSort *bs;
  20
  21         while ((ch = getopt(argc, argv, "cdt:p:")) != -1) {
  22
  23                 switch (ch) {
  24                         case 'c': cflag = 1;
  25                                           break;
  26
  27                         case 'd': dflag = 1;
  28                                           break;
  29
  30                         case 't': tflag = 1;
  31                                           volumesize = atol(optarg);
  32                                           break;
  33
  34                         case 'p': pflag = 1;
  35                                           pagesize = atoi(optarg);
  36                                           break;
  37
  38                         default: fprintf(stderr, "Usage: %s [-cdpt] sourcefile targetfile\n", argv[0]);
  39                                          return(2);
  40                 }
  41         }
  42
  43         if ( (argc == 1) || (cflag & dflag) || !(cflag | dflag) || ((argc - optind) < 2) ) {
  44                 fprintf(stderr, "Usage: %s [-cdt] sourcefile targetfile\n", argv[0]);
  45                 return (3);
  46         }
  47         if ((tflag) && (volumesize <= 0l)) {
  48                 fprintf(stderr,"Error: The volume size must be a non-zero value\n");
  49                 return (4);
  50         }
  51         if ((pflag) && (pagesize <= 1u)) {
  52                 fprintf(stderr,"Error: El tamaño de página debe ser mayor a 1 byte.\n");
  53                 return (5);
  54         }
  55
  56         if (cflag == 1) {
  57                 /* Comprimo */
  58                 /* No me gusta el tmpfile ... es para probar como anda todo junto */
  59                 FILE *fp, *fp_out;
  60                 unsigned long int i, j, total, k;
  61                 int *mtf;
  62                 char *salida, *data, c;
  63                 data = malloc(sizeof(char)*pagesize);
  64                 /* Reservo lugar tambien para guardar el k */
  65                 salida = malloc(sizeof(char)*(pagesize)+sizeof(unsigned long int)*2);
  66                 bs = bs_create(pagesize);
  67                 fp = fopen(argv[optind], "rb");
  68                 fp_out = fopen("tmp.comp", "wb");
  69                 c = fgetc(fp);
  70                 total = 0;
  71                 while (!feof(fp)) {
  72                         i = 0;
  73                         while ((!feof(fp)) && (i < pagesize)) {
  74                                 data[i++] = c;
  75                                 c = fgetc(fp);
  76                                 total++;
  77                         }
  78                         /* Hago el BS */
  79                         bs_solve(data, salida, bs, &k, i);
  80                         /* Le aplico el MTF */
  81                         mtf = jacu_mtf(salida, i+sizeof(unsigned long int)*2);
  82                         for(j=0; j<i; j++)
  83                                 fputc(mtf[j], fp_out);
  84                 }
  85                 fclose(fp);
  86                 fclose(fp_out);
  87                 bs_destroy(bs);
  88                 i = shuff_encode_file("tmp.comp",argv[optind+1], volumesize);
  89                 /* borro el temporal */
  90                 unlink("tmp.comp");
  91                 return i;
  92         }
  93
  94         if (dflag == 1) {
  95                 /* Descomprimo */
  96                 return shuff_decode_file(argv[optind],argv[optind+1]);
  97         }
  98
  99         return 0;
 100 }