openvino/tools/legacy/benchmark_app/progress_bar.hpp

// Copyright (C) 2018-2021 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

#pragma once

#include <memory>
#include "console_progress.hpp"

/// @brief Responsible for progress bar handling within the benchmark_app
class ProgressBar {
public:
    explicit ProgressBar(size_t totalNum, bool streamOutput = false, bool progressEnabled = false) {
        _bar.reset(new ConsoleProgress(totalNum, streamOutput));
        _streamOutput = streamOutput;
        _isFinished = true;
        _progressEnabled = progressEnabled;
    }

    void addProgress(size_t num) {
        _isFinished = false;
        if (_progressEnabled) {
            _bar->addProgress(static_cast<int>(num));
        }
    }

    void finish(size_t num = 0) {
        if (num > 0) {
            addProgress(num);
        }
        _isFinished = true;
        _bar->finish();
        if (_progressEnabled) {
            std::cout << std::endl;
        }
    }

    void newBar(size_t totalNum) {
        if (_isFinished) {
            _bar.reset(new ConsoleProgress(totalNum, _streamOutput));
        } else {
            throw std::logic_error("Cannot create a new bar. Current bar is still in progress");
        }
    }

private:
    std::unique_ptr<ConsoleProgress> _bar;
    bool _streamOutput;
    bool _isFinished;
    bool _progressEnabled;
};