ProgressBarInterceptor.cpp File Reference

#include "ProgressBarInterceptor.h"
#include <spdlog/fmt/chrono.h>
#include <sys/ioctl.h>
#include <filesystem>
#include <iostream>
#include "io/logger/Logger.h"
#include "simulation/SimulationParams.h"
#include "utils/FormatTime.h"

Include dependency graph for ProgressBarInterceptor.cpp:

Go to the source code of this file.

Functions
int	displayed_width (const char *p)
void	printProgress (const std::filesystem::path &input_file_path, size_t percentage, size_t iteration, size_t expected_iterations, int estimated_remaining_seconds=-1, double particle_updates_per_second=-1, bool finished=false)

Function Documentation

displayed_width()

int displayed_width

(

const char *

p

)

Definition at line 13 of file ProgressBarInterceptor.cpp.

                                   {
    int result = 0;
    for (; *p; ++p) {
        if (p[0] == '\e' && p[1] == '[')
            while (*p != 'm')
                if (*p)
                    ++p;
                else
                    throw std::runtime_error("string terminates inside ANSI colour sequence");
        else
            ++result;
    }
    return result;
}

printProgress()

void printProgress	(	const std::filesystem::path &	input_file_path,
		size_t	percentage,
		size_t	iteration,
		size_t	expected_iterations,
		int	estimated_remaining_seconds = -1,
		double	particle_updates_per_second = -1,
		bool	finished = false
	)

Definition at line 28 of file ProgressBarInterceptor.cpp.

                                                                                                                         {
    struct winsize size {};
    ioctl(STDOUT_FILENO, TIOCGWINSZ, &size);
 
    auto should_progress_bar_length = std::min(std::max(size.ws_col - 90, 0), 100);
    auto length_full_progress_bar = static_cast<size_t>(percentage * static_cast<double>(should_progress_bar_length) / 100.0);
 
    auto progress_bar = fmt::format("[{}{}]", ansi_blue_bold + std::string(length_full_progress_bar, '#'),
                                    std::string(should_progress_bar_length - length_full_progress_bar, ' ') + ansi_end);
 
    std::string line = fmt::format("{} {}/{} {:>4} ETA: {} MUP/s: {} [{}]", progress_bar, iteration, expected_iterations,
                                   ansi_bright_white_bold + std::to_string(percentage) + "%" + ansi_end,
                                   ansi_bright_white_bold + format_seconds_eta(estimated_remaining_seconds) + ansi_end,
                                   ansi_bright_white_bold + format_mup_s(particle_updates_per_second) + ansi_end,
                                   ansi_bright_white_bold + std::filesystem::path(input_file_path).stem().string() + ansi_end);
 
    if (displayed_width(line.c_str()) > size.ws_col) {
        line = line.substr(0, size.ws_col - 3) + "...";
    }
 
    std::cout << "\33[2K\r" << line << std::flush;
 
    if (finished) {
        std::cout << std::endl;
    }
}