C2VDAAdaptor.cpp

// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

//#define LOG_NDEBUG 0
#define LOG_TAG "C2VDAAdaptor"

#include "C2VDAAdaptor.h"
#include "bitstream_buffer.h"
#include "native_pixmap_handle.h"
#include "v4l2_slice_video_decode_accelerator.h"

#include <system/graphics.h>
#include <utils/Log.h>

namespace android {

C2VDAAdaptor::C2VDAAdaptor() : mNumOutputBuffers(0u) {
}

C2VDAAdaptor::~C2VDAAdaptor() {
    if (mVDA) {
        destroy();
    }
}

VideoDecodeAcceleratorAdaptor::Result C2VDAAdaptor::initialize(
        media::VideoCodecProfile profile, VideoDecodeAcceleratorAdaptor::Client* client) {
    if (mVDA) {
        ALOGE("Re-initialize() is not allowed");
        return ILLEGAL_STATE;
    }

    media::VideoDecodeAccelerator::Config config;
    config.profile = profile;
    config.output_mode = media::VideoDecodeAccelerator::Config::OutputMode::IMPORT;

    // TODO(johnylin): may need to implement factory to create VDA if there are multiple VDA
    // implementations in the future.
    scoped_refptr<media::V4L2Device> device = new media::V4L2Device();
    std::unique_ptr<media::VideoDecodeAccelerator> vda(
            new media::V4L2SliceVideoDecodeAccelerator(device));
    if (!vda->Initialize(config, this)) {
        ALOGE("Failed to initialize VDA");
        return PLATFORM_FAILURE;
    }

    mVDA = std::move(vda);
    mClient = client;

    return SUCCESS;
}

void C2VDAAdaptor::decode(int32_t bitstreamId, int ashmemFd, off_t offset, uint32_t bytesUsed) {
    CHECK(mVDA);
    mVDA->Decode(media::BitstreamBuffer(
            bitstreamId, base::SharedMemoryHandle(ashmemFd, true), bytesUsed, offset));
}

void C2VDAAdaptor::assignPictureBuffers(uint32_t numOutputBuffers) {
    CHECK(mVDA);
    std::vector<media::PictureBuffer> buffers;
    for (uint32_t id = 0; id < numOutputBuffers; ++id) {
        buffers.push_back(
            media::PictureBuffer(static_cast<int32_t>(id), mPictureSize));
    }
    mVDA->AssignPictureBuffers(buffers);
    mNumOutputBuffers = numOutputBuffers;
}

void C2VDAAdaptor::importBufferForPicture(int32_t pictureBufferId, int dmabufFd,
                                          const std::vector<VideoFramePlane>& planes) {
    CHECK(mVDA);
    CHECK_LT(pictureBufferId, static_cast<int32_t>(mNumOutputBuffers));

    media::NativePixmapHandle handle;
    handle.fds.emplace_back(base::FileDescriptor(dmabufFd, true));
    for (const auto& plane : planes) {
        handle.planes.emplace_back(plane.stride, plane.offset, 0, 0);
    }
    mVDA->ImportBufferForPicture(pictureBufferId, handle);
}

void C2VDAAdaptor::reusePictureBuffer(int32_t pictureBufferId) {
    CHECK(mVDA);
    CHECK_LT(pictureBufferId, static_cast<int32_t>(mNumOutputBuffers));

    mVDA->ReusePictureBuffer(pictureBufferId);
}

void C2VDAAdaptor::flush() {
    CHECK(mVDA);
    mVDA->Flush();
}

void C2VDAAdaptor::reset() {
    CHECK(mVDA);
    mVDA->Reset();
}

void C2VDAAdaptor::destroy() {
    mVDA.reset(nullptr);
    mNumOutputBuffers = 0u;
    mPictureSize = media::Size();
}


void C2VDAAdaptor::ProvidePictureBuffers(uint32_t requested_num_of_buffers,
                                         media::VideoPixelFormat output_format,
                                         const media::Size& dimensions) {
    uint32_t pixelFormat;
    switch (output_format) {
        case media::PIXEL_FORMAT_I420:
        case media::PIXEL_FORMAT_YV12:
        case media::PIXEL_FORMAT_NV12:
        case media::PIXEL_FORMAT_NV21:
            // HAL_PIXEL_FORMAT_YCbCr_420_888 is the flexible pixel format in Android
            // which handles all 420 formats, with both orderings of chroma (CbCr and
            // CrCb) as well as planar and semi-planar layouts.
            pixelFormat = HAL_PIXEL_FORMAT_YCbCr_420_888;
            break;
        case media::PIXEL_FORMAT_ARGB:
            pixelFormat = HAL_PIXEL_FORMAT_BGRA_8888;
            break;
        default:
            ALOGE("Format not supported: %d", output_format);
            mClient->notifyError(PLATFORM_FAILURE);
            return;
    }

    mClient->providePictureBuffers(pixelFormat, requested_num_of_buffers, dimensions);
    mPictureSize = dimensions;
}

void C2VDAAdaptor::DismissPictureBuffer(int32_t picture_buffer_id) {
    mClient->dismissPictureBuffer(picture_buffer_id);
}

void C2VDAAdaptor::PictureReady(const media::Picture& picture) {
    mClient->pictureReady(picture.picture_buffer_id(),
                          picture.bitstream_buffer_id(),
                          picture.visible_rect());
}

void C2VDAAdaptor::NotifyEndOfBitstreamBuffer(int32_t bitstream_buffer_id) {
    mClient->notifyEndOfBitstreamBuffer(bitstream_buffer_id);
}

void C2VDAAdaptor::NotifyFlushDone() {
    mClient->notifyFlushDone();
}

void C2VDAAdaptor::NotifyResetDone() {
    mClient->notifyResetDone();
}

static VideoDecodeAcceleratorAdaptor::Result convertErrorCode(
        media::VideoDecodeAccelerator::Error error) {
    switch (error) {
        case media::VideoDecodeAccelerator::ILLEGAL_STATE:
            return VideoDecodeAcceleratorAdaptor::ILLEGAL_STATE;
        case media::VideoDecodeAccelerator::INVALID_ARGUMENT:
            return VideoDecodeAcceleratorAdaptor::INVALID_ARGUMENT;
        case media::VideoDecodeAccelerator::UNREADABLE_INPUT:
            return VideoDecodeAcceleratorAdaptor::UNREADABLE_INPUT;
        case media::VideoDecodeAccelerator::PLATFORM_FAILURE:
            return VideoDecodeAcceleratorAdaptor::PLATFORM_FAILURE;
        default:
            ALOGE("Unknown error code: %d", error);
            return VideoDecodeAcceleratorAdaptor::PLATFORM_FAILURE;
    }
}

void C2VDAAdaptor::NotifyError(media::VideoDecodeAccelerator::Error error) {
    mClient->notifyError(convertErrorCode(error));
}

}  // namespace android