c++ - 获取发送到 std::cout 的最后一个字符

Question

我需要在 main() 执行结束时检查发送到 stdout(通过 std::cout)的最后一个字符是否为 '\n'(或平台特定的行尾)。如何测试这个？可以安全地假设未使用 C 风格的 io(如 printf)。

程序是 C++ 的 REPL。它计算 C++ 表达式(或语句)并在标准输出上打印结果。输出总是以单个换行符终止是可取的。

Answer 1

这与@Kevin 给出的答案类似。但是我相信它更适合您的需求。除了使用您的一些流代替 cout - 您可以将 std::cout 中的 streambuf 替换为您自己的流:

int main() {
   std::streambuf* cbuf = std::cout.rdbuf(); // back up cout's streambuf
   std::cout.flush();
   keep_last_char_outbuf keep_last_buf(cbuf);
   std::cout.rdbuf(&keep_last_buf);          // assign your streambuf to cout

   std::cout << "ala ma kota\n";

   char last_char = keep_last_buf.get_last_char();
   if (last_char == '\r' || last_char == '\n')
      std::cout << "\nLast char was newline: " << int(last_char) << "\n";
   else
      std::cout << "\nLast char: '" << last_char << "'\n";

   std::cout << "ala ma kota";
   last_char = keep_last_buf.get_last_char();
   if (last_char == '\r' || last_char == '\n')
      std::cout << "\nLast char was newline: " << int(last_char) << "\n";
   else
      std::cout << "\nLast char: '" << last_char << "'\n";

   std::cout.rdbuf(cbuf); // restore cout's original streambuf
}

预期输出:

ala ma kota

Last char was newline: 10
ala ma kota
Last char: 'a'

编写这样的 class keep_last_char_outbuf 的任务不是很容易，寻找装饰器模式和 streambuf界面。

如果你没有时间玩这个 - 看看我的建议 ideone link

class keep_last_char_outbuf : public std::streambuf {
public:
    keep_last_char_outbuf(std::streambuf* buf) : buf(buf), last_char(traits_type::eof()) {
        // no buffering, overflow on every char
        setp(0, 0);
    }
    char get_last_char() const { return last_char; }

    virtual int_type overflow(int_type c) {
        buf->sputc(c);
        last_char = c;
        return c;
    }
private:
    std::streambuf* buf;
    char last_char;
};