com.feilong.excel.writer.ExcelCellConditionStyleInitializer Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2008 feilong
 *
 * 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.feilong.excel.writer;

import static com.feilong.excel.util.CellReferenceUtil.getCellRef;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.feilong.excel.definition.ExcelBlock;
import com.feilong.excel.definition.ExcelCell;
import com.feilong.excel.definition.ExcelCellConditionStyle;
import com.feilong.excel.definition.ExcelSheet;
import com.feilong.excel.util.CellReferenceUtil;

class ExcelCellConditionStyleInitializer{

    private static final Logger LOGGER = LoggerFactory.getLogger(ExcelCellConditionStyleInitializer.class);

    /** Don't let anyone instantiate this class. */
    private ExcelCellConditionStyleInitializer(){
        //AssertionError不是必须的. 但它可以避免不小心在类的内部调用构造器. 保证该类在任何情况下都不会被实例化.
        //see 《Effective Java》 2nd
        throw new AssertionError("No " + getClass().getName() + " instances for you!");
    }

    //---------------------------------------------------------------
    static Map init(Sheet sheet,ExcelSheet excelSheet){
        Map returnMap = new HashMap<>();

        List excelBlocks = excelSheet.getExcelBlocks();
        for (ExcelBlock excelBlock : excelBlocks){
            init(sheet, excelBlock, returnMap);

            ExcelBlock childBlock = excelBlock.getChildBlock();
            if (childBlock != null){
                init(sheet, childBlock, returnMap);
            }
        }
        return returnMap;
    }

    private static void init(Sheet sheet,ExcelBlock excelBlock,Map styleMap){
        for (ExcelCellConditionStyle excelCellConditionStyle : excelBlock.getStyles()){
            init(sheet, excelCellConditionStyle, styleMap);
        }
        for (ExcelCell excelCell : excelBlock.getCells()){
            for (ExcelCellConditionStyle excelCellConditionStyle : excelCell.getStyles()){
                init(sheet, excelCellConditionStyle, styleMap);
            }
        }
    }

    //---------------------------------------------------------------

    private static void init(Sheet sheet,ExcelCellConditionStyle excelCellConditionStyle,Map styleMap){
        //ignore existed style
        int startRow = excelCellConditionStyle.getStartRow();
        int endRow = excelCellConditionStyle.getEndRow();

        int startCol = excelCellConditionStyle.getStartCol();
        int endCol = excelCellConditionStyle.getEndCol();

        String cellIndex = excelCellConditionStyle.getCellIndex();

        if (startRow == 0 && endRow == 0 && startCol == 0 && endCol == 0 && styleMap.containsKey(cellIndex)){
            return;
        }

        //---------------------------------------------------------------
        int[] position = CellReferenceUtil.getCellPosition(cellIndex);
        int rows = endRow - startRow;
        int cols = endCol - startCol;

        //---------------------------------------------------------------
        for (int i = position[0]; i <= position[0] + rows; i++){
            Row row = sheet.getRow(i);
            if (row == null){
                return;
            }

            //---------------------------------------------------------------
            for (int j = position[1]; j <= position[1] + cols; j++){
                Cell cell = row.getCell(j);
                if (cell == null){
                    return;
                }

                String cellIndex2 = getCellRef(i, j);
                styleMap.put(cellIndex2, cell.getCellStyle());

                LOGGER.debug("Condition Style [{}]", cellIndex2);
            }
        }
    }
}