1 files changed, 248 insertions, 0 deletions
diff --git a/src/com/android/camera/mpo/MpoImageData.java b/src/com/android/camera/mpo/MpoImageData.java
new file mode 100644
index 000000000..84d2d0f6e
--- /dev/null
+++ b/src/com/android/camera/mpo/MpoImageData.java
@@ -0,0 +1,248 @@
+/*
+ * Copyright (c) 2016, The Linux Foundation. All rights reserved.
+ * Not a contribution.
+ *
+ * Copyright (C) 2012 The Android Open Source Project
+ *
+ * Licensed 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 com.android.camera.mpo;
+
+import java.nio.ByteOrder;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * This class stores the MPO header in IFDs according to the MPO specification.
+ */
+
+public class MpoImageData {
+    private static final String TAG = "MpoImageData";
+    static final int OFFSET_TO_FIRST_IFD = 8;
+    static final int MP_FORMAT_IDENTIFIER = 0x4D504600; // 'M' 'P' 'F' 'NULL'
+    static final int MP_HEADER_SIZE = 8;
+    static final int APP_HEADER_SIZE = 6;
+
+    private final MpoIfdData mMpIndexIfdData = new MpoIfdData(MpoIfdData.TYPE_MP_INDEX_IFD);
+    private final MpoIfdData mMpAttribIfdData = new MpoIfdData(MpoIfdData.TYPE_MP_ATTRIB_IFD);
+    private final byte[] mJpegData;
+    private final ByteOrder mByteOrder;
+
+    public MpoImageData(byte[] jpegData, ByteOrder byteOrder) {
+        mJpegData = jpegData;
+        mByteOrder = byteOrder;
+    }
+
+    /**
+     * Gets the jpeg data.
+     */
+    protected byte[] getJpegData() {
+        return mJpegData;
+    }
+
+    /**
+     * Gets the byte order.
+     */
+    protected ByteOrder getByteOrder() {
+        return mByteOrder;
+    }
+
+    /**
+     * Returns the {@link mMpAttribIfdData} object if it exists or null.
+     */
+    protected MpoIfdData getAttribIfdData() {
+        return mMpAttribIfdData;
+    }
+
+    /**
+     * Returns the {@link mMpIndexIfdData} object if it exists or null.
+     */
+    protected MpoIfdData getIndexIfdData() {
+        return mMpIndexIfdData;
+    }
+
+    /**
+     * Returns the {@link MpoIfdData} object corresponding to a given IFD.
+     */
+    protected MpoIfdData getMpIfdData(int ifdId) {
+        return (ifdId == MpoIfdData.TYPE_MP_INDEX_IFD) ? mMpIndexIfdData : mMpAttribIfdData;
+    }
+
+    /**
+     * Returns the tag with a given tag ID in the given IFD if the tag exists.
+     * Otherwise returns null.
+     */
+    protected MpoTag getTag(short tag, int ifd) {
+        MpoIfdData mpIfdData = getMpIfdData(ifd);
+        return mpIfdData.getTag(tag);
+    }
+
+    /**
+     * Adds the given MpoTag to its default IFD and returns an existing MpoTag
+     * with the same TID or null if none exist.
+     */
+    protected MpoTag addTag(MpoTag tag) {
+        if (tag != null) {
+            int ifd = tag.getIfd();
+            return addTag(tag, ifd);
+        }
+        return null;
+    }
+
+    /**
+     * Adds the given MpoTag to the given IFD and returns an existing MpoTag
+     * with the same tag ID or null if none exist.
+     */
+    protected MpoTag addTag(MpoTag tag, int ifdId) {
+        if (tag != null && MpoTag.isValidIfd(ifdId)) {
+            return getMpIfdData(ifdId).setTag(tag);
+        }
+        return null;
+    }
+
+    /**
+     * Removes the tag with a given tag ID and IFD.
+     */
+    protected void removeTag(short tagId, int ifdId) {
+        getMpIfdData(ifdId).removeTag(tagId);
+    }
+
+    /**
+     * Returns a list of all {@link MpoTag}s in the ExifData or null if there
+     * are none.
+     */
+    protected List<MpoTag> getAllTags() {
+        ArrayList<MpoTag> ret = new ArrayList<MpoTag>();
+        MpoTag[] tags = mMpIndexIfdData.getAllTags();
+        if (tags != null) {
+            for (MpoTag t : tags) {
+                ret.add(t);
+            }
+        }
+
+        tags = mMpAttribIfdData.getAllTags();
+        if (tags != null) {
+            for (MpoTag t : tags) {
+                ret.add(t);
+            }
+        }
+
+        if (ret.size() == 0) {
+            return null;
+        }
+        return ret;
+    }
+
+    /**
+     * Returns a list of all {@link MpoTag}s in a given IFD or null if there are
+     * none.
+     */
+    protected List<MpoTag> getAllTagsForIfd(int ifd) {
+        MpoTag[] tags = getMpIfdData(ifd).getAllTags();
+        if (tags == null) {
+            return null;
+        }
+        ArrayList<MpoTag> ret = new ArrayList<MpoTag>(tags.length);
+        for (MpoTag t : tags) {
+            ret.add(t);
+        }
+        if (ret.size() == 0) {
+            return null;
+        }
+        return ret;
+    }
+
+    /**
+     * Returns a list of all {@link MpoTag}s with a given TID or null if there
+     * are none.
+     */
+    protected List<MpoTag> getAllTagsForTagId(short tag) {
+        ArrayList<MpoTag> ret = new ArrayList<MpoTag>();
+        MpoTag t = mMpIndexIfdData.getTag(tag);
+        if (t != null) {
+            ret.add(t);
+        }
+
+        t = mMpAttribIfdData.getTag(tag);
+        if (t != null) {
+            ret.add(t);
+        }
+
+        if (ret.size() == 0) {
+            return null;
+        }
+        return ret;
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        if (this == obj) {
+            return true;
+        }
+        if (obj == null) {
+            return false;
+        }
+        if (obj instanceof MpoImageData) {
+            MpoImageData data = (MpoImageData) obj;
+            if (data.mByteOrder != mByteOrder) {
+                return false;
+            }
+
+            MpoIfdData indexIfd1 = data.getMpIfdData(MpoIfdData.TYPE_MP_INDEX_IFD);
+            MpoIfdData indexIfd2 = getMpIfdData(MpoIfdData.TYPE_MP_INDEX_IFD);
+            if (indexIfd1 != indexIfd2 && indexIfd1 != null && !indexIfd1.equals(indexIfd2)) {
+                return false;
+            }
+
+            MpoIfdData attribIfd1 = data.getMpIfdData(MpoIfdData.TYPE_MP_ATTRIB_IFD);
+            MpoIfdData attribIfd2 = getMpIfdData(MpoIfdData.TYPE_MP_ATTRIB_IFD);
+            if (attribIfd1 != attribIfd2 && attribIfd1 != null && !attribIfd1.equals(attribIfd2)) {
+                return false;
+            }
+            return true;
+        }
+        return false;
+    }
+
+    private int calculateOffsetOfIfd(MpoIfdData ifd, int offset) {
+        offset += 2 + ifd.getTagCount() * MpoTag.TAG_SIZE + 4;
+        MpoTag[] tags = ifd.getAllTags();
+        for (MpoTag tag : tags) {
+            if (tag.getDataSize() > 4) {
+                tag.setOffset(offset);
+                offset += tag.getDataSize();
+            }
+        }
+        return offset;
+    }
+
+    public int calculateAllIfdOffsets() {
+        int offset = MP_HEADER_SIZE;
+        MpoIfdData indexIfd = getIndexIfdData();
+        if (indexIfd.getTagCount() > 0)
+            offset = calculateOffsetOfIfd(indexIfd, offset);
+
+        MpoIfdData attribIfd = getAttribIfdData();
+        if (attribIfd.getTagCount() > 0) {
+            indexIfd.setOffsetToNextIfd(offset);
+            offset = calculateOffsetOfIfd(attribIfd, offset);
+        }
+
+        return offset;
+    }
+
+    public int calculateImageSize() {
+        return 2 + APP_HEADER_SIZE + calculateAllIfdOffsets() + mJpegData.length;
+    }
+}