92 lines
3.5 KiB
C
92 lines
3.5 KiB
C
// main.c
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include "compressor.h" // Підключаємо тільки заголовок
|
|
|
|
void display_help();
|
|
|
|
int main(int argc, char *argv[]) {
|
|
// Перевірка кількості аргументів
|
|
if (argc < 4) {
|
|
display_help();
|
|
return -1;
|
|
}
|
|
|
|
// Імена вхідного та вихідного файлів
|
|
char *input_filename = argv[2];
|
|
char *output_filename = argv[3];
|
|
char *algorithm = argc >= 5 ? argv[4] : "rle"; // Алгоритм, за замовчуванням rle
|
|
|
|
// Обробка компресії
|
|
if (argc >= 4 && strcmp(argv[1], "-c") == 0) {
|
|
if (strcmp(algorithm, "rle") == 0) {
|
|
printf("Compressing using RLE...\n");
|
|
int result = rle_compress(input_filename, output_filename);
|
|
if (result >= 0) {
|
|
printf("Compression successful. Compressed size: %d bytes\n", result);
|
|
} else {
|
|
printf("Compression failed with error code %d\n", result);
|
|
}
|
|
} else if (strcmp(algorithm, "lz77") == 0) {
|
|
printf("Compressing using LZ77...\n");
|
|
int result = lz77_compress(input_filename, output_filename);
|
|
if (result >= 0) {
|
|
printf("Compression successful. Compressed size: %d bytes\n", result);
|
|
} else {
|
|
printf("Compression failed with error code %d\n", result);
|
|
}
|
|
} else {
|
|
printf("Unsupported compression algorithm. Use 'rle' or 'lz77'.\n");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// Обробка декомпресії
|
|
else if (argc >= 4 && strcmp(argv[1], "-d") == 0) {
|
|
if (strstr(input_filename, ".rle") != NULL) {
|
|
printf("Decompressing RLE...\n");
|
|
int result = rle_decompress(input_filename, output_filename);
|
|
if (result >= 0) {
|
|
printf("Decompression successful. Decompressed size: %d bytes\n", result);
|
|
} else {
|
|
printf("Decompression failed with error code %d\n", result);
|
|
}
|
|
} else if (strstr(input_filename, ".lz77") != NULL) {
|
|
printf("Decompressing LZ77...\n");
|
|
int result = lz77_decompress(input_filename, output_filename);
|
|
if (result >= 0) {
|
|
printf("Decompression successful. Decompressed size: %d bytes\n", result);
|
|
} else {
|
|
printf("Decompression failed with error code %d\n", result);
|
|
}
|
|
} else {
|
|
printf("Unsupported decompression format.\n");
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
// Виведення довідки
|
|
else if (argc == 3 && strcmp(argv[1], "-h") == 0) {
|
|
display_help();
|
|
} else {
|
|
printf("Invalid arguments\n");
|
|
display_help();
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
// Функція для виведення довідки
|
|
void display_help() {
|
|
printf("Usage:\n");
|
|
printf(" -c infile outfile [algorithm] : Compress infile to outfile. Algorithm can be 'rle' or 'lz77'. Default is 'rle'.\n");
|
|
printf(" -d compressed outfile : Decompress compressed to outfile.\n");
|
|
printf(" -h : Display this help message.\n");
|
|
printf("Examples:\n");
|
|
printf(" compression_tool -c input.txt output.rle rle\n");
|
|
printf(" compression_tool -c input.txt output.lz77 lz77\n");
|
|
printf(" compression_tool -d output.rle decompressed.txt\n");
|
|
}
|