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

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