devutility.external.poi.utils.RowStyleUtils Maven / Gradle / Ivy

Go to download
package devutility.external.poi.utils;

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

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;

import devutility.external.poi.model.RowStyle;

public class RowStyleUtils {
	public static RowStyle clone(Workbook workbook, String sheetName, int rowNum, boolean excludeBold, short fontSize) throws Exception {
		Sheet sheet = SheetUtils.get(workbook, sheetName);
		Row row = sheet.getRow(rowNum);
		RowStyle rowStyle = clone(workbook, row);
		customize(workbook, rowStyle, excludeBold, fontSize);
		return rowStyle;
	}

	public static RowStyle clone(Workbook workbook, String sheetName, int rowNum, boolean excludeBold, int fontSize) throws Exception {
		return clone(workbook, sheetName, rowNum, excludeBold, (short) fontSize);
	}

	public static RowStyle clone(Workbook workbook, String sheetName, int rowNum, boolean excludeBold) throws Exception {
		return clone(workbook, sheetName, rowNum, excludeBold, (short) 0);
	}

	public static RowStyle clone(Workbook workbook, Row row) {
		short firstCellNum = row.getFirstCellNum();
		short lastCellNum = row.getLastCellNum();

		if (firstCellNum == -1 || lastCellNum == -1) {
			return null;
		}

		Map columnStyleMap = new HashMap<>(lastCellNum);

		/**
		 * Map for CellStyle number between cell style and new column style.
		 */
		Map columnStyleIndexMap = new HashMap<>(lastCellNum);

		/**
		 * Map for Font number between cell style and new column style.
		 */
		Map columnfontIndexMap = new HashMap<>(lastCellNum);

		for (int cellNum = firstCellNum; cellNum < lastCellNum; cellNum++) {
			Cell cell = row.getCell(cellNum);

			if (cell == null) {
				continue;
			}

			CellStyle cellStyle = cell.getCellStyle();

			if (cellStyle == null) {
				continue;
			}

			CellStyle newCellStyle = createCellStyle(workbook, columnStyleIndexMap, cellStyle);
			Font newFont = createFont(workbook, columnfontIndexMap, cellStyle);
			newCellStyle.setFont(newFont);
			columnStyleMap.put(cellNum, newCellStyle);
		}

		RowStyle rowStyle = new RowStyle();
		rowStyle.setRowNum(row.getRowNum());
		rowStyle.setRowHeight(row.getHeightInPoints());
		rowStyle.setRowStyle(row.getRowStyle());
		rowStyle.setColumnStyleMap(columnStyleMap);
		return rowStyle;
	}

	public static RowStyle customize(Workbook workbook, RowStyle rowStyle, boolean excludeBold, short fontSize) {
		Map columnStyleMap = rowStyle.getColumnStyleMap();

		for (CellStyle cellStyle : columnStyleMap.values()) {
			int fontIndex = cellStyle.getFontIndexAsInt();
			Font font = workbook.getFontAt(fontIndex);

			if (font != null) {
				customizeFont(font, excludeBold, fontSize);
			}
		}

		return rowStyle;
	}

	private static void customizeFont(Font font, boolean excludeBold, short fontSize) {
		if (excludeBold) {
			font.setBold(false);
		}

		if (fontSize > 0) {
			font.setFontHeightInPoints(fontSize);
		}
	}

	private static CellStyle createCellStyle(Workbook workbook, Map columnStyleIndexMap, CellStyle cellStyle) {
		int cellStyleIndex = cellStyle.getIndex();
		Integer newCellStyleIndex = columnStyleIndexMap.get(cellStyleIndex);

		if (newCellStyleIndex == null) {
			return cloneCellStyle(workbook, columnStyleIndexMap, cellStyle);
		}

		return workbook.getCellStyleAt(newCellStyleIndex);
	}

	private static CellStyle cloneCellStyle(Workbook workbook, Map columnStyleIndexMap, CellStyle cellStyle) {
		CellStyle newCellStyle = CellStyleUtils.clone(workbook, cellStyle);
		int cellStyleIndex = cellStyle.getIndex();
		int newCellStyleIndex = newCellStyle.getIndex();
		columnStyleIndexMap.put(cellStyleIndex, newCellStyleIndex);
		return newCellStyle;
	}

	private static Font createFont(Workbook workbook, Map columnfontIndexMap, CellStyle cellStyle) {
		int fontIndex = cellStyle.getFontIndexAsInt();
		Integer newFontIndex = columnfontIndexMap.get(fontIndex);

		if (newFontIndex == null) {
			return cloneFont(workbook, columnfontIndexMap, fontIndex);
		}

		return workbook.getFontAt(newFontIndex);
	}

	private static Font cloneFont(Workbook workbook, Map columnfontIndexMap, int fontIndex) {
		Font font = workbook.getFontAt(fontIndex);
		Font newFont = FontUtils.clone(workbook, font);
		int newFontIndex = newFont.getIndexAsInt();
		columnfontIndexMap.put(fontIndex, newFontIndex);
		return newFont;
	}
}