Midterm_Sebastien_Dubois_18.../frequencySpectrum.m

63 lines
1.5 KiB
Matlab

function power = frequencySpectrum(signal, fs, resolution, part_signal)
%%%%%%%%%%%%%%%%%%
%function power = frequencySpectrum(signal, fs, pad)
%
% Task: Display the power spectrum (lin and log scale) of a given signal
%
% Input:
% - signal: the input signal to process
% - fs: the sampling rate in Hz
% - resolution: frequency resolution in Hz, signal will be padded with zeros if necessary
%
% Output:
% - power: the power spectrum
%
%
% Guillaume Gibert, guillaume.gibert@ecam.fr
% 15/03/2024
%%%%%%%%%%%%%%%%%%
n = length(signal); % number of samples
current_resolution = fs / n;
if (resolution < current_resolution)
n_original = n;
n = fs / resolution;
signal = [signal zeros(1, n-n_original)];
end
%~ if (pad)
%~ n_original = n;
%~ n = 2^(nextpow2(n));
%~ signal = [signal zeros(1, n-n_original)];
%~ end
y = fft(signal, n);% compute DFT of input signal
power = abs(y).^2/n; % power of the DFT
[val, ind] = max(power); % find the mx value of DFT and its index
% plots
figure;
subplot(1,3,1) % time plot
t=0:1/fs:(n-1)/fs; % time range
plot(t, signal)
xticks(0:0.1fs:nfs);
xticklabels(0:0.1:n/fs);
xlabel('Time (s)');
ylabel('Amplitude (a.u.)');
subplot(1,3,2) % linear frequency plot
f = (0:n-1)(fs/n); % frequency range
plot(f,power, 'b'); hold on;
plot(f,power, 'r');
title('Plot for signal number', part_signal)
xlabel('Frequency (Hz)')
ylabel('Power (a.u.)')
subplot(1,3,3) % log frequency plot
plot(f,10*log10(power/power(ind)));
xlabel('Frequency (Hz)')
ylabel('Power (dB)')