/*
  xacodec.c
  XAnim Video Codec DLL support

  (C) 2001 Alex Beregszaszi <alex@naxine.org>
*/

#include <stdio.h>
#include <dlfcn.h> /* dlsym, dlopen, dlclose */
#include <stdarg.h> /* va_alist, va_start, va_end */
#include <errno.h> /* strerror, errno */

#include "odeon.h"
#include "aclib/byteswap.h"

#include "stream.h"
#include "demuxer.h"
#include "codec-cfg.h"
#include "wine/avifmt.h"
#include "wine/vfw.h"
#include "wine/mmreg.h"
#include "stheader.h"
#include "loader/DirectShow/default.h"
#include "libvo/img_format.h"
#include "xacodec.h"

#ifndef RTLD_NOW
#define RLTD_NOW 2
#endif
#ifndef RTLD_LAZY
#define RLTD_LAZY 1
#endif
#ifndef RTLD_GLOBAL
#define RLTD_GLOBAL 256
#endif

extern int verbose;

typedef struct xacodec_driver
{
    XA_DEC_INFO *decinfo;
    void *file_handler;
    long (*iq_func)(XA_CODEC_HDR *codec_hdr);
    unsigned int (*dec_func)(unsigned char *image, unsigned char *delta,
	unsigned int dsize, XA_DEC_INFO *dec_info);
    void (*close_func)();
} xacodec_driver_t;

xacodec_driver_t *xacodec_driver = NULL;

/* *** XANIM SHIT *** */

typedef struct
{
    unsigned char r0, g0, b0;
    unsigned char r1, g1, b1;
    unsigned char r2, g2, b2;
    unsigned char r3, g3, b3;
    unsigned int clr0_0, clr0_1, clr0_2, clr0_3;
    unsigned int clr1_0, clr1_1, clr1_2, clr1_3;
    unsigned int clr2_0, clr2_1, clr2_2, clr2_3;
    unsigned int clr3_0, clr3_1, clr3_2, clr3_3;
} XA_2x2_Color;

#define ip_OUT_2x2_1BLK(ip, CAST, cmap2x2, rinc) { register CAST d0, d1; \
 *ip++ = d0 = (CAST)(cmap2x2->clr0_0); *ip++ = d0; \
 *ip++ = d1 = (CAST)(cmap2x2->clr1_0); *ip = d1; \
  ip += rinc; \
 *ip++ = d0; *ip++ = d0; *ip++ = d1; *ip = d1; ip += rinc; \
 *ip++ = d0 = (CAST)(cmap2x2->clr2_0); *ip++ = d0; \
 *ip++ = d1 = (CAST)(cmap2x2->clr3_0); *ip = d1; \
  ip += rinc; *ip++ = d0; *ip++ = d0; *ip++ = d1; *ip++ = d1; }

#define ip_OUT_2x2_2BLKS(ip, CAST, c2x2map0, c2x2map1, rinc) { \
 *ip++ = (CAST)(c2x2map0->clr0_0); \
 *ip++ = (CAST)(c2x2map0->clr1_0); \
 *ip++ = (CAST)(c2x2map1->clr0_0); \
 *ip   = (CAST)(c2x2map1->clr1_0); ip += rinc; \
 *ip++ = (CAST)(c2x2map0->clr2_0); \
 *ip++ = (CAST)(c2x2map0->clr3_0); \
 *ip++ = (CAST)(c2x2map1->clr2_0); \
 *ip   = (CAST)(c2x2map1->clr3_0); }

void XA_2x2_OUT_1BLK_clr8(unsigned char *image, unsigned int x, unsigned int y,
    unsigned int imagex, XA_2x2_Color *cmap2x2)
{
    unsigned int row_inc = imagex - 3;
    unsigned char *ip = (unsigned char *)(image + y * imagex + x);

    XA_Print("XA_2x2_OUT_1BLK_clr8('image: %08x', 'x: %d', 'y: %d', 'imagex: %d', 'cmap2x2: %08x')",
	image, x, y, imagex, cmap2x2);
    ip_OUT_2x2_1BLK(ip, unsigned char, cmap2x2, row_inc);
}

void XA_2x2_OUT_4BLKS_clr8(unsigned char *image, unsigned int x, unsigned int y,
    unsigned int imagex, XA_2x2_Color *cm0, XA_2x2_Color *cm1, XA_2x2_Color *cm2,
    XA_2x2_Color *cm3)
{
    unsigned int row_inc = imagex - 3;
    unsigned char *ip = (unsigned char *)(image + y * imagex + x);

    XA_Print("XA_2x2_OUT_1BLK_clr8('image: %08x', 'x: %d', 'y: %d', 'imagex: %d', 'cm0: %08x', 'cm1: %08x', 'cm2: %08x', 'cm3: %08x')",
	image, x, y, imagex, cm0, cm1, cm2, cm3);
    ip_OUT_2x2_2BLKS(ip, unsigned char, cm0, cm1, row_inc);
    ip += row_inc;
    ip_OUT_2x2_2BLKS(ip, unsigned char, cm2, cm3, row_inc);
}

void *YUV2x2_Blk_Func(unsigned int image_type, int blks, unsigned int dith_flag)
{
    void (*color_func)();

    XA_Print("YUV2x2_Blk_Func('image_type: %d', 'blks: %d', 'dith_flag: %d')",
	    image_type, blks, dith_flag);

    if (blks == 1)
    {
	switch(image_type)
	{
	    default:
		color_func = XA_2x2_OUT_1BLK_clr8;
		break;
	}
    }
    else /* blks == 4 */
    {
	switch(image_type)
	{
	    default:
		color_func = XA_2x2_OUT_4BLKS_clr8;
		break;
	}
    }

    XA_Print("YUV2x2_Blk_Func -> %08x", color_func);

    return((void *)color_func);
}

void *YUV2x2_Map_Func(unsigned int image_type, unsigned int dith_type)
{
    XA_Print("YUV2x2_Map_Func('image_type: %d', 'dith_type: %d')",
	    image_type, dith_type);
}

int XA_Add_Func_To_Free_Chain(XA_ANIM_HDR *anim_hdr, void (*function)())
{
    XA_Print("XA_Add_Func_To_Free_Chain('anim_hdr: %08x', 'function: %08x')",
	    anim_hdr, function);
    xacodec_driver->close_func = function;
}

int XA_Gen_YUV_Tabs(XA_ANIM_HDR *anim_hdr)
{
    XA_Print("XA_Gen_YUV_Tabs('anim_hdr: %08x')", anim_hdr);

//    XA_Print("anim type: %d - img[x: %d, y: %d, c: %d, d: %d]",
//	anim_hdr->anim_type, anim_hdr->imagex, anim_hdr->imagey,
//	anim_hdr->imagec, anim_hdr->imaged);
}

void JPG_Setup_Samp_Limit_Table(XA_ANIM_HDR *anim_hdr)
{
    XA_Print("JPG_Setup_Samp_Limit_Table('anim_hdr: %08x')", anim_hdr);
}

void JPG_Alloc_MCU_Bufs(XA_ANIM_HDR *anim_hdr, unsigned int width,
	unsigned int height, unsigned int full_flag)
{
    XA_Print("JPG_Alloc_MCU_Bufs('anim_hdr: %08x', 'width: %d', 'height: %d', 'full_flag: %d')",
	    anim_hdr, width, height, full_flag);
}

typedef struct
{
    unsigned char *Ybuf;
    unsigned char *Ubuf;
    unsigned char *Vbuf;
    unsigned char *the_buf;
    unsigned int the_buf_size;
    unsigned short y_w, y_h;
    unsigned short uv_w, uv_h;
} YUVBufs;

typedef struct
{
    unsigned long Uskip_mask;
    long	*YUV_Y_tab;
    long	*YUV_UB_tab;
    long	*YUV_VR_tab;
    long	*YUV_UG_tab;
    long	*YUV_VG_tab;
} YUVTabs;

#define XA_IMTYPE_RGB	0x0001

void XA_YUV1611_To_RGB(unsigned char *image, unsigned int imagex, unsigned int imagey,
    unsigned int i_x, unsigned int i_y, YUVBufs *yuv_bufs, YUVTabs *yuv_tabs,
    unsigned int map_flag, unsigned int *map, XA_CHDR *chdr)
{
    XA_Print("XA_YUV1611_To_RGB('image: %08x', 'imagex: %d', 'imagey: %d', 'i_x: %d', 'i_y: %d', 'yuv_bufs: %08x', 'yuv_tabs: %08x', 'map_flag: %d', 'map: %08x', 'chdr: %08x')",
	image, imagex, imagey, i_x, i_y, yuv_bufs, yuv_tabs, map_flag, map, chdr);
}

void XA_YUV1611_To_CLR8(unsigned char *image, unsigned int imagex, unsigned int imagey,
    unsigned int i_x, unsigned int i_y, YUVBufs *yuv_bufs, YUVTabs *yuv_tabs,
    unsigned int map_flag, unsigned int *map, XA_CHDR *chdr)
{
    XA_Print("XA_YUV1611_To_CLR8('image: %08x', 'imagex: %d', 'imagey: %d', 'i_x: %d', 'i_y: %d', 'yuv_bufs: %08x', 'yuv_tabs: %08x', 'map_flag: %d', 'map: %08x', 'chdr: %08x')",
	image, imagex, imagey, i_x, i_y, yuv_bufs, yuv_tabs, map_flag, map, chdr);
}

void *XA_YUV1611_Func(unsigned int image_type)
{
    void (*color_func)();

    XA_Print("XA_YUV1611_Func('image_type: %d')", image_type);

    switch(image_type)
    {
	case XA_IMTYPE_RGB:
	    color_func = XA_YUV1611_To_RGB;
	    break;
	default:
	    color_func = XA_YUV1611_To_CLR8;
	    break;
    }

    return((void *)color_func);
}

unsigned long XA_Time_Read()
{
    return(GetRelativeTime());
}

/* YUV 41 11 11 routines */
void XA_YUV411111_To_RGB(unsigned char *image, unsigned int imagex, unsigned int imagey,
    unsigned int i_x, unsigned int i_y, YUVBufs *yuv_bufs, YUVTabs *yuv_tabs,
    unsigned int map_flag, unsigned int *map, XA_CHDR *chdr)
{
    XA_Print("XA_YUV411111_To_RGB('image: %d', 'imagex: %d', 'imagey: %d', 'i_x: %d', 'i_y: %d', 'yuv_bufs: %08x', 'yuv_tabs: %08x', 'map_flag: %d', 'map: %08x', 'chdr: %08x')",
	    image, imagex, imagey, i_x, i_y, yuv_bufs, yuv_tabs, map_flag, map, chdr);
}


void *XA_YUV411111_Func(unsigned int image_type)
{
    void (*color_func)();

    XA_Print("XA_YUV411111_Func('image_type: %d')", image_type);    

    switch(image_type)
    {
	case XA_IMTYPE_RGB:	color_func = XA_YUV411111_To_RGB;	break;
    }
    
    return((void *)color_func);
}


YUVBufs jpg_YUVBufs;
YUVTabs def_yuv_tabs;

/* *** EOF XANIM SHIT *** */

/* Needed by XAnim DLLs */
int XA_Print(char *fmt, ...)
{
    va_list vallist;
    char buf[1024];

    va_start(vallist, fmt);
    vsnprintf(buf, 1024, fmt, vallist);
    odprintf(LOG_NORM, "[xacodec] %s", buf);
    va_end(vallist);
}

int TheEnd1(char *err_mess)
{
    XA_Print("error: %s - exiting\n", err_mess);
    xacodec_exit();
}
/* end of crap */

/* load, init and query */
int xacodec_init(char *filename, xacodec_driver_t *codec_driver)
{
    void *(*what_the)();
    char *error;
    XAVID_MOD_HDR *mod_hdr;
    XAVID_FUNC_HDR *func;
    int i;

    codec_driver->file_handler = dlopen(filename, RTLD_NOW|RTLD_GLOBAL);
    if (!codec_driver->file_handler)
    {
	error = dlerror();
	if (error)
	    printf("xacodec: fialed to init %s while %s\n", filename, error);
	else
	    printf("xacodec: failed ot init (dlopen) %s\n", filename);
	return(0);
    }

    what_the = dlsym(codec_driver->file_handler, "What_The");
    if ((error = dlerror()) != NULL)
    {
	printf("xacodec: failed to init %s while %s\n", filename, error);
	dlclose(codec_driver->file_handler);
	return(0);
    }
	
    mod_hdr = what_the();
    if (!mod_hdr)
    {
	printf("xacodec: 'What_The' function failed in %s\n", filename);
	dlclose(codec_driver->file_handler);
	return(0);
    }
    
    printf("=== XAnim Codec ===\n");
    printf(" Filename: %s (API revision: %x)\n", filename, mod_hdr->api_rev);
    printf(" Codec: %s. Rev: %s\n", mod_hdr->desc, mod_hdr->rev);
    if (mod_hdr->copyright)
	printf(" %s\n", mod_hdr->copyright);
    if (mod_hdr->mod_author)
	printf(" Module Author(s): %s\n", mod_hdr->mod_author);
    if (mod_hdr->authors)
	printf(" Codec Author(s): %s\n", mod_hdr->authors);

    if (mod_hdr->api_rev > XAVID_API_REV)
    {
	printf("xacodec: not supported api revision in %s\n", filename);
	dlclose(codec_driver->file_handler);
	return(0);
    }

    func = mod_hdr->funcs;
    if (!func)
    {
	printf("xacodec: function table error in %s\n", filename);
	dlclose(codec_driver->file_handler);
	return(0);
    }
    
    if (verbose > 2)
	printf("dump: funcs: 0x%08x (num %d)\n", mod_hdr->funcs, mod_hdr->num_funcs);
    for (i = 0; i < mod_hdr->num_funcs; i++)
    {
	if (verbose > 2)
	    printf("dump(%d): %d %d [iq:0x%08x d:0x%08x]\n", i,
		func[i].what, func[i].id, func[i].iq_func, func[i].dec_func);
	if (func[i].what & XAVID_AVI_QUERY)
	{
	    printf("0x%08x: avi init/query func (id: %d)\n",
		func[i].iq_func, func[i].id);
	    codec_driver->iq_func = func[i].iq_func;
	}
	if (func[i].what & XAVID_QT_QUERY)
	{
	    printf("0x%08x: qt init/query func (id: %d)\n",
		func[i].iq_func, func[i].id);
	    codec_driver->iq_func = func[i].iq_func;
	}
	if (func[i].what & XAVID_DEC_FUNC)
	{
	    printf("0x%08x: decoder func (init/query: 0x%08x) (id: %d)\n",
		func[i].dec_func, func[i].iq_func, func[i].id);
	    codec_driver->dec_func = func[i].dec_func;
	}
    }
    return(1);
}

int xacodec_query(xacodec_driver_t *codec_driver, XA_CODEC_HDR *codec_hdr)
{
    long codec_ret;

    codec_ret = codec_driver->iq_func(codec_hdr);
    switch(codec_ret)
    {
	case CODEC_SUPPORTED:
	    codec_driver->dec_func = codec_hdr->decoder;
	    printf("Codec is supported: found decoder for %s at 0x%08x\n",
		codec_hdr->description, codec_hdr->decoder);
	    return(1);
	case CODEC_UNSUPPORTED:
	    printf("Codec is unsupported\n");
	    return(0);
	case CODEC_UNKNOWN:
	default:
	    printf("Codec is unknown\n");
	    return(0);
    }
}

char *xacodec_def_path = XACODEC_PATH;

int xacodec_init_video(sh_video_t *vidinfo, int out_format)
{
    char dll[1024];
    XA_CODEC_HDR codec_hdr;
    
    xacodec_driver = realloc(xacodec_driver, sizeof(struct xacodec_driver));
    if (xacodec_driver == NULL)
	return(0);

    xacodec_driver->iq_func = NULL;
    xacodec_driver->dec_func = NULL;
    xacodec_driver->close_func = NULL;

    snprintf(dll, 1024, "%s/%s", xacodec_def_path, vidinfo->codec->dll);
    if (xacodec_init(dll, xacodec_driver) == 0)
	return(0);

    codec_hdr.anim_hdr = malloc(4096);
    codec_hdr.description = vidinfo->codec->info;
    codec_hdr.compression = bswap_32(vidinfo->format);
    codec_hdr.decoder = NULL;
    codec_hdr.avi_ctab_flag = 0;
    codec_hdr.avi_read_ext = NULL;
    codec_hdr.xapi_rev = 0x0001;
    codec_hdr.extra = NULL;
    codec_hdr.x = vidinfo->disp_w;
    codec_hdr.y = vidinfo->disp_h;

    switch(out_format)
    {
	case IMGFMT_RGB8:
	    codec_hdr.depth = 8;
	    break;
	case IMGFMT_RGB15:
	    codec_hdr.depth = 15;
	    break;
	case IMGFMT_RGB16:
	    codec_hdr.depth = 16;
	    break;
	case IMGFMT_RGB24:
	    codec_hdr.depth = 24;
	    break;
	case IMGFMT_RGB32:
	    codec_hdr.depth = 32;
	    break;
	case IMGFMT_BGR8:
	    codec_hdr.depth = 8;
	    break;
	case IMGFMT_BGR15:
	    codec_hdr.depth = 15;
	    break;
	case IMGFMT_BGR16:
	    codec_hdr.depth = 16;
	    break;
	case IMGFMT_BGR24:
	    codec_hdr.depth = 24;
	    break;
	case IMGFMT_BGR32:
	    codec_hdr.depth = 32;
	    break;
	default:
	    printf("xacodec: not supported image format (%s)\n",
		vo_format_name(out_format));
	    return(0);
    }
    odprintf(LOG_INFO, "xacodec: querying for %dx%d %dbit [fourcc: %4x] (%s)...\n",
	codec_hdr.x, codec_hdr.y, codec_hdr.depth, codec_hdr.compression,
	codec_hdr.description);

    if (xacodec_query(xacodec_driver, &codec_hdr) == 0)
	return(0);

//    free(codec_hdr.anim_hdr);

    xacodec_driver->decinfo = malloc(sizeof(XA_DEC_INFO));
    xacodec_driver->decinfo->cmd = 0;
    xacodec_driver->decinfo->skip_flag = 0;
    xacodec_driver->decinfo->imagex = xacodec_driver->decinfo->xe = codec_hdr.x;
    xacodec_driver->decinfo->imagey = xacodec_driver->decinfo->ye = codec_hdr.y;
    xacodec_driver->decinfo->imaged = codec_hdr.depth;
    xacodec_driver->decinfo->chdr = NULL;
    xacodec_driver->decinfo->map_flag = 0; /* xaFALSE */
    xacodec_driver->decinfo->map = NULL;
    xacodec_driver->decinfo->xs = xacodec_driver->decinfo->ys = 0;
    xacodec_driver->decinfo->special = 0;
    xacodec_driver->decinfo->extra = codec_hdr.extra;

    vidinfo->our_out_buffer = malloc(codec_hdr.y * codec_hdr.x * codec_hdr.depth);

    printf("out_buf size: %d\n", codec_hdr.y * codec_hdr.x * codec_hdr.depth);

    if (vidinfo->our_out_buffer == NULL)
    {
	odprintf(LOG_ERROR, "cannot allocate memory for output: %s",
	    strerror(errno));
	return(0);
    }

    printf("extra: %08x - %dx%d %dbit\n", codec_hdr.extra,
	codec_hdr.x, codec_hdr.y, codec_hdr.depth);
    return(1);
}

#define ACT_DLT_NORM	0x00000000
#define ACT_DLT_BODY	0x00000001
#define ACT_DLT_XOR	0x00000002
#define ACT_DLT_NOP	0x00000004
#define ACT_DLT_MAPD	0x00000008
#define ACT_DLT_DROP	0x00000010
#define ACT_DLT_BAD	0x80000000

//    unsigned int (*dec_func)(unsigned char *image, unsigned char *delta,
//	unsigned int dsize, XA_DEC_INFO *dec_info);

int xacodec_decode_frame(uint8_t *frame, int frame_size,
			uint8_t *dest, int skip_flag)
{
    unsigned int ret;
    int i;
    
    if (skip_flag > 0)
	printf("frame will be dropped..\n");

    xacodec_driver->decinfo->skip_flag = skip_flag;

    printf("frame: %08x (size: %d) - dest: %08x\n", frame, frame_size, dest);

    ret = xacodec_driver->dec_func(dest, frame, frame_size, xacodec_driver->decinfo);

    printf("ret: %lu : ", ret);
    
    for (i = 0; i < 10; i++)
    {
	if (frame[i] != dest[i])
	    printf("%d: [%02x] != [%02x] ", i, frame[i], dest[i]);
	else
	    printf("%d: [%02x] ", frame[i]);
    }

    if (ret == ACT_DLT_NORM)
    {
	printf("norm\n");
	return(TRUE);
    }

/*
    if (!(ret & ACT_DLT_MAPD))
	xacodec_driver->decinfo->map_flag = 0;
    else
    {
	xacodec_driver->decinfo->map_flag = 1;
	xacodec_driver->decinfo->map = ...
    }
*/

    if (ret & ACT_DLT_NOP)
    {
	printf("nop\n");
	return(FALSE); /* dst = 0 */
    }

    if (ret & ACT_DLT_DROP) /* by skip frames and errors */
    {
	printf("drop\n");
	return(FALSE);
    }


    if (ret & ACT_DLT_BAD)
    {
	printf("bad\n");
	return(FALSE);
    }

    if (ret & ACT_DLT_BODY)
    {
	printf("body\n");
	return(TRUE);
    }
    
    if (ret & ACT_DLT_XOR)
    {
	printf("xor\n");
	return(TRUE);
    }

    return(FALSE);
}

int xacodec_exit()
{
    if (xacodec_driver->close_func)
	xacodec_driver->close_func();
    dlclose(xacodec_driver->file_handler);
    if (xacodec_driver->decinfo != NULL)
	free(xacodec_driver->decinfo);
    if (xacodec_driver != NULL)
	free(xacodec_driver);
    return(TRUE);
}