LightExporter.cpp

Go to the documentation of this file.

/*
 * ***** BEGIN GPL LICENSE BLOCK *****
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov, Jan Diederich, Tod Liverseed,
 *                 Nathan Letwory
 *
 * ***** END GPL LICENSE BLOCK *****
 */

#include <string>

#include "COLLADASWColor.h"
#include "COLLADASWLight.h"

#include "BLI_math.h"

#include "LightExporter.h"
#include "collada_internal.h"

template<class Functor>
void forEachLampObjectInScene(Scene *sce, Functor &f, bool export_selected)
{
    Base *base= (Base*) sce->base.first;
    while(base) {
        Object *ob = base->object;
            
        if (ob->type == OB_LAMP && ob->data
            && !(export_selected && !(ob->flag & SELECT))) {
            f(ob);
        }
        base= base->next;
    }
}

LightsExporter::LightsExporter(COLLADASW::StreamWriter *sw, const ExportSettings *export_settings): COLLADASW::LibraryLights(sw), export_settings(export_settings) {}

void LightsExporter::exportLights(Scene *sce)
{
    openLibrary();
    
    forEachLampObjectInScene(sce, *this, this->export_settings->selected);
    
    closeLibrary();
}

void LightsExporter::operator()(Object *ob)
{
    Lamp *la = (Lamp*)ob->data;
    std::string la_id(get_light_id(ob));
    std::string la_name(id_name(la));
    COLLADASW::Color col(la->r * la->energy, la->g * la->energy, la->b * la->energy);
    float d, constatt, linatt, quadatt;
    
    d = la->dist;
    
    constatt = 1.0f;
    
    if(la->falloff_type==LA_FALLOFF_INVLINEAR) {
        linatt = 1.0f / d;
        quadatt = 0.0f;
    }
    else {
        linatt = 0.0f;
        quadatt = 1.0f / (d * d);
    }
    
    // sun
    if (la->type == LA_SUN) {
        COLLADASW::DirectionalLight cla(mSW, la_id, la_name);
        cla.setColor(col,false,"color");
        cla.setConstantAttenuation(constatt);
        exportBlenderProfile(cla, la);
        addLight(cla);
    }
    // hemi
    else if (la->type == LA_HEMI) {
        COLLADASW::AmbientLight cla(mSW, la_id, la_name);
        cla.setColor(col,false,"color");
        cla.setConstantAttenuation(constatt);
        exportBlenderProfile(cla, la);
        addLight(cla);
    }
    // spot
    else if (la->type == LA_SPOT) {
        COLLADASW::SpotLight cla(mSW, la_id, la_name);
        cla.setColor(col,false,"color");
        cla.setFallOffAngle(la->spotsize,false,"fall_off_angle");
        cla.setFallOffExponent(la->spotblend,false,"fall_off_exponent");
        cla.setConstantAttenuation(constatt);
        cla.setLinearAttenuation(linatt);
        cla.setQuadraticAttenuation(quadatt);
        exportBlenderProfile(cla, la);
        addLight(cla);
    }
    // lamp
    else if (la->type == LA_LOCAL) {
        COLLADASW::PointLight cla(mSW, la_id, la_name);
        cla.setColor(col,false,"color");
        cla.setConstantAttenuation(constatt);
        cla.setLinearAttenuation(linatt);
        cla.setQuadraticAttenuation(quadatt);
        exportBlenderProfile(cla, la);
        addLight(cla);
    }
    // area lamp is not supported
    // it will be exported as a local lamp
    else {
        COLLADASW::PointLight cla(mSW, la_id, la_name);
        cla.setColor(col,false,"color");
        cla.setConstantAttenuation(constatt);
        cla.setLinearAttenuation(linatt);
        cla.setQuadraticAttenuation(quadatt);
        exportBlenderProfile(cla, la);
        addLight(cla);
    }
    
}

bool LightsExporter::exportBlenderProfile(COLLADASW::Light &cla, Lamp *la)
{
    cla.addExtraTechniqueParameter("blender", "type", la->type);
    cla.addExtraTechniqueParameter("blender", "flag", la->flag);
    cla.addExtraTechniqueParameter("blender", "mode", la->mode);
    cla.addExtraTechniqueParameter("blender", "gamma", la->k, "blender_gamma");
    cla.addExtraTechniqueParameter("blender", "red", la->r);
    cla.addExtraTechniqueParameter("blender", "green", la->g);
    cla.addExtraTechniqueParameter("blender", "blue", la->b);
    cla.addExtraTechniqueParameter("blender", "shadow_r", la->shdwr, "blender_shadow_r");
    cla.addExtraTechniqueParameter("blender", "shadow_g", la->shdwg, "blender_shadow_g");
    cla.addExtraTechniqueParameter("blender", "shadow_b", la->shdwb, "blender_shadow_b");
    cla.addExtraTechniqueParameter("blender", "energy", la->energy, "blender_energy");
    cla.addExtraTechniqueParameter("blender", "dist", la->dist, "blender_dist");
    cla.addExtraTechniqueParameter("blender", "spotsize", la->spotsize);
    cla.addExtraTechniqueParameter("blender", "spotblend", la->spotblend);
    cla.addExtraTechniqueParameter("blender", "halo_intensity", la->haint, "blnder_halo_intensity");
    cla.addExtraTechniqueParameter("blender", "att1", la->att1);
    cla.addExtraTechniqueParameter("blender", "att2", la->att2);
    // \todo figure out how we can have falloff curve supported here
    cla.addExtraTechniqueParameter("blender", "falloff_type", la->falloff_type);
    cla.addExtraTechniqueParameter("blender", "clipsta", la->clipsta);
    cla.addExtraTechniqueParameter("blender", "clipend", la->clipend);
    cla.addExtraTechniqueParameter("blender", "shadspotsize", la->shadspotsize);
    cla.addExtraTechniqueParameter("blender", "bias", la->bias);
    cla.addExtraTechniqueParameter("blender", "soft", la->soft);
    cla.addExtraTechniqueParameter("blender", "compressthresh", la->compressthresh);
    cla.addExtraTechniqueParameter("blender", "bufsize", la->bufsize);
    cla.addExtraTechniqueParameter("blender", "samp", la->samp);
    cla.addExtraTechniqueParameter("blender", "buffers", la->buffers);
    cla.addExtraTechniqueParameter("blender", "filtertype", la->filtertype);
    cla.addExtraTechniqueParameter("blender", "bufflag", la->bufflag);
    cla.addExtraTechniqueParameter("blender", "buftype", la->buftype);
    cla.addExtraTechniqueParameter("blender", "ray_samp", la->ray_samp);
    cla.addExtraTechniqueParameter("blender", "ray_sampy", la->ray_sampy);
    cla.addExtraTechniqueParameter("blender", "ray_sampz", la->ray_sampz);
    cla.addExtraTechniqueParameter("blender", "ray_samp_type", la->ray_samp_type);
    cla.addExtraTechniqueParameter("blender", "area_shape", la->area_shape);
    cla.addExtraTechniqueParameter("blender", "area_size", la->area_size);
    cla.addExtraTechniqueParameter("blender", "area_sizey", la->area_sizey);
    cla.addExtraTechniqueParameter("blender", "area_sizez", la->area_sizez);
    cla.addExtraTechniqueParameter("blender", "adapt_thresh", la->adapt_thresh);
    cla.addExtraTechniqueParameter("blender", "ray_samp_method", la->ray_samp_method);
    cla.addExtraTechniqueParameter("blender", "shadhalostep", la->shadhalostep);
    cla.addExtraTechniqueParameter("blender", "sun_effect_type", la->shadhalostep);
    cla.addExtraTechniqueParameter("blender", "skyblendtype", la->skyblendtype);
    cla.addExtraTechniqueParameter("blender", "horizon_brightness", la->horizon_brightness);
    cla.addExtraTechniqueParameter("blender", "spread", la->spread);
    cla.addExtraTechniqueParameter("blender", "sun_brightness", la->sun_brightness);
    cla.addExtraTechniqueParameter("blender", "sun_size", la->sun_size);
    cla.addExtraTechniqueParameter("blender", "backscattered_light", la->backscattered_light);
    cla.addExtraTechniqueParameter("blender", "sun_intensity", la->sun_intensity);
    cla.addExtraTechniqueParameter("blender", "atm_turbidity", la->atm_turbidity);
    cla.addExtraTechniqueParameter("blender", "atm_extinction_factor", la->atm_extinction_factor);
    cla.addExtraTechniqueParameter("blender", "atm_distance_factor", la->atm_distance_factor);
    cla.addExtraTechniqueParameter("blender", "skyblendfac", la->skyblendfac);
    cla.addExtraTechniqueParameter("blender", "sky_exposure", la->sky_exposure);
    cla.addExtraTechniqueParameter("blender", "sky_colorspace", la->sky_colorspace);
    
    return true;
}