org.sigmah.server.report.renderer.excel.BaseExcelTableRenderer.java Source code

Introduction

Here is the source code for org.sigmah.server.report.renderer.excel.BaseExcelTableRenderer.java

Source

/*
 * All Sigmah code is released under the GNU General Public License v3
 * See COPYRIGHT.txt and LICENSE.txt.
 */

package org.sigmah.server.report.renderer.excel;

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.ss.util.CellRangeAddress;
import org.sigmah.shared.report.content.TreeNode;
import org.sigmah.shared.report.model.ReportElement;

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

public abstract class BaseExcelTableRenderer<ElementT extends ReportElement, ColumnT extends TreeNode>
        extends BaseExcelRenderer<ElementT> {

    protected Map<ColumnT, Integer> colIndexMap;
    protected CellStyle colHeaderStyle;
    protected CellStyle leafColHeaderStyle;

    public BaseExcelTableRenderer(Workbook book, ElementT element) {
        super(book, element);
    }

    protected void initColHeaderStyles(ColumnT root) {

        Font font = createBaseFont();
        font.setBoldweight(Font.BOLDWEIGHT_BOLD);

        colHeaderStyle = book.createCellStyle();
        colHeaderStyle.setWrapText(true);
        colHeaderStyle.setFont(font);
        colHeaderStyle.setAlignment(CellStyle.ALIGN_CENTER);

        leafColHeaderStyle = book.createCellStyle();
        leafColHeaderStyle.setWrapText(true);
        leafColHeaderStyle.setFont(font);
        leafColHeaderStyle.setAlignment(CellStyle.ALIGN_RIGHT);
    }

    protected void generateColumnHeaders(int firstCol, ColumnT root) {

        /* 
         * Now try building the column headers 
         */

        int depth = root.getDepth();
        colIndexMap = new HashMap<ColumnT, Integer>();

        int startLevel = depth == 0 ? 0 : 1;

        for (int level = startLevel; level <= depth; ++level) {

            Row row = sheet.createRow(rowIndex);

            int colIndex = firstCol;

            List<ColumnT> cols = root.getDescendantsAtDepth(level, true);

            for (ColumnT col : cols) {

                if (col == null) {
                    colIndex++;
                } else {
                    Cell cell = row.createCell(colIndex);
                    cell.setCellValue(factory.createRichTextString(col.getLabel()));
                    cell.setCellStyle(col.isLeaf() ? leafColHeaderStyle : colHeaderStyle);

                    int span = col.getLeaves().size();

                    if (span > 1) {
                        sheet.addMergedRegion(
                                new CellRangeAddress(rowIndex, rowIndex, colIndex, colIndex + span - 1));
                    }
                    if (col.isLeaf()) {
                        colIndexMap.put(col, colIndex);
                    }

                    colIndex += span;
                }
            }

            rowIndex++;
        }

    }

}