aswi2020vldc-gitlab/src/main/java/vldc/aswi/utils/ExporterXLSX.java @ d90fc9de

package vldc.aswi.utils;

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRow;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRowCell;

import java.io.File;
import java.io.FileOutputStream;
import java.util.List;

/**
 * Class used for exporting data to XLSX.
 */
public class ExporterXLSX {

    /**
     * Exports data to .xlsx file
     * @param tableName - name of the table.
     * @param contingencyTableRows - List of contingency table rows.
     */
    public static void export(File tmpFile, String tableName, List<ContingencyTableRow> contingencyTableRows) {

        // Create workbook.
        Workbook workbook = new XSSFWorkbook();

        // Create sheet.
        Sheet sheet = workbook.createSheet(tableName);

        // Create font for row names.
        Font colRowNameFont = workbook.createFont();
        colRowNameFont.setBold(true);
        colRowNameFont.setFontHeightInPoints((short) 16);
        colRowNameFont.setFontName("HELVETICA");
        colRowNameFont.setColor(IndexedColors.BLACK.getIndex());

        // Create font for data.
        Font dataFont = workbook.createFont();
        dataFont.setFontName("HELVETICA");
        dataFont.setBold(false);
        dataFont.setFontHeightInPoints((short) 12);
        dataFont.setColor(IndexedColors.BLACK.getIndex());

        // Set row cell style.
        CellStyle colRowStyle = workbook.createCellStyle();
        colRowStyle.setFont(colRowNameFont);
        colRowStyle.setAlignment(HorizontalAlignment.CENTER);

        // Set data cell style.
        CellStyle dataStyle = workbook.createCellStyle();
        dataStyle.setFont(dataFont);
        dataStyle.setAlignment(HorizontalAlignment.CENTER);

        int rowCounter = 0;

        for (ContingencyTableRow row : contingencyTableRows) {

            Row xlsxRow = sheet.createRow(rowCounter);

            if (row.isHeader()) {
                // Row is header.
                setColRow(sheet, xlsxRow, row.getCells(), colRowStyle);
            }
            else {
                // Row contains data.
                setDataRow(sheet, xlsxRow, row.getCells(), colRowStyle, dataStyle);
            }

            rowCounter++;
        }

        try {

            FileOutputStream out = new FileOutputStream(tmpFile);
            workbook.write(out);
            out.close();

        } catch (Exception e) {

        }
    }

    /**
     * Set column row.
     * @param sheet - Sheet.
     * @param row - Row.
     * @param cells - List of contingency table cells.
     * @param colRowStyle - Column row styles.
     */
    private static void setColRow(Sheet sheet, Row row, List<ContingencyTableRowCell> cells, CellStyle colRowStyle) {

        int cellCounter = 0;

        for (ContingencyTableRowCell cell : cells) {

            // Create new xlsx cell.
            Cell xlsxCell = row.createCell(cellCounter);
            xlsxCell.setCellValue(cell.getValue());
            xlsxCell.setCellStyle(colRowStyle);

            // Calculate last cell to be merged with current.
            int colSpan = cell.getColSpan();
            int newCellPosition = cellCounter + (colSpan - 1);

            // If colSpan is higher than 1, then merge cells.
            if (colSpan > 1) {
                sheet.addMergedRegion(new CellRangeAddress(row.getRowNum(), row.getRowNum(), cellCounter, newCellPosition));
            }

            // If cell has some value, then autosize current column.
            if (!cell.getValue().equals("")) {

                for (int i = cellCounter; i < (newCellPosition + 1); i++) {
                    sheet.autoSizeColumn(cellCounter);
                }
            }

            cellCounter = newCellPosition + 1;
        }
    }

    /**
     * Set data row.
     * @param sheet - Sheet.
     * @param row - Row.
     * @param cells - List of contingency table cells.
     * @param colRowStyle - Column row styles.
     * @param dataStyle - Data row styles.
     */
    private static void setDataRow(Sheet sheet, Row row, List<ContingencyTableRowCell> cells, CellStyle colRowStyle, CellStyle dataStyle) {

        int cellCounter = 0;

        for (ContingencyTableRowCell cell : cells) {

            // Create new xlsx cell.
            Cell xlsxCell = row.createCell(cellCounter);
            xlsxCell.setCellValue(cell.getValue());

            // First cell is always header, so apply different style.
            if (cellCounter == 0) {
                xlsxCell.setCellStyle(colRowStyle);
            } else {
                xlsxCell.setCellStyle(dataStyle);
            }

            sheet.autoSizeColumn(cellCounter);
            cellCounter++;
        }
    }
}