1/*
2 * Copyright (c) 2002-2014 Michael Niedermayer <michaelni@gmx.at>
3 *
4 * see http://www.pcisys.net/~melanson/codecs/huffyuv.txt for a description of
5 * the algorithm used
6 *
7 * This file is part of FFmpeg.
8 *
9 * FFmpeg is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
14 * FFmpeg is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with FFmpeg; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 */
23
24/**
25 * @file
26 * huffyuv codec for libavcodec.
27 */
28
29#ifndef AVCODEC_HUFFYUV_H
30#define AVCODEC_HUFFYUV_H
31
32#include <stdint.h>
33
34#include "avcodec.h"
35#include "bswapdsp.h"
36#include "get_bits.h"
37#include "huffyuvdsp.h"
38#include "huffyuvencdsp.h"
39#include "put_bits.h"
40#include "lossless_videodsp.h"
41
42#define VLC_BITS 12
43
44#define MAX_BITS 16
45#define MAX_N (1<<MAX_BITS)
46#define MAX_VLC_N 16384
47
48typedef enum Predictor {
49    LEFT = 0,
50    PLANE,
51    MEDIAN,
52} Predictor;
53
54typedef struct HYuvContext {
55    AVCodecContext *avctx;
56    Predictor predictor;
57    GetBitContext gb;
58    PutBitContext pb;
59    int interlaced;
60    int decorrelate;
61    int bitstream_bpp;
62    int version;
63    int yuy2;                               //use yuy2 instead of 422P
64    int bgr32;                              //use bgr32 instead of bgr24
65    int bps;
66    int n;                                  // 1<<bps
67    int vlc_n;                              // number of vlc codes (FFMIN(1<<bps, MAX_VLC_N))
68    int alpha;
69    int chroma;
70    int yuv;
71    int chroma_h_shift;
72    int chroma_v_shift;
73    int width, height;
74    int flags;
75    int context;
76    int picture_number;
77    int last_slice_end;
78    uint8_t *temp[3];
79    uint16_t *temp16[3];                    ///< identical to temp but 16bit type
80    uint64_t stats[4][MAX_VLC_N];
81    uint8_t len[4][MAX_VLC_N];
82    uint32_t bits[4][MAX_VLC_N];
83    uint32_t pix_bgr_map[1<<VLC_BITS];
84    VLC vlc[8];                             //Y,U,V,A,YY,YU,YV,AA
85    uint8_t *bitstream_buffer;
86    unsigned int bitstream_buffer_size;
87    BswapDSPContext bdsp;
88    HuffYUVDSPContext hdsp;
89    HuffYUVEncDSPContext hencdsp;
90    LLVidDSPContext llviddsp;
91} HYuvContext;
92
93void ff_huffyuv_common_init(AVCodecContext *s);
94void ff_huffyuv_common_end(HYuvContext *s);
95int  ff_huffyuv_alloc_temp(HYuvContext *s);
96int ff_huffyuv_generate_bits_table(uint32_t *dst, const uint8_t *len_table, int n);
97
98#endif /* AVCODEC_HUFFYUV_H */
99