#include <iostream>

char translacja(char* tablica, int pozycja_bebenka, char znak)
{
	znak -= 'A';
	pozycja_bebenka += znak;
	pozycja_bebenka %= 26;
	return tablica[pozycja_bebenka];
};

char translacja_odwrotna(char* tablica, int pozycja_bebenka, char znak)
{
	for(int i=0; i<26; i++)
	{
		if (tablica[(pozycja_bebenka+i)%26] == znak)
			return i + 'A';
	}
	return -1;
};


int main()
{
	char* tablica1 = "EKMFLGDQVZNTOWYHXUSPAIBRCJ";
	char* tablica2 = "AJDKSIRUXBLHWTMCQGZNPYFVOE";
	char* tablica3 = "YRUHQSLDPXNGOKMIEBFZCWVJAT";

	FILE* plik_zrodlowy = fopen("a.txt", "r");
	FILE* plik_docelowy = fopen("aszyfreed.txt", "w");

	//sprawdzic czy pliki udalo sie otworzyc (=!NULL)

	// poz_bebenka_1 + pozbebenka_2*26 = klucz
	int pozycja_bebenka_1 = 0, pozycja_bebenka_2 = 0;

	char znak;
	while(fread(&znak, 1, 1, plik_zrodlowy))
	{
		znak = toupper(znak);
		if (znak <= 'Z' && znak >= 'A')
		{
			znak = translacja(tablica1, pozycja_bebenka_1, znak);
			znak = translacja(tablica2, pozycja_bebenka_2, znak);
			znak = translacja(tablica3, 0, znak);
			znak = translacja_odwrotna(tablica2, pozycja_bebenka_2, znak);
			znak = translacja_odwrotna(tablica1, pozycja_bebenka_1, znak);

			pozycja_bebenka_1++;
			if (pozycja_bebenka_1 == 26)
			{
				pozycja_bebenka_2 = (pozycja_bebenka_2+1)%26;
				pozycja_bebenka_1 = 0;
			}
		}
		//printf("%c", znak);
		fwrite(&znak, 1, 1, plik_docelowy);
	}

	fclose(plik_zrodlowy);
	fclose(plik_docelowy);
	return 0;
}