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

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