Пытается переключиться между std :: cout и std :: ifstream

Question

Я хочу читать входы от std::cin или std::ifstream, которые определяются из командной строки. Команда выглядит как ./run 1 или ./run 2. Прямо сейчас, я должен написать две почти похожие функции, основанные на режиме чтения.

void read1() 
{ 
    int a, b; 
    while (std::cin >> a >> b) { 
    // do something 
    } 
} 

или

void read2() 
{ 
    int a, b; 
    std::ifstream fin("file.txt"); 
    while (fin >> a >> b) { 
    // do something 
    } 
} 

Для больших петель трудно поддерживать обе функции, так как часть цикла является общим и единственным отличием является входным источником.

Как интегрировать две функции?

Answer 1

std::cin и std::ifstream оба std::istream s, так что вы могли бы решить эту проблему с помощью функции, действует по ссылке std::istream. Это будет работать в std::cin, std::ifstream экземпляров, а также любые другие std::istream S:

void read(std::istream& input) 
{ 
    while (input >> a >> b) { .... } 
} 

затем переключают на стороне вызывающего абонента.

if (something) 
{ 
    read(std::cin); 
} else 
{ 
    isfream input(....); 
    read(input); 
} 

Answer 2

как станд :: ifstream и станд :: КИН inputstreams, вы можете вызвать функцию с IStream в качестве аргумента