/* bitpacker.c
 * pakkaa tai purkaa kokonaislukutaulukon tiedostoon
 * tallettaen vain halutun määrän bittejä per luku.
 *
 * tallennus: bitpacker w lukum bitit tiedostonimi <tiedosto
 * luku:      bitpacker r lukum bitit tiedostonimi >tiedosto
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "psave.h"

int main(int argc, char *argv[])
{
  unsigned *a;
  int nums, bits, count=0, i;
  FILE *f;

  if (argc!=5 || !strchr("rw",*argv[1]) ||
        !sscanf(argv[2],"%d",&nums) ||
        !sscanf(argv[3],"%d",&bits)) {
     fprintf(stderr,"usage: %s r|w numbers bits file\n", argv[0]);
     exit(EXIT_FAILURE);
  }
  if (!(a = malloc(nums * sizeof(int)))) {
     fprintf(stderr,"Out of memory\n");
     exit(EXIT_FAILURE);
  }

  switch(*argv[1]) {
  case 'w':
    while (count<nums && scanf("%u", &a[count++]))
        ;
    if (count<nums)
       fprintf(stderr, "Warning: saving only %d numbers\n", count);
    if (!(f = fopen(argv[4], "wb"))) {
      perror("fopen");
      exit(EXIT_FAILURE);
    }
    psave(a, count, bits, f);
    fclose(f);
    break;
  case 'r':
    if (!(f = fopen(argv[4], "rb"))) {
      perror("fopen");
      exit(EXIT_FAILURE);
    }
    count = prestore(a, nums, bits, f);
    fclose(f);
    if (count < nums)
      fprintf(stderr, "Warning: restored only %d elements\n", count);
    for (i=0; i<count; i++)
      printf("%d\n", a[i]);
    break;
  }
  exit(EXIT_SUCCESS);
}