src/ppmc/ppmcmain.c

   1 /*
   2  Copyright (C) Arturo San Emeterio Campos 1999. All rights reserved.
   3  Permission is granted to make verbatim copies of this file for private
   4  use only. There is ABSOLUTELY NO WARRANTY. Use it at your OWN RISK.
   5
   6  This file is: "ppmcmain.c"
   7  Email: arturo@arturocampos.com
   8  Web: http://www.arturocampos.com
   9
  10  Part of the ppmc encoder only.
  11
  12  This module is the main module and calls the different modules to do
  13  the encoding of a file. When done prints bpb and kbyps.
  14 */
  15
  16
  17 #include <stdio.h>
  18 #include <stdlib.h>
  19 #include "range.h"      //the range coder functions and data
  20 #include "ppmcdata.h"
  21
  22
  23 long filesize(FILE *stream);
  24
  25
  26
  27 //Main
  28 int main (char argc, char *argv[])
  29 {
  30  unsigned long counter, //temporal counter for loops like for or while
  31           counter2,     //another temporal counter for sub loops
  32           size_file_input;     //the size of the input file
  33
  34
  35  // Print title, version and copyright
  36  printf("PPMC using range coder. (without exclusion)\n");
  37  printf("Copyright (C) Arturo San Emeterio Campos 1999. All rights reserved.\n");
  38  printf("Permission is granted to make verbatim copies of this program for private\n");
  39  printf("use only. There is ABSOLUTELY NO WARRANTY. Use it at your OWN RISK.\n");
  40
  41  // Check for correct number of parameters
  42  if(argc!=3)
  43    {
  44    printf("Bad number of arguments.\n");
  45    exit(1);
  46    }
  47
  48
  49  // Try to open input and output files
  50  if((file_input=fopen(argv[1],"r+b"))==NULL)
  51    {
  52    printf("Couldn't open %s.\n",argv[1]);
  53    exit(1);
  54    }
  55
  56  if((file_output=fopen(argv[2],"w+b"))==NULL)
  57    {
  58    printf("Couldn't create %s.\n",argv[2]);
  59    exit(1);
  60    }
  61
  62
  63  // Check input file length and not accept 0 length files
  64  size_file_input=filesize(file_input);
  65
  66  if(size_file_input<5)
  67    {
  68    printf("Can't work with files below than 5 bytes!");
  69    exit(1);
  70    }
  71
  72
  73  // First output file length
  74  fwrite(&size_file_input,1,4,file_output);      //input length
  75
  76
  77  // Initialize ppmc encoder
  78  ppmc_alloc_memory();           //get memory
  79  ppmc_initialize_contexts();    //initialize model
  80  ppmc_encoder_initialize();
  81
  82  // Initialize range coder
  83  range_coder_init(&rc_coder,file_output);
  84
  85
  86  // Start main loop which codes the file
  87  while((byte=fgetc(file_input))!=EOF)
  88  {
  89
  90  // Try to code current byte under order-4 if possible then go to lower orders
  91  if(ppmc_code_byte_order4()==0)
  92  if(ppmc_code_byte_order3()==0)
  93  if(ppmc_code_byte_order2()==0)
  94  if(ppmc_code_byte_order1()==0)
  95    if(ppmc_code_byte_order0()==0) //else try to code under order-0
  96      {
  97      // Code under order-(-1)
  98      ppmc_get_prob_ordern1();
  99      range_coder_encode(&rc_coder,total_cump,symb_cump,symb_prob);
 100      coded_in_order=0;   //update all the tables (unless order-(-1))
 101      }
 102
 103
 104  // Now do update exclusion
 105
 106  switch(coded_in_order)
 107    {
 108    case 0: ppmc_update_order0();        //update only order-0
 109    case 1: ppmc_update_order1();        //update order-0 and order-1
 110    case 2: ppmc_update_order2();        //update order-2 1 and 0...
 111    case 3: ppmc_update_order3();
 112    case 4: ppmc_update_order4();
 113    default: break;
 114    };
 115
 116
 117
 118  // Update order variables
 119
 120  o4_byte=o3_byte;
 121  o3_byte=o2_byte;
 122  o2_byte=o1_byte;
 123  o1_byte=byte;  //current one is next time order-1
 124
 125
 126  // Check if we run out of memory, in that case, flush the encoder
 127
 128  if(ppmc_out_of_memory==1)
 129    {
 130    printf("Flushing memory! Output file might be not decodable.\n");
 131    ppmc_flush_mem_enc();
 132    }
 133
 134
 135  }
 136
 137
 138  // Flush range coder
 139  range_coder_flush(&rc_coder);
 140
 141  // Free memory
 142  ppmc_free_memory();
 143
 144
 145  // Print bpb and kbyps
 146  printf("%s at %f bpb.\n",argv[1],((float)filesize(file_output)/(float)size_file_input)*(float)8);
 147
 148
 149  // Close file handles
 150  fclose(file_input);
 151  fclose(file_output);
 152
 153
 154
 155  // Nicely exit
 156  return 0;
 157 }
 158
 159
 160 // Routines not used by ppmc but rather by main.
 161 // Not including the range coder.
 162
 163
 164 // Returns the file size of a given file.
 165 long filesize(FILE *stream)
 166 {
 167    long curpos, length;
 168
 169    curpos = ftell(stream);
 170    fseek(stream, 0L, SEEK_END);
 171    length = ftell(stream);
 172    fseek(stream, curpos, SEEK_SET);
 173    return length;
 174 }
 175
 176