// etabench
// Copyright (C) 2022 xaizek <xaizek@posteo.net>
//
// This file is part of etabench.
//
// etabench is free software: you can redistribute it and/or modify
// it under the terms of version 3 of the GNU General Public License
// as published by the Free Software Foundation.
//
// etabench is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with etabench.  If not, see <https://www.gnu.org/licenses/>.

#include "EtaBench.hpp"

#include <memory>
#include <stdexcept>
#include <utility>
#include <vector>

#include <fmt/core.h>

#include "Report.hpp"
#include "core.hpp"

void
EtaBench::run(int total, Report &report)
{
    for (auto &alg : algs) {
        for (auto &profile : profiles) {
            bench(total, *alg, *profile, report);
        }
    }
}

void
EtaBench::addProfile(std::unique_ptr<Profile> profile)
{ profiles.emplace_back(std::move(profile)); }

void
EtaBench::addAlg(std::unique_ptr<EtaAlg> alg)
{ algs.emplace_back(std::move(alg)); }

void
EtaBench::bench(int total, EtaAlg &alg, Profile &profile, Report &report)
{
    std::vector<EtaPoint> points;

    profile.reset();
    alg.reset();

    int time = 0;
    int current = 0;
    while (current < total) {
        const int speed = profile(time);
        if (speed < 0) {
            throw std::logic_error(fmt::format("Negative speed from '{}'!",
                                               profile.getName()));
        }

        // Ignore t=0, because no algorithm can handle it meaningfully.
        if (current != 0) {
            const int estimate = alg.estimate(current, total, time);
            if (estimate < 0) {
                throw std::logic_error(
                    fmt::format("Negative ETA from '{}' on '{}'({})!",
                                alg.getName(), profile.getName(), time)
                );
            }

            points.emplace_back(time, estimate, speed);
        }

        current += speed;
        ++time;
    }

    report.addResult(alg, profile, points);
}