src/rcrt/loaders/XMLLoader.cpp

Go to the documentation of this file.

#include "XMLLoader.h"
#include "conversion.hpp"


using namespace std;
using namespace boost::iostreams;

namespace rcrt
{

XMLLoader::XMLLoader(string path):file(path),in(),parser(in),currentFrame(-1),eof(false)
{
        string::size_type loc = path.find(".bz2", path.length()-4);
        if(loc != string::npos) {
                //cout << "[XMLLoader] loading bz2 file:" << path << endl;
                isCompressed = true;
                in.push(bzip2_decompressor());
        } else {
                loc = path.find(".xml", path.length()-4);
                if(loc == string::npos)
                        throw runtime_error("[XMLLoader] could not load file: "+
                                        path+" does not end in .xml or .bz2.");
                isCompressed = false;
        }
        in.push(file);
}

XMLLoader::~XMLLoader()
{
}

bool XMLLoader::nextStartTag(string tagName)
{
        parser.next();
        int eventType = parser.getEventType();
        
        if(tagName.empty()){
                
                while(eventType != XmlPullParser::START_TAG
                                && eventType != XmlPullParser::END_DOCUMENT){
                        parser.next();
                        eventType = parser.getEventType();
                }
        } else {
                while((eventType != XmlPullParser::START_TAG
                                || parser.getName() != tagName)
                                && eventType != XmlPullParser::END_DOCUMENT){
                        parser.next();
                        eventType = parser.getEventType();
                }
        }
        if(eventType == XmlPullParser::END_DOCUMENT){
                return false;
        }
        return true;
}

const int& XMLLoader::getFrame()
{
        return currentFrame;
}

}

---