From 174f2c1fe750349860172e0cbc1a87f06d8cfffb Mon Sep 17 00:00:00 2001 From: Jesse Beder Date: Fri, 10 Jul 2009 04:25:11 +0000 Subject: [PATCH] Set up the parse utility program to read from standard input if no file is specified --- util/parse.cpp | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/util/parse.cpp b/util/parse.cpp index 12f4e6c..8a1080d 100644 --- a/util/parse.cpp +++ b/util/parse.cpp @@ -4,16 +4,17 @@ int main(int argc, char **argv) { - if(argc != 2) { - std::cout << "Usage: " << argv[0] << " input-file\n"; - return 0; - } - - std::ifstream fin(argv[1]); + std::ifstream fin; + if(argc > 1) + fin.open(argv[1]); + + std::istream& input = (argc > 1 ? fin : std::cin); try { - YAML::Parser parser(fin); - YAML::Node doc; - parser.GetNextDocument(doc); + YAML::Parser parser(input); + while(parser) { + YAML::Node doc; + parser.GetNextDocument(doc); + } } catch(const YAML::Exception& e) { std::cerr << e.what() << "\n"; }