decorator.hpp

Go to the documentation of this file.

//====================================================
//     File data
//====================================================
//====================================================
//     Preprocessor settings
//====================================================
#pragma once
#ifndef OSMANIP_MANIPULATORS_DECORATOR_HPP
#define OSMANIP_MANIPULATORS_DECORATOR_HPP
 
//====================================================
//     Headers
//====================================================
 
// My headers
#include <osmanip/manipulators/colsty.hpp>
#include <osmanip/manipulators/common.hpp>
#include <osmanip/utility/generic.hpp>
#include <osmanip/utility/iostream.hpp>
 
// STD headers
#include <string>
#include <string_view>
#include <unordered_map>
#include <vector>
 
namespace osm {
 
    //====================================================
    //     Decorator
    //====================================================
    class Decorator {
        public:
 
            // Constructors and destructor
            Decorator();
            ~Decorator();
 
            // Setters
            void setColor(const std::string &color, std::ostream &os = osm::cout);
            void setStyle(const std::string &style, std::ostream &os = osm::cout);
            void resetColor(std::ostream &os = osm::cout);
            void resetStyle(std::ostream &os = osm::cout);
            void removeStyle(std::string_view style, std::ostream &os = osm::cout);
            void resetFeatures(std::ostream &os = osm::cout);
 
            // Getters
            std::string getColor(std::ostream &os = osm::cout);
            std::string getStyle(std::ostream &os = osm::cout);
            std::unordered_map<std::ostream *, std::string> getColorList();
            std::unordered_map<std::ostream *, std::string> getStyleList();
            std::ostream &getCurrentStream();
 
            // Operators
            const Decorator &operator()(std::ostream &os = osm::cout);
 
        private:
 
            // Members
            std::unordered_map<std::ostream *, std::string> colors, styles;
            std::ostream *current_stream;
    };
 
    //====================================================
    //     Operator <<
    //====================================================
    template <typename T>
    std::ostream &operator<<(Decorator my_shell, const T &elem) {
        if (my_shell.getColor(my_shell.getCurrentStream()) != "") {
            my_shell.getCurrentStream() << feat(col, my_shell.getColor(my_shell.getCurrentStream()));
        }
        if (my_shell.getStyle(my_shell.getCurrentStream()) != "") {
            std::vector<std::string> list_of_styles{
                osm::split_string(my_shell.getStyle(my_shell.getCurrentStream()), " ")};
            for (auto elem: list_of_styles) {
                my_shell.getCurrentStream() << feat(sty, elem);
            }
        }
        my_shell.getCurrentStream() << elem << feat(rst, "all");
 
        return my_shell.getCurrentStream();
    }
}  // namespace osm
 
#endif