include <XMLElement.h>
  
  using namespace std;
  
  XMLElement::XMLElement() : mElementName("unnamed"),  mTextNode("")
  {
  }
  
  XMLElement::~XMLElement()
  {
  }
  
  void XMLElement::setElementName(const string& inName)
  {
    mElementName = inName;
  }
  
  void XMLElement::setAttribute(const string& inAttributeName, 
                                const string& inAttributeValue)
  {
    // set the key/value pair, replacing the existing one if it exists
    mAttributes[inAttributeName] = inAttributeValue;
  }
  
  void XMLElement::addSubElement(const XMLElement* inElement)
  {
    // add the new element to the vector of sub-elements
    mSubElements.push_back(inElement);
  }
  
  void XMLElement::setTextNode(const string& inValue)
  {
    mTextNode = inValue;
  }
  
  ostream& operator<<(ostream& outStream, const XMLElement& inElem)
  {
    inElem.writeToStream(outStream);
    return (outStream);
  }
  
  void XMLElement::writeToStream(ostream& outStream, int inIndentLevel) const
  {
    indentStream(outStream, inIndentLevel);
    outStream << "<" << mElementName;  // open the start tag
    
    // output any attributes
    for (map<string, string>::const_iterator it = mAttributes.begin();
         it != mAttributes.end(); ++it) {
      outStream << " " << it->first << "=\"" << it->second << "\"";
    }
  
    // close the start tag
    outStream << ">";
  
    if (mTextNode != "") {
      // if there's a text node, output it
      outStream << mTextNode;
    } else {
      outStream << endl;
      // call writeToStream at inIndentLevel+1 for any sub-elements
      for (vector<const XMLElement*>::const_iterator it = mSubElements.begin();
           it != mSubElements.end(); ++it) {
        (*it)->writeToStream(outStream, inIndentLevel + 1);
      }
      indentStream(outStream, inIndentLevel);
    }
  
    // write the close tag
    outStream << "</" << mElementName << ">" << endl;
  }
  
  void XMLElement::indentStream(ostream& outStream, int inIndentLevel) const
  {
    for (int i = 0; i < inIndentLevel; i++) {
      outStream << "\t";
    }
  }
  
  int main()
  {
    XMLElement dialogueElement;
    dialogueElement.setElementName("dialogue");
    
    XMLElement sentenceElement1;
    sentenceElement1.setElementName("sentence");
    sentenceElement1.setAttribute("speaker", "Marni");
    sentenceElement1.setTextNode("Let's go get some ice cream.");
  
    XMLElement sentenceElement2;
    sentenceElement2.setElementName("sentence");
    sentenceElement2.setAttribute("speaker", "Scott");
    sentenceElement2.setTextNode("After I'm done writing this C++ book.");
  
    // Add the sentence elements as sub-elements of the dialogue element
    dialogueElement.addSubElement(&sentenceElement1);
    dialogueElement.addSubElement(&sentenceElement2);
  
    // output the dialogue element to stdout
    cout << dialogueElement;
  }