DOM fixup changes

[shibboleth/xmltooling.git] / xmltooling / impl / UnknownElement.h

diff --git a/xmltooling/impl/UnknownElement.h b/xmltooling/impl/UnknownElement.h
index acab528..764e17d 100644 (file)
--- a/xmltooling/impl/UnknownElement.h
+++ b/xmltooling/impl/UnknownElement.h
@@ -112,9 +112,23 @@ namespace xmltooling {
         virtual ~UnknownElementMarshaller() {}\r
     \r
         /**\r
-         * @see Marshaller::marshall()\r
+         * @see Marshaller::marshall(XMLObject*,DOMDocument*)\r
          */\r
         DOMElement* marshall(XMLObject* xmlObject, DOMDocument* document=NULL) const;\r
+\r
+        /**\r
+         * @see Marshaller::marshall(XMLObject*,DOMElement*)\r
+         */\r
+        DOMElement* marshall(XMLObject* xmlObject, DOMElement* parentElement) const;\r
+        \r
+    protected:\r
+        void setDocumentElement(DOMDocument* document, DOMElement* element) const {\r
+            DOMElement* documentRoot = document->getDocumentElement();\r
+            if (documentRoot)\r
+                document->replaceChild(documentRoot, element);\r
+            else\r
+                document->appendChild(element);\r
+        }\r
     };\r
 \r
     /**\r