FFmpeg
wmv2enc.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2002 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 #include "avcodec.h"
22 #include "codec_internal.h"
23 #include "h263.h"
24 #include "mpegvideo.h"
25 #include "mpegvideoenc.h"
26 #include "msmpeg4.h"
27 #include "msmpeg4enc.h"
28 #include "msmpeg4data.h"
29 #include "wmv2.h"
30 #include "wmv2enc.h"
31 
32 typedef struct WMV2EncContext {
36  int j_type;
37  int abt_flag;
38  int abt_type;
40  int mspel_bit;
45 
47 {
48  MpegEncContext *const s = &w->msmpeg4.s;
49  PutBitContext pb;
50  int code;
51 
52  init_put_bits(&pb, s->avctx->extradata, s->avctx->extradata_size);
53 
54  put_bits(&pb, 5, s->avctx->time_base.den / s->avctx->time_base.num); // yes 29.97 -> 29
55  put_bits(&pb, 11, FFMIN(s->bit_rate / 1024, 2047));
56 
57  put_bits(&pb, 1, w->mspel_bit = 1);
58  put_bits(&pb, 1, s->loop_filter);
59  put_bits(&pb, 1, w->abt_flag = 1);
60  put_bits(&pb, 1, w->j_type_bit = 1);
61  put_bits(&pb, 1, w->top_left_mv_flag = 0);
62  put_bits(&pb, 1, w->per_mb_rl_bit = 1);
63  put_bits(&pb, 3, code = 1);
64 
65  flush_put_bits(&pb);
66 
67  s->slice_height = s->mb_height / code;
68 
69  return 0;
70 }
71 
73 {
74  WMV2EncContext *const w = avctx->priv_data;
75  MpegEncContext *const s = &w->msmpeg4.s;
76 
77  s->private_ctx = &w->common;
78  if (ff_mpv_encode_init(avctx) < 0)
79  return -1;
80 
82 
83  avctx->extradata_size = 4;
85  if (!avctx->extradata)
86  return AVERROR(ENOMEM);
87 
89 
90  return 0;
91 }
92 
94 {
95  WMV2EncContext *const w = (WMV2EncContext *) s;
96 
97  put_bits(&s->pb, 1, s->pict_type - 1);
98  if (s->pict_type == AV_PICTURE_TYPE_I)
99  put_bits(&s->pb, 7, 0);
100  put_bits(&s->pb, 5, s->qscale);
101 
102  s->dc_table_index = 1;
103  s->mv_table_index = 1; /* only if P-frame */
104  s->per_mb_rl_table = 0;
105  s->mspel = 0;
106  w->per_mb_abt = 0;
107  w->abt_type = 0;
108  w->j_type = 0;
109 
110  av_assert0(s->flipflop_rounding);
111 
112  if (s->pict_type == AV_PICTURE_TYPE_I) {
113  av_assert0(s->no_rounding == 1);
114  if (w->j_type_bit)
115  put_bits(&s->pb, 1, w->j_type);
116 
117  if (w->per_mb_rl_bit)
118  put_bits(&s->pb, 1, s->per_mb_rl_table);
119 
120  if (!s->per_mb_rl_table) {
121  ff_msmpeg4_code012(&s->pb, s->rl_chroma_table_index);
122  ff_msmpeg4_code012(&s->pb, s->rl_table_index);
123  }
124 
125  put_bits(&s->pb, 1, s->dc_table_index);
126 
127  s->inter_intra_pred = 0;
128  } else {
129  int cbp_index;
130 
131  put_bits(&s->pb, 2, SKIP_TYPE_NONE);
132 
133  ff_msmpeg4_code012(&s->pb, cbp_index = 0);
134  w->cbp_table_index = wmv2_get_cbp_table_index(s, cbp_index);
135 
136  if (w->mspel_bit)
137  put_bits(&s->pb, 1, s->mspel);
138 
139  if (w->abt_flag) {
140  put_bits(&s->pb, 1, w->per_mb_abt ^ 1);
141  if (!w->per_mb_abt)
142  ff_msmpeg4_code012(&s->pb, w->abt_type);
143  }
144 
145  if (w->per_mb_rl_bit)
146  put_bits(&s->pb, 1, s->per_mb_rl_table);
147 
148  if (!s->per_mb_rl_table) {
149  ff_msmpeg4_code012(&s->pb, s->rl_table_index);
150  s->rl_chroma_table_index = s->rl_table_index;
151  }
152  put_bits(&s->pb, 1, s->dc_table_index);
153  put_bits(&s->pb, 1, s->mv_table_index);
154 
155  s->inter_intra_pred = 0; // (s->width * s->height < 320 * 240 && s->bit_rate <= II_BITRATE);
156  }
157  s->esc3_level_length = 0;
158  s->esc3_run_length = 0;
159 
160  return 0;
161 }
162 
163 /* Nearly identical to wmv1 but that is just because we do not use the
164  * useless M$ crap features. It is duplicated here in case someone wants
165  * to add support for these crap features. */
166 void ff_wmv2_encode_mb(MpegEncContext *s, int16_t block[6][64],
167  int motion_x, int motion_y)
168 {
169  WMV2EncContext *const w = (WMV2EncContext *) s;
170  int cbp, coded_cbp, i;
171  int pred_x, pred_y;
172  uint8_t *coded_block;
173 
175 
176  if (!s->mb_intra) {
177  /* compute cbp */
178  cbp = 0;
179  for (i = 0; i < 6; i++)
180  if (s->block_last_index[i] >= 0)
181  cbp |= 1 << (5 - i);
182 
183  put_bits(&s->pb,
184  ff_wmv2_inter_table[w->cbp_table_index][cbp + 64][1],
185  ff_wmv2_inter_table[w->cbp_table_index][cbp + 64][0]);
186 
187  s->misc_bits += get_bits_diff(s);
188  /* motion vector */
189  ff_h263_pred_motion(s, 0, 0, &pred_x, &pred_y);
190  ff_msmpeg4_encode_motion(s, motion_x - pred_x,
191  motion_y - pred_y);
192  s->mv_bits += get_bits_diff(s);
193  } else {
194  /* compute cbp */
195  cbp = 0;
196  coded_cbp = 0;
197  for (i = 0; i < 6; i++) {
198  int val, pred;
199  val = (s->block_last_index[i] >= 1);
200  cbp |= val << (5 - i);
201  if (i < 4) {
202  /* predict value for close blocks only for luma */
203  pred = ff_msmpeg4_coded_block_pred(s, i, &coded_block);
204  *coded_block = val;
205  val = val ^ pred;
206  }
207  coded_cbp |= val << (5 - i);
208  }
209 
210  if (s->pict_type == AV_PICTURE_TYPE_I)
211  put_bits(&s->pb,
212  ff_msmp4_mb_i_table[coded_cbp][1],
213  ff_msmp4_mb_i_table[coded_cbp][0]);
214  else
215  put_bits(&s->pb,
216  ff_wmv2_inter_table[w->cbp_table_index][cbp][1],
217  ff_wmv2_inter_table[w->cbp_table_index][cbp][0]);
218  put_bits(&s->pb, 1, 0); /* no AC prediction yet */
219  if (s->inter_intra_pred) {
220  s->h263_aic_dir = 0;
221  put_bits(&s->pb,
222  ff_table_inter_intra[s->h263_aic_dir][1],
223  ff_table_inter_intra[s->h263_aic_dir][0]);
224  }
225  s->misc_bits += get_bits_diff(s);
226  }
227 
228  for (i = 0; i < 6; i++)
230  if (s->mb_intra)
231  s->i_tex_bits += get_bits_diff(s);
232  else
233  s->p_tex_bits += get_bits_diff(s);
234 }
235 
237  .p.name = "wmv2",
238  CODEC_LONG_NAME("Windows Media Video 8"),
239  .p.type = AVMEDIA_TYPE_VIDEO,
240  .p.id = AV_CODEC_ID_WMV2,
241  .p.priv_class = &ff_mpv_enc_class,
242  .priv_data_size = sizeof(WMV2EncContext),
245  .close = ff_mpv_encode_end,
246  .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
247  .p.pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_YUV420P,
248  AV_PIX_FMT_NONE },
249 };
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
ff_mpv_enc_class
const AVClass ff_mpv_enc_class
Definition: mpegvideo_enc.c:101
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:42
get_bits_diff
static int get_bits_diff(MpegEncContext *s)
Definition: mpegvideoenc.h:139
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
ff_wmv2_encode_picture_header
int ff_wmv2_encode_picture_header(MpegEncContext *s, int picture_number)
Definition: wmv2enc.c:93
ff_msmpeg4_handle_slices
void ff_msmpeg4_handle_slices(MpegEncContext *s)
Definition: msmpeg4enc.c:326
WMV2EncContext::abt_type
int abt_type
Definition: wmv2enc.c:38
mpegvideoenc.h
init_put_bits
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:62
WMV2EncContext::msmpeg4
MSMPEG4EncContext msmpeg4
Definition: wmv2enc.c:33
put_bits
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:221
w
uint8_t w
Definition: llviddspenc.c:38
FFCodec
Definition: codec_internal.h:119
mpegvideo.h
wmv2enc.h
WMV2Context
Definition: wmv2.h:33
ff_msmp4_mb_i_table
const uint16_t ff_msmp4_mb_i_table[64][2]
Definition: msmpeg4data.c:42
wmv2_encode_init
static av_cold int wmv2_encode_init(AVCodecContext *avctx)
Definition: wmv2enc.c:72
ff_mpv_encode_picture
int ff_mpv_encode_picture(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic_arg, int *got_packet)
Definition: mpegvideo_enc.c:1734
WMV2EncContext::top_left_mv_flag
int top_left_mv_flag
Definition: wmv2enc.c:42
FFCodec::p
AVCodec p
The public AVCodec.
Definition: codec_internal.h:123
ff_wmv2_inter_table
const uint32_t(*const [WMV2_INTER_CBP_TABLE_COUNT] ff_wmv2_inter_table)[2]
Definition: msmpeg4data.c:1953
ff_h263_pred_motion
int16_t * ff_h263_pred_motion(MpegEncContext *s, int block, int dir, int *px, int *py)
Definition: h263.c:190
ff_wmv2_encoder
const FFCodec ff_wmv2_encoder
Definition: wmv2enc.c:236
val
static double val(void *priv, double ch)
Definition: aeval.c:77
FF_CODEC_ENCODE_CB
#define FF_CODEC_ENCODE_CB(func)
Definition: codec_internal.h:307
WMV2EncContext
Definition: wmv2enc.c:32
av_cold
#define av_cold
Definition: attributes.h:90
wmv2_get_cbp_table_index
static av_always_inline int wmv2_get_cbp_table_index(MpegEncContext *s, int cbp_index)
Definition: wmv2.h:46
msmpeg4data.h
AVCodecContext::extradata_size
int extradata_size
Definition: avcodec.h:500
s
#define s(width, name)
Definition: cbs_vp9.c:256
init
int(* init)(AVBSFContext *ctx)
Definition: dts2pts_bsf.c:363
WMV2EncContext::j_type_bit
int j_type_bit
Definition: wmv2enc.c:35
AV_CODEC_ID_WMV2
@ AV_CODEC_ID_WMV2
Definition: codec_id.h:70
av_assert0
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
SKIP_TYPE_NONE
#define SKIP_TYPE_NONE
Definition: wmv2.h:27
AV_PIX_FMT_YUV420P
@ AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
WMV2EncContext::per_mb_abt
int per_mb_abt
Definition: wmv2enc.c:39
PutBitContext
Definition: put_bits.h:50
ff_msmpeg4_encode_motion
void ff_msmpeg4_encode_motion(MpegEncContext *s, int mx, int my)
Definition: msmpeg4enc.c:293
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:264
WMV2EncContext::per_mb_rl_bit
int per_mb_rl_bit
Definition: wmv2enc.c:43
AV_PICTURE_TYPE_I
@ AV_PICTURE_TYPE_I
Intra.
Definition: avutil.h:274
ff_mpv_encode_end
av_cold int ff_mpv_encode_end(AVCodecContext *avctx)
Definition: mpegvideo_enc.c:979
wmv2.h
MSMPEG4EncContext
Definition: msmpeg4enc.h:29
codec_internal.h
WMV2EncContext::cbp_table_index
int cbp_table_index
Definition: wmv2enc.c:41
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:269
code
and forward the test the status of outputs and forward it to the corresponding return FFERROR_NOT_READY If the filters stores internally one or a few frame for some it can consider them to be part of the FIFO and delay acknowledging a status change accordingly Example code
Definition: filter_design.txt:178
AVCodecContext::extradata
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:499
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
av_mallocz
void * av_mallocz(size_t size)
Allocate a memory block with alignment suitable for all memory accesses (including vectors if availab...
Definition: mem.c:264
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:211
WMV2EncContext::common
WMV2Context common
Definition: wmv2enc.c:34
avcodec.h
msmpeg4.h
WMV2EncContext::j_type
int j_type
Definition: wmv2enc.c:36
pred
static const float pred[4]
Definition: siprdata.h:259
AV_INPUT_BUFFER_PADDING_SIZE
#define AV_INPUT_BUFFER_PADDING_SIZE
Definition: defs.h:40
AVCodecContext
main external API structure.
Definition: avcodec.h:398
ff_msmpeg4_encode_block
void ff_msmpeg4_encode_block(MpegEncContext *s, int16_t *block, int n)
Definition: msmpeg4enc.c:558
ff_msmpeg4_code012
void ff_msmpeg4_code012(PutBitContext *pb, int n)
Definition: msmpeg4enc.c:68
ff_table_inter_intra
const uint8_t ff_table_inter_intra[4][2]
Definition: msmpeg4data.c:1841
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:65
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
msmpeg4enc.h
ff_mpv_encode_init
av_cold int ff_mpv_encode_init(AVCodecContext *avctx)
Definition: mpegvideo_enc.c:311
flush_put_bits
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:143
WMV2EncContext::abt_flag
int abt_flag
Definition: wmv2enc.c:37
encode_ext_header
static int encode_ext_header(WMV2EncContext *w)
Definition: wmv2enc.c:46
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:425
block
The exact code depends on how similar the blocks are and how related they are to the block
Definition: filter_design.txt:207
MpegEncContext
MpegEncContext.
Definition: mpegvideo.h:62
WMV2EncContext::mspel_bit
int mspel_bit
Definition: wmv2enc.c:40
ff_wmv2_common_init
av_cold void ff_wmv2_common_init(MpegEncContext *s)
Definition: wmv2.c:28
ff_wmv2_encode_mb
void ff_wmv2_encode_mb(MpegEncContext *s, int16_t block[6][64], int motion_x, int motion_y)
Definition: wmv2enc.c:166
ff_msmpeg4_coded_block_pred
int ff_msmpeg4_coded_block_pred(MpegEncContext *s, int n, uint8_t **coded_block_ptr)
Definition: msmpeg4.c:163
h263.h