multi_progress_bar.hpp

Go to the documentation of this file.

//====================================================
//     File data
//====================================================
//====================================================
//     Preprocessor settings
//====================================================
#pragma once
#ifndef OSMANIP_PROGRESSBAR_MULTIPROGRESSBAR_HPP
#define OSMANIP_PROGRESSBAR_MULTIPROGRESSBAR_HPP
 
//====================================================
//     Headers
//====================================================
 
// My headers
#include <osmanip/manipulators/cursor.hpp>
#include <osmanip/utility/iostream.hpp>
 
// STD headers
#include <cstddef>
#include <cstdint>
#include <mutex>
#include <string>
#include <tuple>
#include <type_traits>
#include <utility>
 
namespace osm {
 
    //====================================================
    //     Structs
    //====================================================
 
    // indices
    template <size_t... Is>
    struct indices {};
 
    // gen_indices
    template <size_t N, size_t... Is>
    struct gen_indices : gen_indices<N - 1, N - 1, Is...> {};
 
    // gen_indices specialization
    template <size_t... Is>
    struct gen_indices<0, Is...> : indices<Is...> {};
 
    //====================================================
    //     make_MultiProgressBar
    //====================================================
    template <class... Indicators>
    class make_MultiProgressBar {
        public:
 
            // Parametric constructor
            template <class... Inds>
            make_MultiProgressBar(Inds &&...bars) : bars_{std::forward<Inds>(bars)...}, last_updated_index(0) {}
 
            // size
            static size_t size() { return sizeof...(Indicators); }
 
            // for_one
            template <class Func, class... Args>
            void for_one(size_t idx, Func &&func, Args &&...args) {
                call_one(idx, gen_indices<sizeof...(Indicators)>(), std::forward<Func>(func),
                         std::forward<Args>(args)...);
            }
 
            // for_each
            template <class Func, class... Args>
            void for_each(Func &&func, Args &&...args) {
                call_all(gen_indices<sizeof...(Indicators)>(), std::forward<Func>(func), std::forward<Args>(args)...);
            }
 
        private:
 
            // call_one
            template <size_t... Ids, class Func, class... Args>
            void call_one(size_t idx, indices<Ids...>, Func func, Args &&...args) {
                std::lock_guard<std::mutex> lock{mutex_};
                int32_t idx_delta = idx - last_updated_index;
                std::string direction;
 
                if (idx_delta < 0) {
                    direction = "up";
                    idx_delta = -idx_delta;
                } else {
                    direction = "down";
                }
                for (int32_t i = 0; i < idx_delta; i++) {
                    osm::cout << feat(crs, direction, 1);
                }
                last_updated_index = idx;
                [](...) {}
 
                ((idx == Ids &&
                  ((void)std::forward<Func>(func)(std::get<Ids>(bars_), std::forward<Args>(args)...), false))...);
            }
 
            // call_all
            template <size_t... Ids, class Func, class... Args>
            void call_all(indices<Ids...>, Func func, Args &&...args) {
                std::lock_guard<std::mutex> lock{mutex_};
                auto dummy = {(func(std::get<Ids>(bars_), args...), 0)...};
                (void)dummy;
            }
 
            // Attributes
            std::tuple<Indicators &...> bars_;
            std::mutex mutex_;
            uint32_t last_updated_index;
    };
 
    //====================================================
    //     MultiProgressBar
    //====================================================
    template <class... Indicators>
    make_MultiProgressBar<typename std::remove_reference<Indicators>::type...> MultiProgressBar(Indicators &&...inds) {
        return {std::forward<Indicators>(inds)...};
    }
 
    //====================================================
    //     type_identity
    //====================================================
    template <class T>
    struct type_identity {
            using type = T;
    };
 
    //====================================================
    //     updater
    //====================================================
    struct updater {
            template <template <class> class PB, class bar_type>
            auto operator()(PB<bar_type> &pb, typename type_identity<bar_type>::type v) const
                -> decltype(pb.update(bar_type{})) {
                return pb.update(v);
            }
    };
}  // namespace osm
 
#endif