1 /* 2 * Copyright (C) 2019 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 ANDROID_C2_SOFT_GAV1_DEC_H_ 18 #define ANDROID_C2_SOFT_GAV1_DEC_H_ 19 20 #include <inttypes.h> 21 22 #include <media/stagefright/foundation/ColorUtils.h> 23 24 #include <SimpleC2Component.h> 25 #include <C2Config.h> 26 #include "libgav1/src/gav1/decoder.h" 27 #include "libgav1/src/gav1/decoder_settings.h" 28 29 namespace android { 30 31 struct C2SoftGav1Dec : public SimpleC2Component { 32 class IntfImpl; 33 34 C2SoftGav1Dec(const char* name, c2_node_id_t id, 35 const std::shared_ptr<IntfImpl>& intfImpl); 36 ~C2SoftGav1Dec(); 37 38 // Begin SimpleC2Component overrides. 39 c2_status_t onInit() override; 40 c2_status_t onStop() override; 41 void onReset() override; 42 void onRelease() override; 43 c2_status_t onFlush_sm() override; 44 void process(const std::unique_ptr<C2Work>& work, 45 const std::shared_ptr<C2BlockPool>& pool) override; 46 c2_status_t drain(uint32_t drainMode, 47 const std::shared_ptr<C2BlockPool>& pool) override; 48 // End SimpleC2Component overrides. 49 50 private: 51 std::shared_ptr<IntfImpl> mIntf; 52 std::unique_ptr<libgav1::Decoder> mCodecCtx; 53 54 // configurations used by component in process 55 // (TODO: keep this in intf but make them internal only) 56 std::shared_ptr<C2StreamPixelFormatInfo::output> mPixelFormatInfo; 57 58 uint32_t mHalPixelFormat; 59 uint32_t mWidth; 60 uint32_t mHeight; 61 bool mSignalledOutputEos; 62 bool mSignalledError; 63 64 // Color aspects. These are ISO values and are meant to detect changes in aspects to avoid 65 // converting them to C2 values for each frame 66 struct VuiColorAspects { 67 uint8_t primaries; 68 uint8_t transfer; 69 uint8_t coeffs; 70 uint8_t fullRange; 71 72 // default color aspects VuiColorAspectsC2SoftGav1Dec::VuiColorAspects73 VuiColorAspects() 74 : primaries(C2Color::PRIMARIES_UNSPECIFIED), 75 transfer(C2Color::TRANSFER_UNSPECIFIED), 76 coeffs(C2Color::MATRIX_UNSPECIFIED), 77 fullRange(C2Color::RANGE_UNSPECIFIED) { } 78 79 bool operator==(const VuiColorAspects &o) { 80 return primaries == o.primaries && transfer == o.transfer && coeffs == o.coeffs 81 && fullRange == o.fullRange; 82 } 83 } mBitstreamColorAspects; 84 85 nsecs_t mTimeStart = 0; // Time at the start of decode() 86 nsecs_t mTimeEnd = 0; // Time at the end of decode() 87 88 bool initDecoder(); 89 void getVuiParams(const libgav1::DecoderBuffer *buffer); 90 void destroyDecoder(); 91 void finishWork(uint64_t index, const std::unique_ptr<C2Work>& work, 92 const std::shared_ptr<C2GraphicBlock>& block); 93 bool outputBuffer(const std::shared_ptr<C2BlockPool>& pool, 94 const std::unique_ptr<C2Work>& work); 95 c2_status_t drainInternal(uint32_t drainMode, 96 const std::shared_ptr<C2BlockPool>& pool, 97 const std::unique_ptr<C2Work>& work); 98 99 C2_DO_NOT_COPY(C2SoftGav1Dec); 100 }; 101 102 } // namespace android 103 104 #endif // ANDROID_C2_SOFT_GAV1_DEC_H_ 105