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 private:
58     struct DecodeThread;
59 
60     sp<IDataSource> mDataSource;
61     sp<MediaMetadataRetriever> mRetriever;
62     sp<IMemory> mFrameMemory;
63     HeifFrameInfo mImageInfo;
64     HeifFrameInfo mSequenceInfo;
65     android_pixel_format_t mOutputColor;
66     size_t mCurScanline;
67     size_t mTotalScanline;
68     bool mFrameDecoded;
69     bool mHasImage;
70     bool mHasVideo;
71     size_t mSequenceLength;
72 
73     // Slice decoding only
74     Mutex mLock;
75     Condition mScanlineReady;
76     sp<DecodeThread> mThread;
77     size_t mAvailableLines;
78     size_t mNumSlices;
79     uint32_t mSliceHeight;
80     bool mAsyncDecodeDone;
81 
82     bool decodeAsync();
83     bool getScanlineInner(uint8_t* dst);
84     bool reinit(HeifFrameInfo* frameInfo);
85 };
86 
87 } // namespace android
88 
89 #endif // _HEIF_DECODER_IMPL_
90