1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/* 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;
}
|
