123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- /* ====================================================================
- 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.xssf.usermodel;
-
- import java.util.Arrays;
- import java.util.List;
-
- import junit.framework.TestCase;
-
- import org.apache.poi.xssf.XSSFTestDataSamples;
-
- /**
- * @author Yegor Kozlov
- */
- public final class TestXSSFPictureData extends TestCase {
- public void testRead(){
- XSSFWorkbook wb = XSSFTestDataSamples.openSampleWorkbook("WithDrawing.xlsx");
- List<XSSFPictureData> pictures = wb.getAllPictures();
- //wb.getAllPictures() should return the same instance across multiple calls
- assertSame(pictures, wb.getAllPictures());
-
- assertEquals(5, pictures.size());
- String[] ext = {"jpeg", "emf", "png", "emf", "wmf"};
- String[] mimetype = {"image/jpeg", "image/x-emf", "image/png", "image/x-emf", "image/x-wmf"};
- for (int i = 0; i < pictures.size(); i++) {
- assertEquals(ext[i], pictures.get(i).suggestFileExtension());
- assertEquals(mimetype[i], pictures.get(i).getMimeType());
- }
-
- int num = pictures.size();
-
- byte[] pictureData = {0xA, 0xB, 0XC, 0xD, 0xE, 0xF};
-
- int idx = wb.addPicture(pictureData, XSSFWorkbook.PICTURE_TYPE_JPEG);
- assertEquals(num + 1, pictures.size());
- //idx is 0-based index in the #pictures array
- assertEquals(pictures.size() - 1, idx);
- XSSFPictureData pict = pictures.get(idx);
- assertEquals("jpeg", pict.suggestFileExtension());
- assertTrue(Arrays.equals(pictureData, pict.getData()));
- }
-
- public void testNew(){
- XSSFWorkbook wb = new XSSFWorkbook();
- XSSFSheet sheet = wb.createSheet();
- XSSFDrawing drawing = sheet.createDrawingPatriarch();
-
- byte[] jpegData = "test jpeg data".getBytes();
- byte[] wmfData = "test wmf data".getBytes();
- byte[] pngData = "test png data".getBytes();
-
- List<XSSFPictureData> pictures = wb.getAllPictures();
- assertEquals(0, pictures.size());
-
- int jpegIdx = wb.addPicture(jpegData, XSSFWorkbook.PICTURE_TYPE_JPEG);
- assertEquals(1, pictures.size());
- assertEquals("jpeg", pictures.get(jpegIdx).suggestFileExtension());
- assertTrue(Arrays.equals(jpegData, pictures.get(jpegIdx).getData()));
-
- int wmfIdx = wb.addPicture(wmfData, XSSFWorkbook.PICTURE_TYPE_WMF);
- assertEquals(2, pictures.size());
- assertEquals("wmf", pictures.get(wmfIdx).suggestFileExtension());
- assertTrue(Arrays.equals(wmfData, pictures.get(wmfIdx).getData()));
-
- int pngIdx = wb.addPicture(pngData, XSSFWorkbook.PICTURE_TYPE_PNG);
- assertEquals(3, pictures.size());
- assertEquals("png", pictures.get(pngIdx).suggestFileExtension());
- assertTrue(Arrays.equals(pngData, pictures.get(pngIdx).getData()));
-
- //TODO finish usermodel API for XSSFPicture
- XSSFPicture p1 = drawing.createPicture(new XSSFClientAnchor(), jpegIdx);
- XSSFPicture p2 = drawing.createPicture(new XSSFClientAnchor(), wmfIdx);
- XSSFPicture p3 = drawing.createPicture(new XSSFClientAnchor(), pngIdx);
-
- //check that the added pictures are accessible after write
- wb = XSSFTestDataSamples.writeOutAndReadBack(wb);
- List<XSSFPictureData> pictures2 = wb.getAllPictures();
- assertEquals(3, pictures2.size());
-
- assertEquals("jpeg", pictures2.get(jpegIdx).suggestFileExtension());
- assertTrue(Arrays.equals(jpegData, pictures2.get(jpegIdx).getData()));
-
- assertEquals("wmf", pictures2.get(wmfIdx).suggestFileExtension());
- assertTrue(Arrays.equals(wmfData, pictures2.get(wmfIdx).getData()));
-
- assertEquals("png", pictures2.get(pngIdx).suggestFileExtension());
- assertTrue(Arrays.equals(pngData, pictures2.get(pngIdx).getData()));
-
- }
-
- /**
- * Bug 53568: XSSFPicture.getPictureData() can return null.
- */
- public void test53568(){
- XSSFWorkbook wb = XSSFTestDataSamples.openSampleWorkbook("53568.xlsx");
- List<XSSFPictureData> pictures = wb.getAllPictures();
-
- XSSFSheet sheet1 = wb.getSheetAt(0);
- List<XSSFShape> shapes1 = sheet1.createDrawingPatriarch().getShapes();
-
- for(int i = 0; i < wb.getNumberOfSheets(); i++){
- XSSFSheet sheet = wb.getSheetAt(i);
- XSSFDrawing drawing = sheet.createDrawingPatriarch();
- for(XSSFShape shape : drawing.getShapes()){
- if(shape instanceof XSSFPicture){
- XSSFPicture pic = (XSSFPicture)shape;
- XSSFPictureData picData = pic.getPictureData();
- assertNotNull(picData);
- }
- }
- }
-
- }
- }
|