132 lines
3.6 KiB
C++
Executable File
132 lines
3.6 KiB
C++
Executable File
/* Copyright (C) 2013 Mamadou DIOP
|
|
* Copyright (C) 2013 Doubango Telecom <http://www.doubango.org>
|
|
*
|
|
* This file is part of Open Source Doubango Framework.
|
|
*
|
|
* DOUBANGO is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* DOUBANGO is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with DOUBANGO.
|
|
*/
|
|
#include "plugin_cuda_config.h"
|
|
#include "plugin_cuda_utils.h"
|
|
|
|
#include "tinymedia/tmedia_codec.h"
|
|
|
|
#include "tsk_plugin.h"
|
|
#include "tsk_debug.h"
|
|
|
|
#include <windows.h>
|
|
|
|
#if defined(_MSC_VER)
|
|
# pragma comment(lib, "nvcuvenc")
|
|
# pragma comment(lib, "nvcuvid")
|
|
# pragma comment(lib, "cuda")
|
|
# pragma comment(lib, "cudart")
|
|
|
|
# pragma comment(lib, "d3d9")
|
|
# pragma comment(lib, "d3dx9")
|
|
#endif
|
|
|
|
|
|
#if !defined(PLUGIN_CUDA_H264_ENABLE)
|
|
# define PLUGIN_CUDA_H264_ENABLE 1
|
|
#endif
|
|
|
|
extern const tmedia_codec_plugin_def_t *cuda_codec_h264_main_plugin_def_t;
|
|
extern const tmedia_codec_plugin_def_t *cuda_codec_h264_base_plugin_def_t;
|
|
|
|
PLUGIN_CUDA_BEGIN_DECLS /* BEGIN */
|
|
PLUGIN_CUDA_API int __plugin_get_def_count();
|
|
PLUGIN_CUDA_API tsk_plugin_def_type_t __plugin_get_def_type_at(int index);
|
|
PLUGIN_CUDA_API tsk_plugin_def_media_type_t __plugin_get_def_media_type_at(int index);
|
|
PLUGIN_CUDA_API tsk_plugin_def_ptr_const_t __plugin_get_def_at(int index);
|
|
PLUGIN_CUDA_END_DECLS /* END */
|
|
|
|
BOOL APIENTRY DllMain( HMODULE hModule,
|
|
DWORD ul_reason_for_call,
|
|
LPVOID lpReserved
|
|
)
|
|
{
|
|
switch (ul_reason_for_call) {
|
|
case DLL_PROCESS_ATTACH:
|
|
break;
|
|
case DLL_THREAD_ATTACH:
|
|
break;
|
|
case DLL_THREAD_DETACH:
|
|
break;
|
|
case DLL_PROCESS_DETACH:
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
typedef enum PLUGIN_INDEX_E {
|
|
#if PLUGIN_CUDA_H264_ENABLE
|
|
PLUGIN_INDEX_CODEC_H264_MAIN,
|
|
PLUGIN_INDEX_CODEC_H264_BASE,
|
|
#endif
|
|
|
|
PLUGIN_INDEX_COUNT
|
|
}
|
|
PLUGIN_INDEX_T;
|
|
|
|
|
|
int __plugin_get_def_count()
|
|
{
|
|
return CudaUtils::IsH264Supported() ? PLUGIN_INDEX_COUNT : 0;
|
|
}
|
|
|
|
tsk_plugin_def_type_t __plugin_get_def_type_at(int index)
|
|
{
|
|
#if PLUGIN_CUDA_H264_ENABLE
|
|
switch(index) {
|
|
case PLUGIN_INDEX_CODEC_H264_MAIN:
|
|
case PLUGIN_INDEX_CODEC_H264_BASE: {
|
|
return CudaUtils::IsH264Supported() ? tsk_plugin_def_type_codec : tsk_plugin_def_type_none;
|
|
}
|
|
}
|
|
#endif
|
|
TSK_DEBUG_ERROR("No plugin at index %d", index);
|
|
return tsk_plugin_def_type_none;
|
|
}
|
|
|
|
tsk_plugin_def_media_type_t __plugin_get_def_media_type_at(int index)
|
|
{
|
|
#if PLUGIN_CUDA_H264_ENABLE
|
|
switch(index) {
|
|
case PLUGIN_INDEX_CODEC_H264_MAIN:
|
|
case PLUGIN_INDEX_CODEC_H264_BASE: {
|
|
return CudaUtils::IsH264Supported() ? tsk_plugin_def_media_type_video : tsk_plugin_def_media_type_none;
|
|
}
|
|
}
|
|
#endif
|
|
TSK_DEBUG_ERROR("No plugin at index %d", index);
|
|
return tsk_plugin_def_media_type_none;
|
|
}
|
|
|
|
tsk_plugin_def_ptr_const_t __plugin_get_def_at(int index)
|
|
{
|
|
#if PLUGIN_CUDA_H264_ENABLE
|
|
switch(index) {
|
|
case PLUGIN_INDEX_CODEC_H264_MAIN: {
|
|
return CudaUtils::IsH264Supported() ? cuda_codec_h264_main_plugin_def_t : tsk_null;
|
|
}
|
|
case PLUGIN_INDEX_CODEC_H264_BASE: {
|
|
return CudaUtils::IsH264Supported() ? cuda_codec_h264_base_plugin_def_t : tsk_null;
|
|
}
|
|
}
|
|
#endif
|
|
TSK_DEBUG_ERROR("No plugin at index %d", index);
|
|
return tsk_null;
|
|
}
|