#include "SDL_mxf.h"


void *lock(SDL_Surface *surf, Uint8 type) {
	if(SDL_MUSTLOCK(surf)) {
		if(SDL_LockSurface(surf) < 0)
			return 0;
	}
	return surf->pixels;
}



void unlock(SDL_Surface *surf) {
	if(SDL_MUSTLOCK(surf)) {
		SDL_UnlockSurface(surf);
	}
}

void setpixel(void *buff, Uint32 x, Uint32 y, Uint32 color, Uint8 type, Uint16 pitch) {
	static Uint8 *ubuff8 = 0;

	if(x < 640 && y < 480) {
		switch(type) {
			case 1:
			case 8:
				{ 
					Uint8 *buf = (Uint8*)buff;
					buf += (y * pitch) + x;
					*buf = (Uint8) color;

				}
				break;
			case 2:
			case 16:
				{
					static Uint16 *ubuff16 = 0;
					ubuff8 = (Uint8*) buff;
					ubuff8 += (y * pitch) + (x*2);
					
					ubuff16 = (Uint16*) ubuff8;
					*ubuff16 = (Uint16) color;
				}
				break;
			case 3:
			case 24:
				{
					static char c1 = 0,c2 = 0,c3 = 0;

					ubuff8 = (Uint8*) buff;
					ubuff8 += (y * pitch) + (x*3);
					if(SDL_BYTEORDER == SDL_LIL_ENDIAN) {
						c1 = (color & 0xFF0000) >> 16;
						c2 = (color & 0x00FF00) >> 8;
						c3 = (color & 0x0000FF);
					} else {
						c3 = (color & 0xFF0000) >> 16;
						c2 = (color & 0x00FF00) >> 8;
						c1 = (color & 0x0000FF);
					}
					ubuff8[0] = c3;
					ubuff8[1] = c2;
					ubuff8[2] = c1;
				}
				break;
			case 4:
			case 32:
				{
				static Uint32 *ubuff32 = 0;
				ubuff8 = (Uint8*) buff;
				ubuff8 += (y*pitch) + (x*4);
				ubuff32 = (Uint32*)ubuff8;
				*ubuff32 = color;
				}
				break;
		}

	}
}

Uint32 getpixel(SDL_Surface *surf, int x, int y, Uint8 type, Uint16 pitch, SDL_Color *col) {
	static Uint8 *bitz = 0;
	Uint32 bpp = (Uint32) type;
	bitz = ((Uint8 *)surf->pixels)+y*pitch+x*(bpp/8);
	switch(type) {
		case 8:
		case 1:
			return *((Uint8 *)surf->pixels + y * pitch + x);
			break;
		case 16:
		case 2:
			return *((Uint16 *)surf->pixels + y * pitch/2 + x);
			break;
		case 3:
		case 24:
			{
				Uint8 r, g, b;
				r = *((bitz)+surf->format->Rshift/8);
				g = *((bitz)+surf->format->Gshift/8);
				b = *((bitz)+surf->format->Bshift/8);
				col->r = r;
				col->g = g;
				col->b = b;
				return SDL_MapRGB(surf->format,r,g,b);
			
			}
		case 32:
		case 4:
			return *((Uint32 *)surf->pixels + y * pitch/4 + x);
			break;
	}

	return 0;
}