Merge "Support the dav1d decoder in a new codec2 component"

package {

    // See: http://go/android-license-faq

    // A large-scale-change added 'default_applicable_licenses' to import

    // all of the 'license_kinds' from "frameworks_av_license"

    // to get the below license kinds:

    //   SPDX-license-identifier-Apache-2.0

    default_applicable_licenses: ["frameworks_av_license"],

}

cc_library {

    name: "libcodec2_soft_av1dec_dav1d",

    // TODO: b/277797541 - enable once ready

    enabled: false,

    defaults: [

        "libcodec2_soft-defaults",

        "libcodec2_soft_sanitize_all-defaults",

    ],

    cflags: [

        "-DCODECNAME=\"c2.android.dav1d-av1.decoder\"",

        "-Wno-unused-variable",

    ],

    srcs: ["C2SoftDav1dDec.cpp"],

    static_libs: [

        "libyuv_static",

        "libdav1d_8bit",

        "libdav1d_16bit",

    ],

    apex_available: [

        "//apex_available:platform",

        "com.android.media.swcodec",

    ],

}

/*

 * Copyright (C) 2023 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifndef ANDROID_C2_SOFT_DAV1D_DEC_H_

#define ANDROID_C2_SOFT_DAV1D_DEC_H_

#include <inttypes.h>

#include <memory>

#include <media/stagefright/foundation/ColorUtils.h>

#include <C2Config.h>

#include <SimpleC2Component.h>

#include <dav1d/dav1d.h>

#include <deque>

//#define FILE_DUMP_ENABLE 1

#define DUMP_FILE_PATH "/data/local/tmp/dump"

#define INPUT_DATA_DUMP_EXT "av1"

#define INPUT_SIZE_DUMP_EXT "size"

#define OUTPUT_YUV_DUMP_EXT "yuv"

namespace android {

struct C2SoftDav1dDec : public SimpleC2Component {

    class IntfImpl;

    C2SoftDav1dDec(const char* name, c2_node_id_t id, const std::shared_ptr<IntfImpl>& intfImpl);

    ~C2SoftDav1dDec();

    // Begin SimpleC2Component overrides.

    c2_status_t onInit() override;

    c2_status_t onStop() override;

    void onReset() override;

    void onRelease() override;

    c2_status_t onFlush_sm() override;

    void process(const std::unique_ptr<C2Work>& work,

                 const std::shared_ptr<C2BlockPool>& pool) override;

    c2_status_t drain(uint32_t drainMode, const std::shared_ptr<C2BlockPool>& pool) override;

    // End SimpleC2Component overrides.

  private:

    std::shared_ptr<IntfImpl> mIntf;

    int mInputBufferIndex = 0;

    int mOutputBufferIndex = 0;

    Dav1dContext* mDav1dCtx = nullptr;

    std::deque<Dav1dPicture> mDecodedPictures;

    // configurations used by component in process

    // (TODO: keep this in intf but make them internal only)

    std::shared_ptr<C2StreamPixelFormatInfo::output> mPixelFormatInfo;

    uint32_t mHalPixelFormat;

    uint32_t mWidth;

    uint32_t mHeight;

    bool mSignalledOutputEos;

    bool mSignalledError;

    // Used during 10-bit I444/I422 to 10-bit P010 & 8-bit I420 conversions.

    std::unique_ptr<uint16_t[]> mTmpFrameBuffer;

    size_t mTmpFrameBufferSize = 0;

    C2StreamHdrStaticMetadataInfo::output mHdrStaticMetadataInfo;

    std::unique_ptr<C2StreamHdr10PlusInfo::output> mHdr10PlusInfo = nullptr;

    // Color aspects. These are ISO values and are meant to detect changes in aspects to avoid

    // converting them to C2 values for each frame

    struct VuiColorAspects {

        uint8_t primaries;

        uint8_t transfer;

        uint8_t coeffs;

        uint8_t fullRange;

        // default color aspects

        VuiColorAspects()

            : primaries(C2Color::PRIMARIES_UNSPECIFIED),

              transfer(C2Color::TRANSFER_UNSPECIFIED),

              coeffs(C2Color::MATRIX_UNSPECIFIED),

              fullRange(C2Color::RANGE_UNSPECIFIED) {}

        bool operator==(const VuiColorAspects& o) {

            return primaries == o.primaries && transfer == o.transfer && coeffs == o.coeffs &&

                   fullRange == o.fullRange;

        }

    } mBitstreamColorAspects;

    nsecs_t mTimeStart = 0;  // Time at the start of decode()

    nsecs_t mTimeEnd = 0;    // Time at the end of decode()

    bool initDecoder();

    void getHDRStaticParams(Dav1dPicture* picture, const std::unique_ptr<C2Work>& work);

    void getHDR10PlusInfoData(Dav1dPicture* picture, const std::unique_ptr<C2Work>& work);

    void getVuiParams(Dav1dPicture* picture);

    void destroyDecoder();

    void finishWork(uint64_t index, const std::unique_ptr<C2Work>& work,

                    const std::shared_ptr<C2GraphicBlock>& block);

    // Sets |work->result| and mSignalledError. Returns false.

    void setError(const std::unique_ptr<C2Work>& work, c2_status_t error);

    bool allocTmpFrameBuffer(size_t size);

    bool outputBuffer(const std::shared_ptr<C2BlockPool>& pool,

                      const std::unique_ptr<C2Work>& work);

    c2_status_t drainInternal(uint32_t drainMode, const std::shared_ptr<C2BlockPool>& pool,

                              const std::unique_ptr<C2Work>& work);

    void flushDav1d();

#ifdef FILE_DUMP_ENABLE

    char mInDataFileName[256];

    char mInSizeFileName[256];

    char mDav1dOutYuvFileName[256];

    FILE* mInDataFile = nullptr;

    FILE* mInSizeFile = nullptr;

    FILE* mDav1dOutYuvFile = nullptr;

    void writeDav1dOutYuvFile(const Dav1dPicture& p);

    int num_frames_to_dump = 0;

#endif

    C2_DO_NOT_COPY(C2SoftDav1dDec);

};

}  // namespace android

#endif  // ANDROID_C2_SOFT_DAV1D_DEC_H_