534 lines
15 KiB
C
534 lines
15 KiB
C
/*
|
|
* Copyright (C) 2010-2011 Mamadou Diop.
|
|
*
|
|
* Contact: Mamadou Diop <diopmamadou(at)doubango.org>
|
|
*
|
|
* This file is part of Open Source Doubango Framework.
|
|
*
|
|
* DOUBANGO 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* DOUBANGO 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 DOUBANGO.
|
|
*
|
|
*/
|
|
|
|
/**@file tdav_codec_h261.c
|
|
* @brief H.261 codec plugin.
|
|
* RTP payloader follows RFC 4587
|
|
*
|
|
* @author Mamadou Diop <diopmamadou(at)doubango.org>
|
|
*
|
|
|
|
*/
|
|
#include "tinydav/codecs/h261/tdav_codec_h261.h"
|
|
|
|
#if HAVE_FFMPEG
|
|
|
|
#include "tinydav/video/tdav_converter_video.h"
|
|
|
|
#include "tinyrtp/rtp/trtp_rtp_packet.h"
|
|
|
|
#include "tnet_endianness.h"
|
|
|
|
#include "tsk_time.h"
|
|
#include "tsk_memory.h"
|
|
#include "tsk_debug.h"
|
|
|
|
#define RTP_PAYLOAD_SIZE 700
|
|
#define H261_HEADER_SIZE 4
|
|
|
|
static void *run(void* self);
|
|
static void tdav_codec_h261_rtp_callback(tdav_codec_h261_t *self, const void *data, tsk_size_t size, tsk_bool_t marker);
|
|
static void tdav_codec_h261_encap(const tdav_codec_h261_t* h261, const uint8_t* pdata, tsk_size_t size);
|
|
|
|
/* ============ H.261 Plugin interface ================= */
|
|
|
|
//
|
|
// H.261 object definition
|
|
//
|
|
int tdav_codec_h261_open(tmedia_codec_t* self)
|
|
{
|
|
int ret;
|
|
int size;
|
|
|
|
tdav_codec_h261_t* h261 = (tdav_codec_h261_t*)self;
|
|
|
|
if(!h261){
|
|
TSK_DEBUG_ERROR("Invalid parameter");
|
|
return -1;
|
|
}
|
|
|
|
/* the caller (base class) already checked that the codec is not opened */
|
|
|
|
//
|
|
// Encoder
|
|
//
|
|
if(!(h261->encoder.codec = avcodec_find_encoder(CODEC_ID_H261))){
|
|
TSK_DEBUG_ERROR("Failed to find H.261 encoder");
|
|
return -2;
|
|
}
|
|
h261->encoder.context = avcodec_alloc_context();
|
|
avcodec_get_context_defaults(h261->encoder.context);
|
|
|
|
h261->encoder.context->pix_fmt = PIX_FMT_YUV420P;
|
|
h261->encoder.context->time_base.num = 1;
|
|
h261->encoder.context->time_base.den = TMEDIA_CODEC_VIDEO(h261)->out.fps;
|
|
h261->encoder.context->width = TMEDIA_CODEC_VIDEO(h261)->out.width;
|
|
h261->encoder.context->height = TMEDIA_CODEC_VIDEO(h261)->out.height;
|
|
|
|
h261->encoder.context->mb_qmin = h261->encoder.context->qmin = 4;
|
|
h261->encoder.context->mb_qmax = h261->encoder.context->qmax = 31;
|
|
h261->encoder.context->mb_decision = FF_MB_DECISION_SIMPLE;
|
|
|
|
h261->encoder.context->thread_count = 1;
|
|
h261->encoder.context->rtp_payload_size = RTP_PAYLOAD_SIZE;
|
|
h261->encoder.context->opaque = tsk_null;
|
|
h261->encoder.context->bit_rate = (float) (500000) * 0.80f;
|
|
h261->encoder.context->bit_rate_tolerance = (int) (500000 * 0.20f);
|
|
h261->encoder.context->gop_size = TMEDIA_CODEC_VIDEO(h261)->out.fps*4; /* each 4 seconds */
|
|
|
|
// Picture (YUV 420)
|
|
if(!(h261->encoder.picture = avcodec_alloc_frame())){
|
|
TSK_DEBUG_ERROR("Failed to create encoder picture");
|
|
return -2;
|
|
}
|
|
avcodec_get_frame_defaults(h261->encoder.picture);
|
|
//if((ret = avpicture_alloc((AVPicture*)h261->encoder.picture, PIX_FMT_YUV420P, h261->encoder.context->width, h261->encoder.context->height))){
|
|
// TSK_DEBUG_ERROR("Failed to allocate encoder picture");
|
|
// return ret;
|
|
//}
|
|
|
|
size = avpicture_get_size(PIX_FMT_YUV420P, h261->encoder.context->width, h261->encoder.context->height);
|
|
if(!(h261->encoder.buffer = tsk_calloc(size, sizeof(uint8_t)))){
|
|
TSK_DEBUG_ERROR("Failed to allocate encoder buffer");
|
|
return -2;
|
|
}
|
|
|
|
// Open encoder
|
|
if((ret = avcodec_open(h261->encoder.context, h261->encoder.codec)) < 0){
|
|
TSK_DEBUG_ERROR("Failed to open H.261 encoder");
|
|
return ret;
|
|
}
|
|
|
|
//
|
|
// Decoder
|
|
//
|
|
if(!(h261->decoder.codec = avcodec_find_decoder(CODEC_ID_H261))){
|
|
TSK_DEBUG_ERROR("Failed to find H.261 decoder");
|
|
}
|
|
h261->decoder.context = avcodec_alloc_context();
|
|
avcodec_get_context_defaults(h261->decoder.context);
|
|
|
|
h261->decoder.context->pix_fmt = PIX_FMT_YUV420P;
|
|
h261->decoder.context->width = TMEDIA_CODEC_VIDEO(h261)->in.width;
|
|
h261->decoder.context->height = TMEDIA_CODEC_VIDEO(h261)->in.height;
|
|
|
|
// Picture (YUV 420)
|
|
if(!(h261->decoder.picture = avcodec_alloc_frame())){
|
|
TSK_DEBUG_ERROR("Failed to create decoder picture");
|
|
return -2;
|
|
}
|
|
avcodec_get_frame_defaults(h261->decoder.picture);
|
|
|
|
size = avpicture_get_size(PIX_FMT_YUV420P, h261->decoder.context->width, h261->decoder.context->height);
|
|
if(!(h261->decoder.accumulator = tsk_calloc((size + FF_INPUT_BUFFER_PADDING_SIZE), sizeof(uint8_t)))){
|
|
TSK_DEBUG_ERROR("Failed to allocate decoder buffer");
|
|
return -2;
|
|
}
|
|
|
|
// Open decoder
|
|
if((ret = avcodec_open(h261->decoder.context, h261->decoder.codec)) < 0){
|
|
TSK_DEBUG_ERROR("Failed to open H.261 decoder");
|
|
return ret;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int tdav_codec_h261_close(tmedia_codec_t* self)
|
|
{
|
|
tdav_codec_h261_t* h261 = (tdav_codec_h261_t*)self;
|
|
|
|
if(!h261){
|
|
TSK_DEBUG_ERROR("Invalid parameter");
|
|
return -1;
|
|
}
|
|
|
|
/* the caller (base class) already checked that the codec is opened */
|
|
|
|
//
|
|
// Encoder
|
|
//
|
|
if(h261->encoder.context){
|
|
avcodec_close(h261->encoder.context);
|
|
av_free(h261->encoder.context);
|
|
h261->encoder.context = tsk_null;
|
|
}
|
|
if(h261->encoder.picture){
|
|
av_free(h261->encoder.picture);
|
|
}
|
|
if(h261->encoder.buffer){
|
|
TSK_FREE(h261->encoder.buffer);
|
|
}
|
|
|
|
//
|
|
// Decoder
|
|
//
|
|
if(h261->decoder.context){
|
|
avcodec_close(h261->decoder.context);
|
|
av_free(h261->decoder.context);
|
|
h261->decoder.context = tsk_null;
|
|
}
|
|
if(h261->decoder.picture){
|
|
av_free(h261->decoder.picture);
|
|
h261->decoder.picture = tsk_null;
|
|
}
|
|
if(h261->decoder.accumulator){
|
|
TSK_FREE(h261->decoder.accumulator);
|
|
h261->decoder.accumulator_pos = 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
tsk_size_t tdav_codec_h261_encode(tmedia_codec_t* self, const void* in_data, tsk_size_t in_size, void** out_data, tsk_size_t* out_max_size)
|
|
{
|
|
int ret;
|
|
int size;
|
|
|
|
tdav_codec_h261_t* h261 = (tdav_codec_h261_t*)self;
|
|
|
|
if(!self || !in_data || !in_size || !out_data){
|
|
TSK_DEBUG_ERROR("Invalid parameter");
|
|
return 0;
|
|
}
|
|
|
|
// delete old buffer
|
|
if(*out_data){
|
|
TSK_FREE(*out_data);
|
|
}
|
|
|
|
// wrap yuv420 buffer
|
|
size = avpicture_fill((AVPicture *)h261->encoder.picture, (uint8_t*)in_data, PIX_FMT_YUV420P, h261->encoder.context->width, h261->encoder.context->height);
|
|
if(size != in_size){
|
|
/* guard */
|
|
TSK_DEBUG_ERROR("Invalid size");
|
|
return 0;
|
|
}
|
|
|
|
// Encode data
|
|
h261->encoder.picture->pts = AV_NOPTS_VALUE;
|
|
//h261->encoder.picture->pict_type = FF_I_TYPE;
|
|
ret = avcodec_encode_video(h261->encoder.context, h261->encoder.buffer, size, h261->encoder.picture);
|
|
if(ret > 0){
|
|
tdav_codec_h261_encap(h261, h261->encoder.buffer, (tsk_size_t)ret);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
tsk_size_t tdav_codec_h261_decode(tmedia_codec_t* self, const void* in_data, tsk_size_t in_size, void** out_data, tsk_size_t* out_max_size, const tsk_object_t* proto_hdr)
|
|
{
|
|
uint8_t sbit, ebit;
|
|
const uint8_t* pdata = in_data;
|
|
const uint8_t* pay_ptr;
|
|
tsk_size_t pay_size;
|
|
tsk_size_t xsize, retsize = 0;
|
|
int got_picture_ptr;
|
|
int ret;
|
|
|
|
tdav_codec_h261_t* h261 = (tdav_codec_h261_t*)self;
|
|
const trtp_rtp_header_t* rtp_hdr = proto_hdr;
|
|
|
|
if(!self || !in_data || !in_size || !out_data || !h261->decoder.context){
|
|
TSK_DEBUG_ERROR("Invalid parameter");
|
|
return 0;
|
|
}
|
|
|
|
/* RFC 4587
|
|
0 1 2 3
|
|
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
|
|
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
|
|SBIT |EBIT |I|V| GOBN | MBAP | QUANT | HMVD | VMVD |
|
|
+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
|
|
*/
|
|
sbit = *pdata >> 5;
|
|
ebit = (*pdata >> 2) & 0x07;
|
|
|
|
/* Check size */
|
|
if(in_size < H261_HEADER_SIZE){
|
|
TSK_DEBUG_ERROR("Too short");
|
|
return 0;
|
|
}
|
|
|
|
pay_ptr = (pdata + H261_HEADER_SIZE);
|
|
pay_size = (in_size - H261_HEADER_SIZE);
|
|
xsize = avpicture_get_size(h261->decoder.context->pix_fmt, h261->decoder.context->width, h261->decoder.context->height);
|
|
|
|
/* Packet lost? */
|
|
if(h261->decoder.last_seq != (rtp_hdr->seq_num - 1) && h261->decoder.last_seq){
|
|
TSK_DEBUG_INFO("Packet lost");
|
|
}
|
|
h261->decoder.last_seq = rtp_hdr->seq_num;
|
|
|
|
if((int)(h261->decoder.accumulator_pos + pay_size) <= xsize){
|
|
|
|
if((h261->decoder.ebit + sbit) == 8){ /* Perfect one Byte to clean up */
|
|
if(h261->decoder.accumulator_pos){
|
|
((uint8_t*)h261->decoder.accumulator)[h261->decoder.accumulator_pos-1] =
|
|
(((uint8_t*)h261->decoder.accumulator)[h261->decoder.accumulator_pos-1] & (0xFF << h261->decoder.ebit)) |
|
|
(*pay_ptr << sbit);
|
|
}
|
|
pay_ptr++, pay_size--;
|
|
}
|
|
h261->decoder.ebit = ebit;
|
|
|
|
memcpy(&((uint8_t*)h261->decoder.accumulator)[h261->decoder.accumulator_pos], pay_ptr, pay_size);
|
|
h261->decoder.accumulator_pos += pay_size;
|
|
}
|
|
else{
|
|
TSK_DEBUG_WARN("Buffer overflow");
|
|
h261->decoder.accumulator_pos = 0;
|
|
return 0;
|
|
}
|
|
|
|
if(rtp_hdr->marker){
|
|
AVPacket packet;
|
|
/* allocate destination buffer */
|
|
if(*out_max_size <xsize){
|
|
if(!(*out_data = tsk_realloc(*out_data, xsize))){
|
|
TSK_DEBUG_ERROR("Failed to allocate new buffer");
|
|
*out_max_size = 0;
|
|
h261->decoder.accumulator_pos = 0;
|
|
return 0;
|
|
}
|
|
*out_max_size = xsize;
|
|
}
|
|
|
|
/* decode the picture */
|
|
av_init_packet(&packet);
|
|
packet.size = h261->decoder.accumulator_pos;
|
|
packet.data = h261->decoder.accumulator;
|
|
ret = avcodec_decode_video2(h261->decoder.context, h261->decoder.picture, &got_picture_ptr, &packet);
|
|
|
|
if(ret <0 || !got_picture_ptr){
|
|
TSK_DEBUG_WARN("Failed to decode the buffer");
|
|
}
|
|
else{
|
|
retsize = xsize;
|
|
TMEDIA_CODEC_VIDEO(h261)->in.width = h261->decoder.context->width;
|
|
TMEDIA_CODEC_VIDEO(h261)->in.height = h261->decoder.context->height;
|
|
/* copy picture into a linear buffer */
|
|
avpicture_layout((AVPicture *)h261->decoder.picture, h261->decoder.context->pix_fmt, h261->decoder.context->width, h261->decoder.context->height,
|
|
*out_data, retsize);
|
|
}
|
|
/* in all cases: reset accumulator */
|
|
h261->decoder.accumulator_pos = 0;
|
|
}
|
|
|
|
return retsize;
|
|
}
|
|
|
|
tsk_bool_t tdav_codec_h261_fmtp_match(const tmedia_codec_t* codec, const char* fmtp)
|
|
{
|
|
int ret;
|
|
unsigned maxbr, fps, width, height;
|
|
tmedia_codec_video_t* h261 = (tmedia_codec_video_t*)codec;
|
|
|
|
if(!(ret = tmedia_codec_parse_fmtp(fmtp, &maxbr, &fps, &width, &height))){
|
|
h261->in.max_br = h261->out.max_br = maxbr * 1000;
|
|
h261->in.fps = h261->out.fps = fps;
|
|
h261->in.width = h261->out.width = width;
|
|
h261->in.height = h261->out.height = height;
|
|
return tsk_true;
|
|
}
|
|
else{
|
|
TSK_DEBUG_WARN("Failed to match fmtp [%s]", fmtp);
|
|
return tsk_false;
|
|
}
|
|
}
|
|
|
|
char* tdav_codec_h261_fmtp_get(const tmedia_codec_t* self)
|
|
{
|
|
#if 0
|
|
return tsk_strdup("CIF=2/MaxBR=3840;QCIF=2/MaxBR=1920");
|
|
#else
|
|
return tsk_strdup("QCIF=2");
|
|
#endif
|
|
}
|
|
|
|
int tdav_codec_h261_fmtp_set(tmedia_codec_t* self, const char* fmtp)
|
|
{
|
|
TSK_DEBUG_INFO("remote fmtp=%s", fmtp);
|
|
return 0;
|
|
}
|
|
|
|
/* constructor */
|
|
static tsk_object_t* tdav_codec_h261_ctor(tsk_object_t * self, va_list * app)
|
|
{
|
|
tdav_codec_h261_t *h261 = self;
|
|
if(h261){
|
|
/* init base: called by tmedia_codec_create() */
|
|
/* init self */
|
|
}
|
|
return self;
|
|
}
|
|
/* destructor */
|
|
static tsk_object_t* tdav_codec_h261_dtor(tsk_object_t * self)
|
|
{
|
|
tdav_codec_h261_t *h261 = self;
|
|
if(h261){
|
|
/* deinit base */
|
|
tmedia_codec_video_deinit(h261); // will call close()
|
|
/* deinit self */
|
|
TSK_FREE(h261->rtp.ptr);
|
|
h261->rtp.size = 0;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
/* object definition */
|
|
static const tsk_object_def_t tdav_codec_h261_def_s =
|
|
{
|
|
sizeof(tdav_codec_h261_t),
|
|
tdav_codec_h261_ctor,
|
|
tdav_codec_h261_dtor,
|
|
tmedia_codec_cmp,
|
|
};
|
|
/* plugin definition*/
|
|
static const tmedia_codec_plugin_def_t tdav_codec_h261_plugin_def_s =
|
|
{
|
|
&tdav_codec_h261_def_s,
|
|
|
|
tmedia_video,
|
|
"H261",
|
|
"H261 codec",
|
|
TMEDIA_CODEC_FORMAT_H261,
|
|
tsk_false,
|
|
90000, // rate
|
|
|
|
/* audio */
|
|
{ 0 },
|
|
|
|
/* video */
|
|
{176, 144, 15},
|
|
|
|
tdav_codec_h261_open,
|
|
tdav_codec_h261_close,
|
|
tdav_codec_h261_encode,
|
|
tdav_codec_h261_decode,
|
|
tdav_codec_h261_fmtp_match,
|
|
tdav_codec_h261_fmtp_get,
|
|
tdav_codec_h261_fmtp_set
|
|
};
|
|
const tmedia_codec_plugin_def_t *tdav_codec_h261_plugin_def_t = &tdav_codec_h261_plugin_def_s;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* ============ Callbacks ================= */
|
|
|
|
static void tdav_codec_h261_encap(const tdav_codec_h261_t* h261, const uint8_t* pdata, tsk_size_t size)
|
|
{
|
|
uint32_t i, last_index = 0;
|
|
|
|
if(size < RTP_PAYLOAD_SIZE){
|
|
goto last;
|
|
}
|
|
|
|
for(i = 4; i<(size - 4); i++){
|
|
if(pdata[i] == 0x00 && pdata[i+1] == 0x00 && pdata[i+2]>=0x80){ /* PSC or (GBSC) found */
|
|
if((i - last_index) >= RTP_PAYLOAD_SIZE){
|
|
tdav_codec_h261_rtp_callback((tdav_codec_h261_t*)h261, pdata+last_index,
|
|
(i - last_index), (last_index == size));
|
|
}
|
|
last_index = i;
|
|
}
|
|
}
|
|
last:
|
|
if(last_index < size - 3/*PSC/GBSC size*/){
|
|
tdav_codec_h261_rtp_callback((tdav_codec_h261_t*)h261, pdata + last_index,
|
|
(size - last_index), tsk_true);
|
|
}
|
|
}
|
|
|
|
//static void *run(void* self)
|
|
//{
|
|
// uint32_t i, last_index;
|
|
// tsk_list_item_t *curr;
|
|
//
|
|
// const uint8_t* pdata;
|
|
// tsk_size_t size;
|
|
//
|
|
// const tdav_codec_h261_t* h261 = ((tdav_runnable_video_t*)self)->userdata;
|
|
//
|
|
// TSK_DEBUG_INFO("H261 thread === START");
|
|
//
|
|
// TSK_RUNNABLE_RUN_BEGIN(self);
|
|
//
|
|
// if((curr = TSK_RUNNABLE_POP_FIRST(self))){
|
|
// /* 4 is sizeof(uint32_t) */
|
|
// pdata = ((const tsk_buffer_t*)curr->data)->data;
|
|
// size = ((const tsk_buffer_t*)curr->data)->size;
|
|
// last_index = 0;
|
|
//
|
|
// if(size < RTP_PAYLOAD_SIZE){
|
|
// goto last;
|
|
// }
|
|
//
|
|
// for(i = 4; i<(size - 4); i++){
|
|
// if(pdata[i] == 0x00 && pdata[i+1] == 0x00 && pdata[i+2]>=0x80){ /* PSC or (GBSC) found */
|
|
// if((i - last_index) >= RTP_PAYLOAD_SIZE){
|
|
// tdav_codec_h261_rtp_callback((tdav_codec_h261_t*)h261, pdata+last_index,
|
|
// (i - last_index), (last_index == size));
|
|
// }
|
|
// last_index = i;
|
|
// }
|
|
// }
|
|
//last:
|
|
// if(last_index < size - 3/*PSC/GBSC size*/){
|
|
// tdav_codec_h261_rtp_callback((tdav_codec_h261_t*)h261, pdata + last_index,
|
|
// (size - last_index), tsk_true);
|
|
// }
|
|
//
|
|
// tsk_object_unref(curr);
|
|
// }
|
|
//
|
|
// TSK_RUNNABLE_RUN_END(self);
|
|
//
|
|
// TSK_DEBUG_INFO("H261 thread === STOP");
|
|
//
|
|
// return tsk_null;
|
|
//}
|
|
|
|
static void tdav_codec_h261_rtp_callback(tdav_codec_h261_t *self, const void *data, tsk_size_t size, tsk_bool_t marker)
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif /* HAVE_FFMPEG */ |