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.
121 lines
4.4 KiB
121 lines
4.4 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.
|
|
|
|
#include "host-common/MediaCudaUtils.h"
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
#include "android/main-emugl.h"
|
|
|
|
#ifdef _WIN32
|
|
#define WIN32_LEAN_AND_MEAN 1
|
|
#include <windows.h>
|
|
#include <winioctl.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
extern "C" {
|
|
#define INIT_CUDA_GL 1
|
|
#include "host-common/dynlink_cuda.h"
|
|
#include "host-common/dynlink_cudaGL.h"
|
|
#include "host-common/dynlink_nvcuvid.h"
|
|
}
|
|
|
|
#define MEDIA_CUVID_DEBUG 0
|
|
|
|
#if MEDIA_CUVID_DEBUG
|
|
#define CUVID_DPRINT(fmt, ...) \
|
|
fprintf(stderr, "cuvid-utils: %s:%d " fmt "\n", __func__, __LINE__, \
|
|
##__VA_ARGS__);
|
|
#else
|
|
#define CUVID_DPRINT(fmt, ...)
|
|
#endif
|
|
|
|
#define NVDEC_API_CALL(cuvidAPI) \
|
|
do { \
|
|
CUresult errorCode = cuvidAPI; \
|
|
if (errorCode != CUDA_SUCCESS) { \
|
|
CUVID_DPRINT("%s failed with error code %d\n", #cuvidAPI, \
|
|
(int)errorCode); \
|
|
} \
|
|
} while (0)
|
|
|
|
extern "C" {
|
|
|
|
#define MEDIA_CUDA_COPY_Y_TEXTURE 1
|
|
#define MEDIA_CUDA_COPY_UV_TEXTURE 2
|
|
|
|
static void media_cuda_copy_decoded_frame(void* privData,
|
|
int mode,
|
|
uint32_t dest_texture_handle) {
|
|
media_cuda_utils_copy_context* copy_context =
|
|
static_cast<media_cuda_utils_copy_context*>(privData);
|
|
|
|
const unsigned int GL_TEXTURE_2D = 0x0DE1;
|
|
const unsigned int cudaGraphicsMapFlagsNone = 0x0;
|
|
CUgraphicsResource CudaRes{0};
|
|
CUVID_DPRINT("cuda copy decoded frame testure %d",
|
|
(int)dest_texture_handle);
|
|
NVDEC_API_CALL(cuGraphicsGLRegisterImage(&CudaRes, dest_texture_handle,
|
|
GL_TEXTURE_2D, 0x0));
|
|
CUarray texture_ptr;
|
|
NVDEC_API_CALL(cuGraphicsMapResources(1, &CudaRes, 0));
|
|
NVDEC_API_CALL(
|
|
cuGraphicsSubResourceGetMappedArray(&texture_ptr, CudaRes, 0, 0));
|
|
CUdeviceptr dpSrcFrame = copy_context->src_frame;
|
|
CUDA_MEMCPY2D m = {0};
|
|
m.srcMemoryType = CU_MEMORYTYPE_DEVICE;
|
|
m.srcDevice = dpSrcFrame;
|
|
m.srcPitch = copy_context->src_pitch;
|
|
m.dstMemoryType = CU_MEMORYTYPE_ARRAY;
|
|
m.dstArray = texture_ptr;
|
|
m.dstPitch = copy_context->dest_width * 1;
|
|
m.WidthInBytes = copy_context->dest_width * 1;
|
|
m.Height = copy_context->dest_height;
|
|
CUVID_DPRINT("dstPitch %d, WidthInBytes %d Height %d surface-height %d",
|
|
(int)m.dstPitch, (int)m.WidthInBytes, (int)m.Height,
|
|
(int)copy_context->src_surface_height);
|
|
|
|
if (mode == MEDIA_CUDA_COPY_Y_TEXTURE) { // copy Y data
|
|
NVDEC_API_CALL(cuMemcpy2D(&m));
|
|
} else if (mode == MEDIA_CUDA_COPY_UV_TEXTURE) { // copy UV data
|
|
m.srcDevice =
|
|
(CUdeviceptr)((uint8_t*)dpSrcFrame +
|
|
m.srcPitch * copy_context->src_surface_height);
|
|
m.Height = m.Height / 2;
|
|
NVDEC_API_CALL(cuMemcpy2D(&m));
|
|
}
|
|
NVDEC_API_CALL(cuGraphicsUnmapResources(1, &CudaRes, 0));
|
|
NVDEC_API_CALL(cuGraphicsUnregisterResource(CudaRes));
|
|
}
|
|
|
|
void media_cuda_utils_nv12_updater(void* privData,
|
|
uint32_t type,
|
|
uint32_t* textures) {
|
|
constexpr uint32_t kFRAMEWORK_FORMAT_NV12 = 3;
|
|
if (type != kFRAMEWORK_FORMAT_NV12) {
|
|
return;
|
|
}
|
|
CUVID_DPRINT("copyiong Ytex %d", textures[0]);
|
|
CUVID_DPRINT("copyiong UVtex %d", textures[1]);
|
|
media_cuda_copy_decoded_frame(privData, MEDIA_CUDA_COPY_Y_TEXTURE,
|
|
textures[0]);
|
|
media_cuda_copy_decoded_frame(privData, MEDIA_CUDA_COPY_UV_TEXTURE,
|
|
textures[1]);
|
|
}
|
|
|
|
} // end extern C
|