请设计一个小波包变换的matlab代码

以下是一个简单的小波包变换的 Matlab 代码：

function [coefficients] = wavelet_packet_transform(signal, depth, wavelet)
% 小波包变换函数
% signal: 待变换的信号
% depth: 小波包变换的深度
% wavelet: 小波基函数

% 初始化小波包树
t = wpdec(signal, depth, wavelet);

% 获取小波包树的终端节点
leaves = wpdec2nodes(t, 'all');

% 对每个节点进行小波变换
for i = 1:length(leaves)
    coefficients{i} = wpcoef(t, leaves(i));
end

end

使用方法：

- `signal`：待变换的信号，可以是行向量或列向量。
- `depth`：小波包变换的深度，即小波包树的层数。
- `wavelet`：小波基函数的名称，可以为 `'haar'`、`'dbN'`、`'symN'`、`'coifN'` 等，其中 `N` 为小波基的阶数。

返回值：

- `coefficients`：一个元胞数组，包含了小波包变换后的系数，每个元胞对应小波包树中的一个终端节点。

示例：

% 生成测试信号
t = linspace(0, 1, 1024);
x = sin(2*pi*50*t) + sin(2*pi*120*t);

% 进行小波包变换
depth = 3;
wavelet = 'db4';
coefficients = wavelet_packet_transform(x, depth, wavelet);

% 绘制小波包系数的图像
figure;
for i = 1:length(coefficients)
    subplot(length(coefficients), 1, i);
    plot(coefficients{i});
end

该代码将信号进行了小波包变换，并绘制了小波包系数的图像，可以通过图像观察小波包系数的变化规律，进一步分析信号的特征。