30 #include <schroedinger/schro.h>
31 #include <schroedinger/schrodebug.h>
32 #include <schroedinger/schrovideoformat.h>
34 #include "libavutil/attributes.h"
35 #include "libavutil/avassert.h"
89 for (idx = 0; idx < num_formats; ++idx) {
91 p_schro_params->
format->chroma_format =
98 "This codec currently only supports planar YUV 4:2:0, 4:2:2"
99 " and 4:4:4 formats.\n");
107 SchroVideoFormatEnum
preset;
113 p_schro_params->
encoder = schro_encoder_new();
115 if (!p_schro_params->
encoder) {
117 "Unrecoverable Error: schro_encoder_new failed. ");
124 schro_encoder_get_video_format(p_schro_params->
encoder);
125 schro_video_format_set_std_video_format(p_schro_params->
format, preset);
133 p_schro_params->
format->colour_primaries = SCHRO_COLOUR_PRIMARY_HDTV;
135 p_schro_params->
format->colour_primaries = SCHRO_COLOUR_PRIMARY_SDTV_625;
137 p_schro_params->
format->colour_primaries = SCHRO_COLOUR_PRIMARY_SDTV_525;
141 p_schro_params->
format->colour_matrix = SCHRO_COLOUR_MATRIX_HDTV;
143 p_schro_params->
format->colour_matrix = SCHRO_COLOUR_MATRIX_SDTV;
147 p_schro_params->
format->transfer_function = SCHRO_TRANSFER_CHAR_TV_GAMMA;
153 "This codec currently supports only planar YUV 4:2:0, 4:2:2"
154 " and 4:4:4 formats.\n");
168 schro_encoder_setting_set_double(p_schro_params->
encoder,
170 SCHRO_ENCODER_GOP_INTRA_ONLY);
173 schro_encoder_setting_set_double(p_schro_params->
encoder,
174 "enable_noarith", 1);
176 schro_encoder_setting_set_double(p_schro_params->
encoder,
179 p_schro_params->
dts = -1;
186 schro_encoder_setting_set_double(p_schro_params->
encoder,
188 SCHRO_ENCODER_RATE_CONTROL_LOSSLESS);
191 schro_encoder_setting_set_double(p_schro_params->
encoder,
193 SCHRO_ENCODER_RATE_CONTROL_CONSTANT_QUALITY);
198 schro_encoder_setting_set_double(p_schro_params->
encoder,
202 schro_encoder_setting_set_double(p_schro_params->
encoder,
204 SCHRO_ENCODER_RATE_CONTROL_CONSTANT_BITRATE);
206 schro_encoder_setting_set_double(p_schro_params->
encoder,
213 schro_encoder_setting_set_double(p_schro_params->
encoder,
214 "interlaced_coding", 1);
216 schro_encoder_setting_set_double(p_schro_params->
encoder,
"open_gop",
221 schro_video_format_set_std_signal_range(p_schro_params->
format,
222 SCHRO_SIGNAL_RANGE_8BIT_VIDEO);
225 schro_encoder_set_video_format(p_schro_params->
encoder,
229 schro_debug_set_level(avctx->
debug);
231 schro_encoder_start(p_schro_params->
encoder);
242 SchroFrame *in_frame;
251 in_frame->components[0].data,
270 SchroEncoder *encoder = p_schro_params->
encoder;
273 SchroBuffer *enc_buf;
274 int presentation_frame;
276 int last_frame_in_sequence = 0;
282 schro_encoder_end_of_stream(encoder);
289 schro_encoder_push_frame(encoder, in_frame);
297 SchroStateEnum
state;
298 state = schro_encoder_wait(encoder);
300 case SCHRO_STATE_HAVE_BUFFER:
301 case SCHRO_STATE_END_OF_STREAM:
302 enc_buf = schro_encoder_pull(encoder, &presentation_frame);
304 parse_code = enc_buf->data[4];
314 enc_buf->data, enc_buf->length);
318 if (state == SCHRO_STATE_END_OF_STREAM) {
323 if (!SCHRO_PARSE_CODE_IS_PICTURE(parse_code)) {
324 schro_buffer_unref(enc_buf);
333 if (SCHRO_PARSE_CODE_IS_INTRA(parse_code) &&
334 SCHRO_PARSE_CODE_IS_REFERENCE(parse_code))
344 p_schro_params->
enc_buf = NULL;
346 schro_buffer_unref(enc_buf);
350 case SCHRO_STATE_NEED_FRAME:
354 case SCHRO_STATE_AGAIN:
367 last_frame_in_sequence = 1;
374 pkt_size = p_frame_output->
size;
375 if (last_frame_in_sequence && p_schro_params->
enc_buf_size > 0)
387 pkt->
dts = p_schro_params->
dts++;
388 enc_size = p_frame_output->
size;
392 if (last_frame_in_sequence && p_schro_params->
enc_buf_size > 0) {
393 memcpy(pkt->
data + enc_size, p_schro_params->
enc_buf,
416 schro_encoder_free(p_schro_params->
encoder);
435 .
name =
"libschroedinger",
int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size)
Check AVPacket size and/or allocate data.
AVFrame picture
frame being encoded
This structure describes decoded (raw) audio or video data.
static av_cold int init(AVCodecContext *avctx)
uint16_t key_frame
key frame flag.
SchroFrame * ff_create_schro_frame(AVCodecContext *avctx, SchroFrameFormat schro_frame_fmt)
Create a Schro frame based on the dimensions and frame format passed.
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
void av_log(void *avcl, int level, const char *fmt,...) av_printf_format(3
Send the specified message to the log if the level is less than or equal to the current av_log_level...
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
int avpicture_layout(const AVPicture *src, enum AVPixelFormat pix_fmt, int width, int height, unsigned char *dest, int dest_size)
Copy pixel data from an AVPicture into a buffer.
libschroedinger encoder private data
data structures common to libschroedinger decoder and encoder
enum AVPixelFormat ff_pix_fmt
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
void av_freep(void *ptr)
Free a memory block which has been allocated with av_malloc(z)() or av_realloc() and set the pointer ...
static void libschroedinger_free_frame(void *data)
uint32_t frame_num
encoded frame number.
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
const char * name
Name of the codec implementation.
contains a single encoded frame returned from Dirac or Schroedinger
uint8_t * p_encbuf
encoded frame data
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
static int set_chroma_format(AVCodecContext *avctx)
Works out Schro-compatible chroma format.
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
int has_b_frames
Size of the frame reordering buffer in the decoder.
void av_free(void *ptr)
Free a memory block which has been allocated with av_malloc(z)() or av_realloc(). ...
SchroEncoder * encoder
Schroedinger encoder handle.
#define CODEC_FLAG_INTERLACED_ME
interlaced motion estimation
SchroFrameFormat frame_format
Schroedinger frame format.
#define CODEC_CAP_DELAY
Encoder or decoder requires flushing with NULL input at the end in order to give the complete and cor...
void * ff_schro_queue_pop(FFSchroQueue *queue)
Return the first element in the queue.
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
#define CODEC_FLAG_QSCALE
Use fixed qscale.
FFSchroQueue enc_frame_queue
queue storing encoded frames
SchroVideoFormat * format
Schroedinger video format.
Libavcodec external API header.
AVPixelFormat
Pixel format.
int flags
A combination of AV_PKT_FLAG values.
int ff_get_schro_frame_format(SchroChromaFormat schro_pix_fmt, SchroFrameFormat *schro_frame_fmt)
Sets the Schroedinger frame format corresponding to the Schro chroma format passed.
int enc_buf_size
Size of encoder buffer.
int bit_rate
the average bitrate
int ff_schro_queue_push_back(FFSchroQueue *queue, void *p_data)
Add an element to the end of the queue.
void * av_realloc(void *ptr, size_t size) 1(2)
Allocate or reallocate a block of memory.
void ff_schro_queue_free(FFSchroQueue *queue, void(*free_func)(void *))
Free the queue resources.
av_cold void ff_schro_queue_init(FFSchroQueue *queue)
Initialise the queue.
A simple queue implementation used in libschroedinger.
int width
picture width / height.
static av_cold int libschroedinger_encode_init(AVCodecContext *avctx)
enum AVColorPrimaries color_primaries
Chromaticity coordinates of the source primaries.
#define FF_CODER_TYPE_VLC
static int libschroedinger_encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *frame, int *got_packet)
int eos_signalled
end of sequence signalled
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM
main external API structure.
static void close(AVCodecParserContext *s)
unsigned char * enc_buf
buffer to store encoder output before writing it to the frame queue
enum AVColorSpace colorspace
YUV colorspace type.
enum AVColorTransferCharacteristic color_trc
Color Transfer Characteristic.
AVFrame * coded_frame
the picture in the bitstream
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
SchroVideoFormatEnum ff_get_schro_video_format_preset(AVCodecContext *avctx)
Returns the video format preset matching the input video dimensions and time base.
int eos_pulled
end of sequence pulled
#define CODEC_FLAG_CLOSED_GOP
also ITU-R BT601-6 625 / ITU-R BT1358 625 / ITU-R BT1700 625 PAL & SECAM / IEC 61966-2-4 xvYCC601 ...
int global_quality
Global quality for codecs which cannot change it per frame.
static SchroFrame * libschroedinger_frame_from_data(AVCodecContext *avctx, const AVFrame *frame)
also ITU-R BT601-6 525 / ITU-R BT1358 525 / ITU-R BT1700 NTSC
int gop_size
the number of pictures in a group of pictures, or 0 for intra_only
common internal api header.
static struct @75 schro_pixel_format_map[]
AVCodec ff_libschroedinger_encoder
static int libschroedinger_encode_close(AVCodecContext *avctx)
uint32_t size
encoded frame size
int avpicture_get_size(enum AVPixelFormat pix_fmt, int width, int height)
Calculate the size in bytes that a picture of the given width and height would occupy if stored in th...
int key_frame
1 -> keyframe, 0-> not
#define FF_QP2LAMBDA
factor to convert from H.263 QP to lambda
also ITU-R BT1361 / IEC 61966-2-4 xvYCC709 / SMPTE RP177 Annex B
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed...
also ITU-R BT1361 / IEC 61966-2-4 / SMPTE RP177 Annex B
#define av_assert0(cond)
assert() equivalent, that is always enabled.
This structure stores compressed data.
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
void * av_mallocz(size_t size) av_malloc_attrib 1(1)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...