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
|
/*
* OSD Picture in Picture plugin for the Video Disk Recorder
*
* See the README file for copyright information and how to reach the author.
*/
#include <stdio.h>
#include <stdlib.h>
#include "decoder.h"
#include "config.h"
int cDecoder::Open()
{
m_Codec = avcodec_find_decoder(CODEC_ID_MPEG2VIDEO);
if (!m_Codec)
{
printf("codec not found\n");
return -1;
}
m_Context = avcodec_alloc_context();
if (avcodec_open(m_Context, m_Codec) < 0)
{
printf("could not open codec\n");
return -1;
}
m_PicDecoded = avcodec_alloc_frame();
m_PicResample = avcodec_alloc_frame();
m_BufferResample = new unsigned char[(400 * 300 * 3) / 2]; // size for YUV 420
m_PicConvert = avcodec_alloc_frame();
m_BufferConvert = new unsigned char[400 * 300 * 4]; // size for RGBA32
return 0;
}
int cDecoder::Close()
{
delete[] m_BufferConvert;
free(m_PicConvert);
delete[] m_BufferResample;
free(m_PicResample);
avcodec_close(m_Context);
free(m_Context);
free(m_PicDecoded);
return 0;
}
int cDecoder::Decode(unsigned char * data, int length)
{
int gotPicture, len;
len = avcodec_decode_video(m_Context, m_PicDecoded, &gotPicture, data, length);
if (len < 0)
{
printf("Error while decoding frame\n");
return -1;
}
if (!gotPicture)
{
return -1;
}
return 0;
}
int cDecoder::Resample(int width, int height)
{
ImgReSampleContext * contextResample;
m_Width = width;
m_Height = height;
contextResample = img_resample_full_init(m_Width, m_Height,
m_Context->width, m_Context->height,
OsdPipSetup.CropTop, OsdPipSetup.CropBottom,
OsdPipSetup.CropLeft, OsdPipSetup.CropRight);
if (!contextResample) {
printf("Error initializing resample context.\n");
return -1;
}
avpicture_fill((AVPicture *) m_PicResample, m_BufferResample,
PIX_FMT_YUV420P, m_Width, m_Height);
img_resample(contextResample, (AVPicture *) m_PicResample, (AVPicture *) m_PicDecoded);
img_resample_close(contextResample);
return 0;
}
int cDecoder::ConvertToRGB()
{
avpicture_fill((AVPicture *) m_PicConvert, m_BufferConvert,
PIX_FMT_RGBA32, m_Width, m_Height);
img_convert((AVPicture *) m_PicConvert, PIX_FMT_RGBA32,
(AVPicture *) m_PicResample, PIX_FMT_YUV420P,
m_Width, m_Height);
return 0;
}
|
