I don't understand why the different geometric factors are squared: if `weight_norm` computes the total length of the light-ray, it seems to me that adding the `(1 - di) * (1 - dj) * ds`, `di * dj * ds`, etc. (and not their squared value) should give the right solution. Did I miss something?