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 "libavutil/mem.h"
22 #include "avcodec.h"
23 #include "codec_internal.h"
24 #include "h263.h"
25 #include "mpegvideo.h"
26 #include "mpegvideoenc.h"
27 #include "msmpeg4.h"
28 #include "msmpeg4enc.h"
29 #include "msmpeg4data.h"
30 #include "msmpeg4_vc1_data.h"
31 #include "put_bits.h"
32 #include "wmv2.h"
33 
34 #define WMV2_EXTRADATA_SIZE 4
35 
36 typedef struct WMV2EncContext {
37  MSMPEG4EncContext msmpeg4;
38  int j_type_bit;
39  int j_type;
40  int abt_flag;
41  int abt_type;
42  int per_mb_abt;
43  int mspel_bit;
44  int cbp_table_index;
45  int top_left_mv_flag;
46  int per_mb_rl_bit;
47 } WMV2EncContext;
48 
49 static int encode_ext_header(WMV2EncContext *w)
50 {
51  MPVEncContext *const s = &w->msmpeg4.m.s;
52  PutBitContext pb;
53  int code;
54 
55  init_put_bits(&pb, s->c.avctx->extradata, WMV2_EXTRADATA_SIZE);
56 
57  put_bits(&pb, 5, s->c.avctx->time_base.den / s->c.avctx->time_base.num); // yes 29.97 -> 29
58  put_bits(&pb, 11, FFMIN(w->msmpeg4.m.bit_rate / 1024, 2047));
59 
60  put_bits(&pb, 1, w->mspel_bit = 1);
61  put_bits(&pb, 1, s->loop_filter);
62  put_bits(&pb, 1, w->abt_flag = 1);
63  put_bits(&pb, 1, w->j_type_bit = 1);
64  put_bits(&pb, 1, w->top_left_mv_flag = 0);
65  put_bits(&pb, 1, w->per_mb_rl_bit = 1);
66  put_bits(&pb, 3, code = 1);
67 
68  flush_put_bits(&pb);
69 
70  s->slice_height = s->c.mb_height / code;
71 
72  return 0;
73 }
74 
75 static int wmv2_encode_picture_header(MPVMainEncContext *const m)
76 {
77  WMV2EncContext *const w = (WMV2EncContext *) m;
78  MSMPEG4EncContext *const ms = &w->msmpeg4;
79  MPVEncContext *const s = &m->s;
80 
81  put_bits_assume_flushed(&s->pb);
82 
83  put_bits(&s->pb, 1, s->c.pict_type - 1);
84  if (s->c.pict_type == AV_PICTURE_TYPE_I)
85  put_bits(&s->pb, 7, 0);
86  put_bits(&s->pb, 5, s->c.qscale);
87 
88  ms->dc_table_index = 1;
89  ms->mv_table_index = 1; /* only if P-frame */
90  ms->per_mb_rl_table = 0;
91  s->c.mspel = 0;
92  w->per_mb_abt = 0;
93  w->abt_type = 0;
94  w->j_type = 0;
95 
96  av_assert0(s->flipflop_rounding);
97 
98  if (s->c.pict_type == AV_PICTURE_TYPE_I) {
99  av_assert0(s->c.no_rounding == 1);
100  if (w->j_type_bit)
101  put_bits(&s->pb, 1, w->j_type);
102 
103  if (w->per_mb_rl_bit)
104  put_bits(&s->pb, 1, ms->per_mb_rl_table);
105 
106  if (!ms->per_mb_rl_table) {
107  ff_msmpeg4_code012(&s->pb, ms->rl_chroma_table_index);
108  ff_msmpeg4_code012(&s->pb, ms->rl_table_index);
109  }
110 
111  put_bits(&s->pb, 1, ms->dc_table_index);
112 
113  s->c.inter_intra_pred = 0;
114  } else {
115  int cbp_index;
116 
117  put_bits(&s->pb, 2, SKIP_TYPE_NONE);
118 
119  ff_msmpeg4_code012(&s->pb, cbp_index = 0);
120  w->cbp_table_index = wmv2_get_cbp_table_index(s->c.qscale, cbp_index);
121 
122  if (w->mspel_bit)
123  put_bits(&s->pb, 1, s->c.mspel);
124 
125  if (w->abt_flag) {
126  put_bits(&s->pb, 1, w->per_mb_abt ^ 1);
127  if (!w->per_mb_abt)
128  ff_msmpeg4_code012(&s->pb, w->abt_type);
129  }
130 
131  if (w->per_mb_rl_bit)
132  put_bits(&s->pb, 1, ms->per_mb_rl_table);
133 
134  if (!ms->per_mb_rl_table) {
135  ff_msmpeg4_code012(&s->pb, ms->rl_table_index);
136  ms->rl_chroma_table_index = ms->rl_table_index;
137  }
138  put_bits(&s->pb, 1, ms->dc_table_index);
139  put_bits(&s->pb, 1, ms->mv_table_index);
140 
141  s->c.inter_intra_pred = 0; // (s->c.width * s->c.height < 320 * 240 && m->bit_rate <= II_BITRATE);
142  }
143  s->esc3_level_length = 0;
144  ms->esc3_run_length = 0;
145 
146  return 0;
147 }
148 
149 /* Nearly identical to wmv1 but that is just because we do not use the
150  * useless M$ crap features. It is duplicated here in case someone wants
151  * to add support for these crap features. */
152 static void wmv2_encode_mb(MPVEncContext *const s, int16_t block[][64],
153  int motion_x, int motion_y)
154 {
155  WMV2EncContext *const w = (WMV2EncContext *) s;
156  int cbp, coded_cbp, i;
157  int pred_x, pred_y;
158  uint8_t *coded_block;
159 
160  ff_msmpeg4_handle_slices(s);
161 
162  if (!s->c.mb_intra) {
163  /* compute cbp */
164  cbp = 0;
165  for (i = 0; i < 6; i++)
166  if (s->c.block_last_index[i] >= 0)
167  cbp |= 1 << (5 - i);
168 
169  put_bits(&s->pb,
170  ff_wmv2_inter_table[w->cbp_table_index][cbp + 64][1],
171  ff_wmv2_inter_table[w->cbp_table_index][cbp + 64][0]);
172 
173  s->misc_bits += get_bits_diff(s);
174  /* motion vector */
175  ff_h263_pred_motion(&s->c, 0, 0, &pred_x, &pred_y);
176  ff_msmpeg4_encode_motion(&w->msmpeg4, motion_x - pred_x,
177  motion_y - pred_y);
178  s->mv_bits += get_bits_diff(s);
179  } else {
180  /* compute cbp */
181  cbp = 0;
182  coded_cbp = 0;
183  for (i = 0; i < 6; i++) {
184  int val = (s->c.block_last_index[i] >= 1);
185 
186  cbp |= val << (5 - i);
187  if (i < 4) {
188  /* predict value for close blocks only for luma */
189  int pred = ff_msmpeg4_coded_block_pred(&s->c, i, &coded_block);
190  *coded_block = val;
191  val = val ^ pred;
192  }
193  coded_cbp |= val << (5 - i);
194  }
195 
196  if (s->c.pict_type == AV_PICTURE_TYPE_I)
197  put_bits(&s->pb,
198  ff_msmp4_mb_i_table[coded_cbp][1],
199  ff_msmp4_mb_i_table[coded_cbp][0]);
200  else
201  put_bits(&s->pb,
202  ff_wmv2_inter_table[w->cbp_table_index][cbp][1],
203  ff_wmv2_inter_table[w->cbp_table_index][cbp][0]);
204  put_bits(&s->pb, 1, 0); /* no AC prediction yet */
205  if (s->c.inter_intra_pred) {
206  s->c.h263_aic_dir = 0;
207  put_bits(&s->pb,
208  ff_table_inter_intra[s->c.h263_aic_dir][1],
209  ff_table_inter_intra[s->c.h263_aic_dir][0]);
210  }
211  s->misc_bits += get_bits_diff(s);
212  }
213 
214  for (i = 0; i < 6; i++)
215  ff_msmpeg4_encode_block(s, block[i], i);
216  if (s->c.mb_intra)
217  s->i_tex_bits += get_bits_diff(s);
218  else
219  s->p_tex_bits += get_bits_diff(s);
220 }
221 
222 static av_cold int wmv2_encode_init(AVCodecContext *avctx)
223 {
224  WMV2EncContext *const w = avctx->priv_data;
225  MPVEncContext *const s = &w->msmpeg4.m.s;
226  int ret;
227 
228  w->msmpeg4.m.encode_picture_header = wmv2_encode_picture_header;
229  s->encode_mb = wmv2_encode_mb;
230 
231  ret = ff_mpv_encode_init(avctx);
232  if (ret < 0)
233  return ret;
234 
235  avctx->extradata_size = WMV2_EXTRADATA_SIZE;
236  avctx->extradata = av_mallocz(avctx->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
237  if (!avctx->extradata)
238  return AVERROR(ENOMEM);
239 
240  encode_ext_header(w);
241 
242  return 0;
243 }
244 
245 const FFCodec ff_wmv2_encoder = {
246  .p.name = "wmv2",
247  CODEC_LONG_NAME("Windows Media Video 8"),
248  .p.type = AVMEDIA_TYPE_VIDEO,
249  .p.id = AV_CODEC_ID_WMV2,
250  .p.priv_class = &ff_mpv_enc_class,
251  .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
252  .priv_data_size = sizeof(WMV2EncContext),
253  .init = wmv2_encode_init,
254  FF_CODEC_ENCODE_CB(ff_mpv_encode_picture),
255  .close = ff_mpv_encode_end,
256  .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
257  .color_ranges = AVCOL_RANGE_MPEG,
258  CODEC_PIXFMTS(AV_PIX_FMT_YUV420P),
259 };
CODEC_PIXFMTS
#define CODEC_PIXFMTS(...)
Definition: codec_internal.h:392
ff_mpv_enc_class
const AVClass ff_mpv_enc_class
Definition: mpegvideo_enc.c:101
MPVEncContext
Definition: mpegvideoenc.h:46
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
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
wmv2_get_cbp_table_index
static av_always_inline int wmv2_get_cbp_table_index(int qscale, int cbp_index)
Definition: wmv2.h:33
WMV2EncContext::abt_type
int abt_type
Definition: wmv2enc.c:41
mpegvideoenc.h
ff_msmpeg4_handle_slices
void ff_msmpeg4_handle_slices(MPVEncContext *const s)
Definition: msmpeg4enc.c:332
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:37
put_bits
static void put_bits(Jpeg2000EncoderContext *s, int val, int n)
put n times val bit
Definition: j2kenc.c:154
FFCodec
Definition: codec_internal.h:127
mpegvideo.h
wmv2_encode_init
static av_cold int wmv2_encode_init(AVCodecContext *avctx)
Definition: wmv2enc.c:222
ff_mpv_encode_picture
int ff_mpv_encode_picture(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic_arg, int *got_packet)
Definition: mpegvideo_enc.c:1936
WMV2EncContext::top_left_mv_flag
int top_left_mv_flag
Definition: wmv2enc.c:45
FFCodec::p
AVCodec p
The public AVCodec.
Definition: codec_internal.h:131
ff_wmv2_inter_table
const uint32_t(*const [WMV2_INTER_CBP_TABLE_COUNT] ff_wmv2_inter_table)[2]
Definition: msmpeg4data.c:1129
ff_h263_pred_motion
int16_t * ff_h263_pred_motion(MpegEncContext *s, int block, int dir, int *px, int *py)
Definition: h263.c:182
ff_wmv2_encoder
const FFCodec ff_wmv2_encoder
Definition: wmv2enc.c:245
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:359
WMV2EncContext
Definition: wmv2enc.c:36
av_cold
#define av_cold
Definition: attributes.h:106
msmpeg4data.h
AVCodecContext::extradata_size
int extradata_size
Definition: avcodec.h:523
s
#define s(width, name)
Definition: cbs_vp9.c:198
WMV2EncContext::j_type_bit
int j_type_bit
Definition: wmv2enc.c:38
MSMPEG4EncContext::esc3_run_length
int esc3_run_length
Definition: msmpeg4enc.h:38
AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE
#define AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE
This encoder can reorder user opaque values from input AVFrames and return them with corresponding ou...
Definition: codec.h:144
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:41
SKIP_TYPE_NONE
#define SKIP_TYPE_NONE
Definition: wmv2.h:28
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:73
WMV2EncContext::per_mb_abt
int per_mb_abt
Definition: wmv2enc.c:42
av_mallocz
#define av_mallocz(s)
Definition: tableprint_vlc.h:31
PutBitContext
Definition: put_bits.h:50
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:332
WMV2EncContext::per_mb_rl_bit
int per_mb_rl_bit
Definition: wmv2enc.c:46
MSMPEG4EncContext::per_mb_rl_table
int per_mb_rl_table
Definition: msmpeg4enc.h:37
MSMPEG4EncContext::dc_table_index
int dc_table_index
Definition: msmpeg4enc.h:35
AV_PICTURE_TYPE_I
@ AV_PICTURE_TYPE_I
Intra.
Definition: avutil.h:278
ff_msmp4_mb_i_table
const uint16_t ff_msmp4_mb_i_table[64][2]
Definition: msmpeg4_vc1_data.c:63
ff_mpv_encode_end
av_cold int ff_mpv_encode_end(AVCodecContext *avctx)
Definition: mpegvideo_enc.c:1116
MPVMainEncContext
Definition: mpegvideoenc.h:202
wmv2.h
MSMPEG4EncContext
Definition: msmpeg4enc.h:29
wmv2_encode_mb
static void wmv2_encode_mb(MPVEncContext *const s, int16_t block[][64], int motion_x, int motion_y)
Definition: wmv2enc.c:152
init
int(* init)(AVBSFContext *ctx)
Definition: dts2pts.c:550
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
codec_internal.h
put_bits_assume_flushed
static void put_bits_assume_flushed(const PutBitContext *s)
Inform the compiler that a PutBitContext is flushed (i.e.
Definition: put_bits.h:82
WMV2EncContext::cbp_table_index
int cbp_table_index
Definition: wmv2enc.c:44
WMV2_EXTRADATA_SIZE
#define WMV2_EXTRADATA_SIZE
Definition: wmv2enc.c:34
ff_msmpeg4_encode_motion
void ff_msmpeg4_encode_motion(MSMPEG4EncContext *const ms, int mx, int my)
Definition: msmpeg4enc.c:306
get_bits_diff
static int get_bits_diff(MPVEncContext *s)
Definition: mpegvideoenc.h:411
i
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:256
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
Out-of-band global headers that may be used by some codecs.
Definition: avcodec.h:522
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:179
AVCOL_RANGE_MPEG
@ AVCOL_RANGE_MPEG
Narrow or limited range content.
Definition: pixfmt.h:760
avcodec.h
MSMPEG4EncContext::rl_table_index
int rl_table_index
Definition: msmpeg4enc.h:33
msmpeg4.h
WMV2EncContext::j_type
int j_type
Definition: wmv2enc.c:39
ret
ret
Definition: filter_design.txt:187
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:439
ff_msmpeg4_code012
void ff_msmpeg4_code012(PutBitContext *pb, int n)
Definition: msmpeg4enc.c:81
wmv2_encode_picture_header
static int wmv2_encode_picture_header(MPVMainEncContext *const m)
Definition: wmv2enc.c:75
ff_table_inter_intra
const uint8_t ff_table_inter_intra[4][2]
Definition: msmpeg4data.c:1017
MSMPEG4EncContext::mv_table_index
int mv_table_index
Definition: msmpeg4enc.h:32
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:200
msmpeg4enc.h
mem.h
msmpeg4_vc1_data.h
ff_mpv_encode_init
av_cold int ff_mpv_encode_init(AVCodecContext *avctx)
Definition: mpegvideo_enc.c:553
flush_put_bits
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:153
ff_msmpeg4_encode_block
void ff_msmpeg4_encode_block(MPVEncContext *const s, int16_t *block, int n)
Definition: msmpeg4enc.c:555
w
uint8_t w
Definition: llvidencdsp.c:39
WMV2EncContext::abt_flag
int abt_flag
Definition: wmv2enc.c:40
encode_ext_header
static int encode_ext_header(WMV2EncContext *w)
Definition: wmv2enc.c:49
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:466
block
The exact code depends on how similar the blocks are and how related they are to the block
Definition: filter_design.txt:207
put_bits.h
WMV2EncContext::mspel_bit
int mspel_bit
Definition: wmv2enc.c:43
MPVMainEncContext::s
MPVEncContext s
The main slicecontext.
Definition: mpegvideoenc.h:203
MSMPEG4EncContext::rl_chroma_table_index
int rl_chroma_table_index
Definition: msmpeg4enc.h:34
ff_msmpeg4_coded_block_pred
int ff_msmpeg4_coded_block_pred(MpegEncContext *s, int n, uint8_t **coded_block_ptr)
Definition: msmpeg4.c:164
h263.h
Generated on Sat Jan 3 2026 19:23:10 for FFmpeg by   doxygen 1.8.17