/*
 * Copyright (c) 2004-2010 Marcus Overhagen <marcus@overhagen.de>
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation
 * files (the "Software"), to deal in the Software without restriction, 
 * including without limitation the rights to use, copy, modify, 
 * merge, publish, distribute, sublicense, and/or sell copies of 
 * the Software, and to permit persons to whom the Software is 
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be 
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES 
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT 
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, 
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR 
 * OTHER DEALINGS IN THE SOFTWARE.
 */

#include <media/MediaFormats.h>
#include <media/MediaNode.h>
#include <stdio.h>
#include "MediaFormat.h"

extern "C" {
  #include "avcodec.h"

#ifdef DEBUG
  // Needed to fix debug build, otherwise the linker complains about
  // "undefined reference to `ff_log2_tab'"
  const uint8_t ff_log2_tab[256] = {0};
#endif

}  // extern "C"


#if LIBAVCODEC_VERSION_INT <= ((54 << 16) | (50 << 8))
#define AV_CODEC_ID_AC3 CODEC_ID_AC3
#define AV_CODEC_ID_MP3 CODEC_ID_MP3
#define AV_CODEC_ID_MPEG2VIDEO CODEC_ID_MPEG2VIDEO
#endif


void
PrintFormat(const media_format &format)
{
        char s[200];
        string_for_format(format, s, sizeof(s));
        printf("%s\n", s);
}


void
PrintFormat(const media_output &output)
{
        PrintFormat(output.format);
}


static status_t
GetHeaderFormatAc3Audio(media_format *out_format, const uint8 *header, size_t size)
{
        printf("GetHeaderFormatAc3Audio\n");

        status_t status;
        media_format_description desc;
//      desc.family = B_WAV_FORMAT_FAMILY;
//      desc.u.wav.codec = 0x2000;
        desc.family = B_MISC_FORMAT_FAMILY;
        desc.u.misc.file_format = 'ffmp';
        desc.u.misc.codec = AV_CODEC_ID_AC3;

        BMediaFormats formats;  
        status = formats.InitCheck();
        if (status) {
                printf("formats.InitCheck failed, error %lu\n", status);
                return status;
        }
        
        status = formats.GetFormatFor(desc, out_format);
        if (status) {
                printf("formats.GetFormatFor failed, error %lu\n", status);
                return status;
        }
        
        return B_OK;
}



static status_t
GetHeaderFormatDtsAudio(media_format *out_format, const uint8 *header, size_t size)
{
        printf("GetHeaderFormatDtsAudio: unsupported\n");
        return B_ERROR;
}


static status_t
GetHeaderFormatLpcmAudio(media_format *out_format, const uint8 *header, size_t size)
{
        printf("GetHeaderFormatLpcmAudio: unsupported\n");
        return B_ERROR;
}


static status_t
GetHeaderFormatPrivateStream(media_format *out_format, const uint8 *header, size_t size)
{
        printf("GetHeaderFormatPrivateStream: unsupported, assuming AC3\n");
        return GetHeaderFormatAc3Audio(out_format, header, size);
}


static status_t
GetHeaderFormatMpegAudio(media_format *out_format, const uint8 *header, size_t size)
{
        printf("GetHeaderFormatMpegAudio\n");

        status_t status;
        media_format_description desc;
//      desc.family = B_MPEG_FORMAT_FAMILY;
//      desc.u.mpeg.id = B_MPEG_2_AUDIO_LAYER_2;
        desc.family = B_MISC_FORMAT_FAMILY;
        desc.u.misc.file_format = 'ffmp';
        desc.u.misc.codec = AV_CODEC_ID_MP3;
        
        
        BMediaFormats formats;  
        status = formats.InitCheck();
        if (status) {
                printf("formats.InitCheck failed, error %lu\n", status);
                return status;
        }
        
        status = formats.GetFormatFor(desc, out_format);
        if (status) {
                printf("formats.GetFormatFor failed, error %lu\n", status);
                return status;
        }

        out_format->u.encoded_audio.output.frame_rate = 48000;
        out_format->u.encoded_audio.output.channel_count = 2;
        out_format->u.encoded_audio.output.buffer_size = 1024;
        return B_OK;
}


static status_t
GetHeaderFormatMpegVideo(media_format *out_format, const uint8 *header, size_t size)
{
        printf("GetHeaderFormatMpegVideo\n");

        status_t status;
        media_format_description desc;
//      desc.family = B_MPEG_FORMAT_FAMILY;
//      desc.u.mpeg.id = B_MPEG_2_VIDEO;
        desc.family = B_MISC_FORMAT_FAMILY;
        desc.u.misc.file_format = 'ffmp';
        desc.u.misc.codec = AV_CODEC_ID_MPEG2VIDEO;

        BMediaFormats formats;
        status = formats.InitCheck();
        if (status) {
                printf("formats.InitCheck failed, error %lu\n", status);
                return status;
        }
        
        status = formats.GetFormatFor(desc, out_format);
        if (status) {
                printf("formats.GetFormatFor failed, error %lu\n", status);
                return status;
        }

        return B_OK;
}


status_t
GetHeaderFormat(media_format *out_format, const void *header, size_t size, int stream_id)
{
        const uint8 *h = (const uint8 *)header;
        status_t status;

        printf("GetHeaderFormat: stream_id %02x\n", stream_id);
        printf("inner frame header: "
                   "%02x %02x %02x %02x %02x %02x %02x %02x "
                   "%02x %02x %02x %02x %02x %02x %02x %02x\n",
                   h[0], h[1], h[2], h[3], h[4], h[5], h[6], h[7], 
                   h[8], h[9], h[10], h[11], h[12], h[13], h[14], h[15]);

        if (stream_id >= 0x80 && stream_id <= 0x87)
                status = GetHeaderFormatAc3Audio(out_format, h, size);
        else if (stream_id >= 0x88 && stream_id <= 0x8F)
                status = GetHeaderFormatDtsAudio(out_format, h, size);
        else if (stream_id >= 0xA0 && stream_id <= 0xA7)
                status = GetHeaderFormatLpcmAudio(out_format, h, size);
        else if (stream_id == 0xBD)
                status = GetHeaderFormatPrivateStream(out_format, h, size);
        else if (stream_id >= 0xC0 && stream_id <= 0xDF)
                status = GetHeaderFormatMpegAudio(out_format, h, size);
        else if (stream_id >= 0xE0 && stream_id <= 0xEF)
                status = GetHeaderFormatMpegVideo(out_format, h, size);
        else {
                printf("GetHeaderFormat: don't know what this stream_id means\n");
                status = B_ERROR;
        }
        
        if (status != B_OK) {
                printf("GetHeaderFormat: failed!\n");
        } else {
                printf("GetHeaderFormat: out_format ");
                PrintFormat(*out_format);
        }
        return status;
}