Bugfixes and additional changes

src/compress.cc

@@ -42,11 +42,10 @@ constexpr size_t DICT_MAX = ipow(2, 13) - 256; /* 12 bits */
  *  La chaîne de caractère \p t_text est lue caractère par caractère, et est et
  *  selon la valeur de retour de la fonction \ref dico (permettant dans le même
  *  temps la création du dictionnaire), on rajoute un mot ou pas dans le vecteur
- *  de caractères UTF-8 représentant des mots de chars compressés. La fonction
+ *  de caractères UTF-8 représentant des mots de chars compressés.
- *  renvoie ledit vecteur de uint32_t via le paramètre \p t_res.
  *
- *  \param[in] t_text Chaîne de caractères uint8_t représentant le fichier d'entrée
+ *  \param t_text Chaîne de caractères uint8_t représentant le fichier d'entrée
- *  \param[out] t_res Chaîne de caractères de sortie
+ *  \return Vecteur de chunks (vecteurs de uint32_t)
  */
 vvuint32 lzw_compress(string &&t_text) {
   std::puts("Compressing...");
@@ -88,33 +87,16 @@ vvuint32 lzw_compress(string &&t_text) {
  *  \param[in] t_out_file Chemin vers le fichier de sortie
  */
 void compress(const std::string &t_in_file, const char *t_out_file) {
-  // Fichier d’entrée
   std::ifstream input_file{t_in_file};
-  if (!input_file.is_open()) {
+  assert(input_file.is_open());
-    std::cerr << "Error at " << __FILE__ << ":" << __LINE__ - 2
+  FILE *const out = (t_out_file != nullptr) ? fopen(t_out_file, "wb") : fopen("output.lzw", "wb");
-              << ": could not open output file \"" << t_in_file
-              << "\". Aborting...\n";
-    exit(1);
-  }
-
-  // Fichier de sortie
-  FILE *const out =
-      (t_out_file != nullptr) ? fopen(t_out_file, "wb") : fopen("output.lzw", "wb");
-  // std::ofstream out{(t_out_file != nullptr) ? t_out_file : "output.lzw",
-  //                   std::ios::binary};
   if (out == nullptr) {
-    std::cerr << "Error at " << __FILE__ << ":" << __LINE__ - 4
+    std::cerr << "Error at " << __FILE__ << ":" << __LINE__ - 4 << ": could not open output file. Aborting...\n";
-              << ": could not open output file. Aborting...\n";
     input_file.close();
     exit(1);
   }
-
+  const auto compressed_text{lzw_compress(std::string{std::istreambuf_iterator<char>(input_file), std::istreambuf_iterator<char>()})};
-  const auto compressed_text{
-      lzw_compress(std::string{std::istreambuf_iterator<char>(input_file),
-                               std::istreambuf_iterator<char>()})};
-
   write_file(out, compressed_text);
-
   fclose(out);
   input_file.close();
 }

src/io.cc

@@ -75,7 +75,7 @@ void write_chunk(FILE *const t_out, const vuint32 &t_chunk) {
       data[0] = static_cast<unsigned char>(t_chunk[i] >> 4);
       data[1] = static_cast<unsigned char>(t_chunk[i] << 4);
     } else {
-      data[1] |= static_cast<unsigned char>(t_chunk[i] >> 8) & 0x0F;
+      data[1] |= static_cast<unsigned char>(t_chunk[i] >> 8) & 0xC;
       data[2] = static_cast<unsigned char>(t_chunk[i]);
       fwrite(data.data(), sizeof(data[0]), 3, t_out);
     }

src/io.hh

@@ -1,5 +1,5 @@
 /**
- *   \file io.h
+ *   \file io.hh
  *   \brief Header for file reading and writing
  */
 

src/main.cc

@@ -6,17 +6,15 @@
  *
  */
 
-#ifdef Debug
-constexpr bool debug_mode = true;
-#else
-constexpr bool debug_mode = false;
-#endif
-
 #include "compress.hh"
 #include "getopt.h"
+#include <cassert>
+#include <tuple>
 
 using std::printf;
 using std::puts;
+using std::string;
+using std::tuple;
 
 // custom types ///////////////////////////////////////////////////////////////
 
@@ -53,16 +51,8 @@ void help() {
   puts("\t\textension \".uncompresed\" will be added");
 }
 
-int main(int argc, char *argv[]) {
+std::tuple<string, string, bool> process_args(int t_argc, char *t_argv[]) {
-  if constexpr (debug_mode) {
+  auto ret = std::make_tuple(string{}, string{}, false);
-    for (int i = 0; i < argc; ++i)
-      printf("argv[%d] = %s\n", i, argv[i]);
-  }
-
-  std::string input_path{};
-  std::string output_path{};
-  bool compressing = true;
-
   while (true) {
     int option_index = 0;
     static struct option long_options[] = {
@@ -72,84 +62,41 @@ int main(int argc, char *argv[]) {
         {"compress", no_argument, nullptr, 'c'},
         {"uncompress", no_argument, nullptr, 'u'},
         {nullptr, 0, nullptr, 0}};
-    int c = getopt_long(argc, argv, "hi:o:cu", long_options, &option_index);
+    int c = getopt_long(t_argc, t_argv, "hi:o:cu", long_options, &option_index);
-    if (c == -1)
+    if (c == -1) break;
-      break;
     switch (c) {
-    case 0: {
+    case 0:
-      if constexpr (debug_mode) {
-        printf("\noption %s", long_options[option_index].name);
-        if (optarg) {
-          printf(" with arg %s\n", optarg);
-        }
-      }
       break;
-    }
+    case 'h':
-    case 'h': {
-      if constexpr (debug_mode) {
-        printf("From main - option --help passed\n");
-      }
       help();
-      return 0;
+      exit(0);
-    }
+    case 'i':
-    case 'i': {
+      std::get<0>(ret) = optarg;
-      if constexpr (debug_mode) {
-        printf("From main - option --input with value '%s'\n", optarg);
-      }
-      input_path = optarg;
       break;
-    }
+    case 'o':
-    case 'o': {
+      std::get<1>(ret) = optarg;
-      if constexpr (debug_mode) {
-        printf("From main - option --output with value '%s'\n", optarg);
-      }
-      output_path = optarg;
       break;
-    }
+    case 'c':
-    case 'c': {
+      std::get<2>(ret) = true;
-      if constexpr (debug_mode) {
-        printf("From main - option --compress\n");
-      }
-      compressing = true;
       break;
-    }
+    case 'u':
-    case 'u': {
+      std::get<2>(ret) = false;
-      if constexpr (debug_mode) {
-        printf("From main - option --uncompress\n");
-      }
-      compressing = false;
       break;
-    }
     case '?':
-    default: {
+    default:
       puts("Error: unknown parameter.");
-      if constexpr (debug_mode) {
-        printf("From main - option -?\n");
-      }
       help();
-      return 1;
+      exit(1);
     }
   }
+  return ret;
 }
 
-  if (input_path.empty()) {
+/* TODO: compression multiple : nombre de compressions puis fichier compressé */
-    puts("Error: no input file specified");
+int main(int argc, char *argv[]) {
-    return 2;
+  const auto [input_path, output_path, compressing] = process_args(argc, argv);
-  }
+  assert(!input_path.empty());
-
   if (compressing) {
-    /*
-      TODO:
-      - compresser le fichier d’entrée morceaux par morceaux, 16Ko à la fois
-      - écrire la taille du segment compressé, puis le segment compressé
-      - multithreading
-      - compression multiple : nombre de compressions puis fichier compressé
-      - bit-packing, limiter la taille du dictionnaire pour un certain nombre de
-        bits.
-     */
-    if constexpr (debug_mode) {
-      puts("Beginning compression");
-    }
     if (output_path.empty()) {
       compress(input_path, nullptr);
     } else {
@@ -157,11 +104,7 @@ int main(int argc, char *argv[]) {
     }
   } else {
     puts("Not yet implemented :(");
-    /*
+    /* Inversion des types du dictionnaire pour retrouver les chaînes plus aisément */
-      Inversion des types du dictionnaire pour retrouver les chaînes plus
-      aisément
-     */
   }
-
   return 0;
 }