    001/*
    002 * Licensed to the Apache Software Foundation (ASF) under one or more
    003 * contributor license agreements.  See the NOTICE file distributed with
    004 * this work for additional information regarding copyright ownership.
    005 * The ASF licenses this file to You under the Apache License, Version 2.0
    006 * (the "License"); you may not use this file except in compliance with
    007 * the License.  You may obtain a copy of the License at
    008 *
    009 *      http://www.apache.org/licenses/LICENSE-2.0
    010 *
    011 * Unless required by applicable law or agreed to in writing, software
    012 * distributed under the License is distributed on an "AS IS" BASIS,
    013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    014 * See the License for the specific language governing permissions and
    015 * limitations under the License.
    016 */
    017package org.apache.commons.imaging.formats.jpeg.segments;
    018
    019import static org.apache.commons.imaging.common.BinaryFunctions.readAndVerifyBytes;
    020import static org.apache.commons.imaging.common.BinaryFunctions.readByte;
    021import static org.apache.commons.imaging.common.BinaryFunctions.readBytes;
    022import static org.apache.commons.imaging.common.BinaryFunctions.startsWith;
    023
    024import java.io.ByteArrayInputStream;
    025import java.io.IOException;
    026import java.io.InputStream;
    027
    028import org.apache.commons.imaging.ImageReadException;
    029import org.apache.commons.imaging.formats.jpeg.JpegConstants;
    030
    031public class App2Segment extends AppnSegment implements Comparable<App2Segment> {
    032    private final byte[] iccBytes;
    033    public final int curMarker;
    034    public final int numMarkers;
    035
    036    public App2Segment(final int marker, final byte[] segmentData)
    037            throws ImageReadException, IOException {
    038        this(marker, segmentData.length, new ByteArrayInputStream(segmentData));
    039    }
    040
    041    public App2Segment(final int marker, int markerLength, final InputStream is2)
    042            throws ImageReadException, IOException {
    043        super(marker, markerLength, is2);
    044
    045        if (startsWith(getSegmentData(),
    046                JpegConstants.ICC_PROFILE_LABEL)) {
    047            final InputStream is = new ByteArrayInputStream(getSegmentData());
    048
    049            readAndVerifyBytes(is, JpegConstants.ICC_PROFILE_LABEL,
    050                    "Not a Valid App2 Segment: missing ICC Profile label");
    051
    052            curMarker = readByte("curMarker", is, "Not a valid App2 Marker");
    053            numMarkers = readByte("numMarkers", is, "Not a valid App2 Marker");
    054
    055            markerLength -= JpegConstants.ICC_PROFILE_LABEL.size();
    056            markerLength -= (1 + 1);
    057
    058            iccBytes = readBytes("App2 Data", is, markerLength, "Invalid App2 Segment: insufficient data");
    059        } else {
    060            // debugByteArray("Unknown APP2 Segment Type", bytes);
    061            curMarker = -1;
    062            numMarkers = -1;
    063            iccBytes = null;
    064        }
    065    }
    066
    067    @Override
    068    public boolean equals(final Object obj) {
    069        if (obj instanceof App2Segment) {
    070            final App2Segment other = (App2Segment) obj;
    071            return curMarker == other.curMarker;
    072        }
    073        return false;
    074    }
    075
    076    @Override
    077    public int hashCode() {
    078        return curMarker;
    079    }
    080
    081    @Override
    082    public int compareTo(final App2Segment other) {
    083        return curMarker - other.curMarker;
    084    }
    085
    086    /**
    087     * @return the iccBytes
    088     */
    089    public byte[] getIccBytes() {
    090        return iccBytes.clone();
    091    }
    092}