Update sk1/main.c

This commit is contained in:
Yurii Chechur 2024-12-24 17:17:13 +00:00
parent e0381dccd5
commit e29d9e80b1

View File

@ -7,48 +7,57 @@
#define BUFSIZE 1024 #define BUFSIZE 1024
void print_help() { void print_help() {
printf("Usage:\n"); printf("Available Options:\n");
printf(" ./compressor -c1 infile outfile Compress using algorithm 1 (Huffman)\n"); printf("1 - Compress using algorithm 1 (Huffman)\n");
printf(" ./compressor -d1 infile outfile Decompress using algorithm 1 (Huffman)\n"); printf("2 - Decompress using algorithm 1 (Huffman)\n");
printf(" ./compressor -c2 infile outfile Compress using algorithm 2 (RLE)\n"); printf("3 - Compress using algorithm 2 (RLE)\n");
printf(" ./compressor -d2 infile outfile Decompress using algorithm 2 (RLE)\n"); printf("4 - Decompress using algorithm 2 (RLE)\n");
printf(" ./compressor -h Show this help message\n"); printf("0 - Exit\n");
} }
int main(int argc, char* argv[]) { int main() {
if (argc < 2) { int choice;
char infile[256], outfile[256];
// Покажемо користувачу доступні варіанти
print_help(); print_help();
return 1;
while (1) {
// Запитуємо користувача про вибір
printf("Enter your choice (0 to exit): ");
if (scanf("%d", &choice) != 1) {
fprintf(stderr, "Invalid input. Exiting.\n");
break;
} }
if (strcmp(argv[1], "-h") == 0) { if (choice == 0) {
print_help(); printf("Exiting program.\n");
return 0; break;
} }
// Ensure correct number of arguments // Вводимо імена файлів
if (argc != 4) { printf("Enter input file name: ");
fprintf(stderr, "Error: Invalid number of arguments.\n"); scanf("%s", infile);
print_help(); printf("Enter output file name: ");
return 1; scanf("%s", outfile);
}
const char* mode = argv[1]; switch (choice) {
const char* infile = argv[2]; case 1:
const char* outfile = argv[3];
if (strcmp(mode, "-c1") == 0) {
compress_1(infile, outfile); compress_1(infile, outfile);
} else if (strcmp(mode, "-d1") == 0) { break;
case 2:
decompress_1(infile, outfile); decompress_1(infile, outfile);
} else if (strcmp(mode, "-c2") == 0) { break;
case 3:
compress_2(infile, outfile); compress_2(infile, outfile);
} else if (strcmp(mode, "-d2") == 0) { break;
case 4:
decompress_2(infile, outfile); decompress_2(infile, outfile);
} else { break;
fprintf(stderr, "Error: Unknown mode '%s'.\n", mode); default:
print_help(); printf("Invalid choice. Please select again.\n");
return 1; break;
}
} }
return 0; return 0;