phoenix-firestorm/indra/newview/lggbeammapfloater.cpp

/** 
 * @file lggbeammapfloater.cpp
 * @brief Floater for beam shapes
 * @copyright Copyright (c) 2011 LordGregGreg Back
 *
 * This code is free. It comes
 * WITHOUT ANY WARRANTY, without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. You can redistribute it
 * and/or modify it under the terms of the Do What The Fuck You Want
 * To Public License, Version 2, as published by Sam Hocevar. See
 * http://sam.zoy.org/wtfpl/COPYING for more details.
 */

#include "llviewerprecompiledheaders.h"

#include "lggbeammapfloater.h"

#include "fspanelprefs.h"
#include "lggbeammaps.h"
#include "llcolorswatch.h"
#include "llfilepicker.h"
#include "llsdserialize.h"
#include "llviewercontrol.h"
#include "llviewermenufile.h"

lggBeamMapFloater::lggBeamMapFloater(const LLSD& seed) : LLFloater(seed),
	mContextConeOpacity(0.f),
	mContextConeInAlpha(0.f),
	mContextConeOutAlpha(0.f),
	mContextConeFadeTime(0.f)
{
	mContextConeInAlpha = gSavedSettings.getF32("ContextConeInAlpha");
	mContextConeOutAlpha = gSavedSettings.getF32("ContextConeOutAlpha");
	mContextConeFadeTime = gSavedSettings.getF32("ContextConeFadeTime");
}

lggBeamMapFloater::~lggBeamMapFloater()
{
}

BOOL lggBeamMapFloater::postBuild()
{
	getChild<LLUICtrl>("beamshape_save")->setCommitCallback(boost::bind(&lggBeamMapFloater::onClickSave, this));
	getChild<LLUICtrl>("beamshape_clear")->setCommitCallback(boost::bind(&lggBeamMapFloater::onClickClear, this));
	getChild<LLUICtrl>("beamshape_load")->setCommitCallback(boost::bind(&lggBeamMapFloater::onClickLoad, this));
	getChild<LLUICtrl>("cancel")->setCommitCallback(boost::bind(&lggBeamMapFloater::closeFloater, this, false));

	getChild<LLColorSwatchCtrl>("back_color_swatch")->setCommitCallback(boost::bind(&lggBeamMapFloater::onBackgroundChange, this));
	getChild<LLColorSwatchCtrl>("beam_color_swatch")->setColor(LLColor4::red);

	mBeamshapePanel = getChild<LLPanel>("beamshape_draw");

	return TRUE;
}

void lggBeamMapFloater::draw()
{
	static LLCachedControl<F32> max_opacity(gSavedSettings, "PickerContextOpacity", 0.4f);
	drawConeToOwner(mContextConeOpacity, max_opacity, mFSPanel->getChild<LLButton>("custom_beam_btn"), mContextConeFadeTime, mContextConeInAlpha, mContextConeOutAlpha);

	LLFloater::draw();
	LLRect rec = mBeamshapePanel->getRect();

	gGL.pushMatrix();
	gGL.color4fv(LLColor4::white.mV);
	gl_circle_2d(rec.getCenterX(), rec.getCenterY(), 2.0f, 30, false);
	gGL.color4fv(LLColor4::black.mV);
	gl_circle_2d(rec.getCenterX(), rec.getCenterY(), 30.0f, 30, false);
	gGL.color4fv(LLColor4::white.mV);
	gl_circle_2d(rec.getCenterX(), rec.getCenterY(), 60.0f, 30, false);
	gGL.color4fv(LLColor4::black.mV);
	gl_circle_2d(rec.getCenterX(), rec.getCenterY(), 90.0f, 30, false);
	gGL.color4fv(LLColor4::white.mV);
	gl_circle_2d(rec.getCenterX(), rec.getCenterY(), 120.0f, 30, false);

	for (std::vector<lggPoint>::iterator it = mDots.begin(); it != mDots.end(); ++it)
	{
		lggPoint dot = *it;

		gGL.color4fv(LLColor4::white.mV);
		gl_circle_2d(dot.x, dot.y, 9.0f, 30, true);

		gGL.color4fv(LLColor4::black.mV);
		gl_circle_2d(dot.x, dot.y, 8.0f, 30, true);

		gGL.color4fv(dot.c.mV);
		gl_circle_2d(dot.x, dot.y, 7.0f, 30, true);
	}
	gGL.popMatrix();
}

BOOL lggBeamMapFloater::handleMouseDown(S32 x, S32 y, MASK mask)
{
	if (y > 39 && x > 16 && x < 394 && y < 317)
	{
		lggPoint a;
		a.x = x;
		a.y = y;
		a.c = getChild<LLColorSwatchCtrl>("beam_color_swatch")->get();
		mDots.push_back(a);
	}

	return LLFloater::handleMouseDown(x, y, mask);
}

BOOL lggBeamMapFloater::handleRightMouseDown(S32 x, S32 y, MASK mask)
{
	std::vector<lggPoint> newDots;
	for (std::vector<lggPoint>::iterator it = mDots.begin(); it != mDots.end(); ++it)
	{
		lggPoint dot = *it;
		if (dist_vec(LLVector2(x, y), LLVector2(dot.x, dot.y)) >= 7)
		{
			newDots.push_back(dot);
		}

	}
	mDots = newDots;

	return LLFloater::handleMouseDown(x, y, mask);
}

void lggBeamMapFloater::onBackgroundChange()
{
	mBeamshapePanel->setBackgroundColor(getChild<LLColorSwatchCtrl>("back_color_swatch")->get());
}

LLSD lggBeamMapFloater::getDataSerialized()
{
	LLSD out;
	LLRect r  = mBeamshapePanel->getRect();
	for (S32 i = 0; i < mDots.size(); ++i)
	{
		LLSD point;
		lggPoint t = mDots[i];
		LLVector3 vec = LLVector3(0.f, (F32)t.x, (F32)t.y);
		vec -= LLVector3(0.f, (F32)r.getCenterX(), (F32)r.getCenterY());
		
		point["offset"] = vec.getValue();
		point["color"] = t.c.getValue();

		out[i] = point;
	}
	return out;
}

void lggBeamMapFloater::onClickSave()
{
	std::string filename(gDirUtilp->getExpandedFilename(LL_PATH_USER_SETTINGS , "beams", "NewBeam.xml"));
	LLFilePickerReplyThread::startPicker(boost::bind(&lggBeamMapFloater::onSaveCallback, this, _1), LLFilePicker::FFSAVE_BEAM, filename);
}

void lggBeamMapFloater::onSaveCallback(const std::vector<std::string>& filenames)
{
	std::string filename = filenames[0];

	LLSD export_data;
	export_data["scale"] = 8.0f / (mBeamshapePanel->getRect().getWidth());
	export_data["data"] = getDataSerialized();

	llofstream export_file;
	export_file.open(filename.c_str());
	LLSDSerialize::toPrettyXML(export_data, export_file);
	export_file.close();
	gSavedSettings.setString("FSBeamShape", gDirUtilp->getBaseFileName(filename, true));

	if (mFSPanel)
	{
		mFSPanel->refreshBeamLists();
	}
}

void lggBeamMapFloater::onClickClear()
{
	clearPoints();
}

void lggBeamMapFloater::onClickLoad()
{
	LLFilePickerReplyThread::startPicker(boost::bind(&lggBeamMapFloater::onLoadCallback, this, _1), LLFilePicker::FFLOAD_XML, false);
}

void lggBeamMapFloater::onLoadCallback(const std::vector<std::string>& filenames)
{
	mDots.clear();
	LLSD import_data;
	llifstream importer(filenames[0].c_str());
	LLSDSerialize::fromXMLDocument(import_data, importer);
	LLSD picture = import_data["data"];
	F32 scale = (F32)import_data["scale"].asReal();

	for (LLSD::array_iterator it = picture.beginArray(); it != picture.endArray(); ++it)
	{
		LLRect rec = mBeamshapePanel->getRect();

		LLSD beam_data = *it;
		lggPoint p;
		LLVector3 vec = LLVector3(beam_data["offset"]);
		vec *= scale / (8.0f / rec.getWidth());
		LLColor4 color = LLColor4(beam_data["color"]);
		p.c = color;
		p.x = (S32)(vec.mV[VY] + rec.getCenterX());
		p.y = (S32)(vec.mV[VZ] + rec.getCenterY());

		mDots.push_back(p);
	}
}

void lggBeamMapFloater::setData(FSPanelPrefs* data)
{
	mFSPanel = data;
	if (mFSPanel)
	{
		gFloaterView->getParentFloater(mFSPanel)->addDependentFloater(this);
	}
}

void lggBeamMapFloater::clearPoints()
{
	mDots.clear();
}