com.globalsight.util.ExcelUtil.java Source code

Introduction

Here is the source code for com.globalsight.util.ExcelUtil.java

Source

/**
 *  Copyright 2009 Welocalize, Inc. 
 *  
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  
 *  You may obtain a copy of the License at 
 *  http://www.apache.org/licenses/LICENSE-2.0
 *  
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *  
 */
package com.globalsight.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;

import org.apache.log4j.Logger;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class ExcelUtil {
    private static Logger logger = Logger.getLogger(ExcelUtil.class);

    public static boolean isXls(String filename) {
        if (StringUtil.isEmpty(filename))
            return false;
        return filename.toLowerCase().endsWith(".xls");
    }

    public static boolean isXlsx(String filename) {
        if (StringUtil.isEmpty(filename))
            return false;
        return filename.toLowerCase().endsWith(".xlsx");
    }

    public static boolean isExcel(String filename) {
        if (StringUtil.isEmpty(filename))
            return false;
        return isXls(filename) || isXlsx(filename);
    }

    public static Workbook getWorkbook(File file) {
        return getWorkbook(file.getAbsolutePath());
    }

    public static Workbook getWorkbook(String filename) {
        Workbook workbook = null;
        if (StringUtil.isEmpty(filename) || !isExcel(filename))
            return null;
        File file = null;
        try {
            file = new File(filename);
            if (!file.exists() || file.isDirectory())
                return null;
            InputStream is = new FileInputStream(file);
            if (isXls(filename))
                workbook = new HSSFWorkbook(is);
            else
                workbook = new XSSFWorkbook(OPCPackage.open(file));
        } catch (Exception e) {
            logger.error("Cannot open Excel file correctly.", e);
        }

        return workbook;
    }

    public static Workbook getWorkbook(String filename, InputStream is) {
        Workbook workbook = null;
        if (StringUtil.isEmpty(filename) || !isExcel(filename) || is == null)
            return null;

        File file = null;
        try {
            file = new File(filename);
            if (!file.exists() || file.isDirectory())
                return null;
            if (isXls(filename))
                workbook = new HSSFWorkbook(is);
            else
                workbook = new XSSFWorkbook(OPCPackage.open(is));
        } catch (Exception e) {
            logger.error("Cannot open Excel file correctly.", e);
        }

        return workbook;
    }

    public static Sheet getSheet(Workbook workbook, String sheetName) {
        if (workbook == null || StringUtil.isEmpty(sheetName))
            return null;

        return workbook.getSheet(sheetName);
    }

    public static Sheet getSheet(Workbook workbook, int sheetNumber) {
        if (workbook == null || sheetNumber < 0)
            return null;

        return workbook.getSheetAt(sheetNumber);
    }

    public static Sheet getDefaultSheet(Workbook workbook) {
        if (workbook == null)
            return null;

        return workbook.getSheetAt(0);
    }

    public static String getCellValue(Sheet sheet, int row, int col) {
        String value = "";
        if (sheet == null || row < 0 || col < 0)
            return "";

        Row rowData = sheet.getRow(row);
        if (rowData == null)
            return "";
        Cell cell = rowData.getCell(col);
        if (cell == null)
            return "";
        switch (cell.getCellType()) {
        case Cell.CELL_TYPE_NUMERIC:
            value = String.valueOf((int) cell.getNumericCellValue());
            break;
        case Cell.CELL_TYPE_STRING:
            value = cell.getStringCellValue();
            break;

        default:
            value = cell.toString();
        }

        return value;
    }
}