/*
 * Copyright 2001-2006, Haiku.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Marc Flerackers (mflerackers@androme.be)
 *		Stefano Ceccherini (burton666@libero.it)
 */


#include "DrawingEngine.h"
#include "ServerBitmap.h"
#include "ServerPicture.h"
#include "ServerTokenSpace.h"
#include "ViewLayer.h"
#include "WindowLayer.h"

#include <ServerProtocol.h>
#include <TPicture.h>

#include <Bitmap.h>

#include <stdio.h>


static void
nop()
{
}


static void
move_pen_by(ViewLayer *view, BPoint delta)
{
	view->CurrentState()->SetPenLocation(delta - view->CurrentState()->PenLocation());
}


static void
stroke_line(ViewLayer *view, BPoint start, BPoint end)
{
	view->ConvertToScreenForDrawing(&start);
	view->ConvertToScreenForDrawing(&end);	
	view->Window()->GetDrawingEngine()->StrokeLine(start, end, view->CurrentState()->HighColor());
}


static void
stroke_rect(ViewLayer *view, BRect rect)
{
	view->ConvertToScreenForDrawing(&rect);	
	view->Window()->GetDrawingEngine()->StrokeRect(rect, view->CurrentState()->HighColor());
}


static void
fill_rect(ViewLayer *view, BRect rect)
{
	view->ConvertToScreenForDrawing(&rect);			
	view->Window()->GetDrawingEngine()->FillRect(rect, view->CurrentState()->HighColor());
}


static void
stroke_round_rect(ViewLayer *view, BRect rect, BPoint radii)
{
	view->ConvertToScreenForDrawing(&rect);	
	view->Window()->GetDrawingEngine()->DrawRoundRect(rect, radii.x, radii.y, view->CurrentState(), false);
}


static void
fill_round_rect(ViewLayer *view, BRect rect, BPoint radii)
{
	view->ConvertToScreenForDrawing(&rect);	
	view->Window()->GetDrawingEngine()->DrawRoundRect(rect, radii.x, radii.y, view->CurrentState(), true);
}


static void
stroke_bezier(ViewLayer *view, BPoint *points)
{
	for (int32 i = 0; i < 4; i++)
		view->ConvertToScreenForDrawing(&points[i]);

	view->Window()->GetDrawingEngine()->DrawBezier(points, view->CurrentState(), false);
}


static void
fill_bezier(ViewLayer *view, BPoint *points)
{
	for (int32 i = 0; i < 4; i++)
		view->ConvertToScreenForDrawing(&points[i]);

	view->Window()->GetDrawingEngine()->DrawBezier(points, view->CurrentState(), true);
}


static void
stroke_arc(ViewLayer *view, BPoint center, BPoint radii, float startTheta,
			   float arcTheta)
{
	/*view->ConvertToScreenForDrawing(&r);
	view->Window()->GetDrawingEngine()->DrawArc(r, angle, span, view->CurrentState(), false);*/
}


static void
fill_arc(ViewLayer *view, BPoint center, BPoint radii, float startTheta,
			 float arcTheta)
{
	/*view->ConvertToScreenForDrawing(&r);
	view->Window()->GetDrawingEngine()->DrawArc(r, angle, span, view->CurrentState(), true);*/
}


static void
stroke_ellipse(ViewLayer *view, BPoint center, BPoint radii)
{
	/*view->ConvertToScreenForDrawing(&rect);
	view->Window()->GetDrawingEngine()->DrawEllipse(rect, view->CurrentState(), false);*/
}


static void
fill_ellipse(ViewLayer *view, BPoint center, BPoint radii)
{
	/*view->ConvertToScreenForDrawing(&rect);
	view->Window()->GetDrawingEngine()->DrawEllipse(rect, view->CurrentState(), true);*/
}


static void
stroke_polygon(ViewLayer *view, int32 numPoints, BPoint *points, bool isClosed)
{
	for (int32 i = 0; i < numPoints; i++)
		view->ConvertToScreenForDrawing(&points[i]);

	BRect polyFrame = BRect(points[0], points[0]);

	for (int32 i = 1; i < numPoints; i++) {
		if (points[i].x < polyFrame.left)
			polyFrame.left = points[i].x;
		if (points[i].y < polyFrame.top)
			polyFrame.top = points[i].y;
		if (points[i].x > polyFrame.right)
			polyFrame.right = points[i].x;
		if (points[i].y > polyFrame.bottom)
			polyFrame.bottom = points[i].y;
	}

	view->Window()->GetDrawingEngine()->DrawPolygon(points, numPoints, polyFrame, view->CurrentState(),
							false, isClosed && numPoints > 2);
}


static void
fill_polygon(ViewLayer *view, int32 numPoints, BPoint *points)
{
	for (int32 i = 0; i < numPoints; i++)
		view->ConvertToScreenForDrawing(&points[i]);

	BRect polyFrame = BRect(points[0], points[0]);

	for (int32 i = 1; i < numPoints; i++) {
		if (points[i].x < polyFrame.left)
			polyFrame.left = points[i].x;
		if (points[i].y < polyFrame.top)
			polyFrame.top = points[i].y;
		if (points[i].x > polyFrame.right)
			polyFrame.right = points[i].x;
		if (points[i].y > polyFrame.bottom)
			polyFrame.bottom = points[i].y;
	}

	view->Window()->GetDrawingEngine()->DrawPolygon(points, numPoints, polyFrame, view->CurrentState(),
							true, true);
}


static void
stroke_shape(ViewLayer *view, BShape *shape)
{
	//TODO: Oh-ho... how do I access BShape private members ?
	/*for (int32 i = 0; i < ptCount; i++)
		fCurrentLayer->ConvertToScreenForDrawing(&ptList[i]);

	view->Window()->GetDrawingEngine()->DrawShape(shapeFrame, opCount, opList, ptCount, ptList,
				fCurrentLayer->CurrentState(), code == AS_FILL_SHAPE);*/
}


static void
fill_shape(ViewLayer *view, BShape *shape)
{
	//view->FillShape(shape);

	printf("FillShape\n");
}


static void
draw_string(ViewLayer *view, char *string, float deltaSpace, float deltaNonSpace)
{
	BPoint location = view->CurrentState()->PenLocation();
	escapement_delta delta = {deltaSpace, deltaNonSpace };
	view->ConvertToScreenForDrawing(&location);
	view->Window()->GetDrawingEngine()->DrawString(string, strlen(string), location,
		view->CurrentState(), &delta);
	// TODO: Update pen location ?
	
}


static void
draw_pixels(ViewLayer *view, BRect src, BRect dest, int32 width, int32 height,
				 int32 bytesPerRow, int32 pixelFormat, int32 flags, void *data)
{
	// TODO: Review this
	UtilityBitmap bitmap(BRect(0, 0, width - 1, height - 1), (color_space)pixelFormat, flags, bytesPerRow);
	
	if (!bitmap.IsValid())
		return;
	
	uint8 *pixels = (uint8 *)data;
	uint8 *destPixels = (uint8 *)bitmap.Bits();
	for (int32 h = 0; h < height; h++) {
		memcpy(destPixels, pixels, bytesPerRow);
		pixels += bytesPerRow;
		destPixels += bytesPerRow;
	}

	view->ConvertToScreenForDrawing(&dest);
	
	view->Window()->GetDrawingEngine()->DrawBitmap(&bitmap, src, dest, view->CurrentState());
}


static void
set_clipping_rects(ViewLayer *view, BRect *rects, uint32 numRects)
{
	// TODO: This is too slow, we should copy the rects directly to BRegion's internal data
	BRegion region;
	for (uint32 c = 0; c < numRects; c++)
		region.Include(rects[c]);
	view->SetUserClipping(&region);	
}


static void
clip_to_picture(ViewLayer *view, BPicture *picture, BPoint pt,
				   bool clip_to_inverse_picture)
{
	printf("ClipToPicture(picture, BPoint(%.2f, %.2f), %s)\n", pt.x, pt.y,
		clip_to_inverse_picture ? "inverse" : "");
}


static void
push_state(ViewLayer *view)
{
	view->PushState();
}


static void
pop_state(ViewLayer *view)
{
	view->PopState();
}


static void
enter_state_change(ViewLayer *view)
{
	printf("EnterStateChange\n");
}


static void
exit_state_change(ViewLayer *view)
{
	printf("ExitStateChange\n");
}


static void 
enter_font_state(ViewLayer *view)
{
	printf("EnterFontState\n");
}


static void
exit_font_state(ViewLayer *view)
{
	printf("ExitFontState\n");
}


static void 
set_origin(ViewLayer *view, BPoint pt)
{
	view->CurrentState()->SetOrigin(pt);
}


static void
set_pen_location(ViewLayer *view, BPoint pt)
{
	view->CurrentState()->SetPenLocation(pt);
}


static void
set_drawing_mode(ViewLayer *view, drawing_mode mode)
{
	view->CurrentState()->SetDrawingMode(mode);
}


static void
set_line_mode(ViewLayer *view, cap_mode capMode, join_mode joinMode, float miterLimit)
{
	DrawState *state = view->CurrentState();
	state->SetLineCapMode(capMode);
	state->SetLineJoinMode(joinMode);
	state->SetMiterLimit(miterLimit);
}


static void
set_pen_size(ViewLayer *view, float size)
{
	view->CurrentState()->SetPenSize(size);
}


static void
set_fore_color(ViewLayer *view, rgb_color color)
{
	view->CurrentState()->SetHighColor(RGBColor(color));
}


static void
set_back_color(ViewLayer *view, rgb_color color)
{
	view->CurrentState()->SetLowColor(RGBColor(color));
}


static void
set_stipple_pattern(ViewLayer *view, pattern p)
{
	printf("SetStipplePattern\n");
}


static void
set_scale(ViewLayer *view, float scale)
{
	view->CurrentState()->SetScale(scale);
}


static void
set_font_family(ViewLayer *view, char *family)
{
	printf("SetFontFamily(%s)\n", family);
}


static void
set_font_style(ViewLayer *view, char *style)
{
	printf("SetFontStyle(%s)\n", style);
}


static void
set_font_spacing(ViewLayer *view, int32 spacing)
{
	printf("SetFontSpacing(%ld)\n", spacing);
}


static void
set_font_size(ViewLayer *view, float size)
{
	ServerFont font;
	font.SetSize(size);
	view->CurrentState()->SetFont(font, B_FONT_SIZE);
}


static void
set_font_rotate(ViewLayer *view, float rotation)
{
	printf("SetFontRotate(%.2f)\n", rotation);
}


static void
set_font_encoding(ViewLayer *view, int32 encoding)
{
	printf("SetFontEncoding(%ld)\n", encoding);
}


static void
set_font_flags(ViewLayer *view, int32 flags)
{
	printf("SetFontFlags(%ld)\n", flags);
}


static void
set_font_shear(ViewLayer *view, float shear)
{
	printf("SetFontShear(%.2f)\n", shear);
}


static void
set_font_face(ViewLayer *view, int32 flags)
{
	printf("SetFontFace(%ld)\n", flags);
}


static void
reserved()
{
}


const void *tableEntries[] = {
	(const void *)nop,
	(const void *)move_pen_by,
	(const void *)stroke_line,
	(const void *)stroke_rect,
	(const void *)fill_rect,
	(const void *)stroke_round_rect,
	(const void *)fill_round_rect,
	(const void *)stroke_bezier,
	(const void *)fill_bezier,
	(const void *)stroke_arc,
	(const void *)fill_arc,
	(const void *)stroke_ellipse,
	(const void *)fill_ellipse,
	(const void *)stroke_polygon,
	(const void *)fill_polygon,
	(const void *)stroke_shape,
	(const void *)fill_shape,
	(const void *)draw_string,
	(const void *)draw_pixels,
	(const void *)reserved,
	(const void *)set_clipping_rects,
	(const void *)clip_to_picture,
	(const void *)push_state,
	(const void *)pop_state,
	(const void *)enter_state_change,
	(const void *)exit_state_change,
	(const void *)enter_font_state,
	(const void *)exit_font_state,
	(const void *)set_origin,
	(const void *)set_pen_location,
	(const void *)set_drawing_mode,
	(const void *)set_line_mode,
	(const void *)set_pen_size,
	(const void *)set_fore_color,
	(const void *)set_back_color,
	(const void *)set_stipple_pattern,
	(const void *)set_scale,
	(const void *)set_font_family,
	(const void *)set_font_style,
	(const void *)set_font_spacing,
	(const void *)set_font_size,
	(const void *)set_font_rotate,
	(const void *)set_font_encoding,
	(const void *)set_font_flags,
	(const void *)set_font_shear,
	(const void *)reserved,
	(const void *)set_font_face,
};


// ServerPicture
ServerPicture::ServerPicture()
{
	fToken = gTokenSpace.NewToken(kPictureToken, this);
}


ServerPicture::ServerPicture(const ServerPicture &picture)
{
	fToken = gTokenSpace.NewToken(kPictureToken, this);

	AddData(picture.Data(), picture.DataLength());
}


ServerPicture::~ServerPicture()
{
}


void
ServerPicture::BeginOp(int16 op)
{
	fStack.push(fData.Position());
	fData.Write(&op, sizeof(op));
	
	// Init the size of the opcode block to 0
	size_t size = 0;
	fData.Write(&size, sizeof(size));
}


void
ServerPicture::EndOp()
{
	off_t curPos = fData.Position();
	off_t stackPos = fStack.top();
	fStack.pop();
	
	// The size of the op is calculated like this:
	// current position on the stream minus the position on the stack,
	// minus the space occupied by the op code itself (int16)
	// and the space occupied by the size field (size_t) 
	size_t size = curPos - stackPos - sizeof(size_t) - sizeof(int16);

	// Size was set to 0 in BeginOp(). Now we overwrite it with the correct value
	fData.Seek(stackPos + sizeof(int16), SEEK_SET);
	fData.Write(&size, sizeof(size));
	fData.Seek(curPos, SEEK_SET);
}


void
ServerPicture::EnterStateChange()
{
	// BeginOp(B_PIC_ENTER_STATE_CHANGE);
}


void
ServerPicture::ExitStateChange()
{
	// EndOp();
}


void
ServerPicture::EnterFontChange()
{
	// BeginOp(B_PIC_ENTER_FONT_STATE);
}


void
ServerPicture::ExitFontChange()
{
	// EndOp();
}


void
ServerPicture::AddInt8(int8 data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddInt16(int16 data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddInt32(int32 data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddInt64(int64 data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddFloat(float data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddCoord(BPoint data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddRect(BRect data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddColor(rgb_color data)
{
	fData.Write(&data, sizeof(data));
}


void
ServerPicture::AddString(const char *data)
{
	int32 len = data ? strlen(data) : 0;
	fData.Write(&len, sizeof(int32));
	fData.Write(data, len);
}


void
ServerPicture::AddData(const void *data, int32 size)
{
	fData.Write(data, size);
}


void
ServerPicture::SyncState(ViewLayer *view)
{
	// TODO:
}


void
ServerPicture::Play(ViewLayer *view)
{
	PicturePlayer player(const_cast<void *>(fData.Buffer()), fData.BufferLength(), NULL);
	player.Play(const_cast<void **>(tableEntries), sizeof(tableEntries) / sizeof(void *), view);
}