Chris@16: // ----------------------------------------------------------------------------
Chris@16: // Copyright (C) 2002-2006 Marcin Kalicinski
Chris@16: //
Chris@16: // Distributed under the Boost Software License, Version 1.0. 
Chris@16: // (See accompanying file LICENSE_1_0.txt or copy at 
Chris@16: // http://www.boost.org/LICENSE_1_0.txt)
Chris@16: //
Chris@16: // For more information, see www.boost.org
Chris@16: // ----------------------------------------------------------------------------
Chris@16: #ifndef BOOST_PROPERTY_TREE_DETAIL_INFO_PARSER_WRITE_HPP_INCLUDED
Chris@16: #define BOOST_PROPERTY_TREE_DETAIL_INFO_PARSER_WRITE_HPP_INCLUDED
Chris@16: 
Chris@16: #include "boost/property_tree/ptree.hpp"
Chris@16: #include "boost/property_tree/detail/info_parser_utils.hpp"
Chris@16: #include <string>
Chris@16: 
Chris@16: namespace boost { namespace property_tree { namespace info_parser
Chris@16: {
Chris@16:     template<class Ch>
Chris@16:     void write_info_indent(std::basic_ostream<Ch> &stream,
Chris@16:           int indent,
Chris@16:           const info_writer_settings<Ch> &settings
Chris@16:           )
Chris@16:     {
Chris@16:         stream << std::basic_string<Ch>(indent * settings.indent_count, settings.indent_char);
Chris@16:     }
Chris@16:     
Chris@16:     // Create necessary escape sequences from illegal characters
Chris@16:     template<class Ch>
Chris@16:     std::basic_string<Ch> create_escapes(const std::basic_string<Ch> &s)
Chris@16:     {
Chris@16:         std::basic_string<Ch> result;
Chris@16:         typename std::basic_string<Ch>::const_iterator b = s.begin();
Chris@16:         typename std::basic_string<Ch>::const_iterator e = s.end();
Chris@16:         while (b != e)
Chris@16:         {
Chris@16:             if (*b == Ch('\0')) result += Ch('\\'), result += Ch('0');
Chris@16:             else if (*b == Ch('\a')) result += Ch('\\'), result += Ch('a');
Chris@16:             else if (*b == Ch('\b')) result += Ch('\\'), result += Ch('b');
Chris@16:             else if (*b == Ch('\f')) result += Ch('\\'), result += Ch('f');
Chris@16:             else if (*b == Ch('\n')) result += Ch('\\'), result += Ch('n');
Chris@16:             else if (*b == Ch('\r')) result += Ch('\\'), result += Ch('r');
Chris@16:             else if (*b == Ch('\v')) result += Ch('\\'), result += Ch('v');
Chris@16:             else if (*b == Ch('"')) result += Ch('\\'), result += Ch('"');
Chris@16:             else if (*b == Ch('\\')) result += Ch('\\'), result += Ch('\\');
Chris@16:             else
Chris@16:                 result += *b;
Chris@16:             ++b;
Chris@16:         }
Chris@16:         return result;
Chris@16:     }
Chris@16: 
Chris@16:     template<class Ch>
Chris@16:     bool is_simple_key(const std::basic_string<Ch> &key)
Chris@16:     {
Chris@16:         const static std::basic_string<Ch> chars = convert_chtype<Ch, char>(" \t{};\n\"");
Chris@16:         return !key.empty() && key.find_first_of(chars) == key.npos;
Chris@16:     }
Chris@16:     
Chris@16:     template<class Ch>
Chris@16:     bool is_simple_data(const std::basic_string<Ch> &data)
Chris@16:     {
Chris@16:         const static std::basic_string<Ch> chars = convert_chtype<Ch, char>(" \t{};\n\"");
Chris@16:         return !data.empty() && data.find_first_of(chars) == data.npos;
Chris@16:     }
Chris@16: 
Chris@16:     template<class Ptree>
Chris@16:     void write_info_helper(std::basic_ostream<typename Ptree::key_type::value_type> &stream, 
Chris@16:                            const Ptree &pt, 
Chris@16:                            int indent,
Chris@16:                            const info_writer_settings<typename Ptree::key_type::value_type> &settings)
Chris@16:     {
Chris@16: 
Chris@16:         // Character type
Chris@16:         typedef typename Ptree::key_type::value_type Ch;
Chris@16:         
Chris@16:         // Write data
Chris@16:         if (indent >= 0)
Chris@16:         {
Chris@16:             if (!pt.data().empty())
Chris@16:             {
Chris@16:                 std::basic_string<Ch> data = create_escapes(pt.template get_value<std::basic_string<Ch> >());
Chris@16:                 if (is_simple_data(data))
Chris@16:                     stream << Ch(' ') << data << Ch('\n');
Chris@16:                 else
Chris@16:                     stream << Ch(' ') << Ch('\"') << data << Ch('\"') << Ch('\n');
Chris@16:             }
Chris@16:             else if (pt.empty())
Chris@16:                 stream << Ch(' ') << Ch('\"') << Ch('\"') << Ch('\n');
Chris@16:             else
Chris@16:                 stream << Ch('\n');
Chris@16:         }
Chris@16:         
Chris@16:         // Write keys
Chris@16:         if (!pt.empty())
Chris@16:         {
Chris@16:             
Chris@16:             // Open brace
Chris@16:             if (indent >= 0)
Chris@16:             {
Chris@16:                 write_info_indent( stream, indent, settings);
Chris@16:                 stream << Ch('{') << Ch('\n');
Chris@16:             }
Chris@16:             
Chris@16:             // Write keys
Chris@16:             typename Ptree::const_iterator it = pt.begin();
Chris@16:             for (; it != pt.end(); ++it)
Chris@16:             {
Chris@16: 
Chris@16:                 // Output key
Chris@16:                 std::basic_string<Ch> key = create_escapes(it->first);
Chris@16:                 write_info_indent( stream, indent+1, settings);
Chris@16:                 if (is_simple_key(key))
Chris@16:                     stream << key;
Chris@16:                 else
Chris@16:                     stream << Ch('\"') << key << Ch('\"');
Chris@16: 
Chris@16:                 // Output data and children  
Chris@16:                 write_info_helper(stream, it->second, indent + 1, settings);
Chris@16: 
Chris@16:             }
Chris@16:             
Chris@16:             // Close brace
Chris@16:             if (indent >= 0)
Chris@16:             {
Chris@16:                 write_info_indent( stream, indent, settings);
Chris@16:                 stream << Ch('}') << Ch('\n');
Chris@16:             }
Chris@16: 
Chris@16:         }
Chris@16:     }
Chris@16: 
Chris@16:     // Write ptree to info stream
Chris@16:     template<class Ptree>
Chris@16:     void write_info_internal(std::basic_ostream<typename Ptree::key_type::value_type> &stream, 
Chris@16:                              const Ptree &pt,
Chris@16:                              const std::string &filename,
Chris@16:                              const info_writer_settings<typename Ptree::key_type::value_type> &settings)
Chris@16:     {
Chris@16:         write_info_helper(stream, pt, -1, settings);
Chris@16:         if (!stream.good())
Chris@16:             BOOST_PROPERTY_TREE_THROW(info_parser_error("write error", filename, 0));
Chris@16:     }
Chris@16: 
Chris@16: } } }
Chris@16: 
Chris@16: #endif