Examples with Shape org.apache.poi.ss.usermodel.Shape used on opensource projects

Example 1 with Shape

use of org.apache.poi.ss.usermodel.Shape in project poi by apache.

the class EmbeddedExtractor method extractAll.

protected void extractAll(ShapeContainer<?> parent, List<EmbeddedData> embeddings) throws IOException {
    for (Shape shape : parent) {
        EmbeddedData data = null;
        if (shape instanceof ObjectData) {
            ObjectData od = (ObjectData) shape;
            try {
                if (od.hasDirectoryEntry()) {
                    data = extractOne((DirectoryNode) od.getDirectory());
                } else {
                    String contentType = CONTENT_TYPE_BYTES;
                    if (od instanceof XSSFObjectData) {
                        contentType = ((XSSFObjectData) od).getObjectPart().getContentType();
                    }
                    data = new EmbeddedData(od.getFileName(), od.getObjectData(), contentType);
                }
            } catch (Exception e) {
                LOG.log(POILogger.WARN, "Entry not found / readable - ignoring OLE embedding", e);
            }
        } else if (shape instanceof Picture) {
            data = extractOne((Picture) shape);
        } else if (shape instanceof ShapeContainer) {
            extractAll((ShapeContainer<?>) shape, embeddings);
        }
        if (data == null) {
            continue;
        }
        data.setShape(shape);
        String filename = data.getFilename();
        String extension = (filename == null || filename.lastIndexOf('.') == -1) ? ".bin" : filename.substring(filename.lastIndexOf('.'));
        // try to find an alternative name
        if (filename == null || "".equals(filename) || filename.startsWith("MBD") || filename.startsWith("Root Entry")) {
            filename = shape.getShapeName();
            if (filename != null) {
                filename += extension;
            }
        }
        // default to dummy name
        if (filename == null || "".equals(filename)) {
            filename = "picture_" + embeddings.size() + extension;
        }
        filename = filename.trim();
        data.setFilename(filename);
        embeddings.add(data);
    }
}