zerocodec.c
Go to the documentation of this file.
1 /*
2  * ZeroCodec Decoder
3  *
4  * Copyright (c) 2012, Derek Buitenhuis
5  *
6  * Permission to use, copy, modify, and/or distribute this software for any
7  * purpose with or without fee is hereby granted, provided that the above
8  * copyright notice and this permission notice appear in all copies.
9  *
10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17  */
18 
19 #include <zlib.h>
20 
21 #include "avcodec.h"
22 #include "internal.h"
23 #include "libavutil/common.h"
24 
25 typedef struct {
27  z_stream zstream;
29 
30 static int zerocodec_decode_frame(AVCodecContext *avctx, void *data,
31  int *got_frame, AVPacket *avpkt)
32 {
33  ZeroCodecContext *zc = avctx->priv_data;
34  AVFrame *pic = avctx->coded_frame;
35  AVFrame *prev_pic = &zc->previous_frame;
36  z_stream *zstream = &zc->zstream;
37  uint8_t *prev = prev_pic->data[0];
38  uint8_t *dst;
39  int i, j, zret;
40 
41  pic->reference = 3;
42 
43  if (avpkt->flags & AV_PKT_FLAG_KEY) {
44  pic->key_frame = 1;
46  } else {
47  if (!prev) {
48  av_log(avctx, AV_LOG_ERROR, "Missing reference frame.\n");
49  return AVERROR_INVALIDDATA;
50  }
51 
52  prev += (avctx->height - 1) * prev_pic->linesize[0];
53 
54  pic->key_frame = 0;
56  }
57 
58  zret = inflateReset(zstream);
59  if (zret != Z_OK) {
60  av_log(avctx, AV_LOG_ERROR, "Could not reset inflate: %d.\n", zret);
61  return AVERROR_INVALIDDATA;
62  }
63 
64  if (ff_get_buffer(avctx, pic) < 0) {
65  av_log(avctx, AV_LOG_ERROR, "Could not allocate buffer.\n");
66  return AVERROR(ENOMEM);
67  }
68 
69  zstream->next_in = avpkt->data;
70  zstream->avail_in = avpkt->size;
71 
72  dst = pic->data[0] + (avctx->height - 1) * pic->linesize[0];
73 
79  for (i = 0; i < avctx->height; i++) {
80  zstream->next_out = dst;
81  zstream->avail_out = avctx->width << 1;
82 
83  zret = inflate(zstream, Z_SYNC_FLUSH);
84  if (zret != Z_OK && zret != Z_STREAM_END) {
85  avctx->release_buffer(avctx, pic);
86  av_log(avctx, AV_LOG_ERROR,
87  "Inflate failed with return code: %d.\n", zret);
88  return AVERROR_INVALIDDATA;
89  }
90 
91  if (!(avpkt->flags & AV_PKT_FLAG_KEY))
92  for (j = 0; j < avctx->width << 1; j++)
93  dst[j] += prev[j] & -!dst[j];
94 
95  prev -= prev_pic->linesize[0];
96  dst -= pic->linesize[0];
97  }
98 
99  /* Release the previous buffer if need be */
100  if (prev_pic->data[0])
101  avctx->release_buffer(avctx, prev_pic);
102 
103  *got_frame = 1;
104  *(AVFrame *)data = *pic;
105 
106  /* Store the previous frame for use later.
107  * FFSWAP ensures that e.g. pic->data is NULLed. */
108  FFSWAP(AVFrame, *pic, *prev_pic);
109 
110  return avpkt->size;
111 }
112 
114 {
115  ZeroCodecContext *zc = avctx->priv_data;
116  AVFrame *prev_pic = &zc->previous_frame;
117 
118  inflateEnd(&zc->zstream);
119 
120  /* Release last frame */
121  if (prev_pic->data[0])
122  avctx->release_buffer(avctx, prev_pic);
123 
124  av_freep(&avctx->coded_frame);
125 
126  return 0;
127 }
128 
130 {
131  ZeroCodecContext *zc = avctx->priv_data;
132  z_stream *zstream = &zc->zstream;
133  int zret;
134 
135  avctx->pix_fmt = AV_PIX_FMT_UYVY422;
136  avctx->bits_per_raw_sample = 8;
137 
138  zstream->zalloc = Z_NULL;
139  zstream->zfree = Z_NULL;
140  zstream->opaque = Z_NULL;
141 
142  zret = inflateInit(zstream);
143  if (zret != Z_OK) {
144  av_log(avctx, AV_LOG_ERROR, "Could not initialize inflate: %d.\n", zret);
145  return AVERROR(ENOMEM);
146  }
147 
148  avctx->coded_frame = avcodec_alloc_frame();
149  if (!avctx->coded_frame) {
150  av_log(avctx, AV_LOG_ERROR, "Could not allocate frame buffer.\n");
151  zerocodec_decode_close(avctx);
152  return AVERROR(ENOMEM);
153  }
154 
155  return 0;
156 }
157 
160  .name = "zerocodec",
161  .id = AV_CODEC_ID_ZEROCODEC,
162  .priv_data_size = sizeof(ZeroCodecContext),
166  .capabilities = CODEC_CAP_DR1,
167  .long_name = NULL_IF_CONFIG_SMALL("ZeroCodec Lossless Video"),
168 };
packed YUV 4:2:2, 16bpp, Cb Y0 Cr Y1
Definition: pixfmt.h:82
This structure describes decoded (raw) audio or video data.
Definition: avcodec.h:989
void(* release_buffer)(struct AVCodecContext *c, AVFrame *pic)
Called to release buffers which were allocated with get_buffer.
Definition: avcodec.h:2259
AVFrame * coded_frame
the picture in the bitstream
Definition: avcodec.h:2725
int size
Definition: avcodec.h:916
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1533
z_stream zstream
Definition: zerocodec.c:27
enum AVMediaType type
Definition: avcodec.h:2973
int bits_per_raw_sample
Bits per sample/pixel of internal libavcodec pixel/sample format.
Definition: avcodec.h:2711
AVCodec.
Definition: avcodec.h:2960
void av_freep(void *arg)
Free a memory block which has been allocated with av_malloc(z)() or av_realloc() and set the pointer ...
Definition: mem.c:151
static int decode(MimicContext *ctx, int quality, int num_coeffs, int is_iframe)
Definition: mimic.c:228
uint8_t
static av_cold int zerocodec_decode_close(AVCodecContext *avctx)
Definition: zerocodec.c:113
const char data[16]
Definition: mxf.c:66
uint8_t * data
Definition: avcodec.h:915
static int init(AVCodecParserContext *s)
Definition: h264_parser.c:335
static int zerocodec_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: zerocodec.c:30
AVFrame previous_frame
Definition: zerocodec.c:26
int reference
is this picture used as reference The values for this are the same as the MpegEncContext.picture_structure variable, that is 1->top field, 2->bottom field, 3->frame/both fields.
Definition: avcodec.h:1132
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:88
void av_log(void *avcl, int level, const char *fmt,...)
Definition: log.c:146
int flags
A combination of AV_PKT_FLAG values.
Definition: avcodec.h:921
AVFrame * avcodec_alloc_frame(void)
Allocate an AVFrame and set its fields to default values.
Definition: utils.c:618
enum AVPictureType pict_type
Picture type of the frame, see ?_TYPE below.
Definition: avcodec.h:1065
int width
picture width / height.
Definition: avcodec.h:1508
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame)
Get a buffer for a frame.
Definition: utils.c:464
external API header
int linesize[AV_NUM_DATA_POINTERS]
Size, in bytes, of the data for each picture/channel plane.
Definition: avcodec.h:1008
main external API structure.
Definition: avcodec.h:1339
static void close(AVCodecParserContext *s)
Definition: h264_parser.c:326
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: avcodec.h:997
common internal api header.
common internal and external API header
void * priv_data
Definition: avcodec.h:1382
int key_frame
1 -> keyframe, 0-> not
Definition: avcodec.h:1058
static av_cold int zerocodec_decode_init(AVCodecContext *avctx)
Definition: zerocodec.c:129
This structure stores compressed data.
Definition: avcodec.h:898
for(j=16;j >0;--j)
AVCodec ff_zerocodec_decoder
Definition: zerocodec.c:158
Predicted.
Definition: avutil.h:246