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

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