CoutSink.h

Go to the documentation of this file.

//
// Class CoutSink
//
// Copyright (c) 2010 - 2013, Yves Ineichen, ETH Zürich
// All rights reserved
//
// Implemented as part of the PhD thesis
// "Toward massively parallel multi-objective optimization with application to
// particle accelerators" (https://doi.org/10.3929/ethz-a-009792359)
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#ifndef __TRACE_COUT_SINK_H__
#define __TRACE_COUT_SINK_H__
 
#include <iostream>
#include <string>
 
#include "Util/AnsiColors.h"
#include "Util/Trace/TraceComponent.h"
 
class CoutSink : public TraceComponent {
 
public:
 
 
    CoutSink(std::string prefix = "")
        : TraceComponent("CoutSink")
        , prefix_(prefix) {
 
        clear_color_ = AnsiColors::Reset;
    }
 
    ~CoutSink()
    {}
 
 
    void setColor(std::string color)      { color_ = color; }
    void setClearColor(std::string color) { clear_color_ = color; }
 
 
    void execute(std::ostringstream &dump) {
        std::cout << color_ << prefix_
                  << dump.str()
                  << clear_color_ << std::flush;
    }
 
private:
 
    std::string prefix_;
    std::string color_;
    std::string clear_color_;
 
};
 
#endif