src/CombineShader.h

Go to the documentation of this file.

#ifndef COMBINESHADER_H
#define COMBINESHADER_H

#include <vector>

#include "RGBAColor.h"
#include "defines.h"


class CombineShader : public Shader
{
public:
    
    CombineShader(  Scene * scene,
                    bool    addShaders )
        : Shader(scene),
            mLiShader(NULL, 0.0f),
            mTrShader(NULL, 0.0f),
            mAddShaders(addShaders)
    {
    }

    virtual RGBAColor shade(const Ray & ray) const
    {
        //     for (std::vector<std::pair<Shader *, float> >::const_iterator it=mShaders.begin(); it != mShaders.end(); it++)
//            result = lerp(result, it->first->shade(ray), it->second);
        //    result += it->first->shade(ray) * it->second;
        
        // shade lighting
        RGBAColor lighting;
        if (mLiShader.first)
            lighting = mLiShader.first->shade(ray);
        lighting.setA(1);

        // add Textures
        if (textureCount() > 0)
        {
            std::vector<Texture*>::const_iterator it = mTextures.begin();
            RGBAColor texes = (*it)->texel(ray.hit()->texCoord(ray));     
            ++it;
            for(; it!=mTextures.end(); ++it)
            {
                // get texture color
                const RGBAColor tex = (*it)->texel(ray.hit()->texCoord(ray));

                // combine texture color with the computed result
                texes = lerp(texes, tex, tex.a());
            }

            // add lighting
            lighting.scale(texes);
        }

        // set up adaptive recursion termination
        Ray tray = ray;
        tray.updateInfluence(ray, mAddShaders ? mTrShader.second : mTrShader.second * (1.0f - lighting.a()));
        // shade transparency
        RGBAColor transparency;
        if (mTrShader.first)
            transparency = mTrShader.first->shade(tray);
        transparency.setA(1);

        if (mAddShaders)
            return RGBAColor(transparency.rgb() * mTrShader.second + lighting.rgb() * mLiShader.second);
        else
            return RGBAColor(lerp(transparency.rgb() * mTrShader.second, lighting.rgb() * mLiShader.second, lighting.a()));
    }

    void setLightingShader(Shader * sh, float alpha)
    {
        if (!sh)
            alpha = 0;
        mLiShader = std::make_pair(sh, alpha);
    }


    void setTransparentShader(Shader * sh, float alpha)
    {
        if (!sh)
            alpha = 0;
        mTrShader = std::make_pair(sh, alpha);
    }

private:
    std::pair<Shader *, float>  mLiShader;

    std::pair<Shader *, float>  mTrShader;

    bool                        mAddShaders;
};

#endif

---