Revize d90fc9de

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

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

@@ -1 +1 @@
 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.FileNotFoundException;
+import java.io.File;
 import java.io.FileOutputStream;
-import java.io.IOException;
+import java.util.List;
 
 /**
- * Class used for exporting data to XLSX
+ * Class used for exporting data to XLSX.
  */
 public class ExporterXLSX {
 
     /**
-     * !!!DUMMY!!!
      * Exports data to .xlsx file
-     * @param tableName name of the table
+     * @param tableName - name of the table.
+     * @param contingencyTableRows - List of contingency table rows.
      */
-    public static void export(String tableName) {
-
-        //temporary dummy implementation, used for testing
-        String[] columns = new String[] {"studuje", "nestuduje"};
-        String[] rows = new String[] {"fav", "fst"};
-        String[] data = new String[] {"1", "2", "3", "4"};
+    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);
 
-        Row descRow = sheet.createRow(0);
+        int rowCounter = 0;
 
-        for (int i = 0; i < 2; i++) {
-            Cell cell = descRow.createCell(i + 1);
-            cell.setCellValue(columns[i]);
-            cell.setCellStyle(colRowStyle);
-        }
+        for (ContingencyTableRow row : contingencyTableRows) {
 
-        for (int i = 0; i < 2; i++) {
-            Row row = sheet.createRow(i + 1);
-            Cell nameCell = row.createCell(0);
-            nameCell.setCellValue(rows[i]);
-            nameCell.setCellStyle(colRowStyle);
+            Row xlsxRow = sheet.createRow(rowCounter);
 
-            Cell cell1 = row.createCell(1);
-            cell1.setCellValue(data[i + i + 1]);
-            cell1.setCellStyle(dataStyle);
+            if (row.isHeader()) {
+                // Row is header.
+                setColRow(sheet, xlsxRow, row.getCells(), colRowStyle);
+            }
+            else {
+                // Row contains data.
+                setDataRow(sheet, xlsxRow, row.getCells(), colRowStyle, dataStyle);
+            }
 
-            Cell cell2 = row.createCell(2);
-            cell2.setCellValue(data[i + i + 1]);
-            cell2.setCellStyle(dataStyle);
+            rowCounter++;
         }
 
-        for (int i = 0; i < 3; i++) {
-            sheet.autoSizeColumn(i);
-        }
+        try {
 
-        //merge of columns, will be used for column headers
-        //sheet.addMergedRegion(new CellRangeAddress(0,0,1,2));
+            FileOutputStream out = new FileOutputStream(tmpFile);
+            workbook.write(out);
+            out.close();
 
-        FileOutputStream streamOut = null;
+        } catch (Exception e) {
 
-        try {
-            streamOut = new FileOutputStream("out.xlsx");
-        } catch (FileNotFoundException e) {
-            e.printStackTrace();
         }
+    }
 
-        try {
-            workbook.write(streamOut);
-        } catch (IOException e) {
-            e.printStackTrace();
-        }
+    /**
+     * 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) {
 
-        try {
-            if (streamOut != null) {
-                streamOut.close();
+        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);
+                }
             }
-        } catch (IOException e) {
-            e.printStackTrace();
+
+            cellCounter = newCellPosition + 1;
         }
+    }
 
-        try {
-            workbook.close();
-        } catch (IOException e) {
-            e.printStackTrace();
+    /**
+     * 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++;
         }
     }
 }