clc; clear;
N = 16;
M = 4;
epi = 0;

d_n = floor(rand(N, 1)*M) / M;
A = get_Psi(N, M, d_n, epi);
B = get_Psi_2(N, M, d_n, epi);
C = A - B;
real_C = real(C);
imag_C = imag(C);

% subplot(121)
% heatmap(real_C)
% 
% subplot(122)
% heatmap(imag_C)

r = zeros(N);
for i = 1: N
    for j = 1: N
        r(i, j) = A(i, :) * (A(j, :)');
    end
end
heatmap(abs(r));