source: GTP/trunk/Lib/Illum/IBRBillboardCloudTrees/OGRE/src/LBBCEntityTextureAtlasViewMode.cpp @ 864

#include "LBBCEntityTextureAtlasViewMode.h"

namespace LBBC 
{

EntityTextureAtlasViewMode::EntityTextureAtlasViewMode(Ogre::RenderWindow* win, unsigned int ogreFrameListenerModeHandle, bool useBufferedInputKeys, bool useBufferedInputMouse)
:OBA::OgreFrameListenerMode(win, ogreFrameListenerModeHandle, useBufferedInputKeys, useBufferedInputMouse)
{       
        mTextureAtlasGenerated = false;
        mFrame = 0;
        showDebugOverlay(false);
}

EntityTextureAtlasViewMode::~EntityTextureAtlasViewMode()
{
        // Gametools -- BUG: 06/04/2006
        //destroyScene();
}

void EntityTextureAtlasViewMode::setTextureAtlasFolder(Ogre::String textureAtlasFolder)
{
        mTextureAtlasFolder = textureAtlasFolder;
}

void EntityTextureAtlasViewMode::setTextureName(Ogre::String textureName)
{
        mTextureName = textureName;
}

void EntityTextureAtlasViewMode::setTextureAtlasName(Ogre::String textureAtlasName)
{
        mTextureAtlasName = textureAtlasName;
}

void EntityTextureAtlasViewMode::setTextureAtlasSize(unsigned int size)
{
        mTextureAtlasSize = size;
}

void EntityTextureAtlasViewMode::setTextureAtlasNumSamples(unsigned int numSamples)
{
        mNumSamples = numSamples;
}

void EntityTextureAtlasViewMode::setTextureAtlasBitRange(unsigned int bitRange)
{
        mBitRange = bitRange;
}

bool EntityTextureAtlasViewMode::frameStarted(const Ogre::FrameEvent& evt, Ogre::InputReader *inputReader)
{
        mWindow->getViewport(0)->setBackgroundColour(Ogre::ColourValue(1.0, 1.0, 1.0, 1.0));

        if (mFrame == 1)
        {
                mTextureAtlas->getTextureAtlasSceneNode()->setVisible(false);
                mTextureAtlas->update();
                mTextureAtlas->getTextureAtlasSceneNode()->setVisible(true);
        }
        
        if (mFrame == 2)
        {
                if (!mTextureAtlasGenerated)
                {
                        if (mBitRange == 32)
                        {
                                mDestPixelFormat = Ogre::PF_FLOAT32_RGBA;
                        }
                        else if (mBitRange == 16)
                        {
                                mDestPixelFormat = Ogre::PF_FLOAT16_RGBA;
                        }
                        else  // mBitRange == 8
                        {
                                mDestPixelFormat = Ogre::PF_BYTE_RGBA;
                        }

                        mTextureAtlas->update();
                        //mTextureAtlas->debug();                       
                        mTextureAtlas->save(mTextureAtlasFolder, mTextureAtlasName, mDestPixelFormat);
                        mTextureAtlasGenerated = true;
                }
        }

        mFrame++;

        return OgreFrameListenerMode::frameStarted(evt, inputReader);   
}


void EntityTextureAtlasViewMode::createScene()
{
        IMG::TexturePtr texturePtr ( new IMG::Texture() );
        texturePtr->bind(mTextureName);
        Ogre::ColourValue avgColour = texturePtr->getAvgColour();
        mWindow->getViewport(0)->setBackgroundColour(Ogre::ColourValue(1.0, 1.0, 1.0, 1.0)); 

        mTextureAtlasNode = mSceneMgr->getRootSceneNode()->createChildSceneNode();
        mTextureAtlas = IMG::TextureAtlasPtr( new IMG::TextureAtlas() );

        if (mBitRange == 32)
        {
                mSrcPixelFormat = Ogre::PF_FLOAT32_RGBA;
        }
        else if (mBitRange == 16)
        {
                mSrcPixelFormat = Ogre::PF_FLOAT16_RGBA;
        }
        else  // mBitRange == 8
        {
                mSrcPixelFormat = Ogre::PF_A8R8G8B8;
        }

        mTextureAtlas->create(mTextureAtlasName, mTextureAtlasSize, mTextureAtlasSize, mSrcPixelFormat, mCamera, avgColour);
        mTextureAtlas->setTextureAtlasSceneNode(mTextureAtlasNode);

        unsigned int numSubTextures = mNumSamples;                      
        Ogre::Vector2 maxTexCoords(1.0, 1.0);
        unsigned int irow = 1;
        unsigned int icol = 1;
        unsigned int nrow = Ogre::Math::Sqrt(BBC::Util::nextPowerOf2(numSubTextures));
        Ogre::Real step = 2.0 / nrow;
        Ogre::Real degreeStep = 360.0 / numSubTextures;
        Ogre::Real degree = 0.0;

        Ogre::MaterialPtr materialPtr = Ogre::MaterialManager::getSingleton().create("TextureAtlas" + Ogre::StringConverter::toString(numSubTextures), Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);                
        materialPtr->getTechnique(0)->getPass(0)->setAlphaRejectSettings(Ogre::CMPF_GREATER_EQUAL, 10);
        materialPtr->getTechnique(0)->getPass(0)->createTextureUnitState("transparent.png");
        materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureFiltering(Ogre::FO_NONE, Ogre::FO_NONE, Ogre::FO_NONE);                     
        materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureAddressingMode(Ogre::TextureUnitState::TAM_CLAMP);
        materialPtr->getTechnique(0)->getPass(0)->setLightingEnabled(false);
        mTextureAtlas->setMaterial(materialPtr.getPointer());   
        mTextureAtlas->setCorners(-1.0, 1.0, 1.0, -1.0);        

        for (unsigned int iSubTexture = 0; iSubTexture < numSubTextures; iSubTexture++)
        {                               
                IMG::TexturePtr texturePtr = IMG::TexturePtr( new IMG::Texture() );
                Ogre::MaterialPtr materialPtr = Ogre::MaterialManager::getSingleton().create("TextureAtlas" + Ogre::StringConverter::toString(iSubTexture), Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);           
                //materialPtr->getTechnique(0)->getPass(0)->setAmbient(Ogre::Math::RangeRandom(1.0, 0.0), 1.0, 0.0);
                //materialPtr->getTechnique(0)->getPass(0)->setDiffuse(1.0, 1.0, 1.0, 1.0);
                materialPtr->getTechnique(0)->getPass(0)->setLightingEnabled(false);
                materialPtr->getTechnique(0)->getPass(0)->setAlphaRejectSettings(Ogre::CMPF_GREATER_EQUAL, 10);
                materialPtr->getTechnique(0)->getPass(0)->createTextureUnitState(mTextureName, 0);
                materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureFiltering(Ogre::FO_NONE, Ogre::FO_NONE, Ogre::FO_NONE);
                materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureAddressingMode(Ogre::TextureUnitState::TAM_CLAMP);
                materialPtr->getTechnique(0)->getPass(0)->getTextureUnitState(0)->setTextureRotate(Ogre::Radian(Ogre::Degree(degree)));         
                
                texturePtr->setMaterial(materialPtr.getPointer());
                texturePtr->setCorners(maxTexCoords[0] - step, maxTexCoords[1], maxTexCoords[0], maxTexCoords[1] - step);
        
                mTextureAtlas->addTexture(texturePtr);

                if (irow == nrow)
                {
                        irow = 1;
                        icol++;
                        maxTexCoords = Ogre::Vector2(1.0, maxTexCoords[1] - step);
                }
                else
                {
                        irow++;
                        maxTexCoords = Ogre::Vector2(maxTexCoords[0] - step, maxTexCoords[1]);
                }

                degree = degree + degreeStep;
        }
}

void EntityTextureAtlasViewMode::destroyScene()
{
        mTextureAtlasNode->detachAllObjects();
}

}