terrain.cpp

#include "debug.h"
#include "terrain.h"

using namespace irr;
using namespace core;
using namespace scene;
using namespace video;

extern Debug* dbg;

Terrain::Terrain(IrrlichtDevice* d,
                 stringc fileName,
                 stringc textureFileName_1,
                 stringc textureFileName_2)
{
    //ctor
    dbg->log("create Terrain");
    device = d;
    terrain = NULL;
    collisionTriangles = NULL;

    // load terrain according to fileName extension
    if (stringHasExtension(fileName, stringc("bmp")))
    {
        loadFromHeightmap(fileName, textureFileName_1, textureFileName_2);
        fileType = FILE_TYPE_BMP;
    }
    else if (stringHasExtension(fileName, stringc("3ds")))
    {
        loadFrom3ds(fileName, textureFileName_1, textureFileName_2);
        fileType = FILE_TYPE_3DS;
    }
    else
    {
        dbg->log("Terrain: cannot load file. Filetype unknown.");
    }

    loadSkyBox();
}

Terrain::~Terrain()
{
    //dtor
    if (terrain)
    {
        terrain->remove();
        terrain = NULL;
    }
    if(collisionTriangles)
    {
        collisionTriangles->drop();
        collisionTriangles = NULL;
    }
    dbg->log("Terrain deleted");
}

vector3df Terrain::getCenter()
{
    if (terrain)
    {
        return terrain->getTerrainCenter();
    }
    return vector3df(-1,-1,-1);
}

ITriangleSelector* Terrain::getCollisionTriangles()
{
    if (! collisionTriangles)
    {
        if (terrain)
        {
            s32 LOD = 0;
            collisionTriangles = device->getSceneManager()->createTerrainTriangleSelector(terrain, LOD);
            terrain->setTriangleSelector(collisionTriangles);
        }
    }
    return collisionTriangles;
}

void Terrain::loadFromHeightmap(stringc heightMapFileName,
                   stringc textureFileName_1,
                   stringc textureFileName_2)
{
    ISceneManager* s = device->getSceneManager();

    ISceneNode* parent = s->getRootSceneNode();
    s32 id = 0;
    vector3df pos(0, 0, 0);
    vector3df rot(0, 0, 0);
    vector3df scale(40, 4.4f, 40);

    terrain = s->addTerrainSceneNode(heightMapFileName.c_str(), parent, id, pos, rot, scale);
    if (! terrain)
    {
        dbg->log("Terrain::load() cannot load", heightMapFileName);
        return;
    }

    device->getVideoDriver()->setTextureCreationFlag(ETCF_ALWAYS_32_BIT, true);
    ITexture* texture_1 = device->getVideoDriver()->getTexture(textureFileName_1.c_str());
    if (! texture_1)
    {
        dbg->log("Terrain::load() cannot load", textureFileName_1);
        return;
    }
    ITexture* texture_2 = device->getVideoDriver()->getTexture(textureFileName_2.c_str());
    if (! texture_2)
    {
        dbg->log("Terrain::load() cannot load", textureFileName_2);
        return;
    }
    device->getVideoDriver()->setTextureCreationFlag(ETCF_ALWAYS_32_BIT, false);
    terrain->setMaterialTexture(0, texture_1);
    terrain->setMaterialTexture(1, texture_2);
    terrain->scaleTexture(1, 20);
    //terrain->setMaterialFlag(EMF_WIREFRAME, true);
    terrain->setMaterialType(EMT_DETAIL_MAP);
    terrain->setMaterialFlag(EMF_LIGHTING, false);
}

void Terrain::loadFrom3ds(stringc FileName3ds,
                          stringc textureFileName_1,
                          stringc textureFileName_2)
{
    ISceneManager* s = device->getSceneManager();

    // load mesh
    IMesh* mesh = s->getMesh(FileName3ds.c_str());
    if (! mesh)
    {
        dbg->log("Terrain::load() cannot load", FileName3ds);
        return;
    }

    // create terrain from mesh
    ISceneNode* parent = s->getRootSceneNode();
    s32 id = 0;
    s32 minimalPolysPerNode = 128;
    terrain = s->addOctTreeSceneNode(mesh, parent, id, minimalPolysPerNode);
    if (! terrain)
    {
        dbg->log("Terrain::load() cannot create OctTree", FileName3ds);
        return;
    }

    // setup terrain
    vector3df pos(0, 0, 0);
    vector3df rot(0, 0, 0);
    vector3df scale(40, 4.4f, 40);

    // load textures
    device->getVideoDriver()->setTextureCreationFlag(ETCF_ALWAYS_32_BIT, true);
    ITexture* texture_1 = device->getVideoDriver()->getTexture(textureFileName_1.c_str());
    if (! texture_1)
    {
        dbg->log("Terrain::load() cannot load", textureFileName_1);
        return;
    }
    ITexture* texture_2 = device->getVideoDriver()->getTexture(textureFileName_2.c_str());
    if (! texture_2)
    {
        dbg->log("Terrain::load() cannot load", textureFileName_2);
        return;
    }

    // set textures
    device->getVideoDriver()->setTextureCreationFlag(ETCF_ALWAYS_32_BIT, false);
    terrain->setMaterialTexture(0, texture_1);
    terrain->setMaterialTexture(1, texture_2);
    terrain->scaleTexture(1, 20);
    //terrain->setMaterialFlag(EMF_WIREFRAME, true);
    terrain->setMaterialType(EMT_DETAIL_MAP);
    terrain->setMaterialFlag(EMF_LIGHTING, false);
}

bool Terrain::stringHasExtension(stringc s, stringc extension)
{
    bool result = false;
    s32 s_size = s.size();
    stringc ext(extension.c_str());
    stringc tmp(s.subString(s_size - 3, s_size).c_str());
    result = tmp.equals_ignore_case(ext);

    return result;
}

void Terrain::loadSkyBox()
{
        device->getVideoDriver()->setTextureCreationFlag(ETCF_CREATE_MIP_MAPS, false);
        device->getSceneManager()->addSkyBoxSceneNode(
                device->getVideoDriver()->getTexture("media/terrain/irrlicht2_up.jpg"),
                device->getVideoDriver()->getTexture("media/terrain/irrlicht2_dn.jpg"),
                device->getVideoDriver()->getTexture("media/terrain/irrlicht2_lf.jpg"),
                device->getVideoDriver()->getTexture("media/terrain/irrlicht2_rt.jpg"),
                device->getVideoDriver()->getTexture("media/terrain/irrlicht2_ft.jpg"),
                device->getVideoDriver()->getTexture("media/terrain/irrlicht2_bk.jpg"));
        device->getVideoDriver()->setTextureCreationFlag(ETCF_CREATE_MIP_MAPS, true);
}

---