com.netxforge.netxstudio.data.importer.HSSFMetricValuesImporter.java Source code

Introduction

Here is the source code for com.netxforge.netxstudio.data.importer.HSSFMetricValuesImporter.java

Source

/*******************************************************************************
 * Copyright (c) May 20, 2011 NetXForge.
 *
 * 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 3 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 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/>
 * 
 * Contributors: Martin Taal - initial API and implementation and/or
 * initial documentation
 *******************************************************************************/
package com.netxforge.netxstudio.data.importer;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.text.NumberFormat;
import java.util.Date;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFDateUtil;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import com.netxforge.netxstudio.data.index.IComponentLocator;
import com.netxforge.netxstudio.metrics.MappingXLS;

/**
 * Imports metrics from XLS.
 * 
 * @author Martin Taal
 * @author Christophe Bouhier
 */
public class HSSFMetricValuesImporter extends AbstractMetricValuesImporter {

    private HSSFSheet currentSheet;

    public HSSFMetricValuesImporter(IComponentLocator locator) {
        super(locator);
    }

    @Override
    protected int processFile(File file) throws Exception {
        final InputStream is = new FileInputStream(file);
        final HSSFWorkbook workBook = new HSSFWorkbook(is);
        final HSSFSheet sheet = workBook.getSheetAt(((MappingXLS) getMapping()).getSheetNumber());
        currentSheet = sheet;

        if (sheet.getLastRowNum() < getMapping().getFirstDataRow()) {
            getFailedRecords().add(createMappingRecord(

                    "There is no data in the sheet, first data row is " + getMapping().getFirstDataRow()
                            + " but the sheet has only " + sheet.getLastRowNum() + " rows."));
            return 0;
        }
        is.close();
        final int totalRows = processRows();
        currentSheet = null;
        return totalRows;
    }

    @Override
    protected int getTotalRows() {

        // int lastRow = currentSheet.getLastRowNum();
        int physicalNumberOfRows = currentSheet.getPhysicalNumberOfRows();
        // int totalRows = currentSheet.getLastRowNum();
        // POI reports wrong number of rows as the last row number.
        return physicalNumberOfRows;
    }

    @Override
    protected String getFileExtension() {
        return "xls";
    }

    @Override
    protected String getStringCellValue(int row, int column) {
        return this.getCellValue(currentSheet.getRow(row).getCell(column));
    }

    @Override
    protected double getNumericCellValue(int row, int column) {
        HSSFCell cell = currentSheet.getRow(row).getCell(column);
        if (cell != null) {
            return cell.getNumericCellValue();
        } else {
            return 0.0;
        }
    }

    protected String getCellValue(HSSFCell cell) {
        String value = null;
        // Get the value.
        int cellType = cell.getCellType();

        if (cellType == HSSFCell.CELL_TYPE_NUMERIC) {
            double numericCellValue = cell.getNumericCellValue();
            NumberFormat nf = NumberFormat.getInstance();
            nf.setMaximumFractionDigits(0);// set as you need
            value = nf.format(numericCellValue);

        } else if (cellType == HSSFCell.CELL_TYPE_STRING) {
            value = cell.getStringCellValue();
        }
        return value;
    }

    @Override
    protected Date getDateCellValue(int rowNum, int column) {
        Date value = null;
        HSSFCell cell = currentSheet.getRow(rowNum).getCell(column);
        int cellType = cell.getCellType();
        if (cellType == HSSFCell.CELL_TYPE_NUMERIC) {
            if (HSSFDateUtil.isCellDateFormatted(cell)) {
                double cellValue = cell.getNumericCellValue();
                value = HSSFDateUtil.getJavaDate(cellValue);
            }
        }
        return value;
    }

}