You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

82 lines
2.5 KiB

// Copyright (C) 2020 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.
#pragma once
#include "host-common/GoldfishMediaDefs.h"
#include "host-common/H264NaluParser.h"
#include "host-common/MediaCodec.h"
#include "host-common/MediaH264DecoderFfmpeg.h"
#include "host-common/MediaH264DecoderVideoToolBox.h"
#include <VideoToolbox/VideoToolbox.h>
#include <cstdint>
#include <string>
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stddef.h>
namespace android {
namespace emulation {
class MediaH264DecoderVideoToolBoxProxy : public MediaH264DecoderPlugin {
public:
virtual MediaH264DecoderPlugin* clone() override;
virtual void destroyH264Context() override;
virtual void decodeFrame(void* ptr) override;
virtual void flush(void* ptr) override;
virtual void initH264Context(void* ptr) override;
virtual void reset(void* ptr) override;
virtual void getImage(void* ptr) override;
virtual void save(base::Stream* stream) const override;
virtual bool load(base::Stream* stream) override;
virtual int type() const override { return PLUGIN_TYPE_VIDEO_TOOL_BOX_PROXY; }
explicit MediaH264DecoderVideoToolBoxProxy(uint64_t id,
H264PingInfoParser parser);
virtual ~MediaH264DecoderVideoToolBoxProxy();
private:
using DecoderState = MediaH264DecoderVideoToolBox::DecoderState;
uint64_t mId = 0;
H264PingInfoParser mParser;
unsigned int mWidth;
unsigned int mHeight;
unsigned int mOutputWidth;
unsigned int mOutputHeight;
PixelFormat mOutputPixelFormat;
std::vector<uint8_t> mSPS;
std::vector<uint8_t> mPPS;
bool mIsVideoToolBoxDecoderInGoodState = true;
MediaH264DecoderVideoToolBox mVideoToolBoxDecoder;
MediaH264DecoderFfmpeg mFfmpegDecoder;
MediaH264DecoderPlugin* mCurrentDecoder = nullptr;
std::vector<uint8_t> prefixNaluHeader(std::vector<uint8_t> data);
}; // MediaH264DecoderVideoToolBoxProxy
} // namespace emulation
} // namespace android