de.knowwe.include.export.TableExporter.java Source code

Introduction

Here is the source code for de.knowwe.include.export.TableExporter.java

Source

/*
 * Copyright (C) 2014 denkbares GmbH
 * 
 * This is free software; you can redistribute it and/or modify it under the
 * terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 3 of the License, or (at your option) any
 * later version.
 * 
 * This software is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this software; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA, or see the FSF
 * site: http://www.fsf.org.
 */
package de.knowwe.include.export;

import java.util.List;

import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFTable;
import org.apache.poi.xwpf.usermodel.XWPFTableCell;
import org.apache.poi.xwpf.usermodel.XWPFTableRow;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTR;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTShd;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTText;

import com.denkbares.collections.Matrix;
import com.denkbares.strings.Strings;
import de.knowwe.core.kdom.parsing.Section;
import de.knowwe.core.kdom.parsing.Sections;
import de.knowwe.instantedit.table.WikiTable;
import de.knowwe.jspwiki.types.ParagraphTypeForLists;
import de.knowwe.jspwiki.types.TableCell;
import de.knowwe.jspwiki.types.TableRow;

/**
 * Exports a wiki table to word document
 *
 * @author Volker Belli (denkbares GmbH)
 * @created 07.02.2014
 */
public class TableExporter implements Exporter<WikiTable> {

    @Override
    public boolean canExport(Section<WikiTable> section) {
        return true;
    }

    @Override
    public Class<WikiTable> getSectionType() {
        return WikiTable.class;
    }

    @Override
    public void export(Section<WikiTable> section, DocumentBuilder manager) throws ExportException {
        // initialize table for easier access
        Matrix<Section<TableCell>> matrix = toMatrix(section);

        // create table with correct dimension
        XWPFDocument doc = manager.getDocument();
        XWPFTable table = doc.createTable(matrix.getRowSize(), matrix.getColSize());
        boolean headerCellsOnly = true;

        for (int row = 0; row < matrix.getRowSize(); row++) {
            XWPFTableRow tableRow = table.getRow(row);
            for (int col = 0; col < matrix.getColSize(); col++) {
                Section<TableCell> cell = matrix.get(row, col);
                boolean isHeader = cell.get().isHeader(cell);
                boolean isZebra = row % 2 == 0;
                headerCellsOnly &= isHeader;

                // prepare cell shading
                XWPFTableCell tableCell = tableRow.getCell(col);
                CTShd shade = tableCell.getCTTc().addNewTcPr().addNewShd();
                if (isHeader) {
                    shade.setFill("D0D0D0");
                    tableCell.setVerticalAlignment(XWPFTableCell.XWPFVertAlign.CENTER);
                } else if (isZebra) {
                    shade.setFill("F2F2F2");
                }

                // fill cell contents
                DocumentBuilder cellBuilder = new CellBuilder(manager, tableCell, isHeader);
                Section<?> content = Sections.successor(cell, ParagraphTypeForLists.class);
                if (content != null) {
                    cellBuilder.export(content);
                }

                // clean trailing white-spaces of each cell
                List<CTR> runs = cellBuilder.getParagraph().getCTP().getRList();
                if (runs.isEmpty())
                    continue;
                CTR ctr = runs.get(runs.size() - 1);
                List<CTText> texts = ctr.getTList();
                if (texts.isEmpty())
                    continue;
                CTText ctText = texts.get(texts.size() - 1);
                ctText.setStringValue(Strings.trimRight(ctText.getStringValue()));
            }
            // check if the first row(s) have only headers,
            // repeat those headers until a non-header cell has come
            if (headerCellsOnly) {
                tableRow.setRepeatHeader(true);
            }
            tableRow.setCantSplitRow(true);
        }

        // append empty line after each table
        manager.closeParagraph();
        manager.append("\n\r");
        manager.closeParagraph();
    }

    private Matrix<Section<TableCell>> toMatrix(Section<WikiTable> section) {
        Matrix<Section<TableCell>> matrix = new Matrix<>();
        int row = 0;
        for (Section<TableRow> tableRow : Sections.successors(section, TableRow.class)) {
            int col = 0;
            for (Section<TableCell> cell : Sections.successors(tableRow, TableCell.class)) {
                matrix.set(row, col, cell);
                col++;
            }
            row++;
        }
        return matrix;
    }

}