运算符的一般实现<<根据转储成员函数的功能

时间:2015-12-09 12:08:35

标签: c++

我所有的类都实现了dump成员函数,例如:

struct A {
    template <typename charT>
    std::basic_ostream<charT> &
    dump(std::basic_ostream<charT> &o) const {
        return (o << x);
    }
    int x = 5;
};

我想为所有这些类实现一次operator<<函数:

template<typename charT, typename T>
std::basic_ostream<charT> &
operator<< (std::basic_ostream<charT> &o, const T &t) {
    return t.dump(o);
}

问题是此模板捕获了所有类型,包括标准类型。有办法解决这个问题吗?

4 个答案:

答案 0 :(得分:9)

template <typename T, typename charT>
auto operator<< (std::basic_ostream<charT> & str, const T & t) -> decltype(t.dump(str))
{
    static_assert(std::is_same
                   <decltype(t.dump(str)), 
                    std::basic_ostream<charT> &>::value, 
                  ".dump(ostream&) does not return ostream& !");

    return t.dump(str);
}

仅对定义适当operator<<成员的类型重载dump

编辑:添加了static_assert以获得更好的消息。

答案 1 :(得分:3)

你可以创建一个空的基类,比如说:

struct HasDump {};

HasDump成为所有课程的基础,即:

struct A : HasDump ( ...

然后将operator<<std::enable_ifstd::is_base_of一起打包,以便仅在HasDumpT的基础时适用。

(我没有专注于C ++一两年,所以这个建议可能有点生疏)

答案 2 :(得分:1)

一般来说,这是明智之举,IMO:

struct A {
    int x = 5;

    friend std::ostream & operator<<(std::ostream &os, const A& a){
        return (os << a.x);
    }
};

原因:'friend' functions and << operator overloading: What is the proper way to overload an operator for a class?

如果你真的想要一个专用的转储方法,你可以定义一个基类“收集”可转储的对象。

答案 3 :(得分:1)

添加此内容以获得乐趣。如果您碰巧有多个方法在不同的类上打印/转储:

#include <iostream>
#include <type_traits>

namespace tests {

    // this is a utility class to help us figure out whether a class
    // has a member function called dump that takes a reference to 
    // an ostream
    struct has_dump
    {
        // We will only be checking the TYPE of the returned
        // value of these functions, so there is no need (in fact we
        // *must not*) to provide a definition
        template<class T, class Char>
        static auto test(const T* t, std::basic_ostream<Char>& os)
        -> decltype(t->dump(os), std::true_type());

        // the comma operator in decltype works in the same way as the
        // comma operator everywhere else. It simply evaluates each
        // expression and returns the result of the last one
        // so if t->dump(os) is valid, the expression is equivalent to
        // decltype(std::true_type()) which is the type yielded by default-
        // constructing a true_type... which is true_type!


        // The above decltype will fail to compile if t->dump(os) is not
        // a valid expression. In this case, the compiler will fall back
        // to selecting this next function. this is because the overload
        // that takes a const T* is *more specific* than the one that
        // takes (...) [any arguments] so the compiler will prefer it
        // if it's well formed.

        // this one could be written like this:
        // template<class T, class Char>
        // static std::false_type test(...);
        // I just happen to use the same syntax as the first one to
        // show that they are related.

        template<class T, class Char>
        static auto test(...) -> decltype(std::false_type());
    };

    // ditto for T::print(ostream&) const    
    struct has_print
    {
        template<class T, class Char>
        static auto test(const T* t, std::basic_ostream<Char>& os)
        -> decltype(t->print(os), std::true_type());

        template<class T, class Char>
        static auto test(...) -> decltype(std::false_type());
    };
}

// constexpr means it's evaluated at compile time. This means we can
// use the result in a template expansion.
// depending on whether the expression t->dump(os) is well formed or not
// (see above) it will either return a std::true_type::value (true!) 
// or a std::false_type::value (false!)

template<class T, class Char>
static constexpr bool has_dump() {
    // the reinterpret cast stuff is so we can pass a reference without
    // actually constructing an object. remember we're being evaluated
    // during compile time, so we can't go creating ostream objects here - 
    // they don't have constexpr constructors.
    return decltype(tests::has_dump::test<T, Char>(nullptr,
                                                   *reinterpret_cast<std::basic_ostream<Char>*>(0)))::value;
}

template<class T, class Char>
static constexpr bool has_print() {
    return decltype(tests::has_print::test<T, Char>(nullptr,
                                                   *reinterpret_cast<std::basic_ostream<Char>*>(0)))::value;
}

// so now we can use our constexpr functions has_dump<> and has_print<>
// in a template expansion, because they are known at compile time.
// std::enable_if_t will ensure that the template function is only
// well formed if our condition is true, so we avoid duplicate
// definitions.
// the use of the alternative operator representations make this
// a little more readable IMHO: http://en.cppreference.com/w/cpp/language/operator_alternative

template<class T, class Char>
auto operator<< (std::basic_ostream<Char>& os, const T& t)
-> std::enable_if_t< has_dump<T, Char>() and not has_print<T, Char>(), std::basic_ostream<Char>&>
{
    t.dump(os);
    return os;
}

template<class T, class Char>
auto operator<< (std::basic_ostream<Char>& os, const T& t)
-> std::enable_if_t< has_print<T, Char>() and not has_dump<T, Char>(), std::basic_ostream<Char>&>
{
    t.print(os);
    return os;
}

template<class T, class Char>
auto operator<< (std::basic_ostream<Char>& os, const T& t)
-> std::enable_if_t< has_print<T, Char>() and has_dump<T, Char>(), std::basic_ostream<Char>&>
{
    // because of the above test, this function is only compiled
    // if T has both dump(ostream&) and print(ostream&) defined.

    t.dump(os);
    os << ":";
    t.print(os);
    return os;
}



struct base
{
    template<class Char>
    void dump(std::basic_ostream<Char>& os) const
    {
        os << x;
    }

    int x = 5;
};
namespace animals
{
    class donkey : public base
    {

    public:
        template<class Char>
        void dump(std::basic_ostream<Char>& s) const {
            s << "donkey: ";
            base::dump(s);
        }
    };

    class horse // not dumpable, but is printable
    {
    public:
        template<class Char>
        void print(std::basic_ostream<Char>& s) const {
            s << "horse";
        }
    };

    // this class provides both dump() and print()        
    class banana : public base
    {
    public:

        void dump(std::ostream& os) const {
            os << "banana!?!";
            base::dump(os);
        }

        void print(std::ostream& os) const {
            os << ":printed";
        }

    };
}


auto main() -> int
{
    using namespace std;

    animals::donkey d;
    animals::horse h;

    cout << d << ", " << h << ", " << animals::banana() << endl;

    return 0;
}

预期产出:

donkey: 5, horse, banana!?!5::printed