Context Navigation

source: GTP/trunk/Lib/Illum/IllumModule/OgreIllumModule/src/OgreRenderingRun.cpp @ 2333

Revision 2333, 11.5 KB checked in by szirmay, 17 years ago (diff)

Line
1	#include "OgreRenderingrun.h"
2	#include "OgreIlluminationManager.h"
3
4	MovablePlane* OgreRenderingRun::fullScreenQuad = NULL;
5	Entity* OgreRenderingRun::fullScreenQuadEntity = NULL;
6	SceneNode* OgreRenderingRun::fullScreenQuadNode = NULL;
7	BillboardSet* OgreRenderingRun::pixelSprites = NULL;
8	Entity* OgreRenderingRun::fullscreenGrid = NULL;
9	String OgreRenderingRun::spriteSetName = "";
10
11	class MaterialSetVisitor : public QueuedRenderableVisitor
12	{
13	private:
14	String materialName;
15	std::map<Renderable, String> visibleObjects;
16	public:
17
18	MaterialSetVisitor(String& materialName, std::map<Renderable, String> visibleObjects)
19	{
20	this->materialName = materialName;
21	this->visibleObjects = visibleObjects;
22	}
23
24	void visit(const Renderable* r)
25	{
26	Renderable* rend = (Renderable*) r; //sorry for removing the const
27
28	if( visibleObjects->count(rend) == 0)
29	{
30	(*visibleObjects)[rend] = rend->getMaterial().getPointer()->getName();
31	rend->setMaterialName(materialName);
32	}
33	}
34	bool visit(const Pass* p){return true;}
35	void visit(const RenderablePass* rp)
36	{
37	Renderable* rend = rp->renderable;
38
39	if( visibleObjects->count(rend) == 0)
40	{
41	(*visibleObjects)[rend] = rend->getMaterial().getPointer()->getName();
42	rend->setMaterialName(materialName);
43	}
44	}
45
46	};
47
48	Vector3 OgreRenderingRun::getCubeMapFaceDirection(unsigned char faceId)
49	{
50	switch(faceId)
51	{
52	case 0: //positive X
53	return Vector3(1,0,0);
54	case 1: //negative X
55	return Vector3(-1,0,0);
56	case 2: //positiove Y
57	return Vector3(0,1,0);
58	case 3: //negative Y
59	return Vector3(0,-1,0);
60	case 4: //positiove Z
61	return Vector3(0,0,-1);
62	case 5: //negative Z
63	return Vector3(0,0,1);
64	default:
65	return Vector3(0,0,0);
66	}
67	}
68
69	Texture* OgreRenderingRun::createCubeRenderTexture(String name, const Vector3 position, unsigned int resolution, PixelFormat format, int numMips, ColourValue clearColor)
70	{
71	//create texture
72	TexturePtr texPtr = Ogre::TextureManager::getSingleton().createManual( name,
73	"default",
74	TEX_TYPE_CUBE_MAP,
75	resolution,
76	resolution,
77	0,
78	numMips,
79	format,
80	TU_RENDERTARGET);
81	Texture* result = texPtr.getPointer();
82
83	//add viewports to faces
84	for(unsigned int i = 0; i < 6; i++)
85	{
86	Camera* cam = Root::getSingleton()._getCurrentSceneManager()
87	->createCamera(name + "CUBEMAPFACE_" + StringConverter::toString(i) + "_CAMERA");
88
89	cam->setPosition(position);
90	cam->setAspectRatio(1);
91	cam->setFOVy(Radian(Degree(90)));
92	cam->setNearClipDistance(0.1);
93	cam->setFarClipDistance(OgreIlluminationManager::getSingleton().
94	getMainCamera()->getFarClipDistance());
95
96	//set camera orientation
97	switch(i)
98	{
99	case 2:
100	cam->setOrientation(Quaternion(Radian(Degree(90)),Vector3(1,0,0)));
101	break;
102	case 3:
103	cam->setOrientation(Quaternion(Radian(Degree(90)),Vector3(-1,0,0)));
104	break;
105	default:
106	cam->setDirection(getCubeMapFaceDirection(i));
107	break;
108	}
109
110	HardwarePixelBuffer* hpb = (result->getBuffer(i, 0)).getPointer();
111	RenderTarget* rt = hpb->getRenderTarget(0);
112	Viewport* v = rt->addViewport(cam);
113	v->setOverlaysEnabled(false);
114	rt->setAutoUpdated(false);
115	v->setBackgroundColour(clearColor);
116	Root::getSingleton()._getCurrentSceneManager()->getRenderQueue()->clear();
117	Root::getSingleton()._getCurrentSceneManager()->setFindVisibleObjects(false);
118	rt->update();
119	Root::getSingleton()._getCurrentSceneManager()->setFindVisibleObjects(true);
120
121	Root::getSingleton().getRenderSystem()->_setViewport(v);
122	Root::getSingleton().getRenderSystem()->clearFrameBuffer( FBT_COLOUR, clearColor);
123	}
124
125	return result;
126
127	}
128
129	void OgreRenderingRun::setMaterialForRenderables(String& materialName, RenderQueue* rq, bool solidsonly)
130	{
131	SceneManager* sm = Ogre::Root::getSingleton()._getCurrentSceneManager();
132
133	RenderQueue::QueueGroupIterator queueIt = rq->_getQueueGroupIterator();
134
135	while (queueIt.hasMoreElements())
136	{
137	RenderQueueGroup* pGroup = queueIt.getNext();
138
139	RenderQueueGroup::PriorityMapIterator groupIt = pGroup->getIterator();
140
141	while (groupIt.hasMoreElements())
142	{
143	RenderPriorityGroup* pPriorityGrp = groupIt.getNext();
144	const QueuedRenderableCollection& solids = pPriorityGrp->getSolidsBasic();
145	const QueuedRenderableCollection& transparents = pPriorityGrp->getTransparents();
146	MaterialSetVisitor visitor(materialName, &visibleObjects);
147	solids.acceptVisitor(&visitor, QueuedRenderableCollection::OM_PASS_GROUP);
148	if(!solidsonly)
149	transparents.acceptVisitor(&visitor, QueuedRenderableCollection::OM_SORT_ASCENDING);
150	}
151	}
152
153	//refill renderqueue
154	rq->clear();
155
156	std::map<Renderable*, String>::iterator it = visibleObjects.begin();
157	std::map<Renderable*, String>::iterator itend = visibleObjects.end();
158
159	while(it != itend)
160	{
161	rq->addRenderable((*it).first);
162	it++;
163	}
164
165	sm->setFindVisibleObjects(false);
166	}
167
168	void OgreRenderingRun::setMaterialForVisibles(String& materialName, Camera* cam, bool shadowcastersonly, bool solidsonly)
169	{
170	SceneManager* sm = Ogre::Root::getSingleton()._getCurrentSceneManager();
171
172	RenderQueue* rq = sm->getRenderQueue();
173	rq->clear();
174
175	RenderQueue::QueueGroupIterator groupIter = rq->_getQueueGroupIterator();
176	while (groupIter.hasMoreElements())
177	{
178	RenderQueueGroup* g = groupIter.getNext();
179	g->defaultOrganisationMode();
180	}
181
182	//sm->_findVisibleObjectsOC(cam, cam->getViewport(), false, false, shadowcastersonly);
183	sm->_findVisibleObjects(cam, shadowcastersonly);
184
185	setMaterialForRenderables(materialName, sm->getRenderQueue(), solidsonly);
186	}
187
188	void OgreRenderingRun::restoreMaterials()
189	{
190	SceneManager* sm = Ogre::Root::getSingleton()._getCurrentSceneManager();
191
192	std::map<Renderable*, String>::iterator it = visibleObjects.begin();
193	std::map<Renderable*, String>::iterator itend = visibleObjects.end();
194
195	while(it != itend)
196	{
197	((it).first)->setMaterialName((it).second);
198	it++;
199	}
200
201	visibleObjects.clear();
202
203	sm->setFindVisibleObjects(true);
204	}
205
206	void OgreRenderingRun::renderFullscreenQuad(String materialName, RenderTarget* target)
207	{
208	SceneManager* sm = Ogre::Root::getSingleton()._getCurrentSceneManager();
209
210	if(fullScreenQuad == NULL)
211	{
212	fullScreenQuad = new MovablePlane("PASS_FULLSCREENQUAD_PLANE");
213	fullScreenQuad->d = 0;
214	fullScreenQuad->normal = Vector3::UNIT_Z;
215	MeshManager::getSingleton().createPlane("PASS_FULLSCREENQUAD_MESH",
216	ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME,
217	*fullScreenQuad, 2, 2,
218	1, 1, true, 1, 1, 1, Vector3::UNIT_Y);
219	fullScreenQuadEntity = sm->createEntity( "PASS_FULLSCREENQUAD_ENTITY", "PASS_FULLSCREENQUAD_MESH" );
220	sm->getRootSceneNode()->attachObject(fullScreenQuadEntity);
221	fullScreenQuadEntity->setVisible(false);
222	}
223
224	fullScreenQuadEntity->setMaterialName(materialName);
225	sm->getRenderQueue()->clear();
226	sm->getRenderQueue()->addRenderable(fullScreenQuadEntity->getSubEntity(0));
227	sm->setFindVisibleObjects(false);
228	target->update();
229	sm->setFindVisibleObjects(true);
230	}
231
232	void OgreRenderingRun::renderFullscreenGrid(String& materialName, RenderTarget* rt, int width, int height)
233	{
234	SceneManager* sm = Root::getSingleton()._getCurrentSceneManager();
235
236	if(fullscreenGrid == NULL)
237	{
238	MeshPtr mesh = MeshManager::getSingleton().createManual("FullScreenGrid",ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);
239	SubMesh* subMesh = mesh->createSubMesh();
240
241	mesh->sharedVertexData = new VertexData();
242	VertexData* vertexData = mesh->sharedVertexData;
243
244	// define the vertex format
245	VertexDeclaration* vertexDecl = vertexData->vertexDeclaration;
246	size_t currOffset = 0;
247	// positions
248	vertexDecl->addElement(0, currOffset, VET_FLOAT3, VES_POSITION);
249	currOffset += VertexElement::getTypeSize(VET_FLOAT3);
250
251	// allocate the vertex buffer
252	vertexData->vertexCount = width * height;
253	HardwareVertexBufferSharedPtr vBuf = HardwareBufferManager::getSingleton().createVertexBuffer(vertexDecl->getVertexSize(0), vertexData->vertexCount, HardwareBuffer::HBU_STATIC_WRITE_ONLY, false);
254	VertexBufferBinding* binding = vertexData->vertexBufferBinding;
255	binding->setBinding(0, vBuf);
256	float* pVertex = static_cast<float*>(vBuf->lock(HardwareBuffer::HBL_DISCARD));
257
258	// allocate index buffer
259	subMesh->indexData->indexCount = 6 * (width - 1) * (height - 1);
260	subMesh->indexData->indexBuffer = HardwareBufferManager::getSingleton().createIndexBuffer(HardwareIndexBuffer::IT_16BIT, subMesh->indexData->indexCount, HardwareBuffer::HBU_STATIC_WRITE_ONLY, false);
261	HardwareIndexBufferSharedPtr iBuf = subMesh->indexData->indexBuffer;
262	unsigned short* pIndices = static_cast<unsigned short*>(iBuf->lock(HardwareBuffer::HBL_DISCARD));
263
264	unsigned short wVerticeIndex = 0 ;
265
266
267	for( int j = 0; j < height; j++ )
268	{
269	for( int i = 0; i < width; i++ )
270	{ //positions
271	float x = ((float)i + 0.5) / (float) width;
272	float y = ((float)j + 0.5) / (float) height;
273	*pVertex++ = x;
274	*pVertex++ = y;
275	*pVertex++ = 0;
276
277	//indexes
278	if( j < height - 1 && i < width -1 )
279	{
280	pIndices++ = j width + i;
281	pIndices++ = (j + 1) width + i;
282	pIndices++ = j width + i + 1;
283	pIndices++ = j width + i + 1;
284	pIndices++ = (j + 1) width + i;
285	pIndices++ = (j + 1) width + i + 1;
286	}
287
288	}
289	}
290
291	// Unlock
292	vBuf->unlock();
293	iBuf->unlock();
294	// Generate face list
295	subMesh->useSharedVertices = true;
296
297	// this line makes clear the mesh is loaded (avoids memory leaks)
298	mesh->load();
299
300	fullscreenGrid = sm->createEntity("FullScreenGridEntity", "FullScreenGrid");
301	SceneNode* n = sm->getRootSceneNode()->createChildSceneNode("FullScreenGridNode");
302	n->attachObject(fullscreenGrid);
303	fullscreenGrid->setVisible(false);
304	}
305
306	fullscreenGrid->setMaterialName(materialName);
307
308	sm->getRenderQueue()->clear();
309	sm->getRenderQueue()->addRenderable(fullscreenGrid->getSubEntity(0));
310	sm->setFindVisibleObjects(false);
311	rt->update();
312	sm->setFindVisibleObjects(true);
313
314	}
315
316	void OgreRenderingRun::renderPixelSprites(String& materialName, RenderTarget* rt, int width, int height)
317	{
318	SceneManager* sm = Ogre::Root::getSingleton()._getCurrentSceneManager();
319
320	if(pixelSprites == 0)
321	{
322	int resolutionX = width;
323	int resolutionY = height;
324	pixelSprites = new BillboardSet(spriteSetName, resolutionX * resolutionY, true);
325	pixelSprites->setBillboardsInWorldSpace(true);
326	pixelSprites->setCullIndividually(false);
327	pixelSprites->setPointRenderingEnabled(true);
328	//pixelSprites->setDefaultDimensions(1.0 / (float) resolutionX, 1.0 / resolutionY);
329	pixelSprites->beginBillboards();
330	for(int i = 0; i < resolutionX; i++)
331	{
332	for(int j = 0; j < resolutionY; j++)
333	{
334	Billboard bb( Vector3(
335	( (float)i + 0.5 )/ (float) resolutionX, //position x
336	( (float)j + 0.5 )/ (float) resolutionY, //position y
337	0),
338	pixelSprites);
339	bb.setDimensions(1.0 / (float) resolutionX, 1.0 / resolutionY);
340	pixelSprites->injectBillboard(bb);
341	}
342	}
343	pixelSprites->endBillboards();
344
345	sm->getRootSceneNode()->attachObject(pixelSprites);
346
347	pixelSprites->setVisible(false);
348	}
349
350	pixelSprites->setMaterialName(materialName);
351
352	sm->getRenderQueue()->clear();
353	sm->getRenderQueue()->addRenderable(pixelSprites);
354	sm->setFindVisibleObjects(false);
355	rt->update();
356	sm->setFindVisibleObjects(true);
357
358	}
359

Note: See TracBrowser for help on using the repository browser.

Download in other formats: