cpp/utils/functions.cpp

#include <vector>
#include <array>
#include <chrono>
#include <iostream>
#include <cmath>
#include <utility>


int nearest_index(const std::vector<double> &x_ref, const double x, int start=0) {
    while (x > x_ref[start+1]) {
        start++;
    }
    return start;
}

double lerp(const std::vector<double>& x_ref, const std::vector<double>& y_ref, const double x, const int i) {
    /*
     * Linear interpolation between two
     */
    const double x_left = x_ref[i];
    const double y_left = y_ref[i];
    const double x_right = x_ref[i+1];
    const double y_right = y_ref[i+1];

    const double dydx = (y_right - y_left) / ( x_right - x_left );

    return y_left + dydx * (x - x_left);
}


std::chrono::time_point<std::chrono::system_clock> printSteps(
    /*
     * Prints roughly every 10 seconds how many runs were done and gives a time estimation
     */
    const std::chrono::time_point<std::chrono::system_clock> last_print_out,
    const std::chrono::time_point<std::chrono::system_clock> start,
    const int total,
    const int steps
    ) {
    const auto now = std::chrono::high_resolution_clock::now();

    if (const std::chrono::duration<float> duration = now - last_print_out; duration.count() < 10.) {
        return last_print_out;
    }

    const std::chrono::duration<float> duration = now - start;
    const auto passed = duration.count();


    std::cout << steps << " of " << total << " steps: " << passed << "s passed; ~" << passed * static_cast<float>(total-steps) / static_cast<float>(steps) << "s remaining\n";

    return now;
}