usaa21/sk2a/README.md

296 lines
10 KiB
Markdown
Raw Permalink Normal View History

2022-01-20 07:58:41 +00:00
//Naprogramuj nástroj na kompresiu a dekompresiu. Na kompresiu použite aspoň dva kompresné algoritmy, napr. : Huffmanovo kódovanie, LZ77, LZ78, Run Length kódovanie alebo iný.
2022-01-20 07:45:54 +00:00
//
2022-01-20 07:58:41 +00:00
//Meno vstupného a výstupného súboru načítajte ako argument príkazového riadka. V zadaní by mali byť implementované tieto dve funkcie:
2022-01-20 07:45:54 +00:00
//
2022-01-20 07:58:41 +00:00
//**
// * Skomprimuje súbor in a zapíše do súboru out.
// * @arg in smerník na otvorený vstupný súbor (na čítanie)
// * @arg out smerník na otvorený výstupný súbor (na zápis)
// * @return počet bajtov skomprimovaného súboru
// */
//int compress(FILE* in, FILE* out);
//**
//* Dekomprimuje súbor in a zapíše do súboru out.
//* @arg in smerník na otvorený vstupný súbor (na čítanie)
// * @arg out smerník na otvorený výstupný súbor (na zápis)
// * @return počet bajtov dekomprimovaného súboru
// */
//void decompress(FILE* in, FILE* out);
// najprv robim kompresiu algoritmu runlength a potom lz78, ak sa robi nasledne dekompresia je to naopak najprv dekompresia lz78 a potom runlength.
#include <assert.h>
2022-01-20 07:45:54 +00:00
#include <stdio.h>
2022-01-20 07:58:41 +00:00
#include <stdlib.h>
2022-01-20 07:45:54 +00:00
#include <string.h>
2022-01-20 07:58:41 +00:00
#include "compressor.h"
#define BUFSIZE 1024
#define SIZE 256
#define MAX_WORDS 100000
// One codeword is previous index - new index -character
2022-01-20 07:45:54 +00:00
2022-01-20 07:58:41 +00:00
// Trie (a.k.a prefix tree)
struct dict {
int index;
// Character is encoded as non zero value in characters
struct dict* characters[SIZE];
};
// Free trie
void delete(struct dict* tree){ //skopirovane z lz78, vymaze strukturu co sa pouziva v lz78
if (tree == NULL){ //maze to pamat stromu
return;
}
for (int i = 0 ;i < SIZE; i++){
delete(tree-> characters[i]);
2022-01-20 07:45:54 +00:00
}
}
2022-01-20 07:58:41 +00:00
void get_words (struct dict* tree,char* words[]){ //posielam cely strom a smernik na dvojrozmerne pole
if (tree == NULL){
return;
2022-01-20 07:45:54 +00:00
}
2022-01-20 07:58:41 +00:00
for (int i = 0 ;i < SIZE; i++){ //prejdeme vsetkych potomkov
if (tree-> characters[i] > 0){ //zistime ci tam je nieco naalokovane
int need_size = snprintf(NULL,0,"-%d-%c\n",tree->index,i); //kolko pamate
words[tree->characters[i]->index] = calloc(need_size + 1,sizeof(char)); //naalokujeme pamat
memset(words[tree->characters[i]->index],0,need_size + 1);
sprintf(words[tree->characters[i]->index],"-%d-%c\n",tree->index,i); //napise sa vo formate -index predosleho notu - aktualny znak
2022-01-20 07:45:54 +00:00
2022-01-20 07:58:41 +00:00
get_words(tree-> characters[i],words); //zavolam na kazdeho potomka a potom na dalsich potomkov alebo syna
}
}
}
int mylz78compress(char* buff,int size,char** outbuf){
// https://medium.com/@dbudhrani/how-data-compression-works-exploring-lz78-e97e539138
// The trie represents the current phrase
// The first node represents the empty string with index 0
struct dict* root = calloc(1,sizeof(struct dict));
struct dict* current = root;
int index = 1;
int c = 0;
char *last = NULL;
for(int i = 0; i < size; i++){
c = buff[i];
// Current character is in dictionary, continue searching
if (current->characters[c] > 0){
// ak este nieco ostalo a toto je koniec
if (i == size - 1){ // pridem na koniec suboru a musim si to zapamatat, inac to stratim
int sz_needed = snprintf(NULL,0,"-%d-%c\n",current->index,c); //kolko bajtov to bude zaberat
last = calloc(sz_needed + 1,sizeof(char)); //alokuje do do last
sprintf(last,"-%d-%c\n",current->index,c); //https://stackoverflow.com/questions/3774417/sprintf-with-automatic-memory-allocation
}
2022-01-20 07:45:54 +00:00
2022-01-20 07:58:41 +00:00
current = current->characters[c];
2022-01-20 07:45:54 +00:00
}
2022-01-20 07:58:41 +00:00
else {
current->characters[c] = calloc(1,sizeof(struct dict));
current->characters[c]->index = index;
index += 1;
current = root;
2022-01-20 07:45:54 +00:00
}
2022-01-20 07:58:41 +00:00
}
char **words = calloc(index,sizeof(char*)); // z stromu vytvori format
get_words(root,words);
int size_needed = 0;
for(int i=1; i < index; i++){ //nastavime velkost aku bude mat vystupny buffer
if(words[i] != NULL){
size_needed += strlen(words[i]); //poscitavame velkosti vsetkych stringov
2022-01-20 07:45:54 +00:00
}
2022-01-20 07:58:41 +00:00
}
if (last != NULL){ //mame string last?
size_needed += strlen(last); //priratam
}
char *temp = calloc(size_needed + 10,sizeof(char)); //alokujem velkost, padalo mi to ked som mal mensie cislo ako 10
int pos = 0; //pamatam kde som to uz vypisal
for(int i=0; i < index; i++){
if (words[i] != NULL){
strcpy(temp + pos,words[i]); //nebude sa to pisat na zaciatok ale az za tym
pos += strlen(words[i]);
2022-01-20 07:45:54 +00:00
}
}
2022-01-20 07:58:41 +00:00
if (last != NULL){ // zapisem aj last
strcpy(temp + pos,last);
pos += strlen(last);
}
delete(root); //nepotrebujeme strom, uvolnime pamat
*outbuf = temp; //adresa sa zapise do outbuffer
return strlen(temp);
2022-01-20 07:45:54 +00:00
}
2022-01-20 07:58:41 +00:00
int myrlcompress(char* buff,int size,char** outbuf){
// 2 * size je urcite dost, kedze v najhorsom pripade bude kazdy znak osamote
*outbuf = calloc(2 * size + 1,sizeof(char)); //smernik na smernik, napisem adresu novej pamate
memset(*outbuf,0,size + 1); //vymaze
int head = -1;
int length = 0;
int pos = 0;
for(int i = 0; i < size; i++){
int c = buff[i];
if (c == head){
length += 1;
if (i == size - 1){ //posledne pismenko
(*outbuf)[pos] = head;
(*outbuf)[pos + 1] = length;
pos += 2; //aby som vedel kde to zapisovat
break;
}
}
else {
if (head != -1){
(*outbuf)[pos] = head;
(*outbuf)[pos + 1] = length;
pos += 2;
}
if (i == size - 1){ //posledne pismenko
(*outbuf)[pos] = c;
(*outbuf)[pos + 1] = 1;
pos += 2;
break;
}
head = c;
length = 1;
}
// ak sme blizko preteceniu, vypiseme aktualny znak
if (length == 255){ //nemozme ist viac, vynulujem a idem od nuly
(*outbuf)[pos] = head;
(*outbuf)[pos + 1] = length;
pos += 2;
head = -1;
length = 0;
}
}
return pos; //velkost vystupneho buffera
}
void compress(FILE* infile,FILE* outfile){
fseek(infile, 0, SEEK_END); //vstupny subor, a chceme presunut kurzor na konci suboru
int insize = ftell(infile) + 1; // na akej pozicii je kurzor, zistime velkost suboru
rewind(infile); //vrati kurzor na zaciatok
char *buffer = calloc(insize,sizeof(char)); //alokuje pamat
memset(buffer,0,insize);
insize = fread(buffer,sizeof(char),insize - 1,infile); //nacita obsah do bufferu
if (insize == 0){
assert(!ferror(infile));
}
char *tempbuf = NULL; // az vo funkcii to alokujem
int tempsize = myrlcompress(buffer,insize,&tempbuf); //vstupny buffer, velkost a adresu bufferu
char *outbuf = NULL;
int outsize = mylz78compress(tempbuf,tempsize,&outbuf);
if (outsize > 0){
fwrite(outbuf,sizeof(char),outsize,outfile); //napise sa do suboru, z akeho buffera do akeho suboru
}
}
int myrldecompress(char* buff,int size,char** outbuf){
*outbuf = calloc(size + 1,sizeof(char));
memset(*outbuf,0,size + 1);
int pos = 0;
for(int i = 0; i < size; i+=2){
char c = buff[i];
char length = buff[i + 1];
// ak nam nebude stacit miesto v *outbuf, treba allocovat viac
while((length + pos) > (size - 1)){
size *= 2;
char *tempbuf = calloc(size,sizeof(char));
memset(tempbuf,'\0',size);
strcpy(tempbuf,*outbuf);
*outbuf = tempbuf;
}
for (int j = 0; j < length; j++){ //tolko krat ho pridavam, kolko mam cisielko
(*outbuf)[pos] = c;
pos += 1;
}
}
return pos;
}
int mylz78decompress(char* buff,int size,char** outbuf){ //nepotrebujem pouzivat strom
*outbuf = calloc(size,sizeof(char)); //taka ista velkost ako vstup
memset(*outbuf,0,size);
char *words[MAX_WORDS]; //dvojrozmerne pole
words[0] = "";
char c;
int index = 1;
int number;
int pos = 0;
int bytes_read, bytes_read_all = 0; //citame vstupny buffer, necitame po bajtoch, viac ciferne cisla
while (sscanf(buff + bytes_read_all,"-%d-%c\n%n",&number,&c,&bytes_read) == 2){ //formatovacie citanie, pusaveme o bajty, ktore sa precitali ak vrati ine cislo ako 2 tak je nakocni
bytes_read_all += bytes_read; //o tolko sme sa posunuli
words[index] = calloc(strlen(words[number]) + 2,sizeof(char*)); //vo words sa vytvoria prefixi
strcpy(words[index],words[number]); //nakopirujeme words number a pridame nakoniec c
words[index][strlen(words[number])] = c;
// ak nam nebude stacit miesto v *outbuf, treba allocovat viac
while((strlen(words[index]) + pos) > (size - 1)){
size *= 2;
char *tempbuf = calloc(size,sizeof(char));
memset(tempbuf,'\0',size);
strcpy(tempbuf,*outbuf);
*outbuf = tempbuf;
}
strcpy(*outbuf + pos,words[index]); //nakopiruje najnovsie slovo
pos += strlen(words[index]);
index += 1;
}
return strlen(*outbuf); //vratim velkost
}
void decompress(FILE* infile,FILE* outfile){
fseek(infile, 0, SEEK_END);
int insize = ftell(infile) + 1;
rewind(infile);
char *buffer = calloc(insize,sizeof(char));
memset(buffer,0,insize);
insize = fread(buffer,sizeof(char),insize - 1,infile);
if (insize == 0){
assert(!ferror(infile));
}
char *tempbuf = NULL;
int tempsize = mylz78decompress(buffer,insize,&tempbuf); //najprv lz78
char *outbuf = NULL;
int outsize = myrldecompress(tempbuf,tempsize,&outbuf);
if (outsize > 0){
fwrite(outbuf,sizeof(char),outsize,outfile);
}
2022-01-20 07:45:54 +00:00
}