Dotaz: C++ promenna pro stdin

[MArty]

Dobry den,

program, ktery vytvarim muze svuj vstup nacitat bud ze souboru anebo ze stdin. To urcuje promenne ifileName do ktere pomoci getopt nacitam nazev vstupniho souboru. Pokud neni nazev vstupniho souboru zadan, je hodnota promenne ifileName NULL a cteme ze stdin. Nize uvadim kod, ktery toto resi. Jedna se defacto o skoro dva stejne bloky kodu, coz neni moc dobre. Muzete mi prosim poradit, jakym zpusobem tento fragment prepsat, abych oba bloku co nejvice sloucil? Nejlepe by bylo zavest jednu promennou, ktera by byla kompatibilni s ifstream i cin a z ni jednotne cetl znaky pomoci get().

    char *ifileName;
    ifstream ifile;

    ...

    // cteme ze stdin
    if (ifileName == NULL)
    {
        while (cin)
        {
            c = cin.get();
            if (c != EOF)
            {
                HTInsert(&HT, c);
            }
        }
    }
    // cteme ze souboru
    else
    {
        ifile.open(ifileName);
        if (ifile.is_open())
        {
            while (ifile.good())
            {
                c = ifile.get();
                if (c != EOF)
                {
                    HTInsert(&HT, c);
                }
            }
            ifile.close();
        }
    }

---

---

Odpovědi

[Martin Tůma]

http://www.abclinuxu.cz/poradna/programovani/show/146406

[Sten]

std::ifstream a std::cin mají společného předka, std::istream:

void ctiData(std::istream &stream)
{
   ...
}


if (ifileName == NULL) {
    ctiData(std::cin);
} else {
    std::ifstream file(ifileName);
    ctiData(file);
}

Btw. v Linuxu je standardní vstup dostupný jako /dev/stdin, tzn. lze udělat tohle:

std::ifstream file(ifileName ? ifileName : "/dev/stdin");