LightMap.cc

Go to the documentation of this file.

#include <esg/energy/LightMap.h>

using namespace esg;

LightMap::LightMap(unsigned u, unsigned v, const Vector3& c, const Vector3& e)
    : EnergyMap(u,v), _centroid(c), _extent(e)
{
#ifdef WIN32
        _lightMap = (Color3f**) malloc(6 * _vRes * sizeof(Color3f*));
#else
        _lightMap = new Color3f* [6 * _vRes];
#endif
    for (register unsigned i = 0; i < 6 * _vRes; i++) {
        _lightMap[i] = new Color3f[u];
        for (register unsigned j = 0; j < _uRes; j++)
            _lightMap[i][j].set(.0, .0, .0);
    }
}

LightMap::~LightMap()
{
    for (register unsigned i = 0; i < 6 * _vRes; i++) delete [] _lightMap[i];
    delete [] _lightMap;
}

bool LightMap::getPointEnergy(const Vector3& p, Vector3& e)
{
    unsigned u, v;
    MapIndex i;
    if (!getMapCoords(p, i, u, v)) return false;
    e.set(_lightMap[i*_vRes+v][u]);
    return true;
}

bool LightMap::setPointEnergy(const Vector3& p, const Vector3& e)
{
    unsigned u, v;
    MapIndex i;
    if (!getMapCoords(p, i, u, v)) return false;
    _lightMap[i*_vRes+v][u].set(e);
    return true;
}

bool LightMap::setValue(MapIndex i, unsigned u, unsigned v, const Color3f& c)
{
    if (u >= _uRes || v >= _vRes || i > 5) return false;
    _lightMap[i*_vRes+v][u].set(c);
    return true;
}

bool LightMap::getValue(MapIndex i, unsigned u, unsigned v, Color3f& c) const
{
    if (u >= _uRes || v >= _vRes || i > 5) return false;
    c.set(_lightMap[i*_vRes+v][u]);
    return true;
}

bool LightMap::getMapCoords(const Vector3& p,
                            MapIndex&      i,
                            unsigned&      u,
                            unsigned&      v) const
{
    Vector3 dir(p);
    dir.sub(_centroid);
    Vector3 adir(dir);
    adir.absolute();

    if (adir.x > adir.y) {
        if (adir.x > adir.z) {
            if (dir.x > 0) i = RIGHT_MAP;
            else           i = LEFT_MAP;
        } else {
            if (dir.z > 0) i = NEAR_MAP;
            else           i = FAR_MAP;
        }
    } else {
        if (adir.y > adir.z) {
            if (dir.y > 0) i = TOP_MAP;
            else           i = BOTTOM_MAP;
        } else {
            if (dir.z > 0) i = NEAR_MAP;
            else           i = FAR_MAP;
        }
    }

    if (i == RIGHT_MAP || i == LEFT_MAP) {
        // a = <-cos(PI/4),cos(PI/4)> = <-1/sqrt(2),1/sqrt(2)> :
        float a = dir.y / sqrt(dir.x*dir.x + dir.y*dir.y);
        // a = <0,cos(PI/2)> = <0,2/sqrt(2)>
        a += 1/sqrt(2.0);
        // a = <0,1>
        a /= PI/2;
        // v = <0, _vRes-1>
        v = (unsigned) (a * (_vRes - 1));

        a = dir.z / sqrt(dir.x*dir.x + dir.z*dir.z);
        a += 1/sqrt(2.0);
        a /= PI/2;
        if (i == LEFT_MAP) a = 1 - a;
        u = (unsigned) (a * (_uRes - 1));

        return true;
    }

    if (i == FAR_MAP || i == NEAR_MAP) {
        float a = dir.y / sqrt(dir.z*dir.z + dir.y*dir.y);
        a += 1/sqrt(2.0);
        a /= PI/2;
        v = (unsigned) (a * (_vRes - 1));

        a = dir.x / sqrt(dir.z*dir.z + dir.x*dir.x);
        a += 1/sqrt(2.0);
        a /= PI/2;
        if (i == FAR_MAP) a = 1 - a;
        u = (unsigned) (a * (_uRes - 1));

        return true;
    }

    if (i == BOTTOM_MAP || i == TOP_MAP) {
        float a = dir.z / sqrt(dir.z*dir.z + dir.y*dir.y);
        a += 1/sqrt(2.0);
        a /= PI/2;
        if (i == TOP_MAP) a = 1 - a;
        v = (unsigned) (a * (_vRes - 1));

        a = dir.x / sqrt(dir.y*dir.y + dir.x*dir.x);
        a += 1/sqrt(2.0);
        a /= PI/2;
        u = (unsigned) (a * (_uRes - 1));

        return true;
    }

    return false;
}

---