matlab定义计算THD函数代码

在MATLAB中，总谐波失真（Total Harmonic Distortion, THD）是一个衡量信号波形质量的指标，它表示信号中谐波分量相对于基波（基频）分量的比例。定义一个计算THD的函数，通常需要执行以下步骤：

1. 从信号中提取基频分量。
2. 计算信号的基频分量幅值。
3. 计算信号中所有谐波分量的总幅值。
4. 计算THD值，即谐波分量总幅值与基频分量幅值之比。

以下是一个简单的MATLAB函数示例，用于计算给定信号的THD：

function thd = calculateTHD(signal, fs)
    % calculateTHD - 计算信号的总谐波失真（THD）
    % 输入参数：
    % signal - 输入信号
    % fs - 信号采样频率
    
    % 检查输入参数
    if nargin < 2
        error('请输入信号和采样频率');
    end
    
    % 基频计算
    fundamentalFreq = fs / length(signal);
    
    % 快速傅里叶变换（FFT）计算
    signal_fft = fft(signal);
    
    % 计算幅值谱
    signal_fft_mag = abs(signal_fft);
    
    % 初始化基频分量的幅值
    fundamentalComponent = 0;
    
    % 寻找基频分量的幅值
    for k = 1:length(signal_fft_mag)
        if abs((k-1) * fundamentalFreq - fs/2) < fundamentalFreq/2
            fundamentalComponent = max(fundamentalComponent, signal_fft_mag(k));
        end
    end
    
    % 计算总谐波分量的幅值
    harmonicComponent = sum(signal_fft_mag(fundamentalComponent < signal_fft_mag));
    
    % 计算THD
    thd = harmonicComponent / fundamentalComponent;
end

在使用上述函数之前，你需要有一个信号的样本数组以及该信号的采样频率。将信号和采样频率作为参数传入函数中，就可以得到计算出的THD值。

请记住，这只是一个基本的THD计算示例。在实际应用中，可能需要考虑信号的窗函数处理、基频分量的确切检测方法和信号预处理等因素。