Domanda
Quando provo a compilare il codice
istream in;
if (argc==1)
in=cin;
else
{
ifstream ifn(argv[1]);
in=ifn;
}
gcc non riesce, lamentando che operator=
è privato. C'è un modo per impostare un istream
a valori diversi in base a una condizione?
Soluzione
È possibile sostituire streambuf di cin con un altro, e in alcuni programmi di questo è più semplice che la strategia generale di passare intorno iStreams senza fare riferimento a cin direttamente.
int main(int argc, char* argv[]) {
ifstream input;
streambuf* orig_cin = 0;
if (argc >= 2) {
input.open(argv[1]);
if (!input) return 1;
orig_cin = cin.rdbuf(input.rdbuf());
cin.tie(0); // tied to cout by default
}
try {
// normal program using cin
}
catch (...) {
if (orig_cin) cin.rdbuf(orig_cin);
throw;
}
return 0;
}
Anche se è estremamente raro da usare cin dopo il controllo lascia principale, sopra try-catch evita un comportamento indefinito se questo è qualcosa che il tuo programma potrebbe fare.
Altri suggerimenti
Si potrebbe utilizzare un puntatore per in
, ad es .:
istream *in;
ifstream ifn;
if (argc==1) {
in=&cin;
} else {
ifn.open(argv[1]);
in=&ifn;
}
Quindi, non è forse lamenta "nessun costruttore appropriato disponibile"? In ogni modo, è possibile modificare come di seguito.
void Read(istream& is)
{
string line;
while (getline(is, line))
cout << line;
}
int main(int argc, char* argv[])
{
if (argc == 1)
Read(cin);
else
{
ifstream in("sample.txt");
Read(in);
}
}
Non si può influenzare i flussi come questo. Quello che si vuole raggiungere può essere ottenuto utilizzando un puntatore a un istream però.
#include <fstream>
#include <istream>
#include <iostream>
using namespace std;
int main(int argc, char *argv[])
{
istream *in;
// Must be declared here for scope reasons
ifstream ifn;
// No argument, use cin
if (argc == 1) in = &cin;
// Argument given, open the file and use it
else {
ifn.open(argv[1]);
in = &ifn;
}
return 0;
// You can now use 'in'
// ...
}