Examples with EscherAggregate org.apache.poi.hssf.record.EscherAggregate used on opensource projects

Example 11 with EscherAggregate

use of org.apache.poi.hssf.record.EscherAggregate in project poi by apache.

the class TestShapeGroup method testClearShapes.

@Test
public void testClearShapes() throws IOException {
    HSSFWorkbook wb1 = new HSSFWorkbook();
    HSSFSheet sheet = wb1.createSheet();
    HSSFPatriarch patriarch = sheet.createDrawingPatriarch();
    HSSFShapeGroup group = patriarch.createGroup(new HSSFClientAnchor());
    group.createShape(new HSSFChildAnchor());
    group.createShape(new HSSFChildAnchor());
    EscherAggregate agg = HSSFTestHelper.getEscherAggregate(patriarch);
    assertEquals(agg.getShapeToObjMapping().size(), 5);
    assertEquals(agg.getTailRecords().size(), 0);
    assertEquals(group.getChildren().size(), 2);
    group.clear();
    assertEquals(agg.getShapeToObjMapping().size(), 1);
    assertEquals(agg.getTailRecords().size(), 0);
    assertEquals(group.getChildren().size(), 0);
    HSSFWorkbook wb2 = HSSFTestDataSamples.writeOutAndReadBack(wb1);
    wb1.close();
    sheet = wb2.getSheetAt(0);
    patriarch = sheet.getDrawingPatriarch();
    group = (HSSFShapeGroup) patriarch.getChildren().get(0);
    assertEquals(agg.getShapeToObjMapping().size(), 1);
    assertEquals(agg.getTailRecords().size(), 0);
    assertEquals(group.getChildren().size(), 0);
    wb2.close();
}

Example 12 with EscherAggregate

use of org.apache.poi.hssf.record.EscherAggregate in project poi by apache.

the class InternalWorkbook method cloneDrawings.

/**
     * Check if the cloned sheet has drawings. If yes, then allocate a new drawing group ID and
     * re-generate shape IDs
     *
     * @param sheet the cloned sheet
     */
public void cloneDrawings(InternalSheet sheet) {
    findDrawingGroup();
    if (drawingManager == null) {
        //this workbook does not have drawings
        return;
    }
    //check if the cloned sheet has drawings
    int aggLoc = sheet.aggregateDrawingRecords(drawingManager, false);
    if (aggLoc == -1) {
        return;
    }
    EscherAggregate agg = (EscherAggregate) sheet.findFirstRecordBySid(EscherAggregate.sid);
    EscherContainerRecord escherContainer = agg.getEscherContainer();
    if (escherContainer == null) {
        return;
    }
    EscherDggRecord dgg = drawingManager.getDgg();
    //register a new drawing group for the cloned sheet
    int dgId = drawingManager.findNewDrawingGroupId();
    dgg.addCluster(dgId, 0);
    dgg.setDrawingsSaved(dgg.getDrawingsSaved() + 1);
    EscherDgRecord dg = null;
    for (EscherRecord er : escherContainer) {
        if (er instanceof EscherDgRecord) {
            dg = (EscherDgRecord) er;
            //update id of the drawing in the cloned sheet
            dg.setOptions((short) (dgId << 4));
        } else if (er instanceof EscherContainerRecord) {
            // iterate over shapes and re-generate shapeId
            for (EscherRecord er2 : (EscherContainerRecord) er) {
                for (EscherRecord shapeChildRecord : (EscherContainerRecord) er2) {
                    int recordId = shapeChildRecord.getRecordId();
                    if (recordId == EscherSpRecord.RECORD_ID) {
                        if (dg == null) {
                            throw new RecordFormatException("EscherDgRecord wasn't set/processed before.");
                        }
                        EscherSpRecord sp = (EscherSpRecord) shapeChildRecord;
                        int shapeId = drawingManager.allocateShapeId((short) dgId, dg);
                        //allocateShapeId increments the number of shapes. roll back to the previous value
                        dg.setNumShapes(dg.getNumShapes() - 1);
                        sp.setShapeId(shapeId);
                    } else if (recordId == EscherOptRecord.RECORD_ID) {
                        EscherOptRecord opt = (EscherOptRecord) shapeChildRecord;
                        EscherSimpleProperty prop = (EscherSimpleProperty) opt.lookup(EscherProperties.BLIP__BLIPTODISPLAY);
                        if (prop != null) {
                            int pictureIndex = prop.getPropertyValue();
                            // increment reference count for pictures
                            EscherBSERecord bse = getBSERecord(pictureIndex);
                            bse.setRef(bse.getRef() + 1);
                        }
                    }
                }
            }
        }
    }
}

Example 13 with EscherAggregate

use of org.apache.poi.hssf.record.EscherAggregate in project poi by apache.

the class TestCloneSheet method testCloneSheetWithEmptyDrawingAggregate.

@Test
public void testCloneSheetWithEmptyDrawingAggregate() {
    HSSFWorkbook b = new HSSFWorkbook();
    HSSFSheet s = b.createSheet("Test");
    HSSFPatriarch patriarch = s.createDrawingPatriarch();
    EscherAggregate agg1 = patriarch.getBoundAggregate();
    HSSFSheet s2 = s.cloneSheet(b);
    patriarch = s2.getDrawingPatriarch();
    EscherAggregate agg2 = patriarch.getBoundAggregate();
    EscherSpRecord sp1 = (EscherSpRecord) agg1.getEscherContainer().getChild(1).getChild(0).getChild(1);
    EscherSpRecord sp2 = (EscherSpRecord) agg2.getEscherContainer().getChild(1).getChild(0).getChild(1);
    assertEquals(sp1.getShapeId(), 1024);
    assertEquals(sp2.getShapeId(), 2048);
    EscherDgRecord dg = (EscherDgRecord) agg2.getEscherContainer().getChild(0);
    assertEquals(dg.getLastMSOSPID(), 2048);
    assertEquals(dg.getInstance(), 0x2);
    //everything except id and DgRecord.lastMSOSPID and DgRecord.Instance must be the same
    sp2.setShapeId(1024);
    dg.setLastMSOSPID(1024);
    dg.setInstance((short) 0x1);
    assertEquals(agg1.serialize().length, agg2.serialize().length);
    assertEquals(agg1.toXml(""), agg2.toXml(""));
    assertArrayEquals(agg1.serialize(), agg2.serialize());
}

Example 14 with EscherAggregate

use of org.apache.poi.hssf.record.EscherAggregate in project poi by apache.

the class InternalSheet method cloneSheet.

/**
     * Clones the low level records of this sheet and returns the new sheet instance.
     * This method is implemented by adding methods for deep cloning to all records that
     * can be added to a sheet. The <b>Record</b> object does not implement cloneable.
     * When adding a new record, implement a public clone method if and only if the record
     * belongs to a sheet.
     * 
     * @return the cloned sheet
     */
public InternalSheet cloneSheet() {
    List<Record> clonedRecords = new ArrayList<Record>(_records.size());
    for (int i = 0; i < _records.size(); i++) {
        RecordBase rb = _records.get(i);
        if (rb instanceof RecordAggregate) {
            ((RecordAggregate) rb).visitContainedRecords(new RecordCloner(clonedRecords));
            continue;
        }
        if (rb instanceof EscherAggregate) {
            /**
                 * this record will be removed after reading actual data from EscherAggregate
                 */
            rb = new DrawingRecord();
        }
        try {
            Record rec = (Record) ((Record) rb).clone();
            clonedRecords.add(rec);
        } catch (CloneNotSupportedException e) {
            throw new RecordFormatException(e);
        }
    }
    return createSheet(new RecordStream(clonedRecords, 0));
}

Example 15 with EscherAggregate

use of org.apache.poi.hssf.record.EscherAggregate in project poi by apache.

the class InternalSheet method aggregateDrawingRecords.

/**
     * Finds the DrawingRecord for our sheet, and
     *  attaches it to the DrawingManager (which knows about
     *  the overall DrawingGroup for our workbook).
     * If requested, will create a new DrawRecord
     *  if none currently exist
     * @param drawingManager The DrawingManager2 for our workbook
     * @param createIfMissing Should one be created if missing?
     * @return location of EscherAggregate record. if no EscherAggregate record is found return -1
     */
public int aggregateDrawingRecords(DrawingManager2 drawingManager, boolean createIfMissing) {
    int loc = findFirstRecordLocBySid(DrawingRecord.sid);
    boolean noDrawingRecordsFound = (loc == -1);
    if (noDrawingRecordsFound) {
        if (!createIfMissing) {
            // None found, and not allowed to add in
            return -1;
        }
        EscherAggregate aggregate = new EscherAggregate(true);
        loc = findFirstRecordLocBySid(EscherAggregate.sid);
        if (loc == -1) {
            loc = findFirstRecordLocBySid(WindowTwoRecord.sid);
        } else {
            getRecords().remove(loc);
        }
        getRecords().add(loc, aggregate);
        return loc;
    }
    List<RecordBase> records = getRecords();
    EscherAggregate.createAggregate(records, loc);
    return loc;
}