org.amanzi.splash.importer.ExcelImporter.java Source code

Introduction

Here is the source code for org.amanzi.splash.importer.ExcelImporter.java

Source

/* AWE - Amanzi Wireless Explorer
 * http://awe.amanzi.org
 * (C) 2008-2009, AmanziTel AB
 *
 * This library is provided under the terms of the Eclipse Public License
 * as described at http://www.eclipse.org/legal/epl-v10.html. Any use,
 * reproduction or distribution of the library constitutes recipient's
 * acceptance of this agreement.
 *
 * This library is distributed WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 */

package org.amanzi.splash.importer;

import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.InvocationTargetException;
import java.util.Iterator;

import org.amanzi.neo.services.nodes.SpreadsheetNode;
import org.amanzi.neo.services.ui.NeoUtils;
import org.amanzi.splash.swing.Cell;
import org.apache.log4j.Logger;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.neo4j.graphdb.Transaction;

import com.eteks.openjeks.format.CellFormat;

/**
 * Importer of Excel data into Splash
 * 
 * @author Lagutko_N
 * @since 1.0.0
 */
public class ExcelImporter extends AbstractImporter {
    static private final Logger LOGGER = Logger.getLogger(ExcelImporter.class);
    /**
     * Parent Spreadsheet
     */
    private SpreadsheetNode rootSpreadsheet;

    /**
     * Constructor 
     * 
     * @param containerPath path to Project that will contain new Spreadsheet
     * @param fileName name of File to import
     * @param stream content of File to import
     * @param fileSize size of File to import
     */
    public ExcelImporter(IPath containerPath, String fileName, InputStream stream, long fileSize) {
        super(containerPath, fileName, stream, fileSize);
    }

    @Override
    public void run(IProgressMonitor monitor) throws InvocationTargetException {
        POIFSFileSystem fileSystem = null;

        monitor.beginTask("Importing data from Excel", 100);

        Transaction tx = NeoUtils.beginTransaction();
        try {
            fileSystem = new POIFSFileSystem(fileContent);

            HSSFWorkbook workBook = new HSSFWorkbook(fileSystem);

            createRootSpreadsheet();

            for (int i = 0; i < workBook.getNumberOfSheets(); i++) {
                monitor.subTask("Sheet " + workBook.getSheetName(i));
                createSheet(workBook.getSheetAt(i), workBook.getSheetName(i), tx);
                monitor.worked(100 / workBook.getNumberOfSheets());
            }
            monitor.done();

        } catch (IOException e) {
            throw new InvocationTargetException(e);
        } finally {
            tx.success();
            tx.finish();
        }
    }

    /**
     * Creates a parent spreadsheet
     */
    private void createRootSpreadsheet() {
        createSpreadsheet(null);
        this.rootSpreadsheet = super.spreadsheetNode;
    }

    /**
     * Creates a child spreadsheet inside parent spreadsheet
     *
     * @param sheet sheet to import
     * @param sheetName name of sheet
     * @param monitor monitor
     */
    @SuppressWarnings(value = { "deprecation", "unchecked" })
    private void createSheet(HSSFSheet sheet, String sheetName, Transaction transaction) {
        spreadsheetNode = null;
        spreadsheetName = sheetName;

        createSpreadsheet(rootSpreadsheet);
        try {
            Iterator<HSSFRow> rows = sheet.rowIterator();

            while (rows.hasNext()) {
                HSSFRow row = rows.next();

                //display row number in the console.
                LOGGER.debug("Row No.: " + row.getRowNum());

                //once get a row its time to iterate through cells.
                Iterator<HSSFCell> cells = row.cellIterator();

                int R = row.getRowNum();

                if ((R % 20) == 0) {
                    updateTransaction(transaction);
                }

                while (cells.hasNext()) {
                    HSSFCell cell = cells.next();

                    LOGGER.debug("Cell No.: " + cell.getCellNum());

                    int C = cell.getCellNum();

                    /*
                     * Now we will get the cell type and display the values
                     * accordingly.
                     */
                    switch (cell.getCellType()) {
                    case HSSFCell.CELL_TYPE_NUMERIC: {

                        // cell type numeric.
                        LOGGER.debug("====================================================");
                        LOGGER.debug("Numeric value: " + cell.getNumericCellValue());
                        LOGGER.debug("====================================================");
                        String def = Double.toString(cell.getNumericCellValue());

                        Cell c = new Cell(R, C, def, def, new CellFormat());
                        //TODO: interpet!!!!!!
                        //Cell c = model.interpret(def, R, C);

                        saveCell(c);
                        break;
                    }

                    case HSSFCell.CELL_TYPE_STRING: {
                        // cell type string.
                        HSSFRichTextString richTextString = cell.getRichStringCellValue();
                        LOGGER.debug("====================================================");
                        LOGGER.debug("String value: " + richTextString.getString());
                        LOGGER.debug("====================================================");
                        Cell c = new Cell(R, C, richTextString.getString(), richTextString.getString(),
                                new CellFormat());
                        saveCell(c);
                        break;
                    }

                    case HSSFCell.CELL_TYPE_FORMULA:
                        // cell type string.
                        String cellFormula = "=" + cell.getCellFormula().toLowerCase();

                        Cell c = new Cell(R, C, cellFormula, cellFormula, new CellFormat());
                        //TODO: interpet!!!!!!
                        //Cell c = model.interpret(def, R, C);

                        saveCell(c);

                        LOGGER.debug("====================================================");
                        LOGGER.debug("Formula value: " + cellFormula);
                        LOGGER.debug("====================================================");

                        break;

                    default: {

                        // types other than String and Numeric.
                        LOGGER.debug("Type not supported.");

                        break;
                    }
                    }
                }
            }
        } finally {
            updateTransaction(transaction);
        }
    }

    @Override
    public SpreadsheetNode getSpreadsheet() {
        return rootSpreadsheet;
    }

}