/* -*- c-basic-offset: 8; -*- */
/* theora.c: Ogg Theora data handlers for libshout
 * $Id: theora.c 11416 2006-05-22 00:32:49Z brendan $
 *
 *  Copyright (C) 2004 the Icecast team <team@icecast.org>
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Library General Public
 *  License as published by the Free Software Foundation; either
 *  version 2 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
 *  Library General Public License for more details.
 *
 *  You should have received a copy of the GNU Library General Public
 *  License along with this library; if not, write to the Free
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#ifdef HAVE_INTTYPES_H
# include <inttypes.h>
#endif

#include <theora/theora.h>

#include "shout_private.h"
#include "shout_ogg.h"

/* -- local data structures -- */
typedef struct {
	theora_info ti;
	theora_comment tc;
	uint32_t granule_shift;
	double per_frame;
    uint64_t start_frame;
    int initial_frames;
    int get_start_frame;
} theora_data_t;

/* -- local prototypes -- */
static int read_theora_page(ogg_codec_t *codec, ogg_page *page);
static void free_theora_data(void *codec_data);
static int theora_ilog(unsigned int v);

/* -- theora functions -- */
int _shout_open_theora(ogg_codec_t *codec, ogg_page *page)
{
	ogg_packet packet;

	theora_data_t *theora_data = calloc(1, sizeof(theora_data_t));
	if (! theora_data)
		return SHOUTERR_MALLOC;

	theora_info_init(&theora_data->ti);
	theora_comment_init(&theora_data->tc);

	ogg_stream_packetout(&codec->os, &packet);
	
	if (theora_decode_header(&theora_data->ti, &theora_data->tc, &packet) < 0) {
		free_theora_data(theora_data);

		return SHOUTERR_UNSUPPORTED;
	}

	codec->codec_data = theora_data;
	codec->read_page = read_theora_page;
	codec->free_data = free_theora_data;
	codec->headers = 1;
    theora_data->initial_frames = 0;

	return SHOUTERR_SUCCESS;
}

static int read_theora_page(ogg_codec_t *codec, ogg_page *page)
{
	theora_data_t *theora_data = codec->codec_data;
	ogg_packet packet;
	ogg_int64_t granulepos, iframe, pframe;

	granulepos = ogg_page_granulepos(page);

	if (granulepos == 0)
	{
		while (ogg_stream_packetout(&codec->os, &packet) > 0) {
			if (theora_decode_header(&theora_data->ti, &theora_data->tc, &packet) < 0)
				return SHOUTERR_INSANE;
			codec->headers++;
		}
		if (codec->headers == 3)
        {
			theora_data->granule_shift = theora_ilog(theora_data->ti.keyframe_frequency_force - 1);
            theora_data->per_frame = (double)theora_data->ti.fps_denominator / theora_data->ti.fps_numerator * 1000000;
            theora_data->get_start_frame = 1;
		}

		return SHOUTERR_SUCCESS;
	}

    while (ogg_stream_packetout(&codec->os, &packet) > 0)
    {
        if (theora_data->get_start_frame)
            theora_data->initial_frames++;
    }
	if (granulepos > 0 && codec->headers >= 3)
    {
        iframe = granulepos >> theora_data->granule_shift;
        pframe = granulepos - (iframe << theora_data->granule_shift);

        if (theora_data->get_start_frame)
        {
            /* work out the real start frame, which may not be 0 */
            theora_data->start_frame = iframe + pframe - theora_data->initial_frames;
            codec->senttime = 0;
            theora_data->get_start_frame = 0;
        }
        else
        {
            uint64_t frames = ((iframe + pframe) - theora_data->start_frame);
            codec->senttime = (uint64_t)(frames * theora_data->per_frame);
        }
	}

	return SHOUTERR_SUCCESS;
}

static void free_theora_data(void *codec_data)
{
	theora_data_t *theora_data = (theora_data_t *)codec_data;

	theora_info_clear(&theora_data->ti);
	theora_comment_clear(&theora_data->tc);
	free(theora_data);
}

static int theora_ilog(unsigned int v)
{
	int ret = 0;

	while (v) {
		ret++;
		v >>= 1;
	}

	return ret;
}