#include "LEDAnimation.h"

#include <InterfaceDefs.h>

#include <stdio.h>

#define SNOOZE_TIME 150000


LEDAnimation::LEDAnimation()
	:
	fThread(-1),
	fRunning(false),
	fOrigModifiers(::modifiers())
{
}


LEDAnimation::~LEDAnimation()
{
	Stop();
}


void
LEDAnimation::Start()
{
	// don't do anything if the thread is already running
	if (fThread >= 0)
		return;

	fOrigModifiers = ::modifiers();
	::set_keyboard_locks(0);
	fRunning = true;
	fThread = ::spawn_thread(AnimationThread,"LED thread",B_NORMAL_PRIORITY,this);
	::resume_thread(fThread);
}


void
LEDAnimation::Stop()
{
	// don't do anything if the thread doesn't run
	if (fThread < 0)
		return;

	fRunning = false;
	status_t result;
	::wait_for_thread(fThread,&result);

	::set_keyboard_locks(fOrigModifiers);
}


int32
LEDAnimation::AnimationThread(void* data)
{
	LEDAnimation *anim = (LEDAnimation*)data;

	while (anim->fRunning) {
		LED(B_NUM_LOCK,true);
		LED(B_NUM_LOCK,false);

		LED(B_CAPS_LOCK,true);
		LED(B_CAPS_LOCK,false);

		LED(B_SCROLL_LOCK,true);
		LED(B_SCROLL_LOCK,false);

		LED(B_CAPS_LOCK,true);
		LED(B_CAPS_LOCK,false);
	}
	anim->fThread = -1;
	return 0;
}


void
LEDAnimation::LED(uint32 mod,bool on)
{
	uint32 current_modifiers = ::modifiers();
	if (on)
		current_modifiers |= mod;
	else
		current_modifiers &= ~mod;
	::set_keyboard_locks(current_modifiers);
	if (on)
		::snooze(SNOOZE_TIME);
}