#include "shader.h"
#include "miaux.h"

struct global_lambert {
    miColor ambient;
    miColor diffuse;
    int     i_light;
    int     n_light;
    miTag   light[1];
};

DLLEXPORT
int global_lambert_version(void) { return 1; }

DLLEXPORT
miBoolean global_lambert (
    miColor *result, miState *state, struct global_lambert *params  )
{
    int i, light_count, light_sample_count;
    miColor sum, light_color, global_illumination_color;
    miScalar dot_nl;
    miTag *light;

    miColor *diffuse  = mi_eval_color(&params->diffuse);
    miaux_light_array(&light, &light_count, state,
                      &params->i_light, &params->n_light, params->light);
    *result = *mi_eval_color(&params->ambient);

    for (i = 0; i < light_count; i++, light++) {
        miaux_set_channels(&sum, 0);
        light_sample_count = 0;
        while (mi_sample_light(&light_color, NULL, &dot_nl,
                               state, *light, &light_sample_count))
            miaux_add_diffuse_component(&sum, dot_nl, diffuse, &light_color);
        if (light_sample_count)
            miaux_add_scaled_color(result, &sum, 1.0/light_sample_count);
    }
    mi_compute_avg_radiance(&global_illumination_color, state, 'f', NULL);
    miaux_add_multiplied_colors(result, &global_illumination_color, diffuse);
    result->a = 1.0;

    return miTRUE;
}