external_libraries/yaml-cpp-0.2.6/util/parse.cpp

   1 #include "yaml-cpp/yaml.h"
   2 #include "yaml-cpp/eventhandler.h"
   3 #include <fstream>
   4 #include <iostream>
   5 #include <vector>
   6
   7 struct Params {
   8         bool hasFile;
   9         std::string fileName;
  10 };
  11
  12 Params ParseArgs(int argc, char **argv) {
  13         Params p;
  14
  15         std::vector<std::string> args(argv + 1, argv + argc);
  16
  17         return p;
  18 }
  19
  20 class NullEventHandler: public YAML::EventHandler
  21 {
  22 public:
  23         virtual void OnDocumentStart(const YAML::Mark&) {}
  24         virtual void OnDocumentEnd() {}
  25
  26         virtual void OnNull(const YAML::Mark&, YAML::anchor_t) {}
  27         virtual void OnAlias(const YAML::Mark&, YAML::anchor_t) {}
  28         virtual void OnScalar(const YAML::Mark&, const std::string&, YAML::anchor_t, const std::string&) {}
  29
  30         virtual void OnSequenceStart(const YAML::Mark&, const std::string&, YAML::anchor_t) {}
  31         virtual void OnSequenceEnd() {}
  32
  33         virtual void OnMapStart(const YAML::Mark&, const std::string&, YAML::anchor_t) {}
  34         virtual void OnMapEnd() {}
  35 };
  36
  37 void parse(std::istream& input)
  38 {
  39         try {
  40                 YAML::Parser parser(input);
  41                 YAML::Node doc;
  42                 NullEventHandler handler;
  43                 while(parser.GetNextDocument(doc)) {
  44                         YAML::Emitter emitter;
  45                         emitter << doc;
  46                         std::cout << emitter.c_str() << "\n";
  47                 }
  48         } catch(const YAML::Exception& e) {
  49                 std::cerr << e.what() << "\n";
  50         }
  51 }
  52
  53 int main(int argc, char **argv)
  54 {
  55         Params p = ParseArgs(argc, argv);
  56
  57         if(argc > 1) {
  58                 std::ifstream fin;
  59                 fin.open(argv[1]);
  60                 parse(fin);
  61         } else {
  62                 parse(std::cin);
  63         }
  64
  65         return 0;
  66 }