controllers.transformer.ExcelTransformer.java Source code

Introduction

Here is the source code for controllers.transformer.ExcelTransformer.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.transformer;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.logging.Level;
import java.util.logging.Logger;
import models.Answer;
import models.NdgResult;
import models.Question;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import models.Survey;
import models.constants.QuestionTypesConsts;
import org.apache.commons.collections.CollectionUtils;

public class ExcelTransformer extends ResultsTransformer {

    public ExcelTransformer(Survey survey, Boolean exportWithImages) {
        super(survey, exportWithImages);
    }

    public ExcelTransformer(Survey survey, Collection<NdgResult> results, Boolean exportWithImages) {
        super(survey, results, exportWithImages);
    }

    public byte[] getBytes() {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        HSSFWorkbook wb = new HSSFWorkbook();
        HSSFSheet sheet = wb.createSheet("Sheet1");

        if (results == null) {
            results = survey.resultCollection;
        }

        /** Header **/
        HSSFRow row = sheet.createRow(0);
        int fieldcounter = 0;
        row.createCell(fieldcounter++).setCellValue("ResultId");
        row.createCell(fieldcounter++).setCellValue("SurveyId");
        row.createCell(fieldcounter++).setCellValue("Title");
        row.createCell(fieldcounter++).setCellValue("Start time");
        row.createCell(fieldcounter++).setCellValue("End time");
        row.createCell(fieldcounter++).setCellValue("Date Sent");
        row.createCell(fieldcounter++).setCellValue("User");
        row.createCell(fieldcounter++).setCellValue("Phone Number");
        row.createCell(fieldcounter++).setCellValue("Lat");
        row.createCell(fieldcounter++).setCellValue("Lon");

        /** Header Fields**/
        for (Question question : survey.getQuestions()) {
            row.createCell(fieldcounter++).setCellValue(question.label);
        }

        int countrow = 0;
        row = sheet.createRow(++countrow);

        //SimpleDateFormat dateFormat = new SimpleDateFormat("EEE, d MMM yyyy HH:mm:ss Z");
        SimpleDateFormat dateFormat = new SimpleDateFormat("dd/MM/yyyy HH:mm:ss Z");

        for (NdgResult result : results) {
            fieldcounter = 0;
            row.createCell(fieldcounter++).setCellValue(result.resultId);
            row.createCell(fieldcounter++).setCellValue(result.survey.surveyId);
            row.createCell(fieldcounter++).setCellValue(result.title);
            row.createCell(fieldcounter++).setCellValue(dateFormat.format(result.startTime));
            row.createCell(fieldcounter++).setCellValue(dateFormat.format(result.endTime));

            if (result.dateSent != null) {
                row.createCell(fieldcounter++).setCellValue(dateFormat.format(result.dateSent));
            } else {
                row.createCell(fieldcounter++).setCellValue("");
            }

            row.createCell(fieldcounter++).setCellValue(result.ndgUser.username);
            row.createCell(fieldcounter++).setCellValue(result.ndgUser.phoneNumber);
            row.createCell(fieldcounter++).setCellValue(result.latitude);
            row.createCell(fieldcounter++).setCellValue(result.longitude);

            for (Question question : survey.getQuestions()) {//to ensure right answer order
                Collection<Answer> answers = CollectionUtils.intersection(question.answerCollection,
                        result.answerCollection);//only one should left, hope that it does not modify results
                if (answers.isEmpty()) {
                    row.createCell(fieldcounter++).setCellValue("");
                } else if (answers.size() == 1) {
                    Answer answer = answers.iterator().next();
                    if (answer.question.questionType.typeName.equalsIgnoreCase(QuestionTypesConsts.IMAGE)) {//TODO handle other binary data
                        row.createCell(fieldcounter++).setCellValue(storeImagesAndGetValueToExport(survey.surveyId,
                                result.resultId, answer.id, answer.binaryData));
                    } else if (answer.question.questionType.typeName.equalsIgnoreCase(QuestionTypesConsts.INT)) {
                        Integer value = Integer.valueOf(answer.textData);
                        row.createCell(fieldcounter++).setCellValue(value);
                    } else if (answer.question.questionType.typeName
                            .equalsIgnoreCase(QuestionTypesConsts.DECIMAL)) {
                        Float value = Float.valueOf(answer.textData);
                        row.createCell(fieldcounter++).setCellValue(value);
                    } else {
                        String value = answer.textData;
                        value = value.trim().replaceAll("\n", "");
                        row.createCell(fieldcounter++).setCellValue(value);
                    }
                } else {
                    Logger.getAnonymousLogger().log(Level.WARNING,
                            "to many answers. ResID={0}questioId={1}answerCount={2}",
                            new Object[] { result.resultId, question.id, question.answerCollection.size() });
                    break;
                }
            }
            row = sheet.createRow(++countrow);
        }
        try {
            wb.write(out);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return out.toByteArray();
    }
}