2 * Copyright (c) 2008 BBC, Anuradha Suraparaju <asuraparaju at gmail dot com >
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 * function definitions common to libschroedinger decoder and encoder
26 #include "libavutil/attributes.h"
27 #include "libavutil/mem.h"
28 #include "libschroedinger.h"
30 static const SchroVideoFormatInfo ff_schro_video_format_info
[] = {
31 { 640, 480, 24000, 1001},
32 { 176, 120, 15000, 1001},
34 { 352, 240, 15000, 1001},
36 { 704, 480, 15000, 1001},
38 { 720, 480, 30000, 1001},
40 { 1280, 720, 60000, 1001},
42 { 1920, 1080, 30000, 1001},
43 { 1920, 1080, 25, 1 },
44 { 1920, 1080, 60000, 1001},
45 { 1920, 1080, 50, 1 },
46 { 2048, 1080, 24, 1 },
47 { 4096, 2160, 24, 1 },
50 static unsigned int get_video_format_idx(AVCodecContext
*avctx
)
52 unsigned int ret_idx
= 0;
54 unsigned int num_formats
= sizeof(ff_schro_video_format_info
) /
55 sizeof(ff_schro_video_format_info
[0]);
57 for (idx
= 1; idx
< num_formats
; ++idx
) {
58 const SchroVideoFormatInfo
*vf
= &ff_schro_video_format_info
[idx
];
59 if (avctx
->width
== vf
->width
&&
60 avctx
->height
== vf
->height
) {
62 if (avctx
->time_base
.den
== vf
->frame_rate_num
&&
63 avctx
->time_base
.num
== vf
->frame_rate_denom
)
70 av_cold
void ff_schro_queue_init(FFSchroQueue
*queue
)
72 queue
->p_head
= queue
->p_tail
= NULL
;
76 void ff_schro_queue_free(FFSchroQueue
*queue
, void (*free_func
)(void *))
79 free_func(ff_schro_queue_pop(queue
));
82 int ff_schro_queue_push_back(FFSchroQueue
*queue
, void *p_data
)
84 FFSchroQueueElement
*p_new
= av_mallocz(sizeof(FFSchroQueueElement
));
92 queue
->p_head
= p_new
;
94 queue
->p_tail
->next
= p_new
;
95 queue
->p_tail
= p_new
;
101 void *ff_schro_queue_pop(FFSchroQueue
*queue
)
103 FFSchroQueueElement
*top
= queue
->p_head
;
106 void *data
= top
->data
;
107 queue
->p_head
= queue
->p_head
->next
;
117 * Schroedinger video preset table. Ensure that this tables matches up correctly
118 * with the ff_schro_video_format_info table.
120 static const SchroVideoFormatEnum ff_schro_video_formats
[]={
121 SCHRO_VIDEO_FORMAT_CUSTOM
,
122 SCHRO_VIDEO_FORMAT_QSIF
,
123 SCHRO_VIDEO_FORMAT_QCIF
,
124 SCHRO_VIDEO_FORMAT_SIF
,
125 SCHRO_VIDEO_FORMAT_CIF
,
126 SCHRO_VIDEO_FORMAT_4SIF
,
127 SCHRO_VIDEO_FORMAT_4CIF
,
128 SCHRO_VIDEO_FORMAT_SD480I_60
,
129 SCHRO_VIDEO_FORMAT_SD576I_50
,
130 SCHRO_VIDEO_FORMAT_HD720P_60
,
131 SCHRO_VIDEO_FORMAT_HD720P_50
,
132 SCHRO_VIDEO_FORMAT_HD1080I_60
,
133 SCHRO_VIDEO_FORMAT_HD1080I_50
,
134 SCHRO_VIDEO_FORMAT_HD1080P_60
,
135 SCHRO_VIDEO_FORMAT_HD1080P_50
,
136 SCHRO_VIDEO_FORMAT_DC2K_24
,
137 SCHRO_VIDEO_FORMAT_DC4K_24
,
140 SchroVideoFormatEnum
ff_get_schro_video_format_preset(AVCodecContext
*avctx
)
142 unsigned int num_formats
= sizeof(ff_schro_video_formats
) /
143 sizeof(ff_schro_video_formats
[0]);
145 unsigned int idx
= get_video_format_idx(avctx
);
147 return (idx
< num_formats
) ? ff_schro_video_formats
[idx
] :
148 SCHRO_VIDEO_FORMAT_CUSTOM
;
151 int ff_get_schro_frame_format (SchroChromaFormat schro_pix_fmt
,
152 SchroFrameFormat
*schro_frame_fmt
)
154 unsigned int num_formats
= sizeof(schro_pixel_format_map
) /
155 sizeof(schro_pixel_format_map
[0]);
159 for (idx
= 0; idx
< num_formats
; ++idx
) {
160 if (schro_pixel_format_map
[idx
].schro_pix_fmt
== schro_pix_fmt
) {
161 *schro_frame_fmt
= schro_pixel_format_map
[idx
].schro_frame_fmt
;
168 static void free_schro_frame(SchroFrame
*frame
, void *priv
)
170 AVPicture
*p_pic
= priv
;
175 avpicture_free(p_pic
);
179 SchroFrame
*ff_create_schro_frame(AVCodecContext
*avctx
,
180 SchroFrameFormat schro_frame_fmt
)
184 int y_width
, uv_width
;
185 int y_height
, uv_height
;
188 y_width
= avctx
->width
;
189 y_height
= avctx
->height
;
190 uv_width
= y_width
>> (SCHRO_FRAME_FORMAT_H_SHIFT(schro_frame_fmt
));
191 uv_height
= y_height
>> (SCHRO_FRAME_FORMAT_V_SHIFT(schro_frame_fmt
));
193 p_pic
= av_mallocz(sizeof(AVPicture
));
194 if (!p_pic
|| avpicture_alloc(p_pic
, avctx
->pix_fmt
, y_width
, y_height
) < 0) {
199 p_frame
= schro_frame_new();
200 p_frame
->format
= schro_frame_fmt
;
201 p_frame
->width
= y_width
;
202 p_frame
->height
= y_height
;
203 schro_frame_set_free_callback(p_frame
, free_schro_frame
, (void *)p_pic
);
205 for (i
= 0; i
< 3; ++i
) {
206 p_frame
->components
[i
].width
= i
? uv_width
: y_width
;
207 p_frame
->components
[i
].stride
= p_pic
->linesize
[i
];
208 p_frame
->components
[i
].height
= i
? uv_height
: y_height
;
209 p_frame
->components
[i
].length
=
210 p_frame
->components
[i
].stride
* p_frame
->components
[i
].height
;
211 p_frame
->components
[i
].data
= p_pic
->data
[i
];
214 p_frame
->components
[i
].v_shift
=
215 SCHRO_FRAME_FORMAT_V_SHIFT(p_frame
->format
);
216 p_frame
->components
[i
].h_shift
=
217 SCHRO_FRAME_FORMAT_H_SHIFT(p_frame
->format
);