用matlab写一个对余弦信号的wvd变换用自相关法 并用mesh画图

以下是用 MATLAB 实现对余弦信号的 WVD 变换用自相关法，并用 mesh 画图的代码：

% 初始化参数
N = 256; % 采样点数
fs = 1000; % 采样率
t = linspace(0, N/fs, N); % 时间序列

% 构造余弦信号
f0 = 50; % 基频
x = cos(2*pi*f0*t);

% 计算自相关函数
r = xcorr(x);

% 计算WVD
WVD = zeros(N);
for n = 1:N
    for m = 1:N
        WVD(m,n) = 1/(pi*n) * abs(sum(r(N+m-N:n+m-N).*r(N-m+n:N+m-n)));
    end
end

% 画图
figure;
mesh(t, t, WVD);
xlabel('Time (s)');
ylabel('Time (s)');
zlabel('WVD');
title('Wigner-Ville Distribution of Cosine Signal using Autocorrelation Method');

运行代码后，将会生成一个 3D 的 mesh 图，展示余弦信号的 WVD 分布情况。