campello_algo.m

汪裕民的OFDM关键技术与应用中仿真MIMO-OFDM的源代码

function [bits_alloc,energy_alloc] = campello_algo(bits_alloc,energy_alloc,energytable,total_bits,num_subc,SNR,M)

bt = sum(bits_alloc);
%   不能传输大于M的比特数
if  total_bits>M*num_subc
    total_bits = M*num_subc;
end

while(bt~=total_bits)
    if(bt>total_bits)
        max_val = 0;
        max_ind = ceil(rand(1)*num_subc);
        for i = 1:num_subc
            if bits_alloc(i)~=0
                temp = energytable(i,bits_alloc(i));
            else
                temp = 0;
            end
            if temp>max_val
                max_val = temp;
                max_ind = i;
            end
        end
        if(bits_alloc(max_ind)>0)
            bits_alloc(max_ind) = bits_alloc(max_ind)-1;
%             energy_alloc(max_ind) = energy_alloc(max_ind)-max_val;
            energy_alloc(max_ind) = (2^bits_alloc(max_ind)-1)/SNR(max_ind);
            bt = bt-1;
        end
    else
    min_val =Inf;
    min_ind = ceil(rand(1)*num_subc);
    for i = 1:num_subc
        if bits_alloc(i)~=0&bits_alloc(i)<9
            temp = energytable(i,bits_alloc(i)+1);
        else
            temp = Inf;
        end
        if(temp<min_val)
            min_val = temp;
            min_ind = i;
        end
    end
    if(bits_alloc(min_ind)<8)
        bits_alloc(min_ind) = bits_alloc(min_ind)+1;
        if(min_val==inf)
            min_val = energytable(min_ind,bits_alloc(min_ind));
        end
%         energy_alloc(min_ind) = energy_alloc(min_ind)+min_val;
        energy_alloc(min_ind) = (2^bits_alloc(min_ind)-1)/SNR(max_ind);
        bt = bt+1;
    end
    end
end

for i = 1:length(bits_alloc)
    if(mod(bits_alloc(i),2)==1&bits_alloc(i)~=1)
        [bits_alloc,energy_alloc] = ResolvetheLastBit(bits_alloc,energy_alloc,i,energytable,num_subc);
    end
end