123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271 |
- /* ====================================================================
- Licensed to the Apache Software Foundation (ASF) under one or more
- contributor license agreements. See the NOTICE file distributed with
- this work for additional information regarding copyright ownership.
- The ASF licenses this file to You under the Apache License, Version 2.0
- (the "License"); you may not use this file except in compliance with
- the License. You may obtain a copy of the License at
-
- http://www.apache.org/licenses/LICENSE-2.0
-
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- ==================================================================== */
-
- package org.apache.poi.xwpf.usermodel;
-
- import java.io.IOException;
- import java.io.InputStream;
- import java.util.Arrays;
-
- import org.apache.poi.common.usermodel.PictureType;
- import org.apache.poi.ooxml.POIXMLDocumentPart;
- import org.apache.poi.ooxml.POIXMLException;
- import org.apache.poi.ooxml.POIXMLRelation;
- import org.apache.poi.openxml4j.opc.OPCPackage;
- import org.apache.poi.openxml4j.opc.PackagePart;
- import org.apache.poi.util.IOUtils;
-
- /**
- * Raw picture data, normally attached to a WordprocessingML Drawing.
- * As a rule, pictures are stored in the /word/media/ part of a WordprocessingML package.
- */
- public class XWPFPictureData extends POIXMLDocumentPart {
-
- private static final int DEFAULT_MAX_IMAGE_SIZE = 100_000_000;
- private static int MAX_IMAGE_SIZE = DEFAULT_MAX_IMAGE_SIZE;
-
- /**
- * @param length the max image size allowed for XWPF pictures
- */
- public static void setMaxImageSize(int length) {
- MAX_IMAGE_SIZE = length;
- }
-
- /**
- * @return the max image size allowed for XSSF pictures
- */
- public static int getMaxImageSize() {
- return MAX_IMAGE_SIZE;
- }
-
- /**
- * Relationships for each known picture type
- */
- protected static final POIXMLRelation[] RELATIONS;
-
- static {
- RELATIONS = new POIXMLRelation[15];
- RELATIONS[PictureType.EMF.ooxmlId] = XWPFRelation.IMAGE_EMF;
- RELATIONS[PictureType.WMF.ooxmlId] = XWPFRelation.IMAGE_WMF;
- RELATIONS[PictureType.PICT.ooxmlId] = XWPFRelation.IMAGE_PICT;
- RELATIONS[PictureType.JPEG.ooxmlId] = XWPFRelation.IMAGE_JPEG;
- RELATIONS[PictureType.PNG.ooxmlId] = XWPFRelation.IMAGE_PNG;
- RELATIONS[PictureType.DIB.ooxmlId] = XWPFRelation.IMAGE_DIB;
- RELATIONS[PictureType.GIF.ooxmlId] = XWPFRelation.IMAGE_GIF;
- RELATIONS[PictureType.TIFF.ooxmlId] = XWPFRelation.IMAGE_TIFF;
- RELATIONS[PictureType.EPS.ooxmlId] = XWPFRelation.IMAGE_EPS;
- RELATIONS[PictureType.BMP.ooxmlId] = XWPFRelation.IMAGE_BMP;
- RELATIONS[PictureType.WPG.ooxmlId] = XWPFRelation.IMAGE_WPG;
- RELATIONS[PictureType.WDP.ooxmlId] = XWPFRelation.HDPHOTO_WDP;
- RELATIONS[PictureType.SVG.ooxmlId] = XWPFRelation.IMAGE_SVG;
- }
-
- private Long checksum;
-
- /**
- * Create a new XWPFGraphicData node
- */
- protected XWPFPictureData() {
- super();
- }
-
- /**
- * Construct XWPFPictureData from a package part
- *
- * @param part the package part holding the drawing data,
- *
- * @since POI 3.14-Beta1
- */
- public XWPFPictureData(PackagePart part) {
- super(part);
- }
-
- @Override
- protected void onDocumentRead() throws IOException {
- super.onDocumentRead();
- }
-
- /**
- * Gets the picture data as a byte array.
- * <p>
- * Note, that this call might be expensive since all the picture data is copied into a temporary byte array.
- * You can grab the picture data directly from the underlying package part as follows:
- * <br>
- * <code>
- * InputStream is = getPackagePart().getInputStream();
- * </code>
- * </p>
- *
- * @return the Picture data.
- */
- public byte[] getData() {
- try (InputStream stream = getPackagePart().getInputStream()) {
- return IOUtils.toByteArrayWithMaxLength(stream, getMaxImageSize());
- } catch (IOException e) {
- throw new POIXMLException(e);
- }
- }
-
- /**
- * Returns the file name of the image, eg image7.jpg . The original filename
- * isn't always available, but if it can be found it's likely to be in the
- * CTDrawing
- */
- public String getFileName() {
- String name = getPackagePart().getPartName().getName();
- return name.substring(name.lastIndexOf('/') + 1);
- }
-
- /**
- * Suggests a file extension for this image.
- *
- * @return the file extension.
- */
- public String suggestFileExtension() {
- return getPackagePart().getPartName().getExtension();
- }
-
- /**
- * Return an integer constant that specifies type of this picture
- *
- * @return an integer constant that specifies type of this picture, returns 0 if an unknown type
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_EMF
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_WMF
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_PICT
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_JPEG
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_PNG
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_GIF
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_DIB
- * @see org.apache.poi.xwpf.usermodel.Document#PICTURE_TYPE_SVG
- * @see #getPictureTypeEnum()
- */
- public int getPictureType() {
- String contentType = getPackagePart().getContentType();
- for (int i = 0; i < RELATIONS.length; i++) {
- if (RELATIONS[i] == null) {
- continue;
- }
-
- if (RELATIONS[i].getContentType().equals(contentType)) {
- return i;
- }
- }
- return 0;
- }
-
- /**
- * Return a {@link PictureType} that specifies type of this picture
- *
- * @return a {@link PictureType}, returns null if an unknown type
- * @since POI 5.2.3
- */
- public PictureType getPictureTypeEnum() {
- return PictureType.findByOoxmlId(getPictureType());
- }
-
- public Long getChecksum() {
- if (this.checksum == null) {
- try (InputStream is = getPackagePart().getInputStream()) {
- this.checksum = IOUtils.calculateChecksum(is);
- } catch (IOException e) {
- throw new POIXMLException(e);
- }
- }
- return this.checksum;
- }
-
- @Override
- public boolean equals(Object obj) {
- /*
- * In case two objects ARE equal, but its not the same instance, this
- * implementation will always run through the whole
- * byte-array-comparison before returning true. If this will turn into a
- * performance issue, two possible approaches are available:<br>
- * a) Use the checksum only and take the risk that two images might have
- * the same CRC32 sum, although they are not the same.<br>
- * b) Use a second (or third) checksum algorithm to minimise the chance
- * that two images have the same checksums but are not equal (e.g.
- * CRC32, MD5 and SHA-1 checksums, additionally compare the
- * data-byte-array lengths).
- */
- if (obj == this) {
- return true;
- }
-
- if (obj == null) {
- return false;
- }
-
- if (!(obj instanceof XWPFPictureData)) {
- return false;
- }
-
- XWPFPictureData picData = (XWPFPictureData) obj;
- PackagePart foreignPackagePart = picData.getPackagePart();
- PackagePart ownPackagePart = this.getPackagePart();
-
- if ((foreignPackagePart != null && ownPackagePart == null)
- || (foreignPackagePart == null && ownPackagePart != null)) {
- return false;
- }
-
- if (ownPackagePart != null) {
- OPCPackage foreignPackage = foreignPackagePart.getPackage();
- OPCPackage ownPackage = ownPackagePart.getPackage();
-
- if ((foreignPackage != null && ownPackage == null)
- || (foreignPackage == null && ownPackage != null)) {
- return false;
- }
- if (ownPackage != null) {
-
- if (!ownPackage.equals(foreignPackage)) {
- return false;
- }
- }
- }
-
- Long foreignChecksum = picData.getChecksum();
- Long localChecksum = getChecksum();
-
- if (localChecksum == null) {
- if (foreignChecksum != null) {
- return false;
- }
- } else {
- if (!(localChecksum.equals(foreignChecksum))) {
- return false;
- }
- }
- return Arrays.equals(this.getData(), picData.getData());
- }
-
- @Override
- public int hashCode() {
- Long checksum = getChecksum();
- return checksum == null ? super.hashCode() : checksum.hashCode();
- }
-
- /**
- * *PictureData objects store the actual content in the part directly without keeping a
- * copy like all others therefore we need to handle them differently.
- */
- @Override
- protected void prepareForCommit() {
- // do not clear the part here
- }
- }
|