STORM-1980 Add shake effect

3 files changed, 90 insertions, 13 deletions

diff --git a/indra/newview/llfloaterregionrestarting.cpp b/indra/newview/llfloaterregionrestarting.cpp
index 003ed488c6..111daf93eb 100644
--- a/indra/newview/llfloaterregionrestarting.cpp
+++ b/indra/newview/llfloaterregionrestarting.cpp
@@ -31,15 +31,18 @@
 #include "llfloaterreg.h"
 #include "lluictrl.h"
 #include "llagent.h"
+#include "llagentcamera.h"
+#include "llviewerwindow.h"
 
-static S32 mSeconds;
+static S32 sSeconds;
+static U32 sShakeState;
 
 LLFloaterRegionRestarting::LLFloaterRegionRestarting(const LLSD& key) :
 	LLFloater(key),
 	LLEventTimer(1)
 {
 	mName = (std::string)key["NAME"];
-	mSeconds = (LLSD::Integer)key["SECONDS"];
+	sSeconds = (LLSD::Integer)key["SECONDS"];
 }
 
 LLFloaterRegionRestarting::~LLFloaterRegionRestarting()
@@ -49,6 +52,8 @@ LLFloaterRegionRestarting::~LLFloaterRegionRestarting()
 
 BOOL LLFloaterRegionRestarting::postBuild()
 {
+	mRegionChangedConnection = gAgent.addRegionChangedCallback(boost::bind(&LLFloaterRegionRestarting::regionChange, this));
+
 	LLStringUtil::format_map_t args;
 	std::string text;
 
@@ -57,9 +62,9 @@ BOOL LLFloaterRegionRestarting::postBuild()
 	LLTextBox* textbox = getChild<LLTextBox>("region_name");
 	textbox->setValue(text);
 
-	refresh();
+	sShakeState = SHAKE_START;
 
-	mRegionChangedConnection = gAgent.addRegionChangedCallback(boost::bind(&LLFloaterRegionRestarting::regionChange, this));
+	refresh();
 
 	return TRUE;
 }
@@ -81,13 +86,72 @@ void LLFloaterRegionRestarting::refresh()
 	LLStringUtil::format_map_t args;
 	std::string text;
 
-	args["[SECONDS]"] = llformat("%d", mSeconds);
+	args["[SECONDS]"] = llformat("%d", sSeconds);
 	getChild<LLTextBox>("restart_seconds")->setValue(getString("RestartSeconds", args));
 
-	mSeconds = mSeconds - 1;
-	if(mSeconds < 0.0)
+	sSeconds = sSeconds - 1;
+	if(sSeconds < 0.0)
+	{
+		sSeconds = 0;
+	}
+}
+
+void LLFloaterRegionRestarting::draw()
+{
+	LLFloater::draw();
+
+	const F32 SHAKE_INTERVAL = 0.05;
+	const U32 SHAKE_ITERATIONS = 4;
+
+	if(SHAKE_START == sShakeState)
+	{
+			mShakeTimer.setTimerExpirySec(SHAKE_INTERVAL);
+			sShakeState = SHAKE_LEFT;
+			mIterations = 0;
+	}
+
+	if(SHAKE_DONE != sShakeState && mShakeTimer.hasExpired())
 	{
-		mSeconds = 0;
+		gAgentCamera.unlockView();
+
+		switch(sShakeState)
+		{
+			case SHAKE_LEFT:
+				gAgentCamera.setPanLeftKey(1.0);
+				gAgentCamera.setPanLeftKey(1.0);
+				sShakeState = SHAKE_UP;
+				break;
+
+			case SHAKE_UP:
+				gAgentCamera.setPanUpKey(1.0);
+				gAgentCamera.setPanUpKey(1.0);
+				sShakeState = SHAKE_RIGHT;
+				break;
+
+			case SHAKE_RIGHT:
+				gAgentCamera.setPanRightKey(1.0);
+				gAgentCamera.setPanRightKey(1.0);
+				sShakeState = SHAKE_DOWN;
+				break;
+
+			case SHAKE_DOWN:
+				gAgentCamera.setPanDownKey(1.0);
+				gAgentCamera.setPanDownKey(1.0);
+				mIterations = mIterations + 1;
+				if(SHAKE_ITERATIONS == mIterations)
+				{
+					sShakeState = SHAKE_DONE;
+				}
+				else
+				{
+					sShakeState = SHAKE_LEFT;
+				}
+				break;
+
+			default:
+				break;
+		}
+		mShakeTimer.setTimerExpirySec(SHAKE_INTERVAL);
 	}
 }
 
@@ -103,5 +167,6 @@ void LLFloaterRegionRestarting::close()
 
 void LLFloaterRegionRestarting::updateTime(S32 time)
 {
-	mSeconds = time;
+	sSeconds = time;
+	sShakeState = SHAKE_START;
 }
diff --git a/indra/newview/llfloaterregionrestarting.h b/indra/newview/llfloaterregionrestarting.h
index 6ff067e0a8..95fe772a28 100644
--- a/indra/newview/llfloaterregionrestarting.h
+++ b/indra/newview/llfloaterregionrestarting.h
@@ -45,11 +45,24 @@ private:
 	virtual BOOL postBuild();
 	virtual BOOL tick();
 	virtual void refresh();
+	virtual void draw();
 	virtual void regionChange();
 
 	std::string mName;
+	U32 mIterations;
+	LLTimer mShakeTimer;
 
 	boost::signals2::connection mRegionChangedConnection;
+
+	enum
+	{
+		SHAKE_START,
+		SHAKE_LEFT,
+		SHAKE_UP,
+		SHAKE_RIGHT,
+		SHAKE_DOWN,
+		SHAKE_DONE
+	};
 };
 
 #endif // LL_LLFLOATERREGIONRESTARTING_H
diff --git a/indra/newview/llviewermessage.cpp b/indra/newview/llviewermessage.cpp
index e81de844d1..076419e96a 100755
--- a/indra/newview/llviewermessage.cpp
+++ b/indra/newview/llviewermessage.cpp
@@ -5967,10 +5967,6 @@ bool attempt_standard_notification(LLMessageSystem* msgsystem)
 				seconds = static_cast<S32>(llsdBlock["SECONDS"].asInteger());
 			}
 
-			LLSD params;
-			params["NAME"] = llsdBlock["NAME"];
-			params["SECONDS"] = (LLSD::Integer)seconds;
-
 			LLFloaterRegionRestarting* floaterp = LLFloaterReg::findTypedInstance<LLFloaterRegionRestarting>("region_restarting");
 
 			if (floaterp)
@@ -5979,6 +5975,9 @@ bool attempt_standard_notification(LLMessageSystem* msgsystem)
 			}
 			else
 			{
+				LLSD params;
+				params["NAME"] = llsdBlock["NAME"];
+				params["SECONDS"] = (LLSD::Integer)seconds;
 				LLFloaterReg::showInstance("region_restarting", params);
 			}