org.sejda.impl.sambox.component.excel.DataTable Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sejda-sambox Show documentation
Show all versions of sejda-sambox Show documentation
Package containing tasks implemented using sambox.
/*
* Copyright 2016 by Eduard Weissmann ([email protected]).
*
* This file is part of the Sejda source code
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see > data = new ArrayList<>();
private final TreeSet pageNumbers = new TreeSet<>();
public DataTable(int pageNumber) {
this.pageNumbers.add(pageNumber);
}
public DataTable(Collection pageNumbers) {
this.pageNumbers.addAll(pageNumbers);
}
public DataTable addRow(String... dataRow) {
List row = new ArrayList<>();
row.addAll(Arrays.asList(dataRow));
addRow(row);
return this;
}
public void addRow(List dataRow) {
data.add(dataRow);
}
public void addRows(List> dataRows) {
dataRows.forEach(this::addRow);
}
public List headerRow() {
return data.get(0);
}
public List headerRowIgnoreBlanks() {
return data.get(0).stream().filter(s -> !s.trim().isEmpty()).collect(Collectors.toList());
}
public boolean hasSameHeaderAs(DataTable other) {
String thisHeader = String.join("", this.headerRowIgnoreBlanks()).trim();
String otherHeader = String.join("", other.headerRowIgnoreBlanks()).trim();
LOG.debug("Comparing header columns: '{}' and '{}'", thisHeader, otherHeader);
return thisHeader.equalsIgnoreCase(otherHeader);
}
public boolean hasSameHeaderBlanksIgnoredAs(DataTable other) {
return this.headerRowIgnoreBlanks().equals(other.headerRowIgnoreBlanks());
}
public boolean hasSameColumnCountAs(DataTable other) {
LOG.debug("Comparing header columns size: {} and {}", this.headerRow().size(), other.headerRow().size());
return other.headerRow().size() == this.headerRow().size();
}
public List> getData() {
return data;
}
public TreeSet getPageNumbers() {
return pageNumbers;
}
public DataTable mergeWith(DataTable other) {
TreeSet resultPageNumbers = new TreeSet<>();
resultPageNumbers.addAll(this.pageNumbers);
resultPageNumbers.addAll(other.pageNumbers);
DataTable result = new DataTable(resultPageNumbers);
result.addRows(this.data);
List> otherData = other.data;
if (this.hasSameHeaderAs(other)) {
otherData.remove(0);
}
result.addRows(otherData);
return result;
}
private boolean hasConsecutivePages() {
Integer prev = null;
for(Integer current: pageNumbers) {
if(prev != null) {
if(prev != current - 1) {
return false;
}
}
prev = current;
}
return true;
}
public String getPagesAsString() {
StringBuilder sb = new StringBuilder();
if (this.pageNumbers.size() > 1) {
sb.append("Pages ");
if(pageNumbers.size() > 2 && hasConsecutivePages()) {
sb.append(pageNumbers.first()).append("-").append(pageNumbers.last());
return sb.toString();
}
int i = 0;
for (Integer pageNumber : pageNumbers) {
if (i != 0) {
sb.append(", ");
}
sb.append(pageNumber);
i++;
}
} else {
sb.append("Page ").append(pageNumbers.iterator().next());
}
return sb.toString();
}
public boolean hasData() {
return this.data.size() > 0;
}
public int getColumnsCount() {
int result = 0;
for (List row : data) {
result = Math.max(row.size(), result);
}
return result;
}
public int getRowsCount() {
return data.size();
}
public List getColumn(int c) {
List result = new ArrayList<>(getRowsCount());
for (List row : data) {
result.add(getOrEmpty(row, c));
}
return result;
}
public List getRow(int r) {
return data.get(r);
}
public DataTable mergeColumns(int c1, int c2) {
DataTable result = new DataTable(getPageNumbers());
for (List row : data) {
List newRow = new ArrayList<>(row.size() - 1);
for (int c = 0; c < row.size(); c++) {
if (c == c2) {
// noop, continue
} else if (c == c1) {
String newValue = (getOrEmpty(row, c1) + " " + getOrEmpty(row, c2)).trim();
newRow.add(newValue);
} else {
newRow.add(getOrEmpty(row, c));
}
}
result.addRow(newRow);
}
return result;
}
public void addBlankColumn(int index) {
for(List row : this.data) {
row.add(index, "");
}
}
private static String getOrEmpty(List list, int index) {
if (list.size() <= index) {
return "";
}
return list.get(index);
}
@Override
public String toString() {
StringBuilder sb = new StringBuilder();
int totalWidth = 0;
int colCount = getColumnsCount();
List colWidths = new ArrayList<>();
for (int i = 0; i < colCount; i++) {
List col = getColumn(i);
int colWidth = 0;
for (int j = 0; j < col.size(); j++) {
colWidth = Math.max(colWidth, col.get(j).length());
}
colWidths.add(colWidth);
totalWidth += colWidth;
}
String line = "+" + StringUtils.repeat("-", totalWidth + colCount - 1) + "+";
for(int i = 0; i < getRowsCount(); i++) {
List row = getRow(i);
sb.append("\n").append(line).append("\n");
for(int j = 0; j < colWidths.size(); j++) {
// TODO: fix padding for arabic when unprintable chars are present
// TODO: ensure unprintable characters are not counted when padding (all columns have same width)
String cellPadded = rightPad("", colWidths.get(j));
if(j < row.size()) {
cellPadded = isolateRTLIfRequired(rightPad(row.get(j), colWidths.get(j)));
}
sb.append("|").append(cellPadded);
}
sb.append("|");
}
sb.append("\n").append(line).append("\n");
return sb.toString();
}
}