:- module(inputOutput, [aaa/0, getTest/1, getTraining/1, printEpoch/3]).

aaa.

%% ----------------- Parte com funçoes de Escrita ----------------------------

% -- Escreve os dados da biases e do output
% nos seus respectivos arquivos
writeData(BH, BO, WH, WO):-
    writeFile("biasesHidden.txt", BH),
    writeFile("weightsHidden.txt", WH),

    writeFile("biasesOut.txt", BO),
    writeFile("weightsOut.txt", WO).

% Escreve no arquivo
writeFile(File, Matrix):-
    matrix_to_list(Matrix, Arr),

    clear(File),
    open(File, append, Stream),
    processW(Stream, Arr), nl,
    close(Stream).

% Cria uma lista com os caracteres presentes no Arquivo
processW(_, []).
processW(Stream, [Line|Rest]) :-
    write(Stream, Line), nl,
    processW(Stream, Rest).

% Transforma as listas da matriz em string
matrix_to_list([], []).
matrix_to_list([H|T], [C|R]):-
    atomic_list_concat(H, " ", C),
    matrix_to_list(T, R).

% Limpa o arquivo
clear(File):- open(File, write, Stream), write(Stream, ""), close(Stream).

% ------------------ Parte com funçoes de Leitura -------------------------

% Retorna a matriz de Treino
getTrainings(Train):- readTraining("trainings.txt", Train).

% Retorna a matriz de Testes
getTests(Test):- readTraining("tests.txt", Test).

% Retorna a imagem de execucao
getImage(Image):- readFile("image.txt", Image).

% -- Retorna dados da Hidden
% Retorna a matriz de Treino
getBiasesHidden(Train):- readFile("biasesHidden.txt", Train).
% Retorna a matriz de Treino
getWeightsHidden(Train):- readFile("weightsHidden.txt", Train).

% -- Retorna dados do output
% Retorna a matriz de Treino
getBiasesOut(Train):- readFile("biasesOut.txt", Train).
% Retorna a matriz de Treino
getWeightsOut(Train):- readFile("biasesOut.txt", Train).

% Lê o arquivo e transforma-o em uma matriz de numeros
readFile(File, Arr):-
    open(File, read, Stream),

    read_line_to_string(Stream, Line),

    % Setar a quantidade de linhas a ser lida
    processR(Line, Stream, F, 0, 1000),
    arr_to_matrix(F, Arr),
    close(Stream).

% Lê um arquivo em forma de Treino
readTraining(File, Arr):-
    open(File, read, Stream),

    read_line_to_string(Stream, Line),
    writeln(Line),
    % Setar a quantidade de linhas a ser lida
    processR(Line, Stream, F, 0, 1000),
    arr_to_training(F, Arr),
    close(Stream).

% Transforma as strings ,presentes na lista, em listas
arr_to_matrix([], []).
arr_to_matrix([H|T], [C|R]) :- string_to_arr(H, C), arr_to_matrix(T, R).

% Transforma as strings ,presentes na lista, em listas de treinos
% no formato [resultado, imagem]
arr_to_training([], []).
arr_to_training([H|T], [C|R]) :-
    string_to_matrix_column(H, [[Label|_]|Mt]),
    C = [Label, Mt],
    arr_to_training(T, R).

% Cria uma lista com os caracteres presentes no Arquivo
processR(end_of_file, _, [], _, _).
processR(_, _, [], Limit, Limit).
processR(Line, Stream, [Line|Ans], Cnt, Limit) :-
    read_line_to_string(Stream, Line1),
    Bef is Cnt + 1,
    processR(Line1, Stream, Ans, Bef, Limit).

% Transforma todos os caracteres em Atomos
arr_to_atom([], []).
arr_to_atom([H|T], [C|B]):- string_to_atom(H, C), arr_to_atom(T, B).

% Transforma todos os Atomos da lista em numeros
arr_to_number([], []).
arr_to_number([H|T], [C|B]):- atom_number(H, C), arr_to_number(T, B).

% Transforma todos os Atomos da lista em numeros
arr_to_column_matrix([], []).
arr_to_column_matrix([H|T], [[C]|B]):- atom_number(H, C), arr_to_column_matrix(T, B).

% Transforma uma string em uma lista de numeros
string_to_arr(Str, Arr):-
    split_string(Str, " ", " ", Aux1),

    arr_to_atom(Aux1 , Aux2),
    arr_to_number(Aux2, Arr).

% Transforma uma string em uma lista de numeros
string_to_matrix_column(Str, Arr):-
    split_string(Str, " ", " ", Aux1),

    arr_to_atom(Aux1 , Aux2),
    arr_to_column_matrix(Aux2, Arr).


% Encontra o index de um elemento em uma lista
indexOf([Element|_], Element, 0).
indexOf([_|Tail], Element, Index):-
  indexOf(Tail, Element, Index1),
  Index is Index1+1.

% Pega o melhor sigmoid e retorna a resposta definitiva
getBestSigmoid(Values, Ans):-
    max_list(Values, bigger),
    indexOf(Values, bigger, Idx),
    IdxP is Idx + 1,
    Ans = [IdxP, bigger],
    Sigmoid = (bigger * 100 / sum_list(Values)),

    %imprimindo os resultados finais
    write("Resposta definitiva: "),
    write("["), write(IdxP), write(" - "),
    format('~2f', Sigmoid),
    write("%]").


printEpoch(CorrectAmount, Amount, TotalAmount) :- write("Epoch #"),
                                                  write(Amount),
                                                  write(" - "),
                                                  write(CorrectAmount),
                                                  write(" / "),
                                                  writeln(TotalAmount).
save(Network).