Création Manipulateurs ostream pour une classe spécifique

Question

J'ai une classe qui est dérivée de ostream:

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

Je veux faire un manipulateur (par exemple do_something), qui travaille spécifiquement à cette classe, comme ceci:

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

j'ai fait ce qui suit:

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

Cela fonctionne, mais est plutôt laid. J'ai essayé ce qui suit:

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

Cela ne fonctionne pas. J'ai également essayé une autre approche:

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

Cela ne fonctionne toujours pas.

Answer 1

Vous devez ajouter le support pour les manipulateurs dans votre classe:

#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; 
}