/*
 * Copyright 2012-2013 Tri-Edge AI <triedgeai@gmail.com>
 * Copyright 2014 Haiku, Inc. All rights reserved.
 *
 * Distributed under the terms of the MIT license.
 *
 * Authors:
 *		Tri-Edge AI
 *		John Scipione, jscipione@gmail.com
 */


#include "Gravity.h"

#include "ConfigView.h"
#include "GravityView.h"

#include <ScreenSaver.h>
#include <View.h>

#include <stdlib.h>


Gravity::Gravity(BMessage* prefs, image_id imageID)
	:
	BScreenSaver(prefs, imageID),
	fGravityView(NULL),
	fConfigView(NULL)
{
	srand(time(NULL));

	if (prefs->IsEmpty()) {
		Config.ParticleCount = 1;
		Config.ShadeID = 2;
	} else {
		if (prefs->FindInt32("ParticleCount", &Config.ParticleCount) != B_OK)
			Config.ParticleCount = 1;

		if (prefs->FindInt32("ShadeID", &Config.ShadeID) != B_OK)
			Config.ShadeID = 2;
	}
}


status_t
Gravity::SaveState(BMessage* prefs) const
{
	prefs->AddInt32("ParticleCount", Config.ParticleCount);
	prefs->AddInt32("ShadeID", Config.ShadeID);
	return B_OK;
}


void
Gravity::StartConfig(BView* view)
{
	fConfigView = new ConfigView(view->Bounds(), this);
	view->AddChild(fConfigView);
}


status_t
Gravity::StartSaver(BView* view, bool preview)
{
	SetTickSize((1000 / 20) * 1000);
		// ~20 FPS
	fGravityView = new GravityView(view->Bounds(), this);
	view->AddChild(fGravityView);

	return B_OK;
}


void
Gravity::StopSaver()
{
	if (fGravityView != NULL)
		fGravityView->EnableDirectMode(false);
}


void
Gravity::DirectConnected(direct_buffer_info* info)
{
	if (fGravityView != NULL) {
		fGravityView->DirectConnected(info);
		fGravityView->EnableDirectMode(true);
	}
}


void
Gravity::Draw(BView*, int32 frame)
{
	fGravityView->DirectDraw();
}