/*
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * This program 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 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
 * All rights reserved.
 *
 * The Original Code is: all of this file.
 *
 * Contributor(s): none yet.
 *
 * ***** END GPL LICENSE BLOCK *****
 *
 */

/** \file blender/avi/intern/avi_rgb.c
 *  \ingroup avi
 *
 * This is external code. Converts rgb-type avi-s.
 */

#include <stdlib.h>
#include <string.h>

#include "MEM_guardedalloc.h"

#include "AVI_avi.h"
#include "avi_rgb.h"

/* implementation */

void *avi_converter_from_avi_rgb(AviMovie *movie, int stream, unsigned char *buffer, int *size)
{
	int x, y, i, rowstride;
	unsigned char *buf;
	AviBitmapInfoHeader *bi;
	short bits = 32;
	
	(void)size; /* unused */

	bi = (AviBitmapInfoHeader *) movie->streams[stream].sf;
	if (bi) bits = bi->BitCount;

	if (bits == 16) {
		unsigned short *pxl;
		unsigned char *to;
#ifdef __BIG_ENDIAN__
		unsigned char  *pxla;
#endif
		
		buf = MEM_mallocN(movie->header->Height * movie->header->Width * 3, "fromavirgbbuf");

		y = movie->header->Height;
		to = buf;
				
		while (y--) {
			pxl = (unsigned short *) (buffer + y * movie->header->Width * 2);
			
#ifdef __BIG_ENDIAN__
			pxla = (unsigned char *)pxl;
#endif

			x = movie->header->Width;
			while (x--) {
#ifdef __BIG_ENDIAN__
				i = pxla[0];
				pxla[0] = pxla[1];
				pxla[1] = i;
	
				pxla += 2;
#endif
			
				*(to++) = ((*pxl >> 10) & 0x1f) * 8;
				*(to++) = ((*pxl >> 5) & 0x1f) * 8;
				*(to++) = (*pxl & 0x1f) * 8;
				pxl++;
			}
		}

		MEM_freeN(buffer);
		
		return buf;
	}
	else {
		buf = MEM_mallocN(movie->header->Height * movie->header->Width * 3, "fromavirgbbuf");
	
		rowstride = movie->header->Width * 3;
		if ((bits != 16) && (movie->header->Width % 2)) rowstride++;
	
		for (y = 0; y < movie->header->Height; y++) {
			memcpy(&buf[y * movie->header->Width * 3], &buffer[((movie->header->Height - 1) - y) * rowstride], movie->header->Width * 3);
		}
	
		for (y = 0; y < movie->header->Height * movie->header->Width * 3; y += 3) {
			i = buf[y];
			buf[y] = buf[y + 2];
			buf[y + 2] = i;
		}
	
		MEM_freeN(buffer);
	
		return buf;
	}
}

void *avi_converter_to_avi_rgb(AviMovie *movie, int stream, unsigned char *buffer, int *size)
{
	int y, x, i, rowstride;
	unsigned char *buf;

	(void)stream; /* unused */

	rowstride = movie->header->Width * 3;
	/* AVI files has uncompressed lines 4-byte aligned */
	rowstride = (rowstride + 3) & ~3;

	*size = movie->header->Height * rowstride;
	buf = MEM_mallocN(*size, "toavirgbbuf");

	for (y = 0; y < movie->header->Height; y++) {
		memcpy(&buf[y * rowstride], &buffer[((movie->header->Height - 1) - y) * movie->header->Width * 3], movie->header->Width * 3);
	}

	for (y = 0; y < movie->header->Height; y++) {
		for (x = 0; x < movie->header->Width * 3; x += 3) {
			i = buf[y * rowstride + x];
			buf[y * rowstride + x] = buf[y * rowstride + x + 2];
			buf[y * rowstride + x + 2] = i;
		}
	}

	MEM_freeN(buffer);

	return buf;
}