C ++ общий интерфейс для "cin" и "File"

Question

Существует ли общий интерфейс для ввода cin и файла?

Я хочу создать программу с необязательным параметром

prog [input-file]

Если указан входной файл, то он долженчитать из файла, а если нет - читать из cin.

Из того, что я могу сказать, они оба реализуют istream.Как бы вы настроили его так, чтобы я мог сделать что-то вроде in >> var, где in - это istream.

Answer 1

#include <iostream>
#include <fstream>

int main(int argc, char **argv)
{
    std::ifstream f;
    if (argc >= 2) {
        f.open(argv[1]);
    }
    std::istream &in = (argc >= 2) ? f : std::cin;

    // use in here
}

Вы можете перенести часть этой работы в вспомогательный класс, чтобы прояснить, что происходит (обратите внимание, что это немного отличается в случае, когда файл не может быть открыт):

#include <iostream>
#include <fstream>

class ifstream_or_cin_t {
    std::ifstream f;

public:
    ifstream_or_cin_t(const char *filename)
    {
        if (filename) {
            f.open(filename);
        }
    }

    operator std::istream &() { return f.is_open() ? f : std::cin; }
};

static void do_input(std::istream &in)
{
    // use in...
}

int main(int argc, char **argv)
{
    do_input(
        ifstream_or_cin_t((argc >= 2) ? argv[1] : NULL));
}

Answer 2

Вы можете написать функцию, которая принимает ссылку на std::istream:

void do_input(std::istream& the_istream)
{
    int my_awesome_variable;
    the_istream >> my_awesome_variable;
}

Пример использования:

int main()
{
    // reading from stdin:
    do_input(std::cin);

    // reading from a file:
    std::ifstream fs("test.txt");
    do_input(fs);
}

Answer 3

Мой 2P стоит:

#include <iostream>
#include <fstream>
#include <string>

extern char const* findFlag(int,char*[],char const*);
int main(int argc,char* argv[])
{
    std::string     isFile  = findFlag(argc,argv,"-f");
    std::ifstream   file;

    if (isFile != "")
    {   file.open(isFile.c_str());
    }
    std::istream&   data    = isFile != "" ? file : std::cin;


}

Answer 4

istream& input = cin;

или

inputFileStream ifstream(fileName);
istream& input = inputFileStream;

Однако я думаю, что это не очень хороший метод, поскольку у cin нет close(), тогда как у ifstream есть.