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 _HEIF_DECODER_IMPL_
18 #define _HEIF_DECODER_IMPL_
19 
20 #include "include/HeifDecoderAPI.h"
21 #include <system/graphics.h>
22 #include <utils/Condition.h>
23 #include <utils/Mutex.h>
24 #include <utils/RefBase.h>
25 
26 namespace android {
27 
28 class IDataSource;
29 class IMemory;
30 class MediaMetadataRetriever;
31 
32 /*
33  * An implementation of HeifDecoder based on Android's MediaMetadataRetriever.
34  */
35 class HeifDecoderImpl : public HeifDecoder {
36 public:
37 
38     HeifDecoderImpl();
39     ~HeifDecoderImpl() override;
40 
41     bool init(HeifStream* stream, HeifFrameInfo* frameInfo) override;
42 
43     bool getSequenceInfo(HeifFrameInfo* frameInfo, size_t *frameCount) override;
44 
45     bool getEncodedColor(HeifEncodedColor* outColor) const override;
46 
47     bool setOutputColor(HeifColorFormat heifColor) override;
48 
49     bool decode(HeifFrameInfo* frameInfo) override;
50 
51     bool decodeSequence(int frameIndex, HeifFrameInfo* frameInfo) override;
52 
53     bool getScanline(uint8_t* dst) override;
54 
55     size_t skipScanlines(size_t count) override;
56 
57     uint32_t getColorDepth() override;
58 
59 private:
60     struct DecodeThread;
61 
62     sp<IDataSource> mDataSource;
63     sp<MediaMetadataRetriever> mRetriever;
64     sp<IMemory> mFrameMemory;
65     HeifFrameInfo mImageInfo;
66     HeifFrameInfo mSequenceInfo;
67     android_pixel_format_t mOutputColor;
68     size_t mCurScanline;
69     size_t mTotalScanline;
70     bool mFrameDecoded;
71     bool mHasImage;
72     bool mHasVideo;
73     size_t mSequenceLength;
74 
75     Mutex mRetrieverLock;
76 
77     // Slice decoding only
78     Mutex mLock;
79     Condition mScanlineReady;
80     sp<DecodeThread> mThread;
81     size_t mAvailableLines;
82     size_t mNumSlices;
83     uint32_t mSliceHeight;
84     bool mAsyncDecodeDone;
85 
86     bool decodeAsync();
87     bool getScanlineInner(uint8_t* dst);
88     bool reinit(HeifFrameInfo* frameInfo);
89 };
90 
91 } // namespace android
92 
93 #endif // _HEIF_DECODER_IMPL_
94