#include "../core_include/api.h"
#include "../core_include/rect.h"
#include "../core_include/cmd_target.h"
#include "../core_include/wnd.h"
#include "../core_include/surface.h"
#include "../core_include/display.h"
#include "../core_include/resource.h"
#include <string.h>
#include <stdlib.h>

#define GL_ROUND_RGB_32(rgb) (rgb & 0xFFF8FCF8) //make RGB32 = RGB16

c_surface::c_surface(c_display* display,  unsigned int width, unsigned int height, unsigned int color_bytes)
{
	m_width = width;
	m_height = height;
	m_color_bytes = color_bytes;
	m_display = display;
	m_phy_fb = display->m_phy_fb;
	m_phy_write_index = &display->m_phy_write_index;
	m_fb = m_usr = NULL;
	m_top_zorder = m_max_zorder = Z_ORDER_LEVEL_0;
	m_is_active = false;
	m_frame_layers[Z_ORDER_LEVEL_0].rect = c_rect(0, 0, m_width, m_height);
}

void c_surface::set_surface(void* wnd_root, Z_ORDER_LEVEL max_z_order)
{
	m_usr = wnd_root;
	m_max_zorder = max_z_order;

	if (m_display->m_surface_cnt > 1)
	{
		m_fb = calloc(m_width * m_height, m_color_bytes);
	}
	
	for(int i = Z_ORDER_LEVEL_0; i < m_max_zorder; i++)
	{//Top layber fb always be NULL
		m_frame_layers[i].fb = (unsigned short*)calloc(m_width * m_height, sizeof(unsigned short));
		ASSERT(NULL != m_frame_layers[i].fb);
	}
}

void c_surface::draw_pixel(int x, int y, unsigned int rgb, unsigned int z_order)
{
	if (x >= m_width || y >= m_height || x < 0 || y < 0)
	{
		return;
	}
	if (z_order > m_max_zorder)
	{
		ASSERT(FALSE);
		return;
	}
	rgb = GL_ROUND_RGB_32(rgb);
	if (z_order == m_max_zorder)
	{
		return draw_pixel_on_fb(x, y, rgb);
	}

	if (z_order > m_top_zorder)
	{
		m_top_zorder = (Z_ORDER_LEVEL)z_order;
	}

	if (0 == m_frame_layers[z_order].rect.PtInRect(x, y))
	{
		ASSERT(FALSE);
		return;
	}
	((unsigned short*)(m_frame_layers[z_order].fb))[x + y * m_width] = GL_RGB_32_to_16(rgb);

	if (z_order == m_top_zorder)
	{
		return draw_pixel_on_fb(x, y, rgb);
	}

	bool is_covered = false;
	for (int tmp_z_order = Z_ORDER_LEVEL_MAX - 1; tmp_z_order > z_order; tmp_z_order--)
	{
		if (TRUE == m_frame_layers[tmp_z_order].rect.PtInRect(x, y))
		{
			is_covered = true;
			break;
		}
	}

	if (!is_covered)
	{
		draw_pixel_on_fb(x, y, rgb);
	}
}

void c_surface::draw_pixel_on_fb(int x, int y, unsigned int rgb)
{
	if (m_fb)
	{
		(m_color_bytes == 4) ? ((unsigned int*)m_fb)[y * m_width + x] = rgb : ((unsigned short*)m_fb)[y * m_width + x] = GL_RGB_32_to_16(rgb);
	}

	int display_width = m_display->get_width();
	int display_height = m_display->get_height();
	if (m_is_active && (x < display_width) && (y < display_height))
	{		
		if (m_color_bytes == 4)
		{
			((unsigned int*)m_phy_fb)[y * (m_display->get_width()) + x] = rgb;
		}
		else
		{
			((unsigned short*)m_phy_fb)[y * (m_display->get_width()) + x] = GL_RGB_32_to_16(rgb);
		}
		*m_phy_write_index = *m_phy_write_index + 1;
	}
}

void c_surface::fill_rect(int x0, int y0, int x1, int y1, unsigned int rgb, unsigned int z_order)
{
	rgb = GL_ROUND_RGB_32(rgb);
	if (z_order == m_max_zorder)
	{
		return fill_rect_on_fb(x0, y0, x1, y1, rgb);
	}

	if (z_order == m_top_zorder)
	{
		int x, y;
		unsigned short *mem_fb;
		unsigned int rgb_16 = GL_RGB_32_to_16(rgb);
		for (y = y0; y <= y1; y++)
		{
			x = x0;
			mem_fb = &((unsigned short*)m_frame_layers[z_order].fb)[y * m_width + x];
			for (; x <= x1; x++)
			{
				*mem_fb++ = rgb_16;
			}
		}
		return fill_rect_on_fb(x0, y0, x1, y1, rgb);
	}

	for (; y0 <= y1; y0++)
	{
		draw_hline(x0, x1, y0, rgb, z_order);
	}
}

void c_surface::fill_rect_on_fb(int x0, int y0, int x1, int y1, unsigned int rgb)
{
	if (x0 < 0 || y0 < 0 || x1 < 0 || y1 < 0 ||
		x0 >= m_width || x1 >= m_width || y0 >= m_height || y1 >= m_height)
	{
		ASSERT(FALSE);
	}
	int display_width = m_display->get_width();
	int display_height = m_display->get_height();

	if (m_color_bytes == 4)
	{
		int x;
		unsigned int *fb, *phy_fb;
		for (; y0 <= y1; y0++)
		{
			x = x0;
			fb = m_fb ? &((unsigned int*)m_fb)[y0 * m_width + x] : NULL;
			phy_fb = &((unsigned int*)m_phy_fb)[y0 * display_width + x];
			*m_phy_write_index = *m_phy_write_index + 1;
			for (; x <= x1; x++)
			{
				if (fb)
				{
					*fb++ = rgb;
				}
				if (m_is_active && (x < display_width) && (y0 < display_height))
				{
					*phy_fb++ = rgb;
				}
			}
		}
	}
	else if(m_color_bytes == 2)
	{
		int x;
		unsigned short *fb, *phy_fb;
		rgb = GL_RGB_32_to_16(rgb);
		for (; y0 <= y1; y0++)
		{
			x = x0;
			fb = m_fb ? &((unsigned short*)m_fb)[y0 * m_width + x] : NULL;
			phy_fb = &((unsigned short*)m_phy_fb)[y0 * display_width + x];
			*m_phy_write_index = *m_phy_write_index + 1;
			for (; x <= x1; x++)
			{
				if (fb)
				{
					*fb++ = rgb;
				}
				if (m_is_active && (x < display_width) && (y0 < display_height))
				{
					*phy_fb++ = rgb;
				}
			}
		}
	}
	
}

unsigned int c_surface::get_pixel(int x, int y, unsigned int z_order)
{
	if (x >= m_width || y >= m_height || x < 0 || y < 0 ||
			z_order >= Z_ORDER_LEVEL_MAX)
	{
		ASSERT(FALSE);
		return 0;
	}

	if (z_order == m_max_zorder)
	{
		if (m_fb)
		{
			return (m_color_bytes == 4) ? ((unsigned int*)m_fb)[y * m_width + x] : GL_RGB_16_to_32(((unsigned short*)m_fb)[y * m_width + x]);
		}
		else if(m_phy_fb)
		{
			return (m_color_bytes == 4) ? ((unsigned int*)m_phy_fb)[y * m_width + x] : GL_RGB_16_to_32(((unsigned short*)m_phy_fb)[y * m_width + x]);
		}
		return 0;
	}
	
	unsigned short rgb_16 = ((unsigned short*)(m_frame_layers[z_order].fb))[y * m_width + x];
	return GL_RGB_16_to_32(rgb_16);
}

void c_surface::draw_hline(int x0, int x1, int y, unsigned int rgb, unsigned int z_order)
{
	for (;x0 <= x1; x0++)
	{ draw_pixel(x0, y, rgb, z_order); }
}

void c_surface::draw_vline(int x, int y0, int y1, unsigned int rgb, unsigned int z_order)
{
	for (;y0 <= y1; y0++)
	{ draw_pixel(x, y0, rgb, z_order); }
}

void c_surface::draw_line(int x1, int y1, int x2, int y2, unsigned int rgb, unsigned int z_order)
{
	int dx, dy, e;
	dx = x2 - x1;
	dy = y2 - y1;

	if ((dx >= 0) && (dy >= 0))
	{
		if (dx >= dy)
		{
			e = dy - dx / 2;
			for(; x1 <= x2; x1++, e += dy)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { y1++; e -= dx; }
			}
		}
		else
		{
			e = dx - dy / 2;
			for(; y1 <= y2; y1++, e += dx)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { x1++; e -= dy; }
			}
		}
	}

	else if ((dx >= 0) && (dy < 0))
	{
		dy = -dy;
		if (dx >= dy)
		{
			e = dy - dx / 2;
			for(; x1 <= x2; x1++, e += dy)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { y1--; e -= dx; }
			}
		}
		else
		{
			e = dx - dy / 2;
			for(; y1 >= y2; y1--, e += dx)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { x1++; e -= dy; }
			}
		}
	}

	else if ((dx < 0) && (dy >= 0))
	{
		dx = -dx;
		if (dx >= dy)
		{
			e = dy - dx / 2;
			for(; x1 >= x2; x1--, e += dy)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { y1++; e -= dx; }
			}
		}
		else
		{
			e = dx - dy / 2;
			for(; y1 <= y2; y1++, e += dx)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { x1--; e -= dy; }
			}
		}
	}

	else if ((dx < 0) && (dy < 0))
	{
		dx = -dx;
		dy = -dy;
		if (dx >= dy)
		{
			e = dy - dx / 2;
			for(; x1 >= x2; x1--, e += dy)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { y1--; e -= dx; }
			}
		}
		else
		{
			e = dx - dy / 2;
			while (y1-- >= y2)
			for(; y1 >= y2; y1--, e += dx)
			{
				draw_pixel(x1, y1, rgb, z_order);
				if (e>0) { x1--; e -= dy; }
			}
		}
	}
}

void c_surface::draw_rect(int x0, int y0, int x1, int y1, unsigned int rgb, unsigned int z_order, unsigned int size)
{
	for (unsigned int offset = 0; offset < size; offset++)
	{
		draw_hline(x0 + offset, x1 - offset, y0 + offset, rgb, z_order);
		draw_hline(x0 + offset, x1 - offset, y1 - offset, rgb, z_order);
		draw_vline(x0 + offset, y0 + offset, y1 - offset, rgb, z_order);
		draw_vline(x1 - offset, y0 + offset, y1 - offset, rgb, z_order);
	}
}

int c_surface::set_frame_layer(c_rect& rect, unsigned int z_order)
{
	if (rect == m_frame_layers[z_order].rect)
	{
		return 0;
	}
	if (rect.m_left < 0 || rect.m_left >= m_width ||
		rect.m_right < 0 || rect.m_right >= m_width ||
		rect.m_top < 0 || rect.m_top >= m_height ||
		rect.m_bottom < 0 || rect.m_bottom >=m_height)
	{
		ASSERT(FALSE);
		return -1;
	}
	if (!(z_order > Z_ORDER_LEVEL_0 && z_order < Z_ORDER_LEVEL_MAX))
	{
		ASSERT(FALSE);
		return -2;
	}
	if (z_order < m_top_zorder)
	{
		ASSERT(FALSE);
		return -3;
	}
	m_top_zorder = (Z_ORDER_LEVEL)z_order;
	
	c_rect old_rect = m_frame_layers[z_order].rect;
	//Recover the lower layer
	int src_zorder = (Z_ORDER_LEVEL)(z_order - 1);
	int display_width = m_display->get_width();
	int display_height = m_display->get_height();

	for (int y = old_rect.m_top; y <= old_rect.m_bottom; y++)
	{
		for (int x = old_rect.m_left; x <= old_rect.m_right; x++)
		{
			if (!rect.PtInRect(x, y))
			{
				unsigned int rgb = ((unsigned short*)(m_frame_layers[src_zorder].fb))[x + y * m_width];
				draw_pixel_on_fb(x, y, GL_RGB_16_to_32(rgb));
			}
		}
	}

	m_frame_layers[z_order].rect = rect;
	if (rect.IsEmpty())
	{
		m_top_zorder = (Z_ORDER_LEVEL)(z_order - 1);
	}
	return 0;
}

int c_surface::flush_scrren(int left, int top, int right, int bottom)
{
	if(left < 0 || left >= m_width || right < 0 || right >= m_width ||
		top < 0 || top >= m_height || bottom < 0 || bottom >= m_height)
	{
		ASSERT(FALSE);
	}

	if(!m_is_active || (0 == m_phy_fb) || (0 == m_fb))
	{
		return -1;
	}

	int display_width = m_display->get_width();
	int display_height = m_display->get_height();

	left = (left >= display_width) ? (display_width - 1) : left;
	right = (right >= display_width) ? (display_width - 1) : right;
	top = (top >= display_height) ? (display_height - 1) : top;
	bottom = (bottom >= display_height) ? (display_height - 1) : bottom;

	for (int y = top; y < bottom; y++)
	{
		void* s_addr = (char*)m_fb + ((y * m_width + left) * m_color_bytes);
		void* d_addr = (char*)m_phy_fb + ((y * display_width + left) * m_color_bytes);
		memcpy(d_addr, s_addr, (right - left) * m_color_bytes);
	}
	*m_phy_write_index = *m_phy_write_index + 1;
	return 0;
}

bool c_surface::is_valid(c_rect rect)
{
	if (rect.m_left < 0 || rect.m_top < 0)
	{
		return false;
	}

	if (rect.m_right >= m_width || rect.m_bottom >= m_height)
	{
		return false;
	}
	return true;
}

//////////////////////////////////////////////////////////////////////////////////////
void c_surface_no_fb::fill_rect_on_fb(int x0, int y0, int x1, int y1, unsigned int rgb)
{
	if (!m_gfx_op)
	{
		return;
	}
	if (m_gfx_op->fill_rect)
	{
		return m_gfx_op->fill_rect(x0, y0, x1, y1, rgb);
	}

	if (m_gfx_op->draw_pixel && m_is_active)
	{
		for (int y = y0; y <= y1; y++)
		{
			for (int x = x0; x <= x1; x++)
			{
				m_gfx_op->draw_pixel(x, y, rgb);
			}
		}
	}

	if (!m_fb) { return; }
	if(m_color_bytes == 4)
	{
		unsigned int *fb;
		for (int y = y0; y <= y1; y++)
		{
			fb = &((unsigned int*)m_fb)[y0 * m_width + x0];
			for (int x = x0; x <= x1; x++)
			{
				*fb++ = rgb;
			}
		}
	}
	else if (m_color_bytes == 2)
	{
		unsigned short *fb;
		rgb = GL_RGB_32_to_16(rgb);
		for (int y = y0; y <= y1; y++)
		{
			fb = &((unsigned short*)m_fb)[y0 * m_width + x0];
			for (int x = x0; x <= x1; x++)
			{
				*fb++ = rgb;
			}
		}
	}
}

void c_surface_no_fb::draw_pixel_on_fb(int x, int y, unsigned int rgb)
{
	if (m_gfx_op && m_gfx_op->draw_pixel && m_is_active)
	{
		m_gfx_op->draw_pixel(x, y, rgb);
	}

	if (!m_fb) { return; }
	if (m_color_bytes == 4)
	{
		((unsigned int*)m_fb)[y * m_width + x] = rgb;
	}
	else if (m_color_bytes == 2)
	{
		((unsigned short*)m_fb)[y * m_width + x] = GL_RGB_32_to_16(rgb);
	}
}