Ottenere uno std :: ostream sia da std :: cout o std :: ofstream (file)
-
21-08-2019 - |
Domanda
Come faccio associare un std::ostream
a uno std::cout
o ad un oggetto std::ofstream
, a seconda di una certa condizione del programma? Anche se questo non valida per molte ragioni, vorrei realizzare qualcosa che è semanticamente equivalente al seguente:
std::ostream out = condition ? &std::cout : std::ofstream(filename);
Ho visto alcuni esempi che non sono sicura rispetto alle eccezioni, come ad esempio uno da http://www2.roguewave.com/support/docs/sourcepro/edition9/html/stdlibug/34-2.html :
int main(int argc, char *argv[])
{
std::ostream* fp; //1
if (argc > 1)
fp = new std::ofstream(argv[1]); //2
else
fp = &std::cout //3
*fp << "Hello world!" << std::endl; //4
if (fp!=&std::cout)
delete fp;
}
Qualcuno sa una soluzione migliore, sicura rispetto alle eccezioni?
Soluzione
std::streambuf * buf;
std::ofstream of;
if(!condition) {
of.open("file.txt");
buf = of.rdbuf();
} else {
buf = std::cout.rdbuf();
}
std::ostream out(buf);
che associa lo streambuf sottostante sia cout o il flusso di file di output di fuori. Dopo di che è possibile scrivere a "fuori" e finirà nella destinazione giusta. Se si desidera solo che tutto ciò che accade a std::cout
va in un file, è possibile fare partecipavano
std::ofstream file("file.txt");
std::streambuf * old = std::cout.rdbuf(file.rdbuf());
// do here output to std::cout
std::cout.rdbuf(old); // restore
Questo secondo metodo ha l'inconveniente che non è sicuro eccezione. È forse vuole scrivere una classe che fa questo usando RAII:
struct opiped {
opiped(std::streambuf * buf, std::ostream & os)
:os(os), old_buf(os.rdbuf(buf)) { }
~opiped() { os.rdbuf(old_buf); }
std::ostream& os;
std::streambuf * old_buf;
};
int main() {
// or: std::filebuf of;
// of.open("file.txt", std::ios_base::out);
std::ofstream of("file.txt");
{
// or: opiped raii(&of, std::cout);
opiped raii(of.rdbuf(), std::cout);
std::cout << "going into file" << std::endl;
}
std::cout << "going on screen" << std::endl;
}
Ora, qualunque cosa accada, std :: cout è in stato pulito.
Altri suggerimenti
Questa è sicura rispetto alle eccezioni:
void process(std::ostream &os);
int main(int argc, char *argv[]) {
std::ostream* fp = &cout;
std::ofstream fout;
if (argc > 1) {
fout.open(argv[1]);
fp = &fout;
}
process(*fp);
}
Modifica: Herb Sutter ha affrontato questo nell'articolo commutazione Streams (Guru della settimana) .
std::ofstream of;
std::ostream& out = condition ? std::cout : of.open(filename);
Essendo un novizio di C ++, non so se questo è sicuro rispetto alle eccezioni, ma ecco come faccio di solito è:
std::ostream& output = (condition)?*(new std::ofstream(filename)):std::cout;
Il semplice codice seguente funziona per me:
int main(int argc, char const *argv[]){
std::ofstream outF;
if (argc > 1)
{
outF = std::ofstream(argv[1], std::ofstream::out);
}
std::ostream& os = (argc > 1)? outF : std::cout;
}