帮我修改此代码 % 计算窗口长度和重叠长度的样点数 nwinlen = round(winlen*fs); % 窗口长度（样点数） nwinshift = round(winshift*fs); % 重叠长度（样点数） % 分帧加窗 nx = length(x); nframe = floor((nx-nwinlen)/nwinshift)+1; % 帧数 w = rectwin(nwinlen); % 矩形窗 win = repmat(w, 1, nframe); % 窗矩阵 x_seg_rect = x(1:nwinlen+(nframe-1)*nwinshift) .* win; % 分帧加窗（矩形窗） w = hamming(nwinlen); % 汉明窗 win = repmat(w, 1, nframe); % 窗矩阵 x_seg_ham = x(1:nwinlen+(nframe-1)*nwinshift) .* win; % 分帧加窗（汉明窗） % 时域波形 t = (0:nx-1)/fs; % 时间轴 subplot(2,2,1); plot(t,x,'k');title('原始波形'); subplot(2,2,2); plot(t(1:nwinlen+(nframe-1)*nwinshift),x_seg_rect,'k'); title('加矩形窗后波形'); subplot(2,2,3); plot(t(1:nwinlen+(nframe-1)*nwinshift),x_seg_ham,'k'); title('加汉明窗后波形'); % 短时频谱 nfft = 512; % FFT点数 f = (0:nfft/2-1)*fs/nfft; % 频率轴 subplot(2,2,4); spectrogram(x_seg_rect,winlen*fs,winshift*fs,nfft,fs,'yaxis');title('加矩形窗后短时频谱'); subplot(2,2,5); spectrogram(x_seg_ham,winlen*fs,winshift*fs,nfft,fs,'yaxis'); title('加汉明窗后短时频谱');

修改后的代码如下所示：

% 计算窗口长度和重叠长度的样点数
nwinlen = round(winlen*fs); % 窗口长度（样点数）
nwinshift = round(winshift*fs); % 重叠长度（样点数）

% 分帧加窗
nx = length(x);
nframe = floor((nx-nwinlen)/nwinshift)+1; % 帧数

% 矩形窗
w_rect = rectwin(nwinlen);
win_rect = repmat(w_rect, 1, nframe); % 窗矩阵
x_seg_rect = x(1:nwinlen+(nframe-1)*nwinshift) .* win_rect; % 分帧加窗（矩形窗）

% 汉明窗
w_ham = hamming(nwinlen);
win_ham = repmat(w_ham, 1, nframe); % 窗矩阵
x_seg_ham = x(1:nwinlen+(nframe-1)*nwinshift) .* win_ham; % 分帧加窗（汉明窗）

% 时域波形
t = (0:nx-1)/fs; % 时间轴
subplot(2,2,1); plot(t,x,'k');title('原始波形');
subplot(2,2,2); plot(t(1:nwinlen+(nframe-1)*nwinshift),x_seg_rect,'k'); title('加矩形窗后波形');
subplot(2,2,3); plot(t(1:nwinlen+(nframe-1)*nwinshift),x_seg_ham,'k'); title('加汉明窗后波形');

% 短时频谱
nfft = 512; % FFT点数
f = (0:nfft/2-1)*fs/nfft; % 频率轴
subplot(2,2,4); spectrogram(x_seg_rect,win_rect,winlen*fs-nwinshift,nfft,fs,'yaxis');title('加矩形窗后短时频谱');
subplot(2,2,5); spectrogram(x_seg_ham,win_ham,winlen*fs-nwinshift,nfft,fs,'yaxis'); title('加汉明窗后短时频谱');

修改说明：

1. 将矩形窗和汉明窗的定义分别赋值给 `w_rect` 和 `w_ham` 变量。

2. 将 `win` 变量改为 `win_rect` 和 `win_ham`，分别表示矩形窗和汉明窗的窗矩阵。

3. 将 `winshift*fs` 改为 `winlen*fs-nwinshift`，以保证每个窗口之间有重叠部分。

4. 在 `spectrogram()` 函数中，将 `win` 参数改为相应的窗矩阵变量，以便选择不同的窗函数。

这样修改后的代码应该能够正确执行。