1 |
|
|
/* |
2 |
|
|
* This file is part of FFmpeg. |
3 |
|
|
* |
4 |
|
|
* FFmpeg is free software; you can redistribute it and/or modify |
5 |
|
|
* it under the terms of the GNU General Public License as published by |
6 |
|
|
* the Free Software Foundation; either version 2 of the License, or |
7 |
|
|
* (at your option) any later version. |
8 |
|
|
* |
9 |
|
|
* FFmpeg is distributed in the hope that it will be useful, |
10 |
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 |
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 |
|
|
* GNU General Public License for more details. |
13 |
|
|
* |
14 |
|
|
* You should have received a copy of the GNU General Public License along |
15 |
|
|
* with FFmpeg; if not, write to the Free Software Foundation, Inc., |
16 |
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
17 |
|
|
*/ |
18 |
|
|
|
19 |
|
|
#include <stdint.h> |
20 |
|
|
|
21 |
|
|
// for FF_QSCALE_TYPE_* |
22 |
|
|
#include "libavcodec/internal.h" |
23 |
|
|
|
24 |
|
|
#include "libavutil/frame.h" |
25 |
|
|
#include "libavutil/mem.h" |
26 |
|
|
#include "libavutil/video_enc_params.h" |
27 |
|
|
|
28 |
|
|
#include "qp_table.h" |
29 |
|
|
|
30 |
|
50 |
int ff_qp_table_extract(AVFrame *frame, int8_t **table, int *table_w, int *table_h, |
31 |
|
|
int *qscale_type) |
32 |
|
|
{ |
33 |
|
|
AVFrameSideData *sd; |
34 |
|
|
AVVideoEncParams *par; |
35 |
|
50 |
unsigned int mb_h = (frame->height + 15) / 16; |
36 |
|
50 |
unsigned int mb_w = (frame->width + 15) / 16; |
37 |
|
50 |
unsigned int nb_mb = mb_h * mb_w; |
38 |
|
|
unsigned int block_idx; |
39 |
|
|
|
40 |
|
50 |
*table = NULL; |
41 |
|
|
|
42 |
|
50 |
sd = av_frame_get_side_data(frame, AV_FRAME_DATA_VIDEO_ENC_PARAMS); |
43 |
✓✓ |
50 |
if (!sd) |
44 |
|
23 |
return 0; |
45 |
|
27 |
par = (AVVideoEncParams*)sd->data; |
46 |
✓✗ |
27 |
if (par->type != AV_VIDEO_ENC_PARAMS_MPEG2 || |
47 |
✓✓✗✓
|
27 |
(par->nb_blocks != 0 && par->nb_blocks != nb_mb)) |
48 |
|
|
return AVERROR(ENOSYS); |
49 |
|
|
|
50 |
|
27 |
*table = av_malloc(nb_mb); |
51 |
✗✓ |
27 |
if (!*table) |
52 |
|
|
return AVERROR(ENOMEM); |
53 |
✓✗ |
27 |
if (table_w) |
54 |
|
27 |
*table_w = mb_w; |
55 |
✗✓ |
27 |
if (table_h) |
56 |
|
|
*table_h = mb_h; |
57 |
✓✓ |
27 |
if (qscale_type) |
58 |
|
7 |
*qscale_type = FF_QSCALE_TYPE_MPEG2; |
59 |
|
|
|
60 |
✓✓ |
27 |
if (par->nb_blocks == 0) { |
61 |
|
5 |
memset(*table, par->qp, nb_mb); |
62 |
|
5 |
return 0; |
63 |
|
|
} |
64 |
|
|
|
65 |
✓✓ |
8734 |
for (block_idx = 0; block_idx < nb_mb; block_idx++) { |
66 |
|
8712 |
AVVideoBlockParams *b = av_video_enc_params_block(par, block_idx); |
67 |
|
8712 |
(*table)[block_idx] = par->qp + b->delta_qp; |
68 |
|
|
} |
69 |
|
|
|
70 |
|
22 |
return 0; |
71 |
|
|
} |
72 |
|
|
|