source/textwriter.cpp

   1 #include <msp/io/print.h>
   2 #include <msp/strings/utils.h>
   3 #include "statement.h"
   4 #include "textwriter.h"
   5
   6 using namespace std;
   7
   8 namespace Msp {
   9 namespace DataFile {
  10
  11 TextWriter::TextWriter(IO::Base &o):
  12         WriterMode(o)
  13 { }
  14
  15 void TextWriter::write(const Statement &st)
  16 {
  17         write_(st, 0);
  18 }
  19
  20 void TextWriter::write_(const Statement &st, unsigned level)
  21 {
  22         string indent(level, '\t');
  23
  24         IO::print(out, "%s%s", indent, st.keyword);
  25         for(ValueArray::const_iterator i = st.args.begin(); i!=st.args.end(); ++i)
  26         {
  27                 out.put(' ');
  28                 if(i->get_signature()==StringType::signature)
  29                         IO::print(out, "\"%s\"", c_escape(i->get<StringType::Store>(), false));
  30                 else if(i->get_signature()==BoolType::signature)
  31                         out.write(i->get<BoolType::Store>() ? "true" : "false");
  32                 else if(i->get_signature()==IntType::signature)
  33                         out.write(lexical_cast(i->get<IntType::Store>()));
  34                 else if(i->get_signature()==FloatType::signature)
  35                         out.write(format("%15g", (i->get<FloatType::Store>())));
  36                 else if(i->get_signature()==SymbolType::signature)
  37                 {
  38                         string name = i->get<SymbolType::Store>().name;
  39                         if(isdigit(name[0]))
  40                                 out.write("\\"+name);
  41                         else
  42                                 out.write(name);
  43                 }
  44         }
  45         if(!st.sub.empty())
  46         {
  47                 IO::print(out, "\n%s{\n", indent);
  48                 for(list<Statement>::const_iterator i = st.sub.begin(); i!=st.sub.end(); ++i)
  49                         write_(*i, level+1);
  50                 IO::print(out, "%s}", indent);
  51         }
  52         out.write(";\n", 2);
  53 }
  54
  55 } // namespace DataFile
  56 } // namespace Msp