std::endl 的重载处理？

Question

我想定义一个类MyStream以便：

MyStream myStream;
myStream << 1 << 2 << 3 << std::endl << 5 << 6 << std::endl << 7 << 8 << std::endl;

给出输出

[blah]123
[blah]56
[blah]78

基本上，我想在前面插入一个“[blah]”，然后在每个非终止std::endl之后插入？

这里的困难不是逻辑管理，而是检测和重载std::endl的处理。 有没有一种优雅的方法来做到这一点？

谢谢！

编辑：我不需要关于逻辑管理的建议。 我需要知道如何检测/重载std::endl打印。

Answer 1

您需要做的是编写自己的流缓冲区：当流缓冲区被刷新时，您输出前缀字符和流的内容。

以下工作是因为std::endl导致以下内容。

1. 将'\\n'添加到流中。

2. 在流上调用flush()

3. 这会在流缓冲区上调用pubsync() 。

   1. 这调用了虚方法sync()
   2. 覆盖这个虚拟方法来完成你想要的工作。

#include <iostream>
#include <sstream>

class MyStream: public std::ostream
{
    // Write a stream buffer that prefixes each line with Plop
    class MyStreamBuf: public std::stringbuf
    {
        std::ostream&   output;
        public:
            MyStreamBuf(std::ostream& str)
                :output(str)
            {}
            ~MyStreamBuf() {
                if (pbase() != pptr()) {
                    putOutput();
                }
            }
   
        // When we sync the stream with the output. 
        // 1) Output Plop then the buffer
        // 2) Reset the buffer
        // 3) flush the actual output stream we are using.
        virtual int sync() {
            putOutput();
            return 0;
        }
        void putOutput() {
            // Called by destructor.
            // destructor can not call virtual methods.
            output << "[blah]" << str();
            str("");
            output.flush();
        }
    };

    // My Stream just uses a version of my special buffer
    MyStreamBuf buffer;
    public:
        MyStream(std::ostream& str)
            :std::ostream(&buffer)
            ,buffer(str)
        {
        }
};


int main()
{
    MyStream myStream(std::cout);
    myStream << 1 << 2 << 3 << std::endl << 5 << 6 << std::endl << 7 << 8 << std::endl;
}

> ./a.out
[blah]123 
[blah]56 
[blah]78
>

Answer 2

MyStream类的重载运算符必须设置一个 previous-printed-token-was-endl 标志。

然后，如果打印下一个对象，则可以在其前面插入[blah] 。

std::endl是一个接受并返回对std::ostream的引用的函数。 要检测它是否已转移到您的流中，您必须在您的类型和这样的函数之间重载operator<< ：

MyStream& operator<<( std::ostream&(*f)(std::ostream&) )
{
    std::cout << f;

    if( f == std::endl )
    {
        _lastTokenWasEndl = true;
    }

    return *this;
}

Answer 3

原则上同意尼尔。

您想要更改缓冲区的行为，因为这是扩展 iostream 的唯一方法。 endl这样做：

flush(__os.put(__os.widen('\n')));

widen返回一个字符，所以你不能把你的字符串放在那里。 put调用putc ，它不是一个虚函数，只是偶尔挂钩overflow 。 您可以在调用缓冲区的sync flush处进行拦截。 您需要拦截并更改所有换行符，因为它们已overflow或手动sync ，并将它们转换为您的字符串。

设计覆盖缓冲区类很麻烦，因为basic_streambuf期望直接访问其缓冲区内存。 这可以防止您轻松地将 I/O 请求传递给预先存在的basic_streambuf 。 您需要尝试一下，假设您知道流缓冲区类，并从中派生。 （ cin和cout不能保证使用basic_filebuf ，据我所知。）然后，只需添加virtual overflow和sync 。 （请参阅 §27.5.2.4.5/3 和 27.5.2.4.2/7。）执行替换可能需要额外的空间，因此请注意提前分配。

- 或者 -

只需在您自己的命名空间中声明一个新的endl ，或者更好的是，一个根本不称为endl的操纵器！

Answer 4

我使用函数指针。 对于不习惯 C 的人来说，这听起来很可怕，但在大多数情况下，它的效率要高得多。 下面是一个例子：

#include <iostream>

class Foo
{
public:
    Foo& operator<<(const char* str) { std::cout << str; return *this; }
    // If your compiler allows it, you can omit the "fun" from *fun below.  It'll make it an anonymous parameter, though...
    Foo& operator<<(std::ostream& (*fun)(std::ostream&)) { std::cout << std::endl; }
} foo;

int main(int argc,char **argv)
{
    foo << "This is a test!" << std::endl;
    return 0;
}

如果你真的想要，你可以检查 endl 的地址以确认你没有得到一些其他的 void/void 函数，但我认为在大多数情况下不值得。 我希望这有帮助。

Answer 5

与其尝试修改std::endl的行为， std::endl创建一个过滤流缓冲来完成这项工作。 James Kanze 有一个示例，展示了如何在每个输出行的开头插入时间戳。 只需稍加修改即可将其更改为每行所需的任何前缀。

Answer 6

我有同样的问题，我认为 Potatoswatter 的第二个答案有好处：“只需在你自己的命名空间中声明一个新的 endl，或者更好的是，一个根本不叫 endl 的操纵器！”

所以我发现了如何编写一个并不难的自定义操纵器：

#include <sstream>
#include <iostream>

class log_t : public std::ostringstream
{
    public:
};


std::ostream& custom_endl(std::ostream& out)
{
    log_t *log = dynamic_cast<log_t*>(&out);
    if (log)
    {
        std::cout << "custom endl succeeded.\n";
    }
    out << std::endl;
    return out;
}

std::ostream& custom_flush(std::ostream& out)
{
    log_t *log = dynamic_cast<log_t*>(&out);
    if (log)
    {
        std::cout << "custom flush succeeded.\n";
    }
    out << std::flush;
    return out;
}

int main(int argc, char **argv)
{
    log_t log;

    log << "custom endl test" << custom_endl;
    log << "custom flush test" << custom_flush;

    std::cout << "Contents of log:\n" << log.str() << std::endl;
}

这是输出：

custom endl succeeded.
custom flush succeeded.
Contents of log:
custom endl test
custom flush test

在这里，我创建了两个自定义操纵器，一个处理endl ，一个处理flush 。 您可以向这两个函数添加任何您想要的处理，因为您有一个指向log_t对象的指针。

Answer 7

您不能更改std::endl - 顾名思义，它是 C++ 标准库的一部分，并且其行为是固定的。 当流收到行尾时，您需要更改流本身的行为。 就个人而言，我认为这不值得付出努力，但如果您想涉足这个领域，我强烈建议您阅读Standard C++ IOStreams & Locales一书。