View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  
18  package org.apache.commons.imaging.formats.jpeg.iptc;
19  
20  import java.io.File;
21  import java.io.IOException;
22  import java.util.List;
23  
24  import org.apache.commons.imaging.ImageReadException;
25  import org.apache.commons.imaging.ImagingTest;
26  import org.apache.commons.imaging.common.bytesource.ByteSource;
27  import org.apache.commons.imaging.common.bytesource.ByteSourceFile;
28  import org.apache.commons.imaging.formats.jpeg.JpegImageParser;
29  
30  public abstract class IptcBaseTest extends ImagingTest {
31      protected static boolean hasIptcData(final File file) {
32          // Debug.debug("hasIptcData file", file.getAbsoluteFile());
33  
34          if (!file.getName().toLowerCase().endsWith(".jpg"))
35           {
36              return false;
37          // ImageFormat format = Imaging.guessFormat(file);
38          // if (format != ImageFormat.IMAGE_FORMAT_JPEG)
39          // return false;
40          }
41  
42          try {
43              final ByteSource byteSource = new ByteSourceFile(file);
44              return new JpegImageParser().hasIptcSegment(byteSource);
45          } catch (final Exception e) {
46              // Debug.debug("Error file", file.getAbsoluteFile());
47              // Debug.debug(e, 4);
48              return false;
49          }
50      }
51  
52      private static final ImageFilter HAS_IPTC_IMAGE_FILTER = IptcBaseTest::hasIptcData;
53  
54      private static final ImageFilter JPEG_IMAGE_FILTER = file -> file.getName().toLowerCase().endsWith(".jpg");
55  
56      protected File getImageWithIptcData() throws IOException,
57              ImageReadException {
58          return getTestImage(HAS_IPTC_IMAGE_FILTER);
59      }
60  
61      protected static List<File> getImagesWithIptcData() throws IOException,
62              ImageReadException {
63          return getTestImages(HAS_IPTC_IMAGE_FILTER);
64      }
65  
66      protected List<File> getImagesWithIptcData(final int max) throws IOException,
67              ImageReadException {
68          return getTestImages(HAS_IPTC_IMAGE_FILTER, max);
69      }
70  
71      protected File getJpegImage() throws IOException, ImageReadException {
72          return getTestImage(JPEG_IMAGE_FILTER);
73      }
74  
75      protected static List<File> getJpegImages() throws IOException, ImageReadException {
76          return getTestImages(JPEG_IMAGE_FILTER);
77      }
78  
79      protected List<File> getJpegImages(final int max) throws IOException,
80              ImageReadException {
81          return getTestImages(JPEG_IMAGE_FILTER, max);
82      }
83  
84  }