%% Homework 3.6
clear;
H = [(0.46  + 1i*0.31), (-0.079  + 1i*0.18), (0.43  - 1i*0.68); ...
     (-0.04 + 1i*0.66),  (0.95   - 1i*0.50), (-0.77 - 1i*0.51);...
     (0.45  + 1i*0.46),  (0.90   + 1i*0.56), (0.46  - 1i*0.87);...
     (-0.13 - 1i*0.19),  (0.90   - 1i*0.73), (1.04  + 1i*0.91) ]


 %[m,n] = size(H) %n=t, m=r
 a = -20; %db
 b = 20; %db
 SNRvec_db = a + (b-a).*rand(1, 1000);
 %SNRvec_db = [-10; -9; -8; -7; -6; -5; -4; -3; -2; -1; 0;]

 calcType = 1;
 C = CvsSNR_EqualPowerAllocation(SNRvec_db, H, calcType)
 f1 = figure;
 scatter(SNRvec_db, C, 'filled');

 calcType = 2;
 C = CvsSNR_EqualPowerAllocation(SNRvec_db, H, calcType)
 f2 = figure;
 scatter(SNRvec_db, C, 'filled');

 [C, Weights] = CvsSNR_SingleMode(SNRvec_db, H)
 f3 = figure;
 scatter(SNRvec_db, C, 'filled');