創建ostream的操縱特定類

Question

我有一個是從ostream的派生類：

class my_ostream: public std::ostream 
{ 
    // ... 
} 

我想打一個操縱（例如do_something），該工程具體到這個類，像這樣：

my_ostream s; 
s << "some text" << do_something << "some more text"; 

我做了以下內容：

std::ostream &do_something(std::ostream &os) 
{ 
    my_ostream *s = dynamic_cast<my_ostream*>(&os); 
    if (s != NULL) 
    { 
     // do something 
    } 
    return os; 
} 

這工作，但比較難看。我試過以下內容：

my_ostream &do_something(my_ostream &s) 
{ 
    // do something 
    return s; 
} 

這不起作用。我也嘗試了另一種方法：

class my_ostream: public std::ostream 
{ 
    // ... 
    my_ostream &operator<<(const do_something & x) 
    { 
     // do something 
     return *this; 
    } 
} 

這仍然不起作用。

Answer 1

您需要在您的類中添加用於操縱支持：

#include<iostream> 
class my_ostream : public std::ostream 
{ 
    public: 
    std::string prefix; 

    my_ostream():prefix("*"){} 

    // manipulator support here: 
    my_ostream& operator<<(my_ostream&(*f)(my_ostream&)){ 
     f(*this); 
     return *this; 
    } 
}; 

my_ostream& operator<<(my_ostream &st, const std::string &s){ 
     std::cout << st.prefix << s; 
     return st; 
} 

// manipulator: clear prefix 
my_ostream& noprefix(my_ostream &st){ 
    st.prefix=""; 
} 

int main(){ 
    my_ostream s; 
    std::string str1("text"); 
    std::string str2("text"); 
    s << str1 << noprefix << str2; 
}