1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef ITEM_TABLE_H_ 18 #define ITEM_TABLE_H_ 19 20 #include <set> 21 22 #include <media/NdkMediaFormat.h> 23 24 #include <media/stagefright/foundation/ADebug.h> 25 #include <utils/KeyedVector.h> 26 #include <utils/RefBase.h> 27 28 namespace android { 29 30 class DataSourceHelper; 31 class MetaData; 32 33 namespace heif { 34 35 struct AssociationEntry; 36 struct ImageItem; 37 struct ExternalMetaItem; 38 struct ItemLoc; 39 struct ItemInfo; 40 struct ItemProperty; 41 struct ItemReference; 42 43 /* 44 * ItemTable keeps track of all image items (including coded images, grids and 45 * tiles) inside a HEIF/AVIF still image (ISO/IEC FDIS 23008-12.2:2017(E)). 46 */ 47 48 class ItemTable : public RefBase { 49 public: 50 ItemTable(DataSourceHelper *source, bool isHeif); 51 52 status_t parse(uint32_t type, off64_t offset, size_t size); 53 isValid()54 bool isValid() { return mImageItemsValid; } 55 uint32_t countImages() const; 56 AMediaFormat *getImageMeta(const uint32_t imageIndex); 57 status_t findImageItem(const uint32_t imageIndex, uint32_t *itemIndex); 58 status_t findThumbnailItem(const uint32_t imageIndex, uint32_t *itemIndex); 59 status_t getImageOffsetAndSize( 60 uint32_t *itemIndex, off64_t *offset, size_t *size); 61 status_t getExifOffsetAndSize(off64_t *offset, size_t *size); 62 status_t getXmpOffsetAndSize(off64_t *offset, size_t *size); 63 64 protected: 65 ~ItemTable(); 66 67 private: 68 DataSourceHelper *mDataSource; 69 // If this is true, then this item table is for a HEIF image. Otherwise it is for an AVIF image. 70 bool mIsHeif; 71 72 KeyedVector<uint32_t, ItemLoc> mItemLocs; 73 Vector<ItemInfo> mItemInfos; 74 Vector<AssociationEntry> mAssociations; 75 Vector<sp<ItemProperty> > mItemProperties; 76 Vector<sp<ItemReference> > mItemReferences; 77 78 uint32_t mPrimaryItemId; 79 off64_t mIdatOffset; 80 size_t mIdatSize; 81 82 std::set<uint32_t> mRequiredBoxes; 83 std::set<uint32_t> mBoxesSeen; 84 85 bool mImageItemsValid; 86 uint32_t mCurrentItemIndex; 87 KeyedVector<uint32_t, ImageItem> mItemIdToItemMap; 88 KeyedVector<uint32_t, ExternalMetaItem> mItemIdToMetaMap; 89 Vector<uint32_t> mDisplayables; 90 91 status_t parseIlocBox(off64_t offset, size_t size); 92 status_t parseIinfBox(off64_t offset, size_t size); 93 status_t parsePitmBox(off64_t offset, size_t size); 94 status_t parseIprpBox(off64_t offset, size_t size); 95 status_t parseIdatBox(off64_t offset, size_t size); 96 status_t parseIrefBox(off64_t offset, size_t size); 97 98 void attachProperty(const AssociationEntry &association); 99 status_t buildImageItemsIfPossible(uint32_t type); 100 101 DISALLOW_EVIL_CONSTRUCTORS(ItemTable); 102 }; 103 104 } // namespace heif 105 } // namespace android 106 107 #endif // ITEM_TABLE_H_ 108