LArFormattingHelper.h

Go to the documentation of this file.

#ifndef LAR_FORMATTING_HELPER_H
#define LAR_FORMATTING_HELPER_H 1

#include "Pandora/PandoraInternal.h"
#include "Pandora/StatusCodes.h"

#include <iostream>
#include <sstream>
#include <string>
#include <vector>

namespace lar_content
{

class LArFormattingHelper
{
public:
    enum Color : unsigned int;
    enum Style : unsigned int;

    static void SetStyle(const Style style, std::ostream &stream = std::cout);

    static void SetColor(const Color color, std::ostream &stream = std::cout);

    static void ResetStyle(std::ostream &stream = std::cout);

    static void ResetColor(std::ostream &stream = std::cout);

    static void Reset(std::ostream &stream = std::cout);

    static void PrintFormatCharacter(const unsigned int code, std::ostream &stream = std::cout);

    static std::string GetFormatCharacter(const unsigned int code);

    static void PrintHeader(const std::string &title = "", const unsigned int width = 140);

    static void PrintRule(const unsigned int width = 140);

    enum Style : unsigned int
    {
        REGULAR = 0,
        BOLD = 1,
        DIM = 2,
        UNDERLINED = 4,
        INVERTED = 7
    };

    enum Color : unsigned int
    {
        DEFAULT = 39,
        BLACK = 30,
        RED = 31,
        GREEN = 32,
        YELLOW = 33,
        BLUE = 34,
        MAGENTA = 35,
        CYAN = 36,
        LIGHT_GRAY = 37,
        DARK_GRAY = 90,
        LIGHT_RED = 91,
        LIGHT_GREEN = 92,
        LIGHT_YELLOW = 93,
        LIGHT_BLUE = 94,
        LIGHT_MAGENTA = 95,
        LIGHT_CYAN = 96,
        WHITE = 97
    };

    class Table
    {
    public:
        Table(const pandora::StringVector &columnTitles, const unsigned int precision = 3);

        void Print() const;

        template <typename T>
        void AddElement(const T &value, const Style style = REGULAR, const Color color = DEFAULT);

    private:
        bool IsSeparatorColumn(const unsigned int column) const;

        void PrintColumnTitles() const;

        void PrintHorizontalLine() const;

        void PrintTableElements() const;

        void PrintTableCell(const std::string &value, const std::string &format, const unsigned int index) const;

        void CheckAndSetSeparatorColumn();

        void UpdateColumnWidth();

        const pandora::StringVector m_columnTitles; 
        const unsigned int m_precision;             
        pandora::StringVector m_elements;           
        pandora::StringVector m_format;             
        std::vector<unsigned int> m_widths;         
        std::stringstream m_stringstream;           
    };
};

//------------------------------------------------------------------------------------------------------------------------------------------

template <typename T>
inline void LArFormattingHelper::Table::AddElement(const T &value, const Style style, const Color color)
{
    this->CheckAndSetSeparatorColumn();

    if (!(m_stringstream << value))
        throw pandora::StatusCodeException(pandora::STATUS_CODE_INVALID_PARAMETER);

    m_elements.push_back(static_cast<std::string>(m_stringstream.str()));
    m_format.push_back(LArFormattingHelper::GetFormatCharacter(style) + LArFormattingHelper::GetFormatCharacter(color));
    m_stringstream.str("");

    this->UpdateColumnWidth();
    this->CheckAndSetSeparatorColumn();
}

} // namespace lar_content

#endif // #ifndef LAR_FORMATTING_HELPER_H