zeroxmlreader.cpp

/*!
 * \file zeroxmlreader.cpp
 *  \author Simon Coakley
 *  \date 2012
 *  \copyright Copyright (c) 2012 University of Sheffield
 *  \brief Implementation of zero XML reader
 */
#include <QtGui>
#include "./zeroxmlreader.h"
#include "./visualsettingsitem.h"
#include "./ruleagent.h"

ZeroXMLReader::ZeroXMLReader(QList<Agent *> *a, QList<AgentType> *at,
        VisualSettingsModel *vsm, double r, Dimension * ad,
                             QStringList *sat, QHash<QString, int> *atc,
                             double xo, double yo, double zo) {
    agents = a;
    agentTypes = at;
    vsmodel = vsm;
    ratio = r;
    agentDimension = ad;
    stringAgentTypes = sat;
    agentTypeCounts = atc;
    xoffset = xo;
    yoffset = yo;
    zoffset = zo;

    agentDimension->xmin =  999999.9;
    agentDimension->xmax = -999999.9;
    agentDimension->ymin =  999999.9;
    agentDimension->ymax = -999999.9;
    agentDimension->zmin =  999999.9;
    agentDimension->zmax = -999999.9;
}

bool ZeroXMLReader::read(QIODevice * device) {
    setDevice(device);

    while (!atEnd()) {
    readNext();

    if (isStartElement()) {
        if (name() == "states")
            readZeroXML();
            else
            raiseError(QObject::tr(
                "The file does not contain the root tag 'states'"));
        }
    }

    // For every rule
    for (int i = 0; i < vsmodel->rowCount(); i++) {
        // Populate rule with ruleagents from agents
        vsmodel->getRule(i)->populate(agents);
        // Calculate visual variables for ruleagent
        vsmodel->getRule(i)->
                copyAgentDrawDataToRuleAgentDrawData(agentDimension);
        // Apply offset to ruleagents to centre the scene
        vsmodel->getRule(i)->applyOffset(xoffset, yoffset, zoffset);
        // Apply ratio to ruleagents to go from model space to opengl space
        vsmodel->getRule(i)->applyRatio(ratio);
    }

    return !error();
}

void ZeroXMLReader::readUnknownElement() {
    Q_ASSERT(isStartElement());

     while (!atEnd()) {
         readNext();

         if (isEndElement())
             break;

         if (isStartElement())
             readUnknownElement();
     }
}

/* Read 0.xml ignore tags until an xagent if found and handle it.
*/
void ZeroXMLReader::readZeroXML() {
    Q_ASSERT(isStartElement() && name() == "states");

    // qDebug("readZeroXML()\n");

    while (!atEnd()) {
         readNext();

         if (isEndElement())
             break;

         if (isStartElement()) {
             if (name() == "environment")
                 readEnvironmentXML();
             else if (name() == "agents")
                 readAgentsXML();
             else if (name() == "xagent")
                 readAgentXML();
             else
                 readUnknownElement();
         }
     }
}

void ZeroXMLReader::readEnvironmentXML() {
    // Create 'agent' with type environment
    Agent * agent = new Agent;
    int index = -1;
    agent->isEnvironment = true;
    agent->agentType = "environment";

    if (stringAgentTypes->contains("environment") == false) {
     // qDebug() << "new agent type found: environment";
     stringAgentTypes->append("environment");
     agentTypes->append(AgentType("environment"));
     index = agentTypes->count() - 1;
    }

    /* Make environment count one for iteration info */
    agentTypeCounts->insert("environment", 1);

    while (!atEnd()) {
         readNext();

         if (isEndElement())
             break;

         if (isStartElement()) {
             agent->tags.append(name().toString());
             agent->values.append(readElementText());
             if (index != -1) {
                (*agentTypes)[index].variables.append(name().toString());
             }
         }
     }

    // Add agent to list of agents
    agents->append(agent);
}

void ZeroXMLReader::readAgentsXML() {
    while (!atEnd()) {
         readNext();

         if (isEndElement())
             break;

         if (isStartElement()) {
             if (name() == "xagent")
                 readAgentXML();
             else
                 readUnknownElement();
         }
     }
}

void ZeroXMLReader::readAgentXML() {
    Agent * agent = new Agent;
    QString agentname = "";
    int index = -1;

    while (!atEnd()) {
         readNext();

         if (isEndElement()) break;

         if (isStartElement()) {
             if (name() == "name") {
                 // Agent type
                 agentname = readElementText();
                 agent->agentType = agentname;
                 /* Increment agent counts for iteration info */
                 agentTypeCounts->insert(agentname,
                                        agentTypeCounts->value(agentname) + 1);
                 /* If agent type is unknown then add to agent list */
                 if (stringAgentTypes->contains(agentname) == false) {
                     // qDebug() << "new agent type found: " << agentname;
                     stringAgentTypes->append(agentname);
                     agentTypes->append(AgentType(agentname));
                     index = agentTypes->count() - 1;
                 } else { index = -1; }
             } else {
                 // Agent memory variable
                 agent->tags.append(name().toString());
                 agent->values.append(readElementText());
                 /* If agent is unknown then add variables to agent type */
                 if (index != -1)
                    (*agentTypes)[index].variables.append(name().toString());
             }
         }
     }
    // Add agent to list of agents
    agents->append(agent);
}