boost/property_tree/detail/xml_parser_read_tinyxml.hpp

   1 // ----------------------------------------------------------------------------
   2 // Copyright (C) 2002-2006 Marcin Kalicinski
   3 //
   4 // Distributed under the Boost Software License, Version 1.0.
   5 // (See accompanying file LICENSE_1_0.txt or copy at
   6 // http://www.boost.org/LICENSE_1_0.txt)
   7 //
   8 // For more information, see www.boost.org
   9 // ----------------------------------------------------------------------------
  10 #ifndef BOOST_PROPERTY_TREE_DETAIL_XML_PARSER_READ_TINYXML_HPP_INCLUDED
  11 #define BOOST_PROPERTY_TREE_DETAIL_XML_PARSER_READ_TINYXML_HPP_INCLUDED
  12
  13 #include <boost/property_tree/ptree.hpp>
  14 #include <boost/property_tree/detail/xml_parser_error.hpp>
  15 #include <boost/property_tree/detail/xml_parser_flags.hpp>
  16 #include <boost/property_tree/detail/xml_parser_utils.hpp>
  17
  18 #include <tinyxml.h>
  19
  20 namespace boost { namespace property_tree { namespace xml_parser
  21 {
  22
  23 #ifdef TIXML_USE_STL
  24
  25     template<class Ptree>
  26     void read_xml_node(TiXmlNode *node, Ptree &pt, int flags)
  27     {
  28
  29         typedef typename Ptree::key_type::value_type Ch;
  30
  31         if (TiXmlElement *elem = node->ToElement())
  32         {
  33             Ptree &tmp = pt.push_back(std::make_pair(elem->Value(), Ptree()))->second;
  34             for (TiXmlAttribute *attr = elem->FirstAttribute(); attr; attr = attr->Next())
  35                 tmp.put(xmlattr<Ch>() + Ch('.') + attr->Name(), attr->Value());
  36             for (TiXmlNode *child = node->FirstChild(); child; child = child->NextSibling())
  37                 read_xml_node(child, tmp, flags);
  38         }
  39         else if (TiXmlText *text = node->ToText())
  40         {
  41             if (flags & no_concat_text)
  42                 pt.push_back(std::make_pair(xmltext<Ch>(), Ptree(text->Value())));
  43             else
  44                 pt.data() += text->Value();
  45         }
  46         else if (TiXmlComment *comment = node->ToComment())
  47         {
  48             if (!(flags & no_comments))
  49                 pt.push_back(std::make_pair(xmlcomment<Ch>(), Ptree(comment->Value())));
  50         }
  51     }
  52
  53     template<class Ptree>
  54     void read_xml_internal(std::basic_istream<typename Ptree::key_type::value_type> &stream,
  55                            Ptree &pt,
  56                            int flags,
  57                            const std::string &filename)
  58     {
  59
  60         // Create and load document from stream
  61         TiXmlDocument doc;
  62         stream >> doc;
  63         if (!stream.good())
  64             BOOST_PROPERTY_TREE_THROW(xml_parser_error("read error", filename, 0));
  65         if (doc.Error())
  66             BOOST_PROPERTY_TREE_THROW(xml_parser_error(doc.ErrorDesc(), filename, doc.ErrorRow()));
  67
  68         // Create ptree from nodes
  69         Ptree local;
  70         for (TiXmlNode *child = doc.FirstChild(); child; child = child->NextSibling())
  71             read_xml_node(child, local, flags);
  72
  73         // Swap local and result ptrees
  74         pt.swap(local);
  75
  76     }
  77
  78 #else
  79
  80     template<class Ptree>
  81     void read_xml_node(TiXmlNode *node, Ptree &pt, int flags)
  82     {
  83
  84         typedef typename Ptree::key_type::value_type Ch;
  85
  86         if (TiXmlElement *elem = node->ToElement())
  87         {
  88             Ptree &tmp = pt.push_back(std::make_pair(elem->Value(), Ptree()))->second;
  89             for (TiXmlAttribute *attr = elem->FirstAttribute(); attr; attr = attr->Next())
  90                 tmp.put(xmlattr<Ch>() + Ch('.') + attr->Name(), attr->Value());
  91             for (TiXmlNode *child = node->FirstChild(); child; child = child->NextSibling())
  92                 read_xml_node(child, tmp, flags);
  93         }
  94         else if (TiXmlText *text = node->ToText())
  95         {
  96             if (flags & no_concat_text)
  97                 pt.push_back(std::make_pair(xmltext<Ch>(), Ptree(text->Value())));
  98             else
  99                 pt.data() += text->Value();
 100         }
 101         else if (TiXmlComment *comment = node->ToComment())
 102         {
 103             if (!(flags & no_comments))
 104                 pt.push_back(std::make_pair(xmlcomment<Ch>(), Ptree(comment->Value())));
 105         }
 106     }
 107
 108     template<class Ptree>
 109     void read_xml_internal(std::basic_istream<typename Ptree::key_type::value_type> &stream,
 110                            Ptree &pt,
 111                            int flags,
 112                            const std::string &filename)
 113     {
 114
 115         // Load data into vector
 116         typedef typename Ptree::key_type::value_type Ch;
 117         std::vector<Ch> data(std::istreambuf_iterator<Ch>(stream.rdbuf()),
 118                              std::istreambuf_iterator<Ch>());
 119         if (!stream.good())
 120             BOOST_PROPERTY_TREE_THROW(xml_parser_error("read error", filename, 0));
 121         data.push_back(Ch('\0'));
 122
 123         // Create and load document
 124         TiXmlDocument doc;
 125         doc.Parse(&data.front());
 126         if (doc.Error())
 127             BOOST_PROPERTY_TREE_THROW(xml_parser_error(doc.ErrorDesc(), filename, doc.ErrorRow()));
 128
 129         // Create ptree from nodes
 130         Ptree local;
 131         for (TiXmlNode *child = doc.FirstChild(); child; child = child->NextSibling())
 132             read_xml_node(child, local, flags);
 133
 134         // Swap local and result ptrees
 135         pt.swap(local);
 136
 137     }
 138
 139 #endif
 140
 141 } } }
 142
 143 #endif