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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
|
/* $Id: decmpeg2.c,v 1.12 2005/03/03 16:30:42 titer Exp $
This file is part of the HandBrake source code.
Homepage: <http://handbrake.m0k.org/>.
It may be used under the terms of the GNU General Public License. */
#include "hb.h"
#include "mpeg2dec/mpeg2.h"
/**********************************************************************
* hb_libmpeg2_t
**********************************************************************
* A convenient libmpeg wrapper, used both here and in scan.c
*********************************************************************/
struct hb_libmpeg2_s
{
mpeg2dec_t * libmpeg2;
const mpeg2_info_t * info;
int width;
int height;
int rate;
int aspect_ratio;
int got_iframe;
int look_for_break;
int64_t last_pts;
};
/**********************************************************************
* hb_libmpeg2_init
**********************************************************************
*
*********************************************************************/
hb_libmpeg2_t * hb_libmpeg2_init()
{
hb_libmpeg2_t * m = calloc( sizeof( hb_libmpeg2_t ), 1 );
m->libmpeg2 = mpeg2_init();
m->info = mpeg2_info( m->libmpeg2 );
m->last_pts = -1;
m->look_for_break = 0;
return m;
}
/**********************************************************************
* hb_libmpeg2_decode
**********************************************************************
*
*********************************************************************/
int hb_libmpeg2_decode( hb_libmpeg2_t * m, hb_buffer_t * buf_es,
hb_list_t * list_raw )
{
mpeg2_state_t state;
hb_buffer_t * buf;
uint8_t * data;
int chap_break = 0;
/* Feed libmpeg2 */
if( buf_es->start > -1 )
{
mpeg2_tag_picture( m->libmpeg2, buf_es->start >> 32,
buf_es->start & 0xFFFFFFFF );
}
mpeg2_buffer( m->libmpeg2, buf_es->data,
buf_es->data + buf_es->size );
for( ;; )
{
state = mpeg2_parse( m->libmpeg2 );
if( state == STATE_BUFFER )
{
/* Require some more data */
break;
}
else if( state == STATE_SEQUENCE )
{
if( !( m->width && m->height && m->rate ) )
{
m->width = m->info->sequence->width;
m->height = m->info->sequence->height;
m->rate = m->info->sequence->frame_period;
if( m->rate == 900900 )
{
/* 29.97 fps. 3:2 pulldown might, or might not be
used. I can't find a way to know, so we always
output 23.976 */
m->rate = 1126125;
}
}
if ( m->aspect_ratio <= 0 )
{
// We can parse out the aspect ratio from the Sequence Start Header data in buf_es->data
// Make sure we have the correct data in the buffer
if ((buf_es->data[0] == 0x00) && (buf_es->data[1] == 0x00) && (buf_es->data[2] == 0x01) && (buf_es->data[3] == 0xb3))
{
unsigned char ar_fr = buf_es->data[7]; // Top 4 bits == aspect ratio flag - bottom 4 bits == rate flags
switch ((ar_fr & 0xf0) >> 4)
{
case 2:
m->aspect_ratio = HB_ASPECT_BASE * 4 / 3; // 4:3
break;
case 3:
m->aspect_ratio = HB_ASPECT_BASE * 16 / 9; // 16:9
break;
default:
hb_log("hb_libmpeg2_decode - STATE_SEQUENCE unexpected aspect ratio/frame rate 0x%x\n", ar_fr);
break;
}
}
}
}
else if( state == STATE_GOP && m->look_for_break == 2)
{
printf("MPEG2: Group of pictures found, searching for I-Frame\n");
m->look_for_break = 1;
}
else if( ( state == STATE_SLICE || state == STATE_END ) &&
m->info->display_fbuf )
{
if( ( m->info->display_picture->flags &
PIC_MASK_CODING_TYPE ) == PIC_FLAG_CODING_TYPE_I )
{
m->got_iframe = 1;
// If we are looking for a break, insert the chapter break on an I-Frame
if( m->look_for_break == 1 )
{
printf("MPEG2: I-Frame Found\n");
m->look_for_break = 0;
chap_break = 1;
}
}
if( m->got_iframe )
{
buf = hb_buffer_init( m->width * m->height * 3 / 2 );
data = buf->data;
// Was a good break point found?
if( chap_break )
{
printf("MPEG2: Chapter Break Inserted\n");
chap_break = 0;
buf->new_chap = 1;
}
memcpy( data, m->info->display_fbuf->buf[0],
m->width * m->height );
data += m->width * m->height;
memcpy( data, m->info->display_fbuf->buf[1],
m->width * m->height / 4 );
data += m->width * m->height / 4;
memcpy( data, m->info->display_fbuf->buf[2],
m->width * m->height / 4 );
if( m->info->display_picture->flags & PIC_FLAG_TAGS )
{
buf->start =
( (uint64_t) m->info->display_picture->tag << 32 ) |
( (uint64_t) m->info->display_picture->tag2 );
/*
* Add back in again to track PTS of MPEG2 frames
* hb_log("MPEG2: Normal buf->start = %lld", buf->start);
*/
}
else if( m->last_pts > -1 )
{
/* For some reason nb_fields is sometimes 1 while it
should be 2 */
buf->start = m->last_pts +
MAX( 2, m->info->display_picture->nb_fields ) *
m->info->sequence->frame_period / 600;
}
else
{
buf->start = -1;
}
m->last_pts = buf->start;
/* Store picture flags for later use by filters */
buf->flags = m->info->display_picture->flags;
hb_list_add( list_raw, buf );
}
}
else if( state == STATE_INVALID )
{
mpeg2_reset( m->libmpeg2, 0 );
}
}
return 1;
}
/**********************************************************************
* hb_libmpeg2_info
**********************************************************************
*
*********************************************************************/
void hb_libmpeg2_info( hb_libmpeg2_t * m, int * width, int * height,
int * rate, int *aspect_ratio )
{
*width = m->width;
*height = m->height;
*rate = m->rate;
*aspect_ratio = m->aspect_ratio;
}
/**********************************************************************
* hb_libmpeg2_close
**********************************************************************
*
*********************************************************************/
void hb_libmpeg2_close( hb_libmpeg2_t ** _m )
{
hb_libmpeg2_t * m = *_m;
mpeg2_close( m->libmpeg2 );
free( m );
*_m = NULL;
}
/**********************************************************************
* The decmpeg2 work object
**********************************************************************
*
*********************************************************************/
struct hb_work_private_s
{
hb_libmpeg2_t * libmpeg2;
hb_list_t * list;
};
/**********************************************************************
* hb_work_decmpeg2_init
**********************************************************************
*
*********************************************************************/
int decmpeg2Init( hb_work_object_t * w, hb_job_t * job )
{
hb_work_private_t * pv;
pv = calloc( 1, sizeof( hb_work_private_t ) );
w->private_data = pv;
pv->libmpeg2 = hb_libmpeg2_init();
pv->list = hb_list_init();
return 0;
}
/**********************************************************************
* Work
**********************************************************************
*
*********************************************************************/
int decmpeg2Work( hb_work_object_t * w, hb_buffer_t ** buf_in,
hb_buffer_t ** buf_out )
{
hb_work_private_t * pv = w->private_data;
hb_buffer_t * buf, * last = NULL;
// The reader found a chapter break, consume it completely, and remove it from the
// stream. We need to shift it.
if( (*buf_in)->new_chap )
{
printf("MPEG2: Chapter Break Cell Found, searching for GOP\n");
pv->libmpeg2->look_for_break = 2;
(*buf_in)->new_chap = 0;
}
hb_libmpeg2_decode( pv->libmpeg2, *buf_in, pv->list );
*buf_out = NULL;
while( ( buf = hb_list_item( pv->list, 0 ) ) )
{
hb_list_rem( pv->list, buf );
if( last )
{
last->next = buf;
last = buf;
}
else
{
*buf_out = buf;
last = buf;
}
}
return HB_WORK_OK;
}
/**********************************************************************
* Close
**********************************************************************
*
*********************************************************************/
void decmpeg2Close( hb_work_object_t * w )
{
hb_work_private_t * pv = w->private_data;
hb_list_close( &pv->list );
hb_libmpeg2_close( &pv->libmpeg2 );
free( pv );
}
hb_work_object_t hb_decmpeg2 =
{
WORK_DECMPEG2,
"MPEG-2 decoder (libmpeg2)",
decmpeg2Init,
decmpeg2Work,
decmpeg2Close
};
|
