c++ - 如何实现带参数的自定义流操纵器？

Question

例如，我想要一个流操纵器，我可以将 uint8_t(unsigned char)传递给并让它输出(例如):

00 或 0f 或 a6。

我知道可以使用不带参数的流操纵器:

std::ostream &hex_format(std::ostream &out)
{
    out << std::hex << std::setfill('0') << std::setw(2);
    return out;
}

uint8_t x {15};
std::cout << hex_format << static_cast<int>(x); // should produce "0f"

但是我怎样才能创建一个接受参数的操纵器呢？像这样的东西:

std::ostream &hex_format(std::ostream &out, uint8_t x)
{
    out << std::hex << std::setfill('0') << std::setw(2)
        << static_cast<int>(x);
    return out;
}

uint8_t x {15};
std::cout << hex_format(x); // (want to) produce "0f"

Answer 1

使用操纵器类:

struct hex_format {
    std::uint8_t number;

    hex_format(std::uint8_t x)
        :number{x}
    {
    }
};

然后定义合适的运算符:

template <typename CharT, typename Traits>
decltype(auto) operator<<(std::basic_ostream<CharT, Traits>& os, hex_format rhs)
{
    return os << std::hex << std::setfill('0') << std::setw(2) << static_cast<int>(x);
}

---

题外话:其实这样会全局修改流的状态，影响后面的流操作。您可以通过保存状态来解决此问题:

template <typename CharT, typename Traits>
decltype(auto) operator<<(std::basic_ostream<CharT, Traits>& os, hex_format rhs)
{
    auto flags = os.flags();
    os << std::hex << std::setfill('0') << std::setw(2) << static_cast<int>(x);
    os.flags(flags);
    return os;
}