Water Filling Algorithm for MIMO System

Water filling algorithm is useful in many application, such as finding optimal power allocation among several relay antennas in MIMO.

This water filling algorithm take 3 inputs.
a) Length of each "floor" (column vector).
b) Height of each "floor" (column vector).
c) The water volume (scalar)
The outputs are :
a) The volume above each "floor" (column vector).
b) The water level (scalar)

The loop number implemented inside code is same with the number of "floor" (or the number of antenna relay). The length "floor" may be different each other.

The picture in the upperleft of this website shows the result with following implementation on command window.
>>length = [3 2 2 1]';
>>height = [6 8 1 3]';
>>totalVolume = 19;
>>[eachVolume, waterLevel] = waterFilling(length, height, totalVolume);

Hope This useful.
Any suggestion are welcome.