Info.h

//
// Created by henri on 23.11.23.
//
 
#ifndef IDOL_BRANCHANDBOUND_INFO_H
#define IDOL_BRANCHANDBOUND_INFO_H
 
#include "Factory.h"
#include "idol/general/optimizers/logs.h"
 
namespace idol::Logs::BranchAndBound {
    template<class NodeInfoT> class Info;
}
 
template<class NodeInfoT = idol::DefaultNodeInfo>
class idol::Logs::BranchAndBound::Info : public Factory<NodeInfoT> {
 
    static constexpr auto double_precision = 5;
    static constexpr auto n_sections = 3;
 
    static constexpr auto log_width_explored = 6;
    static constexpr auto log_width_open = 6;
    static constexpr auto log_width_total_time = 12;
 
    static constexpr auto log_width_current_node_depth = 6;
    static constexpr auto log_width_problem_best_bound = 10;
    static constexpr auto log_width_problem_best_obj = 10;
    static constexpr auto log_width_problem_rel_gap = 10;
    static constexpr auto log_width_problem_abs_gap = 10;
 
    static constexpr auto log_width_current_node_status = 12;
    static constexpr auto log_width_current_node_reason = 12;
    static constexpr auto log_width_current_node_obj = 10;
 
    static constexpr auto log_section_width_algorithm = (log_width_explored + log_width_open + log_width_total_time);
    static constexpr auto log_section_width_problem = (log_width_problem_best_bound + log_width_problem_best_obj + log_width_problem_rel_gap + log_width_problem_abs_gap);
    static constexpr auto current_node_space = (log_width_current_node_status + log_width_current_node_reason + log_width_current_node_obj + log_width_current_node_depth);
    static constexpr auto table_space = log_section_width_algorithm + log_section_width_problem + current_node_space + n_sections * 3 + 1;
 
 
    std::optional<double> m_log_frequency_in_seconds;
    std::optional<double> m_node_logs;
public:
    class Strategy : public Factory<NodeInfoT>::Strategy {
        double m_last_log_timestamp = 0;
        double m_frequency_in_seconds;
        bool m_node_logs;
        bool m_header_has_been_printed = false;
        bool m_root_node_has_been_printed = false;
    protected:
        void log_header();
    public:
 
        Strategy(Optimizers::BranchAndBound<NodeInfoT>& t_parent, double t_frequency_in_seconds, bool t_node_logs);
 
        void initialize() override;
 
        void log_node_after_solve(const Node <NodeInfoT> &t_node) override;
 
        void log_root_node(const Node<NodeInfoT>& t_node);
 
        void log_after_termination() override;
    };
 
    Strategy *operator()(Optimizers::BranchAndBound<NodeInfoT>& t_parent) const override;
 
    Factory<NodeInfoT> *clone() const override;
 
    Info& with_frequency_in_seconds(double t_frequency);
 
    Info& with_node_logs(bool t_value);
};
 
template<class NodeInfoT>
idol::Logs::BranchAndBound::Info<NodeInfoT> &idol::Logs::BranchAndBound::Info<NodeInfoT>::with_node_logs(bool t_value) {
 
    if (m_node_logs.has_value()) {
        throw Exception("Node logs have already been configured.");
    }
 
    m_node_logs = t_value;
 
    return *this;
}
 
template<class NodeInfoT>
idol::Logs::BranchAndBound::Info<NodeInfoT> &idol::Logs::BranchAndBound::Info<NodeInfoT>::with_frequency_in_seconds(double t_frequency) {
 
    if (m_log_frequency_in_seconds.has_value()) {
        throw Exception("Log frequency has already been configured.");
    }
 
    m_log_frequency_in_seconds = t_frequency;
 
    return *this;
}
 
template<class NodeInfoT>
typename idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy *idol::Logs::BranchAndBound::Info<NodeInfoT>::operator()(Optimizers::BranchAndBound<NodeInfoT>& t_parent) const {
    return new Strategy(t_parent,
                        m_log_frequency_in_seconds.has_value() ? m_log_frequency_in_seconds.value() : 5,
                        m_node_logs.has_value() ? m_node_logs.value() : false);
}
 
template<class NodeInfoT>
idol::Logs::BranchAndBound::Factory<NodeInfoT> *idol::Logs::BranchAndBound::Info<NodeInfoT>::clone() const {
    return new Info<NodeInfoT>(*this);
}
 
template<class NodeInfoT>
idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy::Strategy(Optimizers::BranchAndBound<NodeInfoT> &t_parent, double t_frequency_in_seconds, bool t_node_logs)
        : Factory<NodeInfoT>::Strategy(t_parent),
          m_frequency_in_seconds(t_frequency_in_seconds),
          m_node_logs(t_node_logs) {
 
}
 
template<class NodeInfoT>
void idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy::initialize() {
 
    m_last_log_timestamp = 0;
    m_header_has_been_printed = false;
    m_root_node_has_been_printed = false;
 
    if (!this->parent().get_param_logs()) {
        return;
    }
 
    Factory<NodeInfoT>::Strategy::initialize();
 
    std::cout << "Solving root node with " << this->parent().relaxation().optimizer().name() << "...\n" << std::endl;
 
    auto& branch_and_bound = this->parent();
    branch_and_bound.relaxation().optimizer().set_param_logs(true);
}
 
template<class NodeInfoT>
void idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy::log_node_after_solve(const Node <NodeInfoT> &t_node) {
 
    if (!this->parent().get_param_logs()) {
        return;
    }
 
    Factory<NodeInfoT>::Strategy::Strategy::log_node_after_solve(t_node);
 
    const auto& parent = this->parent();
 
    const double total_time = parent.time().count();
 
    if (!m_root_node_has_been_printed && t_node.id() == 0) {
        if (t_node.id() == 0) {
            log_root_node(t_node);
        }
        return;
    }
 
    if (t_node.id() > 0 && total_time - m_last_log_timestamp < m_frequency_in_seconds) {
        return;
    }
 
    log_header();
 
    m_last_log_timestamp = total_time;
 
    std::cout << " | ";
    std::cout << std::setw(log_width_explored) << parent.n_solved_nodes();
    std::cout << std::setw(log_width_open) << parent.n_active_nodes();
    std::cout << std::setw(log_width_total_time) << parent.time().count();
 
    // Problem
    std::cout << " | ";
    std::cout << std::setw(log_width_problem_best_bound) << pretty_double(parent.get_best_bound(), double_precision);
    std::cout << std::setw(log_width_problem_best_obj) << pretty_double(parent.get_best_obj(), double_precision);
    std::cout << std::setw(log_width_problem_rel_gap) << pretty_double(parent.get_relative_gap() * 100, double_precision);
    std::cout << std::setw(log_width_problem_abs_gap) << pretty_double(parent.get_absolute_gap(), double_precision);
 
    // Current Node
    const auto value = t_node.id() == -1 ? t_node.info().best_obj() : t_node.info().best_bound();
    std::cout << " | ";
    std::cout << std::setw(log_width_current_node_depth) << t_node.level();
    std::cout << std::setw(log_width_current_node_status) << t_node.info().status();
    std::cout << std::setw(log_width_current_node_reason);
    if (t_node.id() == -1) {
        std::cout << "H";
    } else {
        std::cout << t_node.info().reason();
    }
    std::cout << std::setw(log_width_current_node_obj) << pretty_double(value, double_precision);
    std::cout << " | ";
 
    std::cout << std::endl;
 
}
 
template<class NodeInfoT>
void idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy::log_root_node(const Node<NodeInfoT> &t_node) {
 
    if (!this->parent().get_param_logs()) {
        return;
    }
 
    auto& branch_and_bound = this->parent();
    const double total_time = branch_and_bound.time().count();
 
    std::cout
            << "Root relaxation: objective " << t_node.info().best_bound()
            << ", " << total_time << " seconds\n"
            << std::endl;
 
    if (!m_node_logs) {
        branch_and_bound.relaxation().optimizer().set_param_logs(false);
    }
 
    m_root_node_has_been_printed = true;
 
}
 
template<class NodeInfoT>
void idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy::log_header() {
 
    if (m_header_has_been_printed) {
        return;
    }
 
    std::cout << ' ';center(std::cout, "*", table_space, '*') << '\n';
    std::cout << ' ';center(std::cout, " Branch-and-Bound Algorithm ", table_space) << '\n';
    std::cout << ' ';center(std::cout, "*", table_space, '*') << '\n';
 
    std::cout << " | ";
    center(std::cout, "Algorithm", log_section_width_algorithm);std::cout << " | ";
    center(std::cout, "Problem", log_section_width_problem);std::cout << " | ";
    center(std::cout, "Current Node", current_node_space);std::cout << " | ";
    std::cout << '\n';
 
    std::cout << " | ";
    std::cout << std::setw(log_width_explored) << "Expl.";
    std::cout << std::setw(log_width_open) << "Open";
    std::cout << std::setw(log_width_total_time) << "Time";
 
    // Problem
    std::cout << " | ";
    std::cout << std::setw(log_width_problem_best_bound) << "Bound";
    std::cout << std::setw(log_width_problem_best_obj) << "Obj";
    std::cout << std::setw(log_width_problem_rel_gap) << "Gap (%)";
    std::cout << std::setw(log_width_problem_abs_gap) << "Abs.";
 
    // Current Node
    std::cout << " | ";
    std::cout << std::setw(log_width_current_node_depth) << "Depth";
    std::cout << std::setw(log_width_current_node_status) << "Status";
    std::cout << std::setw(log_width_current_node_reason) << "Reason";
    std::cout << std::setw(log_width_current_node_obj) << "Value";
    std::cout << " | ";
 
    std::cout << std::endl;
 
    m_header_has_been_printed = true;
}
 
template<class NodeInfoT>
void idol::Logs::BranchAndBound::Info<NodeInfoT>::Strategy::log_after_termination() {
 
    if (!this->parent().get_param_logs()) {
        return;
    }
 
    Factory<NodeInfoT>::Strategy::Strategy::log_after_termination();
 
    auto& parent = this->parent();
 
    std::cout << "Explored " << parent.n_solved_nodes() << " nodes in " << parent.time().count() << " seconds\n" << std::endl;
 
    const unsigned int n_solutions = parent.get_n_solutions();
 
    std::cout << "Solution count " << n_solutions << ":";
 
    for (unsigned int i = 0 ; i < n_solutions ; ++i) {
        parent.set_solution_index(i);
        std::cout << '\t' << parent.get_best_obj();
    }
 
    parent.set_solution_index(0);
 
    std::cout << std::endl;
 
    const auto status = parent.get_status();
    const auto reason = parent.get_reason();
 
    switch (reason) {
        case TimeLimit:
            std::cout << "Time limit was reached." << std::endl;
            break;
        case IterLimit:
            std::cout << "Iteration limit was reached." << std::endl;
            break;
        case ObjLimit:
            std::cout << "Objective limit was reached." << std::endl;
            break;
        case Numerical:
            std::cout << "There was unrecoverable numerical troubles during the execution of the algorithm." << std::endl;
            break;
        case NotSpecified:
            std::cout << "The reason for termination is not specified." << std::endl;
            break;
        default:;
    }
 
    switch (status) {
        case Optimal:
            std::cout << "Optimal solution found (tolerance " << parent.get_tol_mip_relative_gap() << ")" << std::endl;
            break;
        case Feasible:
            std::cout << "Feasible solution found" << std::endl;
            break;
        case Infeasible:
            std::cout << "No feasible solution found" << std::endl;
            break;
        case InfOrUnbnd:
            std::cout << "Problem is infeasible or unbounded" << std::endl;
            break;
        case Unbounded:
            std::cout << "Problem is unbounded" << std::endl;
            break;
        case Fail:
            std::cout << "There was unrecoverable exceptions." << std::endl;
            break;
        case SubOptimal:
            std::cout << "Sub-optimal solution found" << std::endl;
            break;
        default:;
    }
 
    std::cout
            << "Best objective " << pretty_double(parent.get_best_obj(), double_precision)
            << ", best bound " << pretty_double(parent.get_best_bound(), double_precision)
            << ", gap " << pretty_double(parent.get_relative_gap() * 100, double_precision) << " %"
            << std::endl;
 
}
 
#endif //IDOL_BRANCHANDBOUND_INFO_H