v308.c

这个库实现了录象功能

/*******************************************************************************
 v308.c

 libquicktime - A library for reading and writing quicktime/avi/mp4 files.
 http://libquicktime.sourceforge.net

 Copyright (C) 2002 Heroine Virtual Ltd.
 Copyright (C) 2002-2007 Members of the libquicktime project.

 This library 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.

 This library 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 this library; if not, write to the Free Software Foundation, Inc., 51
 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*******************************************************************************/ 

#include "lqt_private.h"
#include "workarounds.h"
#include "videocodec.h"
#include <quicktime/colormodels.h>
#include <stdlib.h>

typedef struct
  {
  uint8_t *buffer;
  int buffer_alloc;
  } quicktime_v308_codec_t;

static int delete_codec(quicktime_video_map_t *vtrack)
{
	quicktime_v308_codec_t *codec;

	codec = ((quicktime_codec_t*)vtrack->codec)->priv;
	if(codec->buffer) free(codec->buffer);
	free(codec);
	return 0;
}





static int decode(quicktime_t *file, unsigned char **row_pointers, int track)
{
        uint8_t *in_ptr, *out_y, *out_u, *out_v;
        int i, j;
	int64_t bytes;
	int result = 0;
	quicktime_video_map_t *vtrack = &(file->vtracks[track]);
	quicktime_v308_codec_t *codec = ((quicktime_codec_t*)vtrack->codec)->priv;
	int width = vtrack->track->tkhd.track_width;
	int height = vtrack->track->tkhd.track_height;

        if(!row_pointers)
          {
          //          vtrack->stream_cmodel = BC_VYU888;
          vtrack->stream_cmodel = BC_YUV444P;
          return 0;
          }

        bytes = lqt_read_video_frame(file, &codec->buffer, &codec->buffer_alloc,
                                     vtrack->current_position, NULL, track);

        if(bytes <= 0)
          return -1;
        
        in_ptr = codec->buffer;
	for(i = 0; i < height; i++)
          {
          out_y = row_pointers[0] + i * file->vtracks[track].stream_row_span;
          out_u = row_pointers[1] + i * file->vtracks[track].stream_row_span_uv;
          out_v = row_pointers[2] + i * file->vtracks[track].stream_row_span_uv;
          for(j = 0; j < width; j++)
            {
            *out_y = in_ptr[1]; /* Y */
            *out_u = in_ptr[2]; /* U */
            *out_v = in_ptr[0]; /* V */

            out_y++;
            out_u++;
            out_v++;
            
            in_ptr += 3;
            }
          }
        
	return result;
}







static int encode(quicktime_t *file, unsigned char **row_pointers, int track)
{
        uint8_t *in_y, *in_u, *in_v, *out_ptr;
	quicktime_video_map_t *vtrack = &(file->vtracks[track]);
	quicktime_v308_codec_t *codec = ((quicktime_codec_t*)vtrack->codec)->priv;
	quicktime_trak_t *trak = vtrack->track;
	int width = vtrack->track->tkhd.track_width;
	int height = vtrack->track->tkhd.track_height;
	int bytes = width * height * 3;
	int result = 0;
	int i, j;
	quicktime_atom_t chunk_atom;

        if(!row_pointers)
          {
          vtrack->stream_cmodel = BC_YUV444P;
          //          vtrack->stream_cmodel = BC_VYU888;
          return 0;
          }
        
        if(!codec->buffer)
          {
          lqt_set_fiel_uncompressed(file, track);
          codec->buffer = malloc(width * height * 3);
          }
        out_ptr = codec->buffer;
	for(i = 0; i < height; i++)
          {
          in_y = row_pointers[0] + i * file->vtracks[track].stream_row_span;
          in_u = row_pointers[1] + i * file->vtracks[track].stream_row_span_uv;
          in_v = row_pointers[2] + i * file->vtracks[track].stream_row_span_uv;
          
          for(j = 0; j < width; j++)
            {

            out_ptr[1] = *in_y; /* Y */
            out_ptr[2] = *in_u; /* U */
            out_ptr[0] = *in_v; /* V */
            
            out_ptr += 3;
            in_y ++;
            in_u ++;
            in_v ++;
            }
          }
        
	quicktime_write_chunk_header(file, trak, &chunk_atom);
	result = !quicktime_write_data(file, codec->buffer, bytes);
	quicktime_write_chunk_footer(file, 
		trak,
		vtrack->current_chunk,
		&chunk_atom, 
		1);

	vtrack->current_chunk++;
	
	return result;
}

void quicktime_init_codec_v308(quicktime_video_map_t *vtrack)
{
	quicktime_codec_t *codec_base = (quicktime_codec_t*)vtrack->codec;

/* Init public items */
	codec_base->priv = calloc(1, sizeof(quicktime_v308_codec_t));
	codec_base->delete_vcodec = delete_codec;
	codec_base->decode_video = decode;
	codec_base->encode_video = encode;
	codec_base->decode_audio = 0;
	codec_base->encode_audio = 0;
}