#include <iostream>
#include <fstream>
#include <vector>
#include <thread>
#include <string>

using namespace std;

vector<unsigned char> vektor_obrade;

void pisi(int br_blokova, int offset,char* ulazniFajl, char* kljuc)
{

    char* buffer = new char[256];
    char* kljuc_za_xor = new char[256];
    char* rez = new char[256];


    ifstream file;                                  //otvaranje fajla koji se obradjuje
    file.open(ulazniFajl, ios::binary | ios::in);
    file.seekg(offset);

    ifstream key_file;                              //otvaranje fajla u kom je kljuc za obradu prvog fajla
    key_file.open(kljuc, ios::binary|ios::in);

    if(!file.is_open() || !key_file.is_open())
    {
        cout<<"Greska pri otvaranju nekog od fajlova."<<endl;
    }


    for(int i = 0; i < br_blokova && !file.eof(); i++)
    {
        file.read(buffer,256);
        key_file.read(kljuc_za_xor, 256);


        for(int j = 0; j <= 255; j++)
        {

            vektor_obrade[offset+j+i] = buffer[j] ^ kljuc_za_xor[j];
            //    rez[i] = buffer[i] + 1;
        }

    }

}

void napravi_fajl(char* ulazniFajl, char* izlazni, char* kljuc, int br_niti){

    int br_koraka;
    vector<thread> niti;

    ifstream file;                                                   // otvaranje fajla koji se obradjuje
    file.open(ulazniFajl, ios::binary | ios::in | ios::ate);        //  otvori fajl i pici odmah na kraj da znam koji mi je offset/vel fajla
    vektor_obrade.resize(file.tellg());                            //   uzmi velicinu fajla

    int velicina_fajla = file.tellg();
    cout << "velicina fajla: " << velicina_fajla << endl;
    br_koraka = file.tellg()/(br_niti*256);                         // broj blokova koji ce svaka nit da izvrsi

    file.close();
     for(int i = 0; i < br_niti; i++){
        cout << "kreiram nit: " << i << endl;
        niti.push_back(thread(pisi,br_koraka,br_koraka*i*256,ulazniFajl, izlazni, kljuc));              //offset - odakle krece svaka od niti

    }

    for(auto it = niti.begin(); it != niti.end(); it++){

        it->join();

    }

}


int main(int argc, char* argv[])
{
    char* ulazniFajl = argv[1];
    char* kljuc      = argv[2];
    char* izlazni    = argv[3];
    int br_niti      = atoi(argv[4]);

    vector<thread> niti;

    napravi_fajl(ulazniFajl, izlazni, kljuc,br_niti);

   // int br_koraka = velicina/br_niti;

    return 0;
}