import org.openxml4j.exceptions.OpenXML4JException;
import org.openxml4j.opc.Package;
import org.openxml4j.opc.PackagePart;
+import org.openxml4j.opc.PackagePartName;
import org.openxml4j.opc.PackageRelationship;
import org.openxml4j.opc.PackageRelationshipTypes;
+import org.openxml4j.opc.PackagingURIHelper;
public abstract class POIXMLDocument {
protected PackagePart getCorePart() {
return this.corePart;
}
+
+ protected PackagePart getPart(PackageRelationship rel) throws InvalidFormatException {
+ PackagePartName relName = PackagingURIHelper.createPartName(rel.getTargetURI());
+ PackagePart part = getPackage().getPart(relName);
+ if (part == null) {
+ throw new IllegalArgumentException("No part found for relationship " + rel);
+ }
+ return part;
+ }
}
private static final String TRUE_AS_STRING = "1";
private final CTCell cell;
private final XSSFRow row;
- private SharedStringSource sharedStringSource;
private short cellNum;
+ private SharedStringSource sharedStringSource;
/**
* Create a new XSSFCell. This method is protected to be used only by
if (cell.getR() != null) {
this.cellNum = parseCellNum(cell.getR());
}
+ this.sharedStringSource = row.getSheet().getWorkbook().getSharedStringSource();
}
-
- protected void setSharedStringSource(SharedStringSource sharedStringSource) {
- this.sharedStringSource = sharedStringSource;
+
+ protected SharedStringSource getSharedStringSource() {
+ return this.sharedStringSource;
}
public boolean getBooleanCellValue() {
if(this.cell.getT() == STCellType.S) {
if(this.cell.isSetV()) {
int sRef = Integer.parseInt(this.cell.getV());
- return new XSSFRichTextString(sharedStringSource.getSharedStringAt(sRef));
+ return new XSSFRichTextString(getSharedStringSource().getSharedStringAt(sRef));
} else {
return new XSSFRichTextString("");
}
if(this.cell.getT() != STCellType.S) {
this.cell.setT(STCellType.S);
}
- int sRef = sharedStringSource.putSharedString(value.getString());
+ int sRef = getSharedStringSource().putSharedString(value.getString());
this.cell.setV(Integer.toString(sRef));
}
private List<Cell> cells;
+ private XSSFSheet sheet;
+
/**
* Create a new XSSFRow. This method is protected to be used only by
* tests.
*/
- protected XSSFRow() {
- this(CTRow.Factory.newInstance());
+ protected XSSFRow(XSSFSheet sheet) {
+ this(CTRow.Factory.newInstance(), sheet);
}
- public XSSFRow(CTRow row) {
+ /**
+ * Create a new XSSFRow.
+ *
+ * @param row The underlying XMLBeans row.
+ * @param sheet The parent sheet.
+ */
+ public XSSFRow(CTRow row, XSSFSheet sheet) {
this.row = row;
+ this.sheet = sheet;
this.cells = new LinkedList<Cell>();
for (CTCell c : row.getCArray()) {
this.cells.add(new XSSFCell(this, c));
}
}
+
+ public XSSFSheet getSheet() {
+ return this.sheet;
+ }
public Iterator<Cell> cellIterator() {
return cells.iterator();
private CTWorksheet worksheet;
private List<Row> rows;
private ColumnHelper columnHelper;
+ private XSSFWorkbook workbook;
public static final short LeftMargin = 0;
public static final short RightMargin = 1;
public static final short HeaderMargin = 4;
public static final short FooterMargin = 5;
- public XSSFSheet(CTSheet sheet) {
+ public XSSFSheet(CTSheet sheet, XSSFWorkbook workbook) {
this.sheet = sheet;
this.worksheet = CTWorksheet.Factory.newInstance();
+ this.workbook = workbook;
this.worksheet.addNewSheetData();
initRows(worksheet);
initColumns(worksheet);
}
- public XSSFSheet(CTSheet sheet, CTWorksheet worksheet) {
+ public XSSFSheet(CTSheet sheet, CTWorksheet worksheet, XSSFWorkbook workbook) {
this.sheet = sheet;
this.worksheet = worksheet;
+ this.workbook = workbook;
initRows(worksheet);
initColumns(worksheet);
}
+ public XSSFSheet(XSSFWorkbook workbook) {
+ this.workbook = workbook;
+ }
+
+ public XSSFWorkbook getWorkbook() {
+ return this.workbook;
+ }
+
private void initRows(CTWorksheet worksheet) {
this.rows = new LinkedList<Row>();
for (CTRow row : worksheet.getSheetData().getRowArray()) {
- this.rows.add(new XSSFRow(row));
+ this.rows.add(new XSSFRow(row, this));
}
}
}
- protected XSSFRow addRow(int index, int rownum) {
- CTRow row = this.worksheet.getSheetData().insertNewRow(index);
- XSSFRow xrow = new XSSFRow(row);
- xrow.setRowNum(rownum);
-// xrow.setHeight(13.41);
- return xrow;
- }
+ protected XSSFRow addRow(int index, int rownum) {
+ CTRow row = this.worksheet.getSheetData().insertNewRow(index);
+ XSSFRow xrow = new XSSFRow(row, this);
+ xrow.setRowNum(rownum);
+ return xrow;
+ }
public Row createRow(int rownum) {
int index = 0;
}
protected XSSFSheet cloneSheet() {
- return new XSSFSheet((CTSheet) sheet.copy());
+ return new XSSFSheet((CTSheet) sheet.copy(), this.workbook);
}
}
import java.io.IOException;
import java.io.OutputStream;
+import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import org.apache.poi.ss.usermodel.Font;
import org.apache.poi.ss.usermodel.Name;
import org.apache.poi.ss.usermodel.Palette;
+import org.apache.poi.ss.usermodel.SharedStringSource;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.xssf.strings.SharedStringsTable;
import org.apache.xmlbeans.XmlException;
import org.apache.xmlbeans.XmlObject;
import org.apache.xmlbeans.XmlOptions;
import org.openxml4j.opc.PackagePart;
import org.openxml4j.opc.PackagePartName;
import org.openxml4j.opc.PackageRelationship;
+import org.openxml4j.opc.PackageRelationshipCollection;
import org.openxml4j.opc.PackageRelationshipTypes;
import org.openxml4j.opc.PackagingURIHelper;
import org.openxml4j.opc.TargetMode;
private static final String WORKSHEET_RELATIONSHIP = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/worksheet";
+ private static final String SHARED_STRINGS_RELATIONSHIP = "http://schemas.openxmlformats.org/officeDocument/2006/relationships/sharedStrings";
+
private CTWorkbook workbook;
private List<XSSFSheet> sheets = new LinkedList<XSSFSheet>();
-
+
+ private SharedStringSource sharedStringSource;
+
public XSSFWorkbook() {
this.workbook = CTWorkbook.Factory.newInstance();
CTBookViews bvs = this.workbook.addNewBookViews();
try {
WorkbookDocument doc = WorkbookDocument.Factory.parse(getCorePart().getInputStream());
this.workbook = doc.getWorkbook();
+ // Load shared strings
+ PackageRelationshipCollection prc = getCorePart().getRelationshipsByType(SHARED_STRINGS_RELATIONSHIP);
+ Iterator<PackageRelationship> it = prc.iterator();
+ if (it.hasNext()) {
+ PackageRelationship rel = it.next();
+ PackagePart part = getPart(rel);
+ this.sharedStringSource = new SharedStringsTable(part);
+ }
+ // Load individual sheets
for (CTSheet ctSheet : this.workbook.getSheets().getSheetArray()) {
PackageRelationship rel = this.getCorePart().getRelationship(ctSheet.getId());
- PackagePartName relName = PackagingURIHelper.createPartName(rel.getTargetURI());
- PackagePart part = getPackage().getPart(relName);
- if (part == null) {
- throw new IllegalArgumentException("No part found for relationship " + rel);
- }
+ PackagePart part = getPart(rel);
WorksheetDocument worksheetDoc = WorksheetDocument.Factory.parse(part.getInputStream());
- XSSFSheet sheet = new XSSFSheet(ctSheet, worksheetDoc.getWorksheet());
+ XSSFSheet sheet = new XSSFSheet(ctSheet, worksheetDoc.getWorksheet(), this);
this.sheets.add(sheet);
}
} catch (XmlException e) {
throw new IOException(e.toString());
} catch (InvalidFormatException e) {
throw new IOException(e.toString());
-/*
- } catch (OpenXML4JException e) {
- throw new IOException(e.toString());
-*/
}
}
if (sheetname != null) {
sheet.setName(sheetname);
}
- XSSFSheet wrapper = new XSSFSheet(sheet);
+ XSSFSheet wrapper = new XSSFSheet(sheet, this);
this.sheets.add(wrapper);
return wrapper;
}
}
+ public SharedStringSource getSharedStringSource() {
+ return this.sharedStringSource;
+ }
+
+ protected void setSharedStringSource(SharedStringSource sharedStringSource) {
+ this.sharedStringSource = sharedStringSource;
+ }
+
}
Cell cell = row.getCell((short) 1);
assertNotNull(cell);
assertEquals(111.0, cell.getNumericCellValue());
+ cell = row.getCell((short) 0);
+ assertEquals("Lorem", cell.getRichStringCellValue().getString());
}
}
* Test setting and getting boolean values.
*/
public void testSetGetBoolean() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
cell.setCellValue(true);
assertEquals(Cell.CELL_TYPE_BOOLEAN, cell.getCellType());
assertTrue(cell.getBooleanCellValue());
* Test setting and getting numeric values.
*/
public void testSetGetNumeric() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
cell.setCellValue(10d);
assertEquals(Cell.CELL_TYPE_NUMERIC, cell.getCellType());
assertEquals(10d, cell.getNumericCellValue());
* Test setting and getting numeric values.
*/
public void testSetGetDate() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
Date now = new Date();
cell.setCellValue(now);
assertEquals(Cell.CELL_TYPE_NUMERIC, cell.getCellType());
}
public void testSetGetError() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
cell.setCellErrorValue((byte)255);
assertEquals(Cell.CELL_TYPE_ERROR, cell.getCellType());
}
public void testSetGetFormula() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
String formula = "SQRT(C2^2+D2^2)";
cell.setCellFormula(formula);
public void testSetGetStringInline() throws Exception {
CTCell rawCell = CTCell.Factory.newInstance();
- XSSFCell cell = new XSSFCell(new XSSFRow(), rawCell);
- cell.setSharedStringSource(new DummySharedStringSource());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row, rawCell);
// Default is shared string mode, so have to do this explicitly
rawCell.setT(STCellType.INLINE_STR);
}
public void testSetGetStringShared() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
- cell.setSharedStringSource(new DummySharedStringSource());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
cell.setCellValue(new XSSFRichTextString(""));
assertEquals(Cell.CELL_TYPE_STRING, cell.getCellType());
* Test that empty cells (no v element) return default values.
*/
public void testGetEmptyCellValue() throws Exception {
- XSSFCell cell = new XSSFCell(new XSSFRow());
+ XSSFRow row = createParentObjects();
+ XSSFCell cell = new XSSFCell(row);
cell.setCellType(Cell.CELL_TYPE_BOOLEAN);
assertFalse(cell.getBooleanCellValue());
cell.setCellType(Cell.CELL_TYPE_NUMERIC);
}
public void testFormatPosition() {
- XSSFRow row = new XSSFRow();
+ XSSFRow row = createParentObjects();
row.setRowNum(0);
XSSFCell cell = new XSSFCell(row);
cell.setCellNum((short) 0);
row.setRowNum(32767);
assertEquals("IV32768", cell.formatPosition());
}
+
+ private XSSFRow createParentObjects() {
+ XSSFWorkbook wb = new XSSFWorkbook();
+ wb.setSharedStringSource(new DummySharedStringSource());
+ XSSFSheet sheet = new XSSFSheet(wb);
+ XSSFRow row = new XSSFRow(sheet);
+ return row;
+ }
public static class DummySharedStringSource implements SharedStringSource {
ArrayList<String> strs = new ArrayList<String>();
import junit.framework.TestCase;
import org.apache.poi.ss.usermodel.Cell;
+import org.apache.poi.xssf.usermodel.TestXSSFCell.DummySharedStringSource;
public class TestXSSFRow extends TestCase {
* Test adding cells to a row in various places and see if we can find them again.
*/
public void testAddAndIterateCells() {
- XSSFRow row = new XSSFRow();
+ XSSFRow row = new XSSFRow(createParentObjects());
// One cell at the beginning
Cell cell1 = row.createCell((short) 1);
assertFalse(row.getFirstCellNum() == (short) 2);
// Test a row without cells
- XSSFRow emptyRow = new XSSFRow();
+ XSSFRow emptyRow = new XSSFRow(createParentObjects());
assertEquals(-1, emptyRow.getFirstCellNum());
}
* Method that returns a row with some sample cells
* @return row
*/
- public static XSSFRow getSampleRow() {
- XSSFRow row = new XSSFRow();
+ public XSSFRow getSampleRow() {
+ XSSFRow row = new XSSFRow(createParentObjects());
row.createCell((short) 2);
row.createCell((short) 3, Cell.CELL_TYPE_NUMERIC);
row.createCell((short) 4);
row.createCell((short) 100);
return row;
}
+
+ private XSSFSheet createParentObjects() {
+ XSSFWorkbook wb = new XSSFWorkbook();
+ wb.setSharedStringSource(new DummySharedStringSource());
+ return new XSSFSheet(wb);
+ }
}
projects / poi.git / commitdiff