#include <string>
#include <stdexcept>
#include "xmlwordcount.h"
#include "getword.h"

XmlWordCount::XmlWordCount() : wcd ( new WordCountData() )
{
    wcd->wordCount = 0;
    XML_SetUserData ( p, wcd.get() );
    XML_SetCharacterDataHandler ( p, characterdata );
    XML_SetEndElementHandler ( p, end );
}

XmlWordCount::~XmlWordCount()
{}

int XmlWordCount::getWordCount()
{
    return wcd->wordCount;
}

void XMLCALL XmlWordCount::characterdata (
    void *data,
    const XML_Char *s,
    int len )
{
    WordCountData *wcd;
    wcd = ( WordCountData * ) data;
    wcd->buffer.append ( s, len );
}

void XMLCALL XmlWordCount::end ( void *data, const XML_Char *el )
{
    WordCountData *wcd;
    wcd = ( WordCountData * ) data;

    char *s = ( char * ) wcd->buffer.c_str();
    size_t len, myCount;
    myCount = 0;
    while ( GetWord::run ( &s, &len ) )
        ++myCount;
    wcd->wordCount += myCount;
    wcd->buffer = "";
}