//
// StickIt
// File: JoystickWindow.cpp
// Joystick window.
// Sampel code used in "Getting a Grip on BJoystick" by Eric Shepherd
//

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

#include <Application.h>
#include <Joystick.h>
#include <String.h>
#include <StringView.h>

#include "JoystickWindow.h"

rgb_color rgb_black = {0, 0, 0, 255};
rgb_color rgb_red = {255, 0, 0, 255};
rgb_color rgb_grey = {216, 216, 216};

int32 hatX[9] = {10, 10, 20, 20, 20, 10, 0, 0, 0};
int32 hatY[9] = {10, 0, 0, 10, 20, 20, 20, 10, 0};

JoystickWindow::JoystickWindow(const char *deviceName, BJoystick *stick,
	BRect rect)
	: BWindow(rect, deviceName, B_TITLED_WINDOW,
	B_NOT_RESIZABLE|B_NOT_ZOOMABLE)
{
	fView = new JoystickView(Bounds(), stick);
	fView->SetViewColor(rgb_grey);
	fView->SetLowColor(rgb_grey);
	ResizeTo(fView->Bounds().Width(), fView->Bounds().Height());
	AddChild(fView);
	SetPulseRate(100000.0);
}


bool
JoystickWindow::QuitRequested(void) {
	return true;
}


JoystickView::JoystickView(BRect frame, BJoystick *stick)
	:BView(frame, "jview", B_FOLLOW_ALL_SIDES, B_WILL_DRAW|B_PULSE_NEEDED),
	fStick(stick)
{
	// Set up the controls
	BString name;
	fStick->GetControllerName(&name);
	BRect rect(5, 5, 350, 25);
	BStringView *stickName = new BStringView(rect, "stickname", name.String());
	stickName->SetFontSize(18);
	AddChild(stickName);

	rect = _BuildButtons(stick);
	rect.top += 15;
	rect.bottom += 15;
	rect = _BuildHats(stick, rect);
	_BuildAxes(stick, rect);
}


JoystickView::~JoystickView() {
	delete fStick;
}


void
JoystickView::Pulse(void) {
	Window()->Lock();
	Draw(Bounds());
	Window()->Unlock();
}


void
JoystickView::Draw(BRect updateRect)
{
	int32 numButtons = fStick->CountButtons();
	int32 numHats = fStick->CountHats();
	int32 numAxes = fStick->CountAxes();
	bool *buttons = (bool *) malloc(sizeof(bool) * numButtons);
	int16 *axes = (int16 *) malloc(sizeof(int16) * numAxes);
	uint8 *hats = (uint8 *) malloc(numHats);
	fStick->Update();

	// Buttons first
	BRect r(105, 50, 115, 60);
	fStick->GetButtonValues(buttons);
	for (int32 i = 0; i < numButtons; i++) {
		if (buttons[i]) {
			FillRect(r, B_SOLID_HIGH);
		} else {
			r.InsetBy(1, 1);
			FillRect(r, B_SOLID_LOW);
			r.InsetBy(-1, -1);
			StrokeRect(r, B_SOLID_HIGH);
		}
		r.top += 18;
		r.bottom += 18;
	}

	// Now hats
	fStick->GetHatValues(hats);
	r.top += 15;
	r.bottom = r.top + 30;
	r.left = 105;
	r.right = r.left + 30;
	BRect curHatRect;
	for (int32 i = 0; i < numHats; i++) {
		BeginLineArray(8);
		AddLine(r.LeftTop(), r.RightTop(), rgb_black);
		AddLine(r.LeftTop(), r.LeftBottom(), rgb_black);
		AddLine(r.RightTop(), r.RightBottom(), rgb_black);
		AddLine(r.LeftBottom(), r.RightBottom(), rgb_black);
		AddLine(BPoint(r.left+10, r.top), BPoint(r.left+10, r.bottom),
			rgb_black);
		AddLine(BPoint(r.left+20, r.top), BPoint(r.left+20, r.bottom),
			rgb_black);
		AddLine(BPoint(r.left, r.top+10), BPoint(r.right, r.top+10),
			rgb_black);
		AddLine(BPoint(r.left, r.top+20), BPoint(r.right, r.top+20),
			rgb_black);
		EndLineArray();
		curHatRect.Set(r.left, r.top, r.left+10, r.top+10);
		curHatRect.OffsetBy(hatX[hats[i]], hatY[hats[i]]);
		fLastHatRect.InsetBy(1, 1);
		FillRect(fLastHatRect, B_SOLID_LOW);
		fLastHatRect = curHatRect;
		fLastHatRect.InsetBy(1, 1);
		FillRect(fLastHatRect, B_SOLID_HIGH);
		fLastHatRect.InsetBy(-1, -1);
		r.top += 20;
		r.bottom += 20;
	}

	// Now the joystick
	r.Set(200, 50, 350, 200);
	FillRect(r, B_SOLID_HIGH);
	fStick->GetAxisValues(axes);
	float x = axes[0];
	float y = axes[1];
	SetHighColor(rgb_red);
	x += 32768;
	y -= 32768;
	x *= 0.0021362304;
	y *= 0.0021362304;
	x += 205;
	y += 195;
	FillEllipse(BPoint(x,y), 5, 5);
	SetHighColor(rgb_black);

	// Finally, other axes
	r.Set(200, 220, 350, 234);
	for (int32 i = 2; i < numAxes; i++) {
		FillRect(r, B_SOLID_HIGH);
		x = axes[i];
		x += 32768;
		x *= 0.0021362304;
		x += 205;
		BRect thumbRect(x - 3, r.top, x + 3, r.bottom);
		SetHighColor(rgb_red);
		FillRoundRect(thumbRect, 3, 3);
		SetHighColor(rgb_black);
		r.top += 20;
		r.bottom += 20;
	}
	free(buttons);
	free(axes);
	free(hats);
}

//----------------------- Pragma mark ------------------------------------//

BRect
JoystickView::_BuildButtons(BJoystick *stick)
{
	BString name;
	int32 numButtons = stick->CountButtons();
	BRect rect(5, 50, 100, 64);
	for (int32 i = 0; i < numButtons; i++) {
		stick->GetButtonNameAt(i, &name);
		rect = _BuildString(name, "buttonlabel", 18, rect);
	}
	return rect;
}


BRect
JoystickView::_BuildHats(BJoystick *stick, BRect rect)
{
	BString name;
	int32 numHats = stick->CountHats();
	for (int32 i = 0; i < numHats; i++) {
		stick->GetHatNameAt(i, &name);
		rect = _BuildString(name, "hatlabel", 40, rect);
	}
	return rect;
}


BRect
JoystickView::_BuildString(BString name, const char* strName, int number,
	BRect rect)
{
	name.Append(":");
	BStringView *sview = new BStringView(rect, strName, name.String());
	sview->SetAlignment(B_ALIGN_RIGHT);
	sview->SetFont(be_bold_font);
	AddChild(sview);
	rect.top += number;
	rect.bottom += number;
	return rect;
}


void
JoystickView::_BuildAxes(BJoystick *stick, BRect rect)
{
	float col1bot = rect.bottom - 30;
	int32 numAxes = stick->CountAxes();
	// We assume that the first two axes are the x and y axes.
	rect.Set(130, 50, 195, 64);
	BStringView *sview = new BStringView(rect, "sticklabel", "Stick:");
	sview->SetFont(be_bold_font);
	sview->SetAlignment(B_ALIGN_RIGHT);
	AddChild(sview);

	BString name;
	int32 i;
	// Now make labels for all the solitary axes
	rect.Set(130, 200, 195, 234);
	for (i = 2; i < numAxes; i++) {
		stick->GetAxisNameAt(i, &name);
		rect = _BuildString(name, "hatlabel", 20, rect);
	}

	fLastHatRect.Set(0, 0, 0, 0);
	if (rect.bottom-10 > col1bot) {
		if (i == 2) {
			col1bot = 205;
		} else {
			col1bot = rect.bottom - 10;
		}
	}
	ResizeTo(355, col1bot);
}