use of io.jmix.uiexport.exporter.ExportMode in project jmix by jmix-framework.
the class ExcelExporter method exportDataGrid.
@Override
public void exportDataGrid(Downloader downloader, DataGrid<Object> dataGrid, ExportMode exportMode) {
if (downloader == null) {
throw new IllegalArgumentException("Downloader is null");
}
createWorkbookWithSheet();
createFonts();
createFormats();
List<DataGrid.Column<Object>> columns = dataGrid.getColumns();
int r = 0;
Row row = sheet.createRow(r);
createAutoColumnSizers(columns.size());
float maxHeight = sheet.getDefaultRowHeightInPoints();
CellStyle headerCellStyle = wb.createCellStyle();
headerCellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
for (DataGrid.Column column : columns) {
String caption = column.getCaption();
int countOfReturnSymbols = StringUtils.countMatches(caption, "\n");
if (countOfReturnSymbols > 0) {
maxHeight = Math.max(maxHeight, (countOfReturnSymbols + 1) * sheet.getDefaultRowHeightInPoints());
headerCellStyle.setWrapText(true);
}
}
row.setHeightInPoints(maxHeight);
for (int c = 0; c < columns.size(); c++) {
DataGrid.Column column = columns.get(c);
String caption = column.getCaption();
Cell cell = row.createCell(c);
RichTextString richTextString = createStringCellValue(caption);
richTextString.applyFont(boldFont);
cell.setCellValue(richTextString);
ExcelAutoColumnSizer sizer = new ExcelAutoColumnSizer();
sizer.notifyCellValue(caption, boldFont);
sizers[c] = sizer;
cell.setCellStyle(headerCellStyle);
}
EntityDataGridItems<Object> dataGridSource = (EntityDataGridItems) dataGrid.getItems();
if (dataGridSource == null) {
throw new IllegalStateException("DataGrid is not bound to data");
}
if (exportMode == ExportMode.SELECTED && dataGrid.getSelected().size() > 0) {
Set<Object> selected = dataGrid.getSelected();
List<Object> ordered = dataGridSource.getItems().filter(selected::contains).collect(Collectors.toList());
for (Object item : ordered) {
if (checkIsRowNumberExceed(r)) {
break;
}
createDataGridRow(dataGrid, columns, 0, ++r, Id.of(item).getValue());
}
} else {
if (dataGrid instanceof TreeDataGrid) {
TreeDataGrid treeDataGrid = (TreeDataGrid) dataGrid;
TreeDataGridItems<Object> treeDataGridItems = (TreeDataGridItems) dataGridSource;
List<Object> items = treeDataGridItems.getChildren(null).collect(Collectors.toList());
for (Object item : items) {
if (checkIsRowNumberExceed(r)) {
break;
}
r = createDataGridHierarchicalRow(treeDataGrid, treeDataGridItems, columns, 0, r, item);
}
} else {
for (Object itemId : dataGridSource.getItems().map(entity -> Id.of(entity).getValue()).collect(Collectors.toList())) {
if (checkIsRowNumberExceed(r)) {
break;
}
createDataGridRow(dataGrid, columns, 0, ++r, itemId);
}
}
}
for (int c = 0; c < columns.size(); c++) {
sheet.setColumnWidth(c, sizers[c].getWidth() * COL_WIDTH_MAGIC);
}
ByteArrayOutputStream out = new ByteArrayOutputStream();
try {
wb.write(out);
} catch (IOException e) {
throw new RuntimeException("Unable to write document", e);
}
ByteArrayDataProvider dataProvider = new ByteArrayDataProvider(out.toByteArray(), uiProperties.getSaveExportedByteArrayDataThresholdBytes(), coreProperties.getTempDir());
switch(exportFormat) {
case XLSX:
downloader.download(dataProvider, getFileName(dataGrid) + "." + XLSX.getFileExt(), XLSX);
break;
case XLS:
downloader.download(dataProvider, getFileName(dataGrid) + "." + XLS.getFileExt(), XLS);
break;
}
}
use of io.jmix.uiexport.exporter.ExportMode in project jmix by jmix-framework.
the class ExcelExporter method exportTable.
@Override
public void exportTable(Downloader downloader, Table<Object> table, ExportMode exportMode) {
if (downloader == null) {
throw new IllegalArgumentException("Downloader is null");
}
if (table.getItems() == null) {
throw new IllegalStateException("Table items should not be null");
}
@SuppressWarnings("unchecked") List<Table.Column<Object>> columns = Collections.unmodifiableList(table.getNotCollapsedColumns()).stream().map(c -> (Table.Column<Object>) c).collect(Collectors.toList());
createWorkbookWithSheet();
createFonts();
createFormats();
int r = 0;
Row row = sheet.createRow(r);
createAutoColumnSizers(columns.size());
float maxHeight = sheet.getDefaultRowHeightInPoints();
CellStyle headerCellStyle = wb.createCellStyle();
headerCellStyle.setVerticalAlignment(VerticalAlignment.CENTER);
for (Table.Column<Object> column : columns) {
String caption = column.getCaption();
int countOfReturnSymbols = StringUtils.countMatches(caption, "\n");
if (countOfReturnSymbols > 0) {
maxHeight = Math.max(maxHeight, (countOfReturnSymbols + 1) * sheet.getDefaultRowHeightInPoints());
headerCellStyle.setWrapText(true);
}
}
row.setHeightInPoints(maxHeight);
for (int c = 0; c < columns.size(); c++) {
Table.Column<Object> column = columns.get(c);
String caption = column.getCaption();
Cell cell = row.createCell(c);
RichTextString richTextString = createStringCellValue(caption);
richTextString.applyFont(boldFont);
cell.setCellValue(richTextString);
ExcelAutoColumnSizer sizer = new ExcelAutoColumnSizer();
sizer.notifyCellValue(caption, boldFont);
sizers[c] = sizer;
cell.setCellStyle(headerCellStyle);
}
TableItems<Object> tableItems = table.getItems();
if (exportMode == ExportMode.SELECTED && table.getSelected().size() > 0) {
Set<Object> selected = table.getSelected();
List<Object> ordered = tableItems.getItemIds().stream().map(tableItems::getItem).filter(selected::contains).collect(Collectors.toList());
for (Object item : ordered) {
if (checkIsRowNumberExceed(r)) {
break;
}
createRow(table, columns, 0, ++r, Id.of(item).getValue());
}
} else {
if (table.isAggregatable() && exportAggregation && hasAggregatableColumn(table)) {
if (table.getAggregationStyle() == Table.AggregationStyle.TOP) {
r = createAggregatableRow(table, columns, ++r, 1);
}
}
if (table instanceof TreeTable) {
TreeTable<Object> treeTable = (TreeTable<Object>) table;
TreeTableItems<Object> treeTableSource = (TreeTableItems<Object>) treeTable.getItems();
if (treeTableSource != null) {
for (Object itemId : treeTableSource.getRootItemIds()) {
if (checkIsRowNumberExceed(r)) {
break;
}
r = createHierarchicalRow(treeTable, columns, exportExpanded, r, itemId);
}
}
} else if (table instanceof GroupTable && tableItems instanceof GroupTableItems && ((GroupTableItems<Object>) tableItems).hasGroups()) {
GroupTableItems<Object> groupTableSource = (GroupTableItems<Object>) tableItems;
for (Object item : groupTableSource.rootGroups()) {
if (checkIsRowNumberExceed(r)) {
break;
}
r = createGroupRow((GroupTable<Object>) table, columns, ++r, (GroupInfo<?>) item, 0);
}
} else {
if (tableItems != null) {
for (Object itemId : tableItems.getItemIds()) {
if (checkIsRowNumberExceed(r)) {
break;
}
createRow(table, columns, 0, ++r, itemId);
}
}
}
if (table.isAggregatable() && exportAggregation && hasAggregatableColumn(table)) {
if (table.getAggregationStyle() == Table.AggregationStyle.BOTTOM) {
r = createAggregatableRow(table, columns, ++r, 1);
}
}
}
for (int c = 0; c < columns.size(); c++) {
sheet.setColumnWidth(c, sizers[c].getWidth() * COL_WIDTH_MAGIC);
}
ByteArrayOutputStream out = new ByteArrayOutputStream();
try {
wb.write(out);
} catch (IOException e) {
throw new RuntimeException("Unable to write document", e);
}
if (isXlsMaxRowNumberExceeded()) {
Notifications notifications = ComponentsHelper.getScreenContext(table).getNotifications();
notifications.create(Notifications.NotificationType.WARNING).withCaption(messages.getMessage("actions.warningExport.title")).withDescription(messages.getMessage("actions.warningExport.message")).show();
}
ByteArrayDataProvider dataProvider = new ByteArrayDataProvider(out.toByteArray(), uiProperties.getSaveExportedByteArrayDataThresholdBytes(), coreProperties.getTempDir());
switch(exportFormat) {
case XLSX:
downloader.download(dataProvider, getFileName(table) + ".xlsx", XLSX);
break;
case XLS:
downloader.download(dataProvider, getFileName(table) + ".xls", XLS);
break;
}
}
Aggregations