1/* 2 * LCL (LossLess Codec Library) Codec 3 * Copyright (c) 2002-2004 Roberto Togni 4 * 5 * This file is part of FFmpeg. 6 * 7 * FFmpeg is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * FFmpeg is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with FFmpeg; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 */ 21 22/** 23 * @file 24 * LCL (LossLess Codec Library) Video Codec 25 * Decoder for MSZH and ZLIB codecs 26 * Experimental encoder for ZLIB RGB24 27 * 28 * Fourcc: MSZH, ZLIB 29 * 30 * Original Win32 dll: 31 * Ver2.23 By Kenji Oshima 2000.09.20 32 * avimszh.dll, avizlib.dll 33 * 34 * A description of the decoding algorithm can be found here: 35 * http://www.pcisys.net/~melanson/codecs 36 * 37 * Supports: BGR24 (RGB 24bpp) 38 */ 39 40#include <stdio.h> 41#include <stdlib.h> 42 43#include "libavutil/avassert.h" 44#include "avcodec.h" 45#include "codec_internal.h" 46#include "encode.h" 47#include "lcl.h" 48#include "zlib_wrapper.h" 49#include "libavutil/internal.h" 50#include "libavutil/mem.h" 51 52#include <zlib.h> 53 54typedef struct LclEncContext { 55 56 AVCodecContext *avctx; 57 58 // Image type 59 int imgtype; 60 // Compression type 61 int compression; 62 // Flags 63 int flags; 64 FFZStream zstream; 65} LclEncContext; 66 67static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, 68 const AVFrame *p, int *got_packet) 69{ 70 LclEncContext *c = avctx->priv_data; 71 z_stream *const zstream = &c->zstream.zstream; 72 int i, ret; 73 int zret; // Zlib return code 74 int max_size = deflateBound(zstream, avctx->width * avctx->height * 3); 75 76 if ((ret = ff_alloc_packet(avctx, pkt, max_size)) < 0) 77 return ret; 78 79 if(avctx->pix_fmt != AV_PIX_FMT_BGR24){ 80 av_log(avctx, AV_LOG_ERROR, "Format not supported!\n"); 81 return -1; 82 } 83 84 zret = deflateReset(zstream); 85 if (zret != Z_OK) { 86 av_log(avctx, AV_LOG_ERROR, "Deflate reset error: %d\n", zret); 87 return -1; 88 } 89 zstream->next_out = pkt->data; 90 zstream->avail_out = pkt->size; 91 92 for(i = avctx->height - 1; i >= 0; i--) { 93 zstream->next_in = p->data[0] + p->linesize[0] * i; 94 zstream->avail_in = avctx->width * 3; 95 zret = deflate(zstream, Z_NO_FLUSH); 96 if (zret != Z_OK) { 97 av_log(avctx, AV_LOG_ERROR, "Deflate error: %d\n", zret); 98 return -1; 99 } 100 } 101 zret = deflate(zstream, Z_FINISH); 102 if (zret != Z_STREAM_END) { 103 av_log(avctx, AV_LOG_ERROR, "Deflate error: %d\n", zret); 104 return -1; 105 } 106 107 pkt->size = zstream->total_out; 108 *got_packet = 1; 109 110 return 0; 111} 112 113static av_cold int encode_init(AVCodecContext *avctx) 114{ 115 LclEncContext *c = avctx->priv_data; 116 117 c->avctx= avctx; 118 119 av_assert0(avctx->width && avctx->height); 120 121 avctx->extradata = av_mallocz(8 + AV_INPUT_BUFFER_PADDING_SIZE); 122 if (!avctx->extradata) 123 return AVERROR(ENOMEM); 124 125 c->compression = avctx->compression_level == FF_COMPRESSION_DEFAULT ? 126 COMP_ZLIB_NORMAL : 127 av_clip(avctx->compression_level, 0, 9); 128 c->flags = 0; 129 c->imgtype = IMGTYPE_RGB24; 130 avctx->bits_per_coded_sample= 24; 131 132 avctx->extradata[0]= 4; 133 avctx->extradata[1]= 0; 134 avctx->extradata[2]= 0; 135 avctx->extradata[3]= 0; 136 avctx->extradata[4]= c->imgtype; 137 avctx->extradata[5]= c->compression; 138 avctx->extradata[6]= c->flags; 139 avctx->extradata[7]= CODEC_ZLIB; 140 c->avctx->extradata_size= 8; 141 142 return ff_deflate_init(&c->zstream, c->compression, avctx); 143} 144 145static av_cold int encode_end(AVCodecContext *avctx) 146{ 147 LclEncContext *c = avctx->priv_data; 148 149 ff_deflate_end(&c->zstream); 150 151 return 0; 152} 153 154const FFCodec ff_zlib_encoder = { 155 .p.name = "zlib", 156 .p.long_name = NULL_IF_CONFIG_SMALL("LCL (LossLess Codec Library) ZLIB"), 157 .p.type = AVMEDIA_TYPE_VIDEO, 158 .p.id = AV_CODEC_ID_ZLIB, 159 .priv_data_size = sizeof(LclEncContext), 160 .init = encode_init, 161 FF_CODEC_ENCODE_CB(encode_frame), 162 .close = encode_end, 163 .p.capabilities = AV_CODEC_CAP_FRAME_THREADS, 164 .p.pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_BGR24, AV_PIX_FMT_NONE }, 165 .caps_internal = FF_CODEC_CAP_INIT_THREADSAFE | 166 FF_CODEC_CAP_INIT_CLEANUP, 167}; 168