controllers.logic.ResultParser.java Source code

Introduction

Here is the source code for controllers.logic.ResultParser.java

Source

/*
*  Nokia Data Gathering
*
*  Copyright (C) 2011 Nokia Corporation
*
*  This program is free software; you can redistribute it and/or
*  modify it under the terms of the GNU Lesser General Public
*  License as published by the Free Software Foundation; either
*  version 2.1 of the License, or (at your option) any later version.
*
*  This program is distributed in the hope that it will be useful,
*  but WITHOUT ANY WARRANTY; without even the implied warranty of
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
*  Lesser General Public License for more details.
*
*  You should have received a copy of the GNU Lesser General Public License
*  along with this program.  If not, see <http://www.gnu.org/licenses/
*/

package controllers.logic;

import controllers.exceptions.ResultSaveException;
import java.io.ByteArrayInputStream;
import models.Answer;
import models.Category;
import models.NdgResult;
import java.io.Reader;
import java.util.Hashtable;
import models.NdgUser;
import models.Question;
import models.Survey;
import models.constants.QuestionTypesConsts;
import org.javarosa.xform.parse.XFormParser;
import org.joda.time.format.ISODateTimeFormat;
import org.kxml2.kdom.Document;
import org.kxml2.kdom.Element;
import play.db.jpa.Blob;
import play.libs.Codec;

public class ResultParser {

    private static final String OPEN_ROSA_ROOT = "data";
    private static final String OPEN_ROSA_INSTANCE_ID = "instanceID";
    private static final String OPEN_ROSA_TIME_START = "timeStart";
    private static final String OPEN_ROSA_TIME_FINISH = "timeEnd";
    private static final String OPEN_ROSA_GEOSTAMP = "geostamp"; //Not OpenROSA
    private static final String OPEN_ROSA_NAMESPACE = "http://openrosa.org/xforms/metadata";
    private NdgResult result;
    private final Reader reader;
    private Hashtable<String, ResultElementHandler> elementHandlers;

    ResultParser(Reader reader, NdgResult result, String surveyId) throws ResultSaveException {
        this.reader = reader;
        this.result = result;
        initElementHandlers();
    }

    private void initElementHandlers() {
        elementHandlers = new Hashtable<String, ResultElementHandler>();
        elementHandlers.put(OPEN_ROSA_INSTANCE_ID, new ResultElementHandler() {

            public void handleElement(ResultParser parser, Element element) {
                result.resultId = element.getText(0);
            }
        });

        elementHandlers.put(OPEN_ROSA_TIME_START, new ResultElementHandler() {

            public void handleElement(ResultParser parser, Element element) {
                result.startTime = ISODateTimeFormat.dateTimeNoMillis().parseDateTime(element.getText(0)).toDate();
            }
        });

        elementHandlers.put(OPEN_ROSA_TIME_FINISH, new ResultElementHandler() {

            @Override
            public void handleElement(ResultParser parser, Element element) {
                result.endTime = ISODateTimeFormat.dateTimeNoMillis().parseDateTime(element.getText(0)).toDate();
            }
        });

        elementHandlers.put(OPEN_ROSA_GEOSTAMP, new ResultElementHandler() {

            public void handleElement(ResultParser parser, Element element) {
                String[] coordinates = element.getText(0).split(" ");
                result.latitude = coordinates[0];
                result.longitude = coordinates[1];
            }
        });

    }

    public void parse() {
        Document xmlDoc = XFormParser.getXMLDocument(reader);
        Element root = xmlDoc.getRootElement();

        parseXmlElement(root, null);

        if (result.title == null) {
            result.title = result.resultId;
        }
        result.save();
    }

    private void parseXmlElement(Element element, Category category) {
        String name = element.getName();
        if (element.getNamespace().equals(OPEN_ROSA_NAMESPACE)) {
            ResultElementHandler handler = elementHandlers.get(name);
            if (handler != null) {
                handler.handleElement(this, element);
            } else {
                parseChilds(element, null);
            }
        } else if (name.equals(OPEN_ROSA_ROOT)) {
            parseChilds(element, null);
        } else if (element.getChildCount() > 0) {
            if (element.getText(0).trim().equals("")) {
                parseCategory(element);
            } else {
                parseAnswer(element, category);
            }
        }
    }

    private void parseChilds(Element element, Category category) {
        for (int i = 0; i < element.getChildCount(); i++) {
            if (element.getType(i) == Element.ELEMENT) {
                parseXmlElement(element.getElement(i), category);
            }
        }
    }

    private void parseCategory(Element element) {
        Category category = Category.find("bySurveyAndObjectName", result.survey, element.getName()).first();
        if (category != null) {
            parseChilds(element, category);
        }
    }

    private void parseAnswer(Element element, Category category) {
        if (category == null) {
            category = result.survey.categoryCollection.get(0);//default category
        }
        Question answeredQuestion = Question.find("byObjectNameAndCategory_id", element.getName(), category.id)
                .first();
        if (answeredQuestion != null) {
            Answer answer = new Answer(result.answerCollection.size());
            answer.ndgResult = result;
            answer.question = answeredQuestion;
            if (answeredQuestion.questionType.typeName.equalsIgnoreCase(QuestionTypesConsts.IMAGE)) {
                answer.binaryData = new Blob();
                answer.binaryData.set(new ByteArrayInputStream(Codec.decodeBASE64(element.getText(0))),
                        "image/jpeg");
            } else {
                answer.textData = element.getText(0);
                if (result.title == null && answer.textData != null
                        && !answeredQuestion.questionType.typeName
                                .equalsIgnoreCase(QuestionTypesConsts.EXCLUSIVECHOICE)
                        && !answeredQuestion.questionType.typeName
                                .equalsIgnoreCase(QuestionTypesConsts.MULTIPLECHOICE)) {
                    result.title = answer.textData;
                }
            }
            result.answerCollection.add(answer);
        }
    }
}