//
// arma-flow/executor.cpp
//
// @author CismonX
//

#include "executor.hpp"
#include "factory.hpp"
#include "writer.hpp"

namespace flow
{
    executor::executor() : factory_(factory::get()) {}

    void executor::execute(int argc, char** argv) const
    {
        // Get components.
        auto args = factory_->get_args();
        auto input = factory_->get_reader();
        auto calc = factory_->get_calc();
        auto writer = factory_->get_writer();

        // Parse options.
        args->parse(argc, argv);

        // Read data from file.
        std::string path_to_nodes, path_to_edges;
        if (!args->input_file_path(path_to_nodes, path_to_edges)) {
            args->help();
        }
        const auto remove = args->remove_first_line();
        if (!input->from_csv_file(path_to_nodes, remove)) {
            writer::error("Failed to read node data from file.");
        }
        const auto nodes = input->get_mat();
        if (!input->from_csv_file(path_to_edges, remove)) {
            writer::error("Failed to read edge data from file.");
        }
        const auto edges = input->get_mat();

        // Get options.
        const auto verbose = args->verbose();
        auto max = 0U;
        if (!args->max_iterations(max) && verbose) {
            writer::notice("Max number of iterations not specified. Defaulted to 100.");
        }
        auto epsilon = 0.0;
        if (!args->accuracy(epsilon) && verbose) {
            writer::notice("Accuracy not specified. Defaulted to 0.00001.");
        }
        if (epsilon < 0 || epsilon > 1) {
            writer::error("Invalid accuracy.");
        }
        std::string output_path;
        if (!args->output_file_path(output_path) && verbose) {
            writer::notice("Output file path not specified. Defaulted to result-*.csv.");
        }
        writer->set_output_path_prefix(output_path);
        unsigned short_circuit_node;
        const auto short_circuit = args->short_circuit(short_circuit_node);
        std::complex<double> transition_impedance;
        if (short_circuit && !args->transition_impedance(transition_impedance) && verbose) {
            writer::notice("Transition impedance not specified, Defaulted to 0.");
        }
        const auto ignore_load = args->ignore_load();

        // Initialize calculation.
        calc->init(nodes, edges, verbose, epsilon, short_circuit, ignore_load,
            short_circuit_node, transition_impedance);
        const auto admittance = calc->node_admittance();
        writer->to_csv_file("node-admittance-real.csv", admittance.first);
        writer->to_csv_file("node-admittance-imag.csv", admittance.second);
        calc->iterate_init();

        // Do iteration.
        unsigned num_iterations;
        do {
            num_iterations = calc->solve();
            if (calc->get_max() <= epsilon) {
                writer::println("Finished. Total number of iterations: ", num_iterations);
                const auto result = calc->result();
                if (verbose) {
                    writer::println("Result [V, theta(in rads), P, Q]:");
                    writer::print_mat(result);
                }
                writer->to_csv_file("flow.csv", result, "V,theta,P,Q");
                goto flow_done;
            }
        } while (num_iterations < max);
        writer::error("Exceeds max number of iterations. Aborted.");

        flow_done:

        // Calculate three-phase short circuit.
        if (!short_circuit) {
            return;
        }
        const auto impedance = calc->node_impedance();
        writer->to_csv_file("node-impedance-real.csv", impedance.first);
        writer->to_csv_file("node-impedance-imag.csv", impedance.second);
        const auto i_f = calc->short_circuit_current();
        writer::print_complex("Three-phase short circuit current: ", i_f);
        const auto u_f = calc->short_circuit_voltage();
        writer->to_csv_file("short-circuit-voltage.csv", u_f, "Ui(real),Ui(imag)");
        const auto i = calc->short_circuit_edge_current();
        writer->to_csv_file("short-circuit-edge-current.csv", i, "Iij(real),Iij(imag)");
    }
}