src/rcrt/bxdf/BSDF.cpp

Go to the documentation of this file.

#include "BSDF.h"

using namespace std;

namespace rcrt
{

BSDF::BSDF()
{
}

BSDF::~BSDF()
{
}

void BSDF::addBXDF(BXDF* bxdf)
{
        bxdfs.push_back(bxdf);
}

RGBColor BSDF::eval(const Vec3D& wOut, const Vec3D& wInc, const Vec3D& gNormal, const Vec3D& sNormal) const
{
        bool ignoreBTDFs = false;
        if ((wInc * gNormal) * (wOut * sNormal) < 0){
                // ignore BTDFs
                ignoreBTDFs = true;
        }
        RGBColor col(0);
        for (unsigned int i = 0; i < bxdfs.size(); i++){
                if(ignoreBTDFs){
                        switch(bxdfs[i]->type){
                        case REFLECTIVE:
                                col = col + bxdfs[i]->eval(wOut, wInc, sNormal); break;
                        case TRANSMISSIVE:
                                break;
                        default://BOTH
                                col = col + bxdfs[i]->eval(wOut, wInc, sNormal);
                        }
                } else {
                        switch(bxdfs[i]->type){
                        case REFLECTIVE:
                                break;
                        case TRANSMISSIVE:
                                col = col + bxdfs[i]->eval(wOut, wInc, sNormal); break;
                        default://BOTH
                                col = col + bxdfs[i]->eval(wOut, wInc, sNormal);
                        }                       
                }
        }
        return col;
}

bool BSDF::hasDiffuse() const
{
        for(unsigned int i = 0; i < bxdfs.size(); i++)
                if(bxdfs[i]->diffuse())
                        return true;
        return false;
}

bool BSDF::hasSpecular() const
{
        for(unsigned int i = 0; i < bxdfs.size(); i++)
                if(bxdfs[i]->specular())
                        return true;
        return false;
}

bool BSDF::hasTransmissive() const
{
        for(unsigned int i = 0; i < bxdfs.size(); i++)
                if(bxdfs[i]->type == TRANSMISSIVE
                                || bxdfs[i]->type == BOTH)
                        return true;
        return false;
}

RGBColor BSDF::evalSpecular(const Vec3D& wOut, const Vec3D& wInc, const Vec3D& gNormal, const Vec3D& sNormal) const
{
        RGBColor col(0);
        for (unsigned int i = 0; i < bxdfs.size(); i++){
                if ((bxdfs[i]->type == REFLECTIVE || bxdfs[i]->type == BOTH)
                                && bxdfs[i]->specular())
                        col = col + bxdfs[i]->evalSpecular(wOut, wInc, sNormal);
        }
        return col;
}

RGBColor BSDF::evalDiffuse(const Vec3D& wOut, const Vec3D& wInc, const Vec3D& gNormal, const Vec3D& sNormal) const
{
        RGBColor col(0);
        for (unsigned int i = 0; i < bxdfs.size(); i++)
                if ((bxdfs[i]->type == REFLECTIVE || bxdfs[i]->type == BOTH)
                                && bxdfs[i]->diffuse())
                        col = col + bxdfs[i]->evalDiffuse(wOut, wInc, sNormal);
        return col;
}


RGBColor BSDF::getKdiffuse() const
{
        RGBColor diff(0);
        int n = 0;
        for (unsigned int i = 0; i < bxdfs.size(); i++){
                if((bxdfs[i]->type == REFLECTIVE || bxdfs[i]->type == BOTH)
                                && bxdfs[i]->diffuse()){
                        diff = diff + bxdfs[i]->getKd();
                        n++;
                }
        }
        if(n == 0) return 0;
        return diff / n;
}

RGBColor BSDF::getKspecular() const
{
        RGBColor spec(0);
        int n = 0;
        for (unsigned int i = 0; i < bxdfs.size(); i++){
                if((bxdfs[i]->type == REFLECTIVE || bxdfs[i]->type == BOTH)
                                && bxdfs[i]->specular()){
                        spec = spec + bxdfs[i]->getKs();
                        n++;
                }               
        }
        if(n == 0) return 0;
        return spec / n;
}

RGBColor BSDF::getKtransmit() const
{
        RGBColor trans(0);
        int n = 0;
        for (unsigned int i = 0; i < bxdfs.size(); i++){
                if((bxdfs[i]->type == TRANSMISSIVE || bxdfs[i]->type == BOTH)){
                        trans = trans + bxdfs[i]->getKt();
                        n++;
                }               
        }
        if(n == 0) return 0;
        return trans / n;       
}

}

---