Revize d90fc9de

re #8133 #8132 Export contingency table to XLSX + PDF (not fully complete).

src/main/java/vldc/aswi/utils/ExporterPDF.java

@@ -1 +1 @@
 package vldc.aswi.utils;
 
-import com.lowagie.text.Cell;
-import com.lowagie.text.Document;
+import be.quodlibet.boxable.BaseTable;
+import com.lowagie.text.*;
 import com.lowagie.text.Font;
-import com.lowagie.text.Phrase;
+import com.lowagie.text.alignment.HorizontalAlignment;
+import com.lowagie.text.alignment.VerticalAlignment;
 import com.lowagie.text.pdf.PdfWriter;
+import com.sun.media.jfxmedia.logging.Logger;
+import org.apache.pdfbox.pdmodel.PDPage;
+import org.apache.pdfbox.pdmodel.font.PDType1Font;
+import vldc.aswi.model.table.contingencyTable.ContingencyTableRow;
+import vldc.aswi.model.table.contingencyTable.ContingencyTableRowCell;
 
-import java.io.FileNotFoundException;
+import java.awt.*;
+import java.io.File;
 import java.io.FileOutputStream;
+import java.util.List;
 
 /**
- * Class used for exporting data to PDF files
+ * Class used for exporting data to PDF files.
  */
 public class ExporterPDF {
 
     /**
-     * !!!DUMMY!!!
      * Exports data to .pdf file
-     * @param tableName name of the table
+     * @param tmpFile - Temporary file.
+     * @param contingencyTableRows - Contingency table rows.
      */
-    public static void export(String tableName) throws FileNotFoundException {
-        Document doc = new Document();
+    public static void export(File tmpFile, List<ContingencyTableRow> contingencyTableRows) {
 
-        PdfWriter.getInstance(doc, new FileOutputStream("out.pdf"));
+        try {
+            if (contingencyTableRows.size() == 0) {
 
-        Font font = new Font(com.lowagie.text.Font.COURIER, 14);
-        com.lowagie.text.Table table = new com.lowagie.text.Table(3);
+                return;
+            }
 
-        Cell cell = new Cell(new Phrase(tableName, font));
-        cell.setColspan(2);
-        cell.setHeader(true);
+            // Create new document.
+            Document doc = new Document();
 
-        Cell cell2 = new Cell(new Phrase("SUM", font));
-        cell2.setColspan(1);
-        cell2.setHeader(true);
+            // Get instance of PdfWriter.
+            PdfWriter.getInstance(doc, new FileOutputStream(tmpFile));
 
-        table.addCell(cell);
-        table.addCell(cell2);
+            // Create fonts.
+            Font colRowFont = new Font(Font.HELVETICA, 14, Font.BOLD);
+            Font dataFont = new Font(Font.HELVETICA, 12);
 
-        table.addCell(new Phrase("Val1", font));
-        table.addCell(new Phrase("Val2", font));
-        table.addCell(new Phrase("", font));
+            // Calculate column and row count.
+            int columnsCount = getColumnCount(contingencyTableRows.get(0));
+            int rowsCount = contingencyTableRows.size();
 
-        table.addCell("1");
-        table.addCell("2");
-        table.addCell("3");
+            // Create table.
+            Table table = new Table(columnsCount, rowsCount);
 
-        table.addCell("4");
-        table.addCell("5");
-        table.addCell("9");
+            for (ContingencyTableRow row : contingencyTableRows) {
 
-        doc.open();
-        doc.add(table);
+                if (row.isHeader()) {
+                    // Row is header.
+                    setColRow(table, row.getCells(), colRowFont);
+                } else {
+                    // Row contains data.
+                    setDataRow(table, row.getCells(), colRowFont, dataFont);
+                }
+            }
 
-        doc.close();
+            // Open document and write created table into it.
+            doc.open();
+            doc.add(table);
+
+            doc.close();
+        } catch (Exception e) {
+
+            Logger.logMsg(Logger.WARNING, "Error while exporting contingency table to PDF.");
+            Logger.logMsg(Logger.WARNING, e.getStackTrace().toString());
+        }
+    }
+
+    /**
+     * Set column header row.
+     * @param table - Table.
+     * @param cells - List of contingency table cells.
+     * @param colRowFont - Column row styles.
+     * @throws BadElementException If cell is not created successfully.
+     */
+    private static void setColRow(Table table, List<ContingencyTableRowCell> cells, Font colRowFont) throws BadElementException {
+
+        for (ContingencyTableRowCell cell : cells) {
+
+            // Create and prepare pdf cell.
+            Cell pdfCell = prepareCell(cell.getValue(), colRowFont, cell.getColSpan());
+            pdfCell.setHeader(true);
+            table.addCell(pdfCell);
+        }
+    }
+
+    /**
+     * Set data row.
+     * @param table - Table.
+     * @param cells - List of contingency table cells.
+     * @param colRowFont - Column row styles.
+     * @param dataFont - Data row styles.
+     * @throws BadElementException If cell is not created successfully.
+     */
+    private static void setDataRow(Table table, List<ContingencyTableRowCell> cells, Font colRowFont, Font dataFont) throws BadElementException {
+
+        boolean headerWrited = false;
+
+        for (ContingencyTableRowCell cell : cells) {
+
+            // Create pdf cell.
+            Cell pdfCell;
+
+            // First cell is always header, so apply different style.
+            if (!headerWrited) {
+
+                headerWrited = true;
+
+                // Prepare pdf cell.
+                pdfCell = prepareCell(cell.getValue(), colRowFont, cell.getColSpan());
+                pdfCell.setHeader(true);
+
+            } else {
+
+                // Prepare pdf cell.
+                pdfCell = prepareCell(cell.getValue(), dataFont, cell.getColSpan());
+            }
+
+            table.addCell(pdfCell);
+        }
+    }
+
+    /**
+     * Prepare pdf cell.
+     * @param value - Cell value.
+     * @param font - Cell font.
+     * @param colSpan - Cell colspan.
+     * @return Prepared pdf cell.
+     */
+    private static Cell prepareCell(String value, Font font, int colSpan) {
+
+        Cell cell = new Cell(new Phrase(value, font));
+        cell.setColspan(colSpan);
+        cell.setHorizontalAlignment(HorizontalAlignment.CENTER);
+        cell.setVerticalAlignment(VerticalAlignment.CENTER);
+
+        return cell;
+    }
+
+    /**
+     * Calculate column count in contingency table row.
+     * @param firstRow - First row of contingency table.
+     * @return Column count in contingency table.
+     */
+    private static int getColumnCount(ContingencyTableRow firstRow) {
+
+        int columnCount = 0;
+
+        for (ContingencyTableRowCell cell : firstRow.getCells()) {
+
+            columnCount += cell.getColSpan();
+        }
+
+        return columnCount;
+    }
+
+
+    public static void exportt(File tmpFile, List<ContingencyTableRow> contingencyTableRows) {
+        /*
+        //Dummy Table
+        float margin = 50;
+
+        // starting y position is whole page height subtracted by top and bottom margin
+        float yStartNewPage = myPage.getMediaBox().getHeight() - (2 * margin);
+
+        // we want table across whole page width (subtracted by left and right margin ofcourse)
+        float tableWidth = myPage.getMediaBox().getWidth() - (2 * margin);
+
+        boolean drawContent = true;
+        float yStart = yStartNewPage;
+        float bottomMargin = 70;
+        // y position is your coordinate of top left corner of the table
+        float yPosition = 550;
+
+        BaseTable table = new BaseTable(yPosition, yStartNewPage, bottomMargin, tableWidth, margin, mainDocument, myPage, true, drawContent);
+
+
+        Row<PDPage> headerRow = table.createRow(15f);
+        Cell<PDPage> cell = headerRow.createCell(100, "Header");
+        table.addHeaderRow(headerRow);
+
+
+        Row<PDPage> row = table.createRow(12);
+        cell = row.createCell(30, "Data 1");
+        cell = row.createCell(70, "Some value");
+
+        table.draw();
+
+
+        contentStream.close();
+        mainDocument.addPage(myPage);
+        mainDocument.save("testfile.pdf");
+        mainDocument.close();
+         */
     }
 }