source/parser.cpp

   1 #include <msp/strings/format.h>
   2 #include "binaryparser.h"
   3 #include "dataerror.h"
   4 #include "parser.h"
   5 #include "statement.h"
   6 #include "textparser.h"
   7
   8 using namespace std;
   9
  10 namespace Msp {
  11 namespace DataFile {
  12
  13 Parser::Parser(IO::Base &i, const string &s):
  14         in(i),
  15         main_src(s),
  16         src(s),
  17         good(true),
  18         mode(new TextParser(in, src))
  19 { }
  20
  21 Parser::~Parser()
  22 {
  23         delete mode;
  24 }
  25
  26 Statement Parser::parse(bool raw)
  27 {
  28         if(!good)
  29                 throw logic_error("Parser::parse() !good");
  30
  31         try
  32         {
  33                 while(1)
  34                 {
  35                         Statement st = mode->parse();
  36                         if(!st.keyword.compare(0, 2, "__"))
  37                         {
  38                                 st.control = true;
  39                                 process_control_statement(st);
  40                         }
  41
  42                         if(raw || !st.control)
  43                                 return st;
  44                         else if(!good)  // This will occur with an __end statement
  45                                 return Statement();
  46                 }
  47         }
  48         catch(const exception &e)
  49         {
  50                 good = false;
  51                 if(dynamic_cast<const data_error *>(&e))
  52                         throw;
  53                 else
  54                         throw data_error(src, in.get_line_number(), e);
  55         }
  56 }
  57
  58 void Parser::process_control_statement(const Statement &st)
  59 {
  60         if(st.keyword=="__bin")
  61         {
  62                 delete mode;
  63                 mode = new BinaryParser(in, src);
  64         }
  65         else if(st.keyword=="__text")
  66         {
  67                 delete mode;
  68                 mode = new TextParser(in, src);
  69         }
  70         else if(st.keyword=="__z")
  71                 in.set_decompress();
  72         else if(st.keyword=="__src")
  73         {
  74                 string s = st.args[0].get<string>();
  75                 if(s.empty())
  76                         src = main_src;
  77                 else
  78                         src = format("%s[%s]", main_src, s);
  79         }
  80         else if(st.keyword=="__end")
  81                 good = false;
  82         else
  83                 mode->process_control_statement(st);
  84 }
  85
  86 } // namespace DataFile
  87 } // namespace Msp