FFmpeg
roqvideodec.c
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2003 The FFmpeg project
3  *
4  * This file is part of FFmpeg.
5  *
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.
10  *
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.
15  *
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
19  */
20 
21 /**
22  * @file
23  * id RoQ Video Decoder by Dr. Tim Ferguson
24  * For more information about the id RoQ format, visit:
25  * http://www.csse.monash.edu.au/~timf/
26  */
27 
28 #include "libavutil/avassert.h"
29 #include "libavutil/imgutils.h"
30 
31 #include "avcodec.h"
32 #include "bytestream.h"
33 #include "codec_internal.h"
34 #include "internal.h"
35 #include "roqvideo.h"
36 
38 {
39  unsigned int chunk_id = 0, chunk_arg = 0;
40  unsigned long chunk_size = 0;
41  int i, j, k, nv1, nv2, vqflg = 0, vqflg_pos = -1;
42  int vqid, xpos, ypos, xp, yp, x, y, mx, my;
43  roq_qcell *qcell;
44  int64_t chunk_start;
45 
46  while (bytestream2_get_bytes_left(gb) >= 8) {
47  chunk_id = bytestream2_get_le16(gb);
48  chunk_size = bytestream2_get_le32(gb);
49  chunk_arg = bytestream2_get_le16(gb);
50 
51  if(chunk_id == RoQ_QUAD_VQ)
52  break;
53  if(chunk_id == RoQ_QUAD_CODEBOOK) {
54  if((nv1 = chunk_arg >> 8) == 0)
55  nv1 = 256;
56  if((nv2 = chunk_arg & 0xff) == 0 && nv1 * 6 < chunk_size)
57  nv2 = 256;
58  for(i = 0; i < nv1; i++) {
59  ri->cb2x2[i].y[0] = bytestream2_get_byte(gb);
60  ri->cb2x2[i].y[1] = bytestream2_get_byte(gb);
61  ri->cb2x2[i].y[2] = bytestream2_get_byte(gb);
62  ri->cb2x2[i].y[3] = bytestream2_get_byte(gb);
63  ri->cb2x2[i].u = bytestream2_get_byte(gb);
64  ri->cb2x2[i].v = bytestream2_get_byte(gb);
65  }
66  for(i = 0; i < nv2; i++)
67  for(j = 0; j < 4; j++)
68  ri->cb4x4[i].idx[j] = bytestream2_get_byte(gb);
69  }
70  }
71 
73  xpos = ypos = 0;
74 
75  if (chunk_size > bytestream2_get_bytes_left(gb)) {
76  av_log(ri->avctx, AV_LOG_ERROR, "Chunk does not fit in input buffer\n");
77  chunk_size = bytestream2_get_bytes_left(gb);
78  }
79 
80  while (bytestream2_tell(gb) < chunk_start + chunk_size) {
81  for (yp = ypos; yp < ypos + 16; yp += 8)
82  for (xp = xpos; xp < xpos + 16; xp += 8) {
83  if (bytestream2_tell(gb) >= chunk_start + chunk_size) {
84  av_log(ri->avctx, AV_LOG_VERBOSE, "Chunk is too short\n");
85  return;
86  }
87  if (vqflg_pos < 0) {
88  vqflg = bytestream2_get_le16(gb);
89  vqflg_pos = 7;
90  }
91  vqid = (vqflg >> (vqflg_pos * 2)) & 0x3;
92  vqflg_pos--;
93 
94  switch(vqid) {
95  case RoQ_ID_MOT:
96  break;
97  case RoQ_ID_FCC: {
98  int byte = bytestream2_get_byte(gb);
99  mx = 8 - (byte >> 4) - ((signed char) (chunk_arg >> 8));
100  my = 8 - (byte & 0xf) - ((signed char) chunk_arg);
101  ff_apply_motion_8x8(ri, xp, yp, mx, my);
102  break;
103  }
104  case RoQ_ID_SLD:
105  qcell = ri->cb4x4 + bytestream2_get_byte(gb);
106  ff_apply_vector_4x4(ri, xp, yp, ri->cb2x2 + qcell->idx[0]);
107  ff_apply_vector_4x4(ri, xp + 4, yp, ri->cb2x2 + qcell->idx[1]);
108  ff_apply_vector_4x4(ri, xp, yp + 4, ri->cb2x2 + qcell->idx[2]);
109  ff_apply_vector_4x4(ri, xp + 4, yp + 4, ri->cb2x2 + qcell->idx[3]);
110  break;
111  case RoQ_ID_CCC:
112  for (k = 0; k < 4; k++) {
113  x = xp; y = yp;
114  if(k & 0x01) x += 4;
115  if(k & 0x02) y += 4;
116 
117  if (bytestream2_tell(gb) >= chunk_start + chunk_size) {
118  av_log(ri->avctx, AV_LOG_VERBOSE, "Chunk is too short\n");
119  return;
120  }
121  if (vqflg_pos < 0) {
122  vqflg = bytestream2_get_le16(gb);
123  vqflg_pos = 7;
124  }
125  vqid = (vqflg >> (vqflg_pos * 2)) & 0x3;
126  vqflg_pos--;
127  switch(vqid) {
128  case RoQ_ID_MOT:
129  break;
130  case RoQ_ID_FCC: {
131  int byte = bytestream2_get_byte(gb);
132  mx = 8 - (byte >> 4) - ((signed char) (chunk_arg >> 8));
133  my = 8 - (byte & 0xf) - ((signed char) chunk_arg);
134  ff_apply_motion_4x4(ri, x, y, mx, my);
135  break;
136  }
137  case RoQ_ID_SLD:
138  qcell = ri->cb4x4 + bytestream2_get_byte(gb);
139  ff_apply_vector_2x2(ri, x, y, ri->cb2x2 + qcell->idx[0]);
140  ff_apply_vector_2x2(ri, x + 2, y, ri->cb2x2 + qcell->idx[1]);
141  ff_apply_vector_2x2(ri, x, y + 2, ri->cb2x2 + qcell->idx[2]);
142  ff_apply_vector_2x2(ri, x + 2, y + 2, ri->cb2x2 + qcell->idx[3]);
143  break;
144  case RoQ_ID_CCC:
145  ff_apply_vector_2x2(ri, x, y, ri->cb2x2 + bytestream2_get_byte(gb));
146  ff_apply_vector_2x2(ri, x + 2, y, ri->cb2x2 + bytestream2_get_byte(gb));
147  ff_apply_vector_2x2(ri, x, y + 2, ri->cb2x2 + bytestream2_get_byte(gb));
148  ff_apply_vector_2x2(ri, x + 2, y + 2, ri->cb2x2 + bytestream2_get_byte(gb));
149  break;
150  }
151  }
152  break;
153  default:
154  av_assert2(0);
155  }
156  }
157 
158  xpos += 16;
159  if (xpos >= ri->width) {
160  xpos -= ri->width;
161  ypos += 16;
162  }
163  if(ypos >= ri->height)
164  break;
165  }
166 }
167 
168 
170 {
171  RoqContext *s = avctx->priv_data;
172 
173  s->avctx = avctx;
174 
175  if (avctx->width % 16 || avctx->height % 16) {
176  avpriv_request_sample(avctx, "Dimensions not being a multiple of 16");
177  return AVERROR_PATCHWELCOME;
178  }
179 
180  s->width = avctx->width;
181  s->height = avctx->height;
182 
183  s->last_frame = av_frame_alloc();
184  s->current_frame = av_frame_alloc();
185  if (!s->current_frame || !s->last_frame)
186  return AVERROR(ENOMEM);
187 
188  avctx->pix_fmt = AV_PIX_FMT_YUVJ444P;
189  avctx->color_range = AVCOL_RANGE_JPEG;
190 
191  return 0;
192 }
193 
194 static int roq_decode_frame(AVCodecContext *avctx, AVFrame *rframe,
195  int *got_frame, AVPacket *avpkt)
196 {
197  const uint8_t *buf = avpkt->data;
198  int buf_size = avpkt->size;
199  RoqContext *s = avctx->priv_data;
200  int copy = !s->current_frame->data[0] && s->last_frame->data[0];
201  GetByteContext gb;
202  int ret;
203 
204  if ((ret = ff_reget_buffer(avctx, s->current_frame, 0)) < 0)
205  return ret;
206 
207  if (copy) {
208  ret = av_frame_copy(s->current_frame, s->last_frame);
209  if (ret < 0)
210  return ret;
211  }
212 
213  bytestream2_init(&gb, buf, buf_size);
214  roqvideo_decode_frame(s, &gb);
215 
216  if ((ret = av_frame_ref(rframe, s->current_frame)) < 0)
217  return ret;
218  *got_frame = 1;
219 
220  /* shuffle frames */
221  FFSWAP(AVFrame *, s->current_frame, s->last_frame);
222 
223  return buf_size;
224 }
225 
227 {
228  RoqContext *s = avctx->priv_data;
229 
230  av_frame_free(&s->current_frame);
231  av_frame_free(&s->last_frame);
232 
233  return 0;
234 }
235 
237  .p.name = "roqvideo",
238  .p.long_name = NULL_IF_CONFIG_SMALL("id RoQ video"),
239  .p.type = AVMEDIA_TYPE_VIDEO,
240  .p.id = AV_CODEC_ID_ROQ,
241  .priv_data_size = sizeof(RoqContext),
243  .close = roq_decode_end,
245  .p.capabilities = AV_CODEC_CAP_DR1,
246  .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
247 };
RoqContext::cb4x4
roq_qcell cb4x4[256]
Definition: roqvideo.h:48
FF_CODEC_CAP_INIT_CLEANUP
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
Definition: codec_internal.h:41
chunk_start
static int chunk_start(AVFormatContext *s)
Definition: webm_chunk.c:162
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
roqvideo.h
RoQ_ID_SLD
#define RoQ_ID_SLD
Definition: roqvideo.h:59
GetByteContext
Definition: bytestream.h:33
av_frame_free
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:116
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:325
RoqContext
Definition: roqvideo.h:40
AVCOL_RANGE_JPEG
@ AVCOL_RANGE_JPEG
Full range content.
Definition: pixfmt.h:606
internal.h
AVPacket::data
uint8_t * data
Definition: packet.h:374
roq_cell::y
unsigned char y[4]
Definition: roqvideo.h:28
RoqContext::avctx
AVCodecContext * avctx
Definition: roqvideo.h:42
RoQ_ID_FCC
#define RoQ_ID_FCC
Definition: roqvideo.h:58
FFCodec
Definition: codec_internal.h:118
ff_apply_motion_8x8
void ff_apply_motion_8x8(RoqContext *ri, int x, int y, int deltax, int deltay)
Definition: roqvideo.c:140
AV_LOG_VERBOSE
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:196
RoqContext::height
int height
Definition: roqvideo.h:45
init
static int init
Definition: av_tx.c:47
FFCodec::p
AVCodec p
The public AVCodec.
Definition: codec_internal.h:122
RoQ_ID_CCC
#define RoQ_ID_CCC
Definition: roqvideo.h:60
av_frame_alloc
AVFrame * av_frame_alloc(void)
Allocate an AVFrame and set its fields to default values.
Definition: frame.c:104
roqvideo_decode_frame
static void roqvideo_decode_frame(RoqContext *ri, GetByteContext *gb)
Definition: roqvideodec.c:37
avassert.h
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:180
av_cold
#define av_cold
Definition: attributes.h:90
ff_apply_vector_2x2
void ff_apply_vector_2x2(RoqContext *ri, int x, int y, roq_cell *cell)
Definition: roqvideo.c:42
FF_CODEC_DECODE_CB
#define FF_CODEC_DECODE_CB(func)
Definition: codec_internal.h:260
s
#define s(width, name)
Definition: cbs_vp9.c:256
RoQ_QUAD_CODEBOOK
#define RoQ_QUAD_CODEBOOK
Definition: roqvideo.h:52
RoQ_ID_MOT
#define RoQ_ID_MOT
Definition: roqvideo.h:57
roq_cell::v
unsigned char v
Definition: roqvideo.h:29
AV_PIX_FMT_YUVJ444P
@ AV_PIX_FMT_YUVJ444P
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:80
AVERROR_PATCHWELCOME
#define AVERROR_PATCHWELCOME
Not yet implemented in FFmpeg, patches welcome.
Definition: error.h:64
AVCodecContext::color_range
enum AVColorRange color_range
MPEG vs JPEG YUV range.
Definition: avcodec.h:982
bytestream2_get_bytes_left
static av_always_inline int bytestream2_get_bytes_left(GetByteContext *g)
Definition: bytestream.h:158
bytestream2_tell
static av_always_inline int bytestream2_tell(GetByteContext *g)
Definition: bytestream.h:192
roq_qcell
Definition: roqvideo.h:32
AV_CODEC_ID_ROQ
@ AV_CODEC_ID_ROQ
Definition: codec_id.h:88
AV_CODEC_CAP_DR1
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
Definition: codec.h:52
AVPacket::size
int size
Definition: packet.h:375
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
copy
static void copy(const float *p1, float *p2, const int length)
Definition: vf_vaguedenoiser.c:187
av_frame_ref
int av_frame_ref(AVFrame *dst, const AVFrame *src)
Set up a new reference to the data described by the source frame.
Definition: frame.c:353
codec_internal.h
av_frame_copy
int av_frame_copy(AVFrame *dst, const AVFrame *src)
Copy the frame data from src to dst.
Definition: frame.c:771
ff_apply_vector_4x4
void ff_apply_vector_4x4(RoqContext *ri, int x, int y, roq_cell *cell)
Definition: roqvideo.c:72
roq_decode_frame
static int roq_decode_frame(AVCodecContext *avctx, AVFrame *rframe, int *got_frame, AVPacket *avpkt)
Definition: roqvideodec.c:194
roq_qcell::idx
int idx[4]
Definition: roqvideo.h:33
av_assert2
#define av_assert2(cond)
assert() equivalent, that does lie in speed critical code.
Definition: avassert.h:64
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:269
roq_decode_init
static av_cold int roq_decode_init(AVCodecContext *avctx)
Definition: roqvideodec.c:169
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:211
ff_roq_decoder
const FFCodec ff_roq_decoder
Definition: roqvideodec.c:236
AVCodecContext::height
int height
Definition: avcodec.h:571
AVCodecContext::pix_fmt
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:608
avcodec.h
RoQ_QUAD_VQ
#define RoQ_QUAD_VQ
Definition: roqvideo.h:53
ff_reget_buffer
int ff_reget_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Identical in function to ff_get_buffer(), except it reuses the existing buffer if available.
Definition: decode.c:1580
ret
ret
Definition: filter_design.txt:187
FFSWAP
#define FFSWAP(type, a, b)
Definition: macros.h:52
AVCodecContext
main external API structure.
Definition: avcodec.h:398
ff_apply_motion_4x4
void ff_apply_motion_4x4(RoqContext *ri, int x, int y, int deltax, int deltay)
Definition: roqvideo.c:134
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
avpriv_request_sample
#define avpriv_request_sample(...)
Definition: tableprint_vlc.h:36
AVPacket
This structure stores compressed data.
Definition: packet.h:351
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:425
roq_cell::u
unsigned char u
Definition: roqvideo.h:29
AVCodecContext::width
int width
picture width / height.
Definition: avcodec.h:571
bytestream.h
imgutils.h
bytestream2_init
static av_always_inline void bytestream2_init(GetByteContext *g, const uint8_t *buf, int buf_size)
Definition: bytestream.h:137
roq_decode_end
static av_cold int roq_decode_end(AVCodecContext *avctx)
Definition: roqvideodec.c:226
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
RoqContext::width
int width
Definition: roqvideo.h:45
RoqContext::cb2x2
roq_cell cb2x2[256]
Definition: roqvideo.h:47