Home | History | Annotate | Download | only in libheif
      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 getEncodedColor(HeifEncodedColor* outColor) const override;
     44 
     45     bool setOutputColor(HeifColorFormat heifColor) override;
     46 
     47     bool decode(HeifFrameInfo* frameInfo) override;
     48 
     49     bool getScanline(uint8_t* dst) override;
     50 
     51     size_t skipScanlines(size_t count) override;
     52 
     53 private:
     54     struct DecodeThread;
     55 
     56     sp<IDataSource> mDataSource;
     57     sp<MediaMetadataRetriever> mRetriever;
     58     sp<IMemory> mFrameMemory;
     59     android_pixel_format_t mOutputColor;
     60     size_t mCurScanline;
     61     uint32_t mWidth;
     62     uint32_t mHeight;
     63     bool mFrameDecoded;
     64     bool mHasImage;
     65     bool mHasVideo;
     66 
     67     // Slice decoding only
     68     Mutex mLock;
     69     Condition mScanlineReady;
     70     sp<DecodeThread> mThread;
     71     size_t mAvailableLines;
     72     size_t mNumSlices;
     73     uint32_t mSliceHeight;
     74     bool mAsyncDecodeDone;
     75 
     76     bool decodeAsync();
     77     bool getScanlineInner(uint8_t* dst);
     78 };
     79 
     80 } // namespace android
     81 
     82 #endif // _HEIF_DECODER_IMPL_
     83