CWE-ou-minkata/MOULOpenSourceClientPlugin/Plasma20/Sources/Plasma/FeatureLib/pfSurface/plGrabCubeMap.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/

#include "hsTypes.h"

#include "plGrabCubeMap.h"

#include "plPipeline.h"
#include "plDrawable.h"

#include "hsMatrix44.h"
#include "hsGeometry3.h"
#include "hsColorRGBA.h"
#include "hsBounds.h"

#include "../pnSceneObject/plSceneObject.h"
#include "../pnSceneObject/plDrawInterface.h"

#include "../plGImage/plMipmap.h"

#include "../plJPEG/plJPEG.h"

#include "../plMessage/plRenderRequestMsg.h"

plGrabCubeRenderRequest::plGrabCubeRenderRequest()
:	fQuality(75)
{
}

void plGrabCubeRenderRequest::Render(plPipeline* pipe, plPageTreeMgr* pageMgr)
{
	if( !fFileName[0] )
		return;

	plRenderRequest::Render(pipe, pageMgr);

	pipe->EndRender();

	plMipmap		mipmap;
	if( pipe->CaptureScreen(&mipmap) )
	{
		plJPEG::Instance().SetWriteQuality(fQuality);

		plJPEG::Instance().WriteToFile(fFileName, &mipmap);
	}

	pipe->BeginRender();
}


void plGrabCubeMap::GrabCube(plPipeline* pipe, plSceneObject* obj, const char* pref, const hsColorRGBA& clearColor, UInt8 q)
{
	hsPoint3 center;
	if( obj && !(obj->GetLocalToWorld().fFlags & hsMatrix44::kIsIdent) )
	{
		center = obj->GetLocalToWorld().GetTranslate();
	}
	else if( obj && obj->GetDrawInterface() )
	{
		center = obj->GetDrawInterface()->GetWorldBounds().GetCenter();
	}
	else
	{
		center = pipe->GetCameraToWorld().GetTranslate();
	}
	ISetupRenderRequests(pipe, center, pref, clearColor, q);
}

void plGrabCubeMap::GrabCube(plPipeline* pipe, const hsPoint3& center, const char* pref, const hsColorRGBA& clearColor, UInt8 q)
{
	ISetupRenderRequests(pipe, center, pref, clearColor, q);
}

void plGrabCubeMap::ISetupRenderRequests(plPipeline* pipe, const hsPoint3& center, const char* pref, const hsColorRGBA& clearColor, UInt8 q) const
{
	hsMatrix44 worldToCameras[6];
	hsMatrix44 cameraToWorlds[6];
	hsMatrix44::MakeEnvMapMatrices(center, worldToCameras, cameraToWorlds);

	UInt32 renderState 
		= plPipeline::kRenderNormal
		| plPipeline::kRenderClearColor
		| plPipeline::kRenderClearDepth;

	float hither;
	float yon;
	pipe->GetDepth(hither, yon);

	const char* suff[6] = {
		"LF",
		"RT",
		"BK",
		"FR",
		"UP",
		"DN" };

	int i;
	for( i = 0; i < 6; i++ )
	{
		plGrabCubeRenderRequest* req = TRACKED_NEW plGrabCubeRenderRequest;
		req->SetRenderState(renderState);

		req->SetDrawableMask(plDrawable::kNormal);
		req->SetSubDrawableMask(plDrawable::kSubAllTypes);

		req->SetHither(hither);
		req->SetYon(yon);

		req->SetFovX(90.f);
		req->SetFovY(90.f);

		req->SetClearColor(clearColor);
		req->SetClearDepth(1.f);

		req->SetClearDrawable(nil);
		req->SetRenderTarget(nil);

		req->SetCameraTransform(worldToCameras[i], cameraToWorlds[i]);

		req->fQuality = q;
		sprintf(req->fFileName, "%s_%s.jpg", pref, suff[i]);

		plRenderRequestMsg* reqMsg = TRACKED_NEW plRenderRequestMsg(nil, req);
		reqMsg->Send();
		hsRefCnt_SafeUnRef(req);
	}
}