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
|
#include "nsv_h264_decoder.h"
#include "../nsv/nsvlib.h"
#include "../nsv/dec_if.h"
#include <assert.h>
#include <Mferror.h>
H264_Decoder::H264_Decoder()
{
vidbufdec=0;
last_pic = 0;
decoder.Open();
}
H264_Decoder::~H264_Decoder()
{
for (size_t i=0;i<buffered_frames.size();i++) {
nullsoft_h264_frame_data frame_data = buffered_frames[i];
decoder.FreeFrame((YV12_PLANES *)frame_data.data, frame_data.decoder_data);
}
decoder.FreeFrame(vidbufdec, last_pic);
}
int H264_Decoder::decode(int need_kf,
void *_in, int _in_len,
void **out, // out is set to a pointer to data
unsigned int *out_type, // 'Y','V','1','2' is currently defined
int *is_kf)
{
*out_type=NSV_MAKETYPE('Y','V','1','2');
if (last_pic)
{
decoder.FreeFrame(vidbufdec, last_pic);
vidbufdec=0;
last_pic=0;
}
if (_in_len) {
for (;;) {
HRESULT hr = decoder.FeedRaw(_in, _in_len, 0);
if (hr == MF_E_NOTACCEPTING) {
nullsoft_h264_frame_data frame_data;
if (FAILED(decoder.GetFrame((YV12_PLANES **)&frame_data.data, &frame_data.decoder_data, &frame_data.local_timestamp))) {
continue;
}
buffered_frames.push_back(frame_data);
} else if (FAILED(hr)) {
return -1;
} else {
break;
}
}
} else {
decoder.Drain();
}
if (SUCCEEDED(decoder.GetFrame(&vidbufdec, &last_pic, 0))) {
*out = vidbufdec;
*is_kf = 1;
} else {
*out = 0;
}
return 0;
}
void H264_Decoder::flush()
{
for ( size_t i = 0; i < buffered_frames.size(); i++ )
{
nullsoft_h264_frame_data frame_data = buffered_frames[ i ];
decoder.FreeFrame( (YV12_PLANES *)frame_data.data, frame_data.decoder_data );
}
decoder.Flush();
}
|
