diff options
Diffstat (limited to 'debian/transcode/transcode-1.1.7/import/import_lzo.c')
| -rw-r--r-- | debian/transcode/transcode-1.1.7/import/import_lzo.c | 275 |
1 files changed, 275 insertions, 0 deletions
diff --git a/debian/transcode/transcode-1.1.7/import/import_lzo.c b/debian/transcode/transcode-1.1.7/import/import_lzo.c new file mode 100644 index 00000000..5e5fe3f1 --- /dev/null +++ b/debian/transcode/transcode-1.1.7/import/import_lzo.c @@ -0,0 +1,275 @@ +/* + * import_lzo.c + * + * Copyright (C) Thomas Oestreich - October 2002 + * + * This file is part of transcode, a video stream processing tool + * + * transcode is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2, or (at your option) + * any later version. + * + * transcode 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 General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with GNU Make; see the file COPYING. If not, write to + * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ + +#define MOD_NAME "import_lzo.so" +#define MOD_VERSION "v0.1.0 (2005-10-16)" +#define MOD_CODEC "(video) LZO" + +#include "transcode.h" +#include "magic.h" +#include "libtc/tc_lzo.h" + +static int verbose_flag = TC_QUIET; +static int capability_flag = TC_CAP_PCM | TC_CAP_YUV | TC_CAP_RGB | + TC_CAP_AUD | TC_CAP_VID; + +#define MOD_PRE lzo +#include "import_def.h" + + +static avi_t *avifile1=NULL; +static avi_t *avifile2=NULL; + +static uint32_t video_codec; +static int audio_codec; +static int aframe_count=0, vframe_count=0; + +#define BUFFER_SIZE SIZE_RGB_FRAME<<1 + +static int r; +static lzo_byte *out; +static lzo_byte *wrkmem; +static lzo_uint out_len; + +static int done_seek=0; + +/* ------------------------------------------------------------ + * + * open stream + * + * ------------------------------------------------------------*/ + +MOD_open +{ + int width=0, height=0; + double fps=0; + char *codec=NULL; + + param->fd = NULL; + + if(param->flag == TC_AUDIO) return(TC_IMPORT_ERROR); + + if(param->flag == TC_VIDEO) { + + param->fd = NULL; + + if(avifile2==NULL) { + if(vob->nav_seek_file) { + if(NULL == (avifile2 = AVI_open_input_indexfile(vob->video_in_file, + 0,vob->nav_seek_file))){ + AVI_print_error("avi open error"); + return(TC_IMPORT_ERROR); + } + } else { + if(NULL == (avifile2 = AVI_open_input_file(vob->video_in_file,1))){ + AVI_print_error("avi open error"); + return(TC_IMPORT_ERROR); + } + } + } + + // vob->offset contains the last keyframe + if (!done_seek && vob->vob_offset>0) { + AVI_set_video_position(avifile2, vob->vob_offset); + done_seek=1; + } + + //read all video parameter from input file + width = AVI_video_width(avifile2); + height = AVI_video_height(avifile2); + + fps = AVI_frame_rate(avifile2); + codec = AVI_video_compressor(avifile2); + + if (strcmp(codec,"LZO1") == 0) { + video_codec = TC_CODEC_LZO1; + } else if (strcmp(codec,"LZO2") == 0) { + video_codec = TC_CODEC_LZO2; + } else { + tc_log_warn(MOD_NAME, "Unsupported video codec %s", codec); + return(TC_IMPORT_ERROR); + } + + tc_log_info(MOD_NAME, "codec=%s, fps=%6.3f, width=%d, height=%d", + codec, fps, width, height); + + /* + * Step 1: initialize the LZO library + */ + + if (lzo_init() != LZO_E_OK) { + tc_log_warn(MOD_NAME, "lzo_init() failed"); + return(TC_IMPORT_ERROR); + } + + wrkmem = (lzo_bytep) lzo_malloc(LZO1X_1_MEM_COMPRESS); + out = (lzo_bytep) lzo_malloc(BUFFER_SIZE); + + if (wrkmem == NULL || out == NULL) { + tc_log_warn(MOD_NAME, "out of memory"); + return(TC_IMPORT_ERROR); + } + + return(TC_IMPORT_OK); + } + + return(TC_IMPORT_ERROR); +} + + +/* ------------------------------------------------------------ + * + * decode stream + * + * ------------------------------------------------------------*/ + +MOD_decode +{ + + int key; + lzo_uint size; + long bytes_read=0; + + if(param->flag == TC_VIDEO) { + // If we are using tccat, then do nothing here + if (param->fd != NULL) { + return(TC_IMPORT_OK); + } + + out_len = AVI_read_frame(avifile2, out, &key); + + if(verbose & TC_STATS && key) + tc_log_info(MOD_NAME, "keyframe %d", vframe_count); + + if(out_len<=0) { + if(verbose & TC_DEBUG) AVI_print_error("AVI read video frame"); + return(TC_IMPORT_ERROR); + } + + if (video_codec == TC_CODEC_LZO1) { + r = lzo1x_decompress(out, out_len, param->buffer, &size, wrkmem); + } else { + tc_lzo_header_t *h = (tc_lzo_header_t *)out; + uint8_t *compdata = out + sizeof(*h); + int compsize = out_len - sizeof(*h); + if (h->magic != video_codec) { + tc_log_warn(MOD_NAME, "frame with invalid magic 0x%08X", h->magic); + return (TC_IMPORT_ERROR); + } + if (h->flags & TC_LZO_NOT_COMPRESSIBLE) { + ac_memcpy(param->buffer, compdata, compsize); + size = compsize; + r = LZO_E_OK; + } else { + r = lzo1x_decompress(compdata, compsize, param->buffer, &size, wrkmem); + } + } + + if (r == LZO_E_OK) { + if(verbose & TC_DEBUG) + tc_log_info(MOD_NAME, "decompressed %lu bytes into %lu bytes", + (long) out_len, (long) param->size); + } else { + /* this should NEVER happen */ + tc_log_warn(MOD_NAME, "internal error - decompression failed: %d", r); + return(TC_IMPORT_ERROR); + } + + param->size = size; + //transcode v.0.5.0-pre8 addition + if(key) param->attributes |= TC_FRAME_IS_KEYFRAME; + + ++vframe_count; + + return(TC_IMPORT_OK); + } + + if(param->flag == TC_AUDIO) { + + switch(audio_codec) { + + case CODEC_RAW: + + bytes_read = AVI_audio_size(avifile1, aframe_count); + + if(bytes_read<=0) { + if(verbose & TC_DEBUG) AVI_print_error("AVI audio read frame"); + return(TC_IMPORT_ERROR); + } + + if(AVI_read_audio(avifile1, param->buffer, bytes_read) < 0) { + AVI_print_error("AVI audio read frame"); + return(TC_IMPORT_ERROR); + } + + param->size = bytes_read; + ++aframe_count; + + break; + + default: + + bytes_read = AVI_read_audio(avifile1, param->buffer, param->size); + + if(bytes_read<0) { + if(verbose & TC_DEBUG) AVI_print_error("AVI audio read frame"); + return(TC_IMPORT_ERROR); + } + + if(bytes_read < param->size) param->size=bytes_read; + } + + return(TC_IMPORT_OK); + } + + return(TC_IMPORT_ERROR); +} + +/* ------------------------------------------------------------ + * + * close stream + * + * ------------------------------------------------------------*/ + +MOD_close +{ + + if(param->fd != NULL) pclose(param->fd); + + if(param->flag == TC_AUDIO) return(TC_IMPORT_ERROR); + + if(param->flag == TC_VIDEO) { + + lzo_free(wrkmem); + lzo_free(out); + + if(avifile2!=NULL) { + AVI_close(avifile2); + avifile2=NULL; + } + done_seek = 0; + return(TC_IMPORT_OK); + } + + return(TC_IMPORT_ERROR); +} |