avcodec/nvenc: support external context in sw mode
[ffmpeg.git] / libavcodec / wmv2.h
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 #ifndef AVCODEC_WMV2_H
22 #define AVCODEC_WMV2_H
23
24 #include "avcodec.h"
25 #include "intrax8.h"
26 #include "mpegvideo.h"
27 #include "wmv2dsp.h"
28
29 #define SKIP_TYPE_NONE 0
30 #define SKIP_TYPE_MPEG 1
31 #define SKIP_TYPE_ROW  2
32 #define SKIP_TYPE_COL  3
33
34
35 typedef struct Wmv2Context {
36     MpegEncContext s;
37     IntraX8Context x8;
38     WMV2DSPContext wdsp;
39     int j_type_bit;
40     int j_type;
41     int abt_flag;
42     int abt_type;
43     int abt_type_table[6];
44     int per_mb_abt;
45     int per_block_abt;
46     int mspel_bit;
47     int cbp_table_index;
48     int top_left_mv_flag;
49     int per_mb_rl_bit;
50     int skip_type;
51     int hshift;
52
53     ScanTable abt_scantable[2];
54     DECLARE_ALIGNED(16, int16_t, abt_block2)[6][64];
55 } Wmv2Context;
56
57 void ff_wmv2_common_init(Wmv2Context *w);
58
59 int ff_wmv2_decode_mb(MpegEncContext *s, int16_t block[6][64]);
60 int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
61 void ff_wmv2_encode_mb(MpegEncContext * s, int16_t block[6][64],
62                        int motion_x, int motion_y);
63 int ff_wmv2_decode_picture_header(MpegEncContext * s);
64 int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
65 void ff_wmv2_add_mb(MpegEncContext *s, int16_t block[6][64],
66                     uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
67
68 void ff_mspel_motion(MpegEncContext *s,
69                      uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
70                      uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
71                      int motion_x, int motion_y, int h);
72
73
74 static av_always_inline int wmv2_get_cbp_table_index(MpegEncContext *s, int cbp_index)
75 {
76     static const uint8_t map[3][3] = {
77         { 0, 2, 1 },
78         { 1, 0, 2 },
79         { 2, 1, 0 },
80     };
81
82     return map[(s->qscale > 10) + (s->qscale > 20)][cbp_index];
83 }
84
85 #endif /* AVCODEC_WMV2_H */