Shader phong_framebuffer

Click on the filename to display or download the file.

phong_framebuffer.mi
declare shader 
    color "phong_framebuffer" ( 
        color  "ambient"   default 0 0 0, 
        color  "diffuse"   default 1 1 1, 
	color  "specular"  default 0 0 0, 
	scalar "exponent"  default 30, 
        array light "lights" ) 
    version 1 
    apply material 
end declare 

phong_framebuffer.c
#include "shader.h" 
#include "miaux.h" 
 
struct phong_framebuffer { 
    miColor ambient;  
    miColor diffuse; 
    miColor specular; 
    miScalar exponent; 
    int      i_light; 
    int      n_light; 
    miTag    light[1]; 
}; 
 
DLLEXPORT 
int phong_framebuffer_version(void) { return 1; } 
 
DLLEXPORT 
miBoolean phong_framebuffer ( 
    miColor *result, miState *state, struct phong_framebuffer *params ) 
{ 
    miColor light_color, diffuse_from_light, specular_from_light, 
        diffuse_component, specular_component; 
    int i, light_count, light_sample_count; 
    miVector direction_toward_light; 
    miScalar dot_nl; 
    miTag *light; 
 
    miColor *diffuse = mi_eval_color(&params->diffuse); 
    miColor *specular = mi_eval_color(&params->specular); 
    miScalar exponent = *mi_eval_scalar(&params->exponent); 
    miaux_light_array(&light, &light_count, state, 
                      &params->i_light, &params->n_light, params->light); 
 
    *result = *mi_eval_color(&params->ambient); 
    miaux_set_channels(&diffuse_component, 0.0); 
    miaux_set_channels(&specular_component, 0.0); 
 
    for (i = 0; i < light_count; i++, light++) { 
        miaux_set_channels(&diffuse_from_light, 0.0); 
        miaux_set_channels(&specular_from_light, 0.0); 
        light_sample_count = 0; 
 
        while (mi_sample_light(&light_color, &direction_toward_light, 
                               &dot_nl, state, *light, &light_sample_count)) { 
            miaux_add_diffuse_component( 
                &diffuse_from_light, dot_nl, diffuse, &light_color); 
            miaux_add_phong_specular_component( 
                &specular_from_light, state, exponent, 
                &direction_toward_light, specular, &light_color); 
        } 
 
        if (light_sample_count > 0) { 
            miScalar scale_factor = 1.0 / light_sample_count; 
            miaux_add_scaled_color( 
                &diffuse_component, &diffuse_from_light, scale_factor); 
            miaux_add_scaled_color( 
                &specular_component, &specular_from_light, scale_factor); 
        } 
    } 
    mi_fb_put(state, 0, &diffuse_component); 
    mi_fb_put(state, 1, &specular_component); 
 
    miaux_add_color(result, &diffuse_component); 
    miaux_add_color(result, &specular_component); 
 
    return miTRUE; 
} 

phong_framebuffer_util.c
void miaux_light_array(miTag **lights, int *light_count, miState *state, 
		       int *offset_param, int *count_param, miTag *lights_param) 
{ 
    int array_offset = *mi_eval_integer(offset_param); 
    *light_count = *mi_eval_integer(count_param); 
    *lights = mi_eval_tag(lights_param) + array_offset; 
} 
 
void miaux_set_channels(miColor *c, miScalar new_value) 
{ 
    c->r = c->g = c->b = c->a = new_value; 
} 
 
void miaux_add_diffuse_component( 
    miColor *result,  
    miScalar light_and_surface_cosine,  
    miColor *diffuse, miColor *light_color) 
{ 
    result->r += light_and_surface_cosine * diffuse->r * light_color->r; 
    result->g += light_and_surface_cosine * diffuse->g * light_color->g; 
    result->b += light_and_surface_cosine * diffuse->b * light_color->b; 
} 
 
void miaux_add_phong_specular_component( 
    miColor *result, miState *state, miScalar exponent, 
    miVector *direction_toward_light,  
    miColor *specular, miColor *light_color) 
{ 
    miScalar specular_amount = 
	mi_phong_specular(exponent, state, direction_toward_light); 
    if (specular_amount > 0.0) { 
	result->r += specular_amount * specular->r * light_color->r; 
	result->g += specular_amount * specular->g * light_color->g; 
	result->b += specular_amount * specular->b * light_color->b; 
    } 
} 
 
void miaux_add_scaled_color(miColor *result, miColor *color, miScalar scale) 
{ 
    result->r += color->r * scale; 
    result->g += color->g * scale; 
    result->b += color->b * scale; 
} 
 
void miaux_add_color(miColor *result, miColor *c) 
{ 
    result->r += c->r; 
    result->g += c->g; 
    result->b += c->b; 
    result->a += c->a; 
} 

22 April 2008 23:41:01