source: GTP/trunk/Lib/Illum/IBRBillboardCloudTrees/OGRE/src/LBBCBillboardCloudIndirectTextureViewMode.cpp @ 778

Revision 778, 16.6 KB checked in by igarcia, 18 years ago (diff)
Line 
1
2#include "LBBCBillboardCloudIndirectTextureViewMode.h"
3#include "LBBCLeaves.h"
4#include "LBBCManager.h"
5
6namespace LBBC {
7
8BillboardCloudIndirectTextureViewMode::BillboardCloudIndirectTextureViewMode(Ogre::RenderWindow* win, unsigned int ogreFrameListenerModeHandle, bool useBufferedInputKeys, bool useBufferedInputMouse)
9:OBA::OgreFrameListenerMode(win, ogreFrameListenerModeHandle, useBufferedInputKeys, useBufferedInputMouse)
10{       
11        mFrame = 0;
12        mCurrentBillboardGroup = 0;
13        mCurrentBillboard = 0;
14        showDebugOverlay(false);
15        mDebugBillboardGeneration = false;
16        mDebugTextureAtlasGeneration = false;
17}
18
19BillboardCloudIndirectTextureViewMode::~BillboardCloudIndirectTextureViewMode()
20{
21
22}
23
24
25void BillboardCloudIndirectTextureViewMode::setBillboardCloudGroupedTextureAtlasDebug(bool debugTextureAtlasGeneration)
26{
27        mDebugTextureAtlasGeneration = debugTextureAtlasGeneration;
28}
29
30void BillboardCloudIndirectTextureViewMode::configureBillboardOrthogonalAlignedCamera(unsigned int iBillboard)
31{
32        BBC::BillboardPtr billboardPtr = ((LBBC::LBBCManager*)LBBC::LBBCManager::getSingletonPtr())->getLeaves()->getBillboardCloud()->getBillboard(iBillboard);
33        Ogre::Vector3 vTopLeft = billboardPtr->getBillboardClusterData()->getQuadTopLeftCorner();
34        Ogre::Vector3 vTopRight = billboardPtr->getBillboardClusterData()->getQuadTopRightCorner();
35        Ogre::Vector3 vBottomLeft = billboardPtr->getBillboardClusterData()->getQuadBottomLeftCorner();
36        Ogre::Vector3 vBottomRight = billboardPtr->getBillboardClusterData()->getQuadBottomRightCorner();
37
38        Ogre::Vector3 vMidPoint = vTopLeft.midPoint(vBottomRight);
39        Ogre::Vector3 vWidth(vTopRight - vTopLeft);
40        Ogre::Vector3 vHeight(vBottomLeft - vTopLeft);
41
42        Ogre::Real w = vWidth.length();
43        Ogre::Real h = vHeight.length();
44        Ogre::Real aspect = w / h;
45        Ogre::Radian fovy = Ogre::Radian(Ogre::Degree(90.0));
46        Ogre::Real nearPlane =  h / Ogre::Math::Tan(fovy/2.0,false);
47        Ogre::Real farPlane = 100000;
48
49        Ogre::Vector3 vUp = vTopLeft - vBottomLeft;
50        vUp.normalise();
51
52        Ogre::Vector3 vI(vTopRight-vTopLeft);
53        vI.normalise();
54
55        Ogre::Vector3 vJ(vBottomLeft-vTopLeft);
56        vJ.normalise();
57
58        Ogre::Vector3 vDir = -vI.crossProduct(vJ);
59        vDir.normalise();
60
61        Ogre::Vector3 vRight = vDir.crossProduct(vUp);
62        vRight.normalise();     
63
64        Ogre::Quaternion qOrientation(vRight, vUp, -vDir);
65       
66        mCamera->setNearClipDistance(nearPlane);
67        mCamera->setOrientation(qOrientation);
68        mCamera->setPosition(vMidPoint - (vDir * (nearPlane*nearPlane)));
69
70        if (Ogre::Root::getSingleton().getRenderSystem()->getName() != "Direct3D9 Rendering SubSystem")
71        {
72                Ogre::Radian thetaY (fovy / 2.0f);
73                Ogre::Real tanThetaY = Ogre::Math::Tan(thetaY);
74
75                Ogre::Real tanThetaX = tanThetaY * aspect;
76
77                Ogre::Real half_w,half_h;
78                half_w = w / 2.0;
79                half_h = h / 2.0;
80
81                Ogre::Real iw = 1.0 / half_w;
82                Ogre::Real ih = 1.0 / half_h;
83                Ogre::Real q;
84                if (farPlane == 0)
85                {
86                        q = 0;
87                }
88                else
89                {
90                        q = 2.0 / (farPlane - nearPlane);
91                }
92                mCustomProjMatrix = Ogre::Matrix4::ZERO;
93                mCustomProjMatrix[0][0] = iw;
94                mCustomProjMatrix[1][1] = ih;
95                mCustomProjMatrix[2][2] = -q;
96                mCustomProjMatrix[2][3] = - (farPlane + nearPlane)/(farPlane - nearPlane);
97                mCustomProjMatrix[3][3] = 1;   
98        }
99        else
100        {
101                Ogre::Radian thetaY (fovy / 2.0f);
102                Ogre::Real tanThetaY = Ogre::Math::Tan(thetaY);
103
104                //Real thetaX = thetaY * aspect;
105                Ogre::Real tanThetaX = tanThetaY * aspect; //Math::Tan(thetaX);
106                Ogre::Real half_w = w / 2;
107                Ogre::Real half_h = h / 2;
108                Ogre::Real iw = 1.0 / half_w;
109                Ogre::Real ih = 1.0 / half_h;
110                Ogre::Real q;
111                if (farPlane == 0)
112                {
113                        q = 0;
114                }
115                else
116                {
117                        q = 1.0 / (farPlane - nearPlane);
118                }
119
120                mCustomProjMatrix = Ogre::Matrix4::ZERO;
121                mCustomProjMatrix[0][0] = iw;
122                mCustomProjMatrix[1][1] = ih;
123                mCustomProjMatrix[2][2] = q;
124                mCustomProjMatrix[2][3] = -nearPlane / (farPlane - nearPlane);
125                mCustomProjMatrix[3][3] = 1;
126
127                mCustomProjMatrix[2][2] = -mCustomProjMatrix[2][2];
128        }       
129}
130
131void BillboardCloudIndirectTextureViewMode::saveTextureAtlas()
132{
133        if (mBitRange == 32)
134        {
135                mDestPixelFormat = Ogre::PF_FLOAT32_RGBA;
136        }
137        else if (mBitRange == 16)
138        {
139                mDestPixelFormat = Ogre::PF_FLOAT16_RGBA;
140        }
141        else  // mBitRange == 8
142        {
143                mDestPixelFormat = Ogre::PF_BYTE_RGBA;
144        }
145
146        Ogre::String textureAtlasFileName = BBC::Util::getBaseName(mTextureAtlasName) + Ogre::StringConverter::toString(mCurrentBillboardGroup) + "." + BBC::Util::getExtensionName(mTextureAtlasName);
147        mTextureAtlas->save(mTextureAtlasFolder, textureAtlasFileName, mDestPixelFormat);
148}
149
150void BillboardCloudIndirectTextureViewMode::configureTexture()
151{
152        mTextureAtlas->getRectangle2D()->setVisible(false);
153
154        //mClusterTexture->getRectangle2D()->setVisible(true);
155        //mClusterTexture->getMaterial()->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureName("customNormalMap128x128.png");
156        //mClusterTexture->setCorners(-1.0, 1.0, 1.0, -1.0);
157
158        mBillboardCloudPointClusters->getSubEntity((mCurrentTexture-1) % mBillboardCloudPointClusters->getNumSubEntities())->setVisible(false);
159        mBillboardCloudPointClusters->getSubEntity(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->setVisible(true);
160
161        configureBillboardOrthogonalAlignedCamera(mCurrentTexture % mBillboardCloudSplitted->getNumSubEntities());
162        mCamera->setCustomProjectionMatrix(true,mCustomProjMatrix);
163
164        Ogre::GpuProgramParametersSharedPtr fragParams;
165       
166        fragParams = mBillboardCloudPointClusters->getSubEntity(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->getMaterial()->getTechnique(0)->getPass(0)->getFragmentProgramParameters();
167        fragParams->setNamedConstant("textureSize", (Ogre::Real)mTextureSize);
168        mBillboardCloudPointClusters->getSubEntity(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->getMaterial()->reload();
169       
170        if (mDebugTextureAtlasGeneration)
171        {
172                mClusterTexture->getMaterial()->getTechnique(0)->getPass(0)->setAmbient(Ogre::Math::RangeRandom(0.0,1.0), Ogre::Math::RangeRandom(0.0,1.0), Ogre::Math::RangeRandom(0.0,1.0));
173                mClusterTexture->getMaterial()->getTechnique(0)->getPass(0)->setDiffuse(Ogre::Math::RangeRandom(0.0,1.0), Ogre::Math::RangeRandom(0.0,1.0), Ogre::Math::RangeRandom(0.0,1.0), 1.0);
174        }
175}
176
177void BillboardCloudIndirectTextureViewMode::generateTexture()
178{
179        configureTexture();
180       
181        mClusterTexture->update();     
182        //mClusterTexture->debug();
183}
184
185void BillboardCloudIndirectTextureViewMode::setEntityClustersGroupedName(Ogre::String entityClustersGroupedName)
186{
187        mEntityClustersGroupedName = entityClustersGroupedName;
188}
189
190void BillboardCloudIndirectTextureViewMode::initializeEntityClustersGrouped()
191{
192        if (mFrame == 1)
193        {
194                for (unsigned int iSubEntity = 0; iSubEntity < mBillboardCloudPointClusters->getNumSubEntities(); iSubEntity++)
195                {
196                        mEntityClustersGrouped->getSubEntity(iSubEntity)->setVisible(false);
197                }
198        }
199}
200
201void BillboardCloudIndirectTextureViewMode::initializeBillboardCloudPointClusters()
202{
203        if (mFrame == 1)
204        {
205                for (unsigned int iSubEntity = 0; iSubEntity < mBillboardCloudPointClusters->getNumSubEntities(); iSubEntity++)
206                {
207                        mBillboardCloudPointClusters->getSubEntity(iSubEntity)->setVisible(false);
208                }
209        }
210}
211
212void BillboardCloudIndirectTextureViewMode::initializeBillboardCloudSplitted()
213{
214        if (mFrame == 1)
215        {
216                for (unsigned int iSubEntity = 0; iSubEntity < mBillboardCloudSplitted->getNumSubEntities(); iSubEntity++)
217                {
218                        mBillboardCloudSplitted->getSubEntity(iSubEntity)->setVisible(false);
219                }
220        }
221}
222
223bool BillboardCloudIndirectTextureViewMode::processUnbufferedKeyInput(const Ogre::FrameEvent& evt)
224{
225        if (mInputDevice->isKeyDown(Ogre::KC_SPACE))
226    {
227        mCurrentTexture++;
228    }
229
230        return OgreFrameListenerMode::processUnbufferedKeyInput(evt);
231}
232
233void BillboardCloudIndirectTextureViewMode::configureTextureAtlas()
234{
235        mBillboardCloudPointClusters->getSubEntity(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->setVisible(false);
236
237        mTextureAtlas->getRectangle2D()->setVisible(true);
238
239        Ogre::Vector2 uvMapMax = ((LBBC::LBBCManager*)LBBC::LBBCManager::getSingletonPtr())->getLeaves()->getBillboardCloud()->getBillboard(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->getBillboardClusterData()->getBillboardUVMapMax(0);
240        Ogre::Vector2 uvMapMin = ((LBBC::LBBCManager*)LBBC::LBBCManager::getSingletonPtr())->getLeaves()->getBillboardCloud()->getBillboard(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->getBillboardClusterData()->getBillboardUVMapMin(0);
241
242        mClusterTexture->setCorners(uvMapMin[0], uvMapMax[1], uvMapMax[0], uvMapMin[1]);
243        mClusterTexture->getRectangle2D()->setVisible(true);
244        //mClusterTexture->getMaterial()->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureName(mClusterTexture->getTextureName());
245}
246
247void BillboardCloudIndirectTextureViewMode::generateTextureAtlas()
248{       
249        configureTextureAtlas();
250
251        mTextureAtlas->update();               
252        //mTextureAtlas->debug();
253       
254        mClusterTexture->getRectangle2D()->setVisible(false);
255}
256
257bool BillboardCloudIndirectTextureViewMode::frameStarted(const Ogre::FrameEvent& evt, Ogre::InputReader *inputReader)
258{
259        initializeEntityClustersGrouped();
260    initializeBillboardCloudSplitted();
261        initializeBillboardCloudPointClusters();
262
263        if (mFrame > 1)
264        {               
265                if (!mDebugBillboardGeneration)
266                {
267                        mWindow->getViewport(0)->setBackgroundColour(Ogre::ColourValue(1.0, 1.0, 1.0, 1.0));
268
269                        if (mCurrentTexture < mNumTextures)
270                        {
271                                generateTexture();
272
273                                generateTextureAtlas();
274
275                                mCurrentTexture++;
276                                mCurrentBillboard++;
277
278                                unsigned int numBillboardsCurrentGroups = ((LBBC::LBBCManager*)LBBC::LBBCManager::getSingletonPtr())->getLeaves()->getBillboardCloud()->getBillboardGroup(mCurrentBillboardGroup)->getNumBillboards();
279
280                                if (mCurrentBillboard == numBillboardsCurrentGroups)
281                                {
282                                        saveTextureAtlas();
283                                       
284                                        unsigned int numBillboardGroups = ((LBBC::LBBCManager*)LBBC::LBBCManager::getSingletonPtr())->getLeaves()->getBillboardCloud()->getNumBillboardGroups();
285
286                                        if (mCurrentBillboardGroup < numBillboardGroups)
287                                        {
288                                                mCurrentBillboard = 0;
289                                                mCurrentBillboardGroup++;                                               
290
291                                                mTextureAtlas->getRectangle2D()->setVisible(false);
292                                                mTextureAtlas->update();
293                                                //mTextureAtlas->debug();
294                                        }
295                                }
296                        }
297
298                        if (mCurrentTexture == mNumTextures)
299                        {                               
300                                mDebugBillboardGeneration = true;
301
302                                mWindow->getViewport(0)->setBackgroundColour(Ogre::ColourValue(0.0, 0.0, 0.0, 1.0));
303
304                                mTextureAtlas->getRectangle2D()->setVisible(false);
305
306                                mCurrentTexture = 0;
307                        }
308                }
309                else
310                {
311                        mWindow->getViewport(0)->setBackgroundColour(Ogre::ColourValue(0.0, 0.0, 0.0, 1.0));
312
313                        configureBillboard();
314
315                        configureTexture();
316
317                        mCurrentTexture = mCurrentTexture % mNumTextures;
318                }
319        }
320
321        mFrame++;
322
323        return OgreFrameListenerMode::frameStarted(evt, inputReader);   
324}
325
326void BillboardCloudIndirectTextureViewMode::configureBillboard()
327{
328        mEntityClustersGrouped->getSubEntity((mCurrentTexture-1) % mEntityClustersGrouped->getNumSubEntities())->setVisible(false);
329        mEntityClustersGrouped->getSubEntity(mCurrentTexture % mEntityClustersGrouped->getNumSubEntities())->setVisible(true);
330        mBillboardCloudSplitted->getSubEntity((mCurrentTexture-1) % mBillboardCloudPointClusters->getNumSubEntities())->setVisible(false);
331        mBillboardCloudSplitted->getSubEntity(mCurrentTexture % mBillboardCloudPointClusters->getNumSubEntities())->setVisible(true);
332}
333
334void BillboardCloudIndirectTextureViewMode::setTextureAtlasSize(unsigned int size)
335{
336        mTextureAtlasSize = size;
337}
338
339void BillboardCloudIndirectTextureViewMode::setBillboardCloudPointClustersName(Ogre::String billboardCloudPointClustersName)
340{
341        mBillboardCloudPointClustersName = billboardCloudPointClustersName;
342}
343
344void BillboardCloudIndirectTextureViewMode::setBillboardCloudSplittedName(Ogre::String billboardCloudSplittedName)
345{
346        mBillboardCloudSplittedName = billboardCloudSplittedName;
347}
348
349void BillboardCloudIndirectTextureViewMode::setTextureAtlasBitRange(unsigned int bitRange)
350{
351        mBitRange = bitRange;
352}
353
354void BillboardCloudIndirectTextureViewMode::setTextureSize(unsigned int size)
355{
356        mTextureSize = size;
357}
358
359void BillboardCloudIndirectTextureViewMode::setTextureAtlasFolder(Ogre::String textureAtlasFolder)
360{
361        mTextureAtlasFolder = textureAtlasFolder;
362}
363
364void BillboardCloudIndirectTextureViewMode::setTextureAtlasName(Ogre::String textureAtlasName)
365{
366        mTextureAtlasName = textureAtlasName;
367}
368
369void BillboardCloudIndirectTextureViewMode::createScene()
370{
371        if (mBitRange == 32)
372        {
373                mSrcPixelFormat = Ogre::PF_FLOAT32_RGBA;
374        }
375        else if (mBitRange == 16)
376        {
377                mSrcPixelFormat = Ogre::PF_FLOAT16_RGBA;
378        }
379        else  // mBitRange == 8
380        {
381                mSrcPixelFormat = Ogre::PF_A8R8G8B8;
382        }
383
384        LBBC::Leaves *mLeaves = ((LBBC::LBBCManager*)LBBC::LBBCManager::getSingletonPtr())->getLeaves();
385        Ogre::Vector2 uvMapMax = mLeaves->getBillboardCloud()->getBillboard(0)->getBillboardClusterData()->getBillboardUVMapMax(0);
386        Ogre::Vector2 uvMapMin = mLeaves->getBillboardCloud()->getBillboard(0)->getBillboardClusterData()->getBillboardUVMapMin(0);
387
388        mWindow->getViewport(0)->setBackgroundColour(Ogre::ColourValue(1.0, 1.0, 1.0, 1.0));
389
390        mBillboardCloudPointClustersSceneNode = mSceneMgr->getRootSceneNode()->createChildSceneNode();
391        mBillboardCloudSplittedSceneNode = mSceneMgr->getRootSceneNode()->createChildSceneNode();
392
393        mBillboardCloudPointClusters = mSceneMgr->createEntity(mBillboardCloudPointClustersName, mBillboardCloudPointClustersName);
394    mBillboardCloudPointClustersSceneNode->attachObject(mBillboardCloudPointClusters);
395
396        mBillboardCloudSplitted = mSceneMgr->createEntity(mBillboardCloudSplittedName, mBillboardCloudSplittedName);
397    mBillboardCloudSplittedSceneNode->attachObject(mBillboardCloudSplitted);
398
399        mEntityClustersGroupedSceneNode = mSceneMgr->getRootSceneNode()->createChildSceneNode();
400        mEntityClustersGrouped = mSceneMgr->createEntity(mEntityClustersGroupedName, mEntityClustersGroupedName);
401    mEntityClustersGroupedSceneNode->attachObject(mEntityClustersGrouped);
402
403        mTextureAtlas = IMG::TextureAtlasPtr( new IMG::TextureAtlas() );
404        mTextureAtlas->create("IndirectTextureAtlas", mTextureAtlasSize, mTextureAtlasSize, mSrcPixelFormat, mCamera, Ogre::ColourValue(0.0, 0.0, 0.0));
405        mTextureAtlas->setTextureAtlasSceneNode(mBillboardCloudPointClustersSceneNode);
406        Ogre::MaterialPtr materialPtr1 = Ogre::MaterialManager::getSingleton().create("IndirectTextureAtlasMaterial", Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);         
407        materialPtr1->getTechnique(0)->getPass(0)->setAlphaRejectSettings(Ogre::CMPF_GREATER_EQUAL, 10);
408        materialPtr1->getTechnique(0)->getPass(0)->createTextureUnitState("transparent.png");
409        materialPtr1->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureFiltering(Ogre::FO_NONE, Ogre::FO_NONE, Ogre::FO_NONE);                   
410        materialPtr1->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureAddressingMode(Ogre::TextureUnitState::TAM_CLAMP);
411        materialPtr1->getTechnique(0)->getPass(0)->setLightingEnabled(false);
412        mTextureAtlas->setMaterial(materialPtr1.getPointer());
413        mTextureAtlas->setCorners(-1.0, 1.0, 1.0, -1.0);       
414
415        mClusterTexture = IMG::TexturePtr( new IMG::Texture() );
416        mClusterTexture->create("ClusterIndirectTexture", mTextureSize, mTextureSize, mSrcPixelFormat, mCamera, Ogre::ColourValue(0.0, 0.0, 0.0));
417        Ogre::MaterialPtr materialPtr = Ogre::MaterialManager::getSingleton().create("IndirectTextureMaterial", Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);                       
418        if (!mDebugTextureAtlasGeneration)
419        {
420                materialPtr->getTechnique(0)->getPass(0)->createTextureUnitState("customNormalMap128x128.png", 0);
421                materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureFiltering(Ogre::FO_NONE, Ogre::FO_NONE, Ogre::FO_NONE);                     
422                materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureAddressingMode(Ogre::TextureUnitState::TAM_CLAMP);         
423
424                materialPtr->getTechnique(0)->createPass();
425                materialPtr->getTechnique(0)->getPass(1)->setSceneBlending(Ogre::SBT_ADD);
426                materialPtr->getTechnique(0)->getPass(1)->setDepthWriteEnabled(false);
427
428                materialPtr->getTechnique(0)->getPass(1)->setAlphaRejectSettings(Ogre::CMPF_GREATER_EQUAL, 10);
429                materialPtr->getTechnique(0)->getPass(1)->createTextureUnitState("ClusterIndirectTexture", 0);
430                materialPtr->getTechnique(0)->getPass(1)->getTextureUnitState(0)->setTextureFiltering(Ogre::FO_NONE, Ogre::FO_NONE, Ogre::FO_NONE);                     
431                materialPtr->getTechnique(0)->getPass(1)->getTextureUnitState(0)->setTextureAddressingMode(Ogre::TextureUnitState::TAM_CLAMP);         
432
433                materialPtr->getTechnique(0)->getPass(0)->setLightingEnabled(false);
434        }
435
436        mClusterTexture->setMaterial(materialPtr.getPointer());
437
438        mTextureAtlas->addTexture(mClusterTexture);
439
440        mNumTextures = mBillboardCloudPointClusters->getNumSubEntities();
441        mCurrentTexture = 0;
442}
443
444void BillboardCloudIndirectTextureViewMode::destroyScene()
445{
446       
447}
448
449}
Note: See TracBrowser for help on using the repository browser.