mjpegenc.h

Go to the documentation of this file.

/*
 * MJPEG encoder
 * Copyright (c) 2000, 2001 Fabrice Bellard
 * Copyright (c) 2003 Alex Beregszaszi
 * Copyright (c) 2003-2004 Michael Niedermayer
 *
 * Support for external huffman table, various fixes (AVID workaround),
 * aspecting, new decode_frame mechanism and apple mjpeg-b support
 *                                  by Alex Beregszaszi
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#ifndef AVCODEC_MJPEGENC_H
#define AVCODEC_MJPEGENC_H

#include <stdint.h>

#include "mjpeg.h"
#include "mpegvideo.h"
#include "put_bits.h"

typedef struct MJpegHuffmanCode {
    // 0=DC lum, 1=DC chrom, 2=AC lum, 3=AC chrom
    uint8_t table_id; 
    uint8_t code;     
    uint16_t mant;    
} MJpegHuffmanCode;

typedef struct MJpegContext {
    //FIXME use array [3] instead of lumi / chroma, for easier addressing
    uint8_t huff_size_dc_luminance[12];     
    uint16_t huff_code_dc_luminance[12];    
    uint8_t huff_size_dc_chrominance[12];   
    uint16_t huff_code_dc_chrominance[12];  

    uint8_t huff_size_ac_luminance[256];    
    uint16_t huff_code_ac_luminance[256];   
    uint8_t huff_size_ac_chrominance[256];  
    uint16_t huff_code_ac_chrominance[256]; 

    uint8_t uni_ac_vlc_len[64 * 64 * 2];
    uint8_t uni_chroma_ac_vlc_len[64 * 64 * 2];

    // Default DC tables have exactly 12 values
    uint8_t bits_dc_luminance[17];   
    uint8_t val_dc_luminance[12];    
    uint8_t bits_dc_chrominance[17]; 
    uint8_t val_dc_chrominance[12];  

    // 8-bit JPEG has max 256 values
    uint8_t bits_ac_luminance[17];   
    uint8_t val_ac_luminance[256];   
    uint8_t bits_ac_chrominance[17]; 
    uint8_t val_ac_chrominance[256]; 

    size_t huff_ncode;               
    MJpegHuffmanCode *huff_buffer;   
} MJpegContext;

enum HuffmanTableOption {
    HUFFMAN_TABLE_DEFAULT = 0, 
    HUFFMAN_TABLE_OPTIMAL = 1, 
    NB_HUFFMAN_TABLE_OPTION = 2
};

static inline void put_marker(PutBitContext *p, enum JpegMarker code)
{
    put_bits(p, 8, 0xff);
    put_bits(p, 8, code);
}

int  ff_mjpeg_encode_init(MpegEncContext *s);
void ff_mjpeg_encode_close(MpegEncContext *s);
void ff_mjpeg_encode_mb(MpegEncContext *s, int16_t block[12][64]);

#endif /* AVCODEC_MJPEGENC_H */