jeffqjiangNew
ef06f12dfa
* * rocDecode: Removed the hard coded HW decoder capability info. - We now probe HW decoder capabilities through VA-API from the driver. * * rocDecode: Added number of decoder prob and a few missing tear down calls. * * rocDecode/HW cap change: Added a new singleton class, GpuVaContext, to handle HIP and VA initialization and VA attributes probe for both HW capability check and decoder initialization. * * rocDecode/HW caps: Removed GpuVaContext class out of vaapi_videodecoder.h and into a new file. Removed debug logs. Removed roc_decoder_caps.h. * * rocDecode/HW caps: Fixed a crash issue with multi-thread cases. Called to vaInitialize() and vaTerminte() should be paired. * * rocDecode/HW caps: Added multi-GPU support. * * rocDecode/HW cap: Moved GpuVaContext class implementation back to VAAPI layer. * * rocDecode/HW cap: Added changed based on review comments. * * rocDecode/HW cap: Added changes based on review comments. * * rocDecode/HW cap: GPU VA context class name change based on review comment. --------- Co-authored-by: Kiriti Gowda <kiritigowda@gmail.com>
63 líneas
2.5 KiB
C++
63 líneas
2.5 KiB
C++
/*
|
|
Copyright (c) 2023 - 2025 Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <assert.h>
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <string.h>
|
|
#include <map>
|
|
#include "../api/rocdecode.h"
|
|
#include <hip/hip_runtime.h>
|
|
#include "vaapi/vaapi_videodecoder.h"
|
|
|
|
struct HipInteropDeviceMem {
|
|
hipExternalMemory_t hip_ext_mem; // Interface to the vaapi-hip interop
|
|
uint8_t* hip_mapped_device_mem; // Mapped device memory for the YUV plane
|
|
uint32_t width; // Width of the surface in pixels.
|
|
uint32_t height; // Height of the surface in pixels.
|
|
uint32_t offset[3]; // Offset of each plane
|
|
uint32_t pitch[3]; // Pitch of each plane
|
|
uint32_t num_layers; // Number of layers making up the surface
|
|
};
|
|
|
|
class RocDecoder {
|
|
public:
|
|
RocDecoder(RocDecoderCreateInfo &decoder_create_info);
|
|
~RocDecoder();
|
|
rocDecStatus InitializeDecoder();
|
|
rocDecStatus DecodeFrame(RocdecPicParams *pic_params);
|
|
rocDecStatus GetDecodeStatus(int pic_idx, RocdecDecodeStatus* decode_status);
|
|
rocDecStatus ReconfigureDecoder(RocdecReconfigureDecoderInfo *reconfig_params);
|
|
rocDecStatus GetVideoFrame(int pic_idx, void *dev_mem_ptr[3], uint32_t horizontal_pitch[3], RocdecProcParams *vid_postproc_params);
|
|
|
|
private:
|
|
rocDecStatus FreeVideoFrame(int pic_idx);
|
|
int num_devices_;
|
|
RocDecoderCreateInfo decoder_create_info_;
|
|
VaapiVideoDecoder va_video_decoder_;
|
|
std::vector<HipInteropDeviceMem> hip_interop_;
|
|
}; |