% Beam with uniform distribution
function u = uniform(r, w0, r_max)
    u = ones(size(r));
end