Документация Engee
Notebook

Сравнение MATLAB и Engee в задачах спектрального и частотно-временного анализа

Сравнивая MATLAB и Engee в областях спектрального анализа и частотно-временного анализа, рассмотрим их возможности и подходы для двух задач: прямое и обратное преобразование Фурье и частотно-временной анализ.

Зададим случайный входной сигнал.

In [ ]:
Pkg.add(["DSP"])
   Resolving package versions...
Warning: detected a stack overflow; program state may be corrupted, so further execution might be unreliable.
     Project No packages added to or removed from `~/.project/Project.toml`
    Manifest No packages added to or removed from `~/.project/Manifest.toml`
In [ ]:
x_in = randn(300)
plot(x_in)
Out[0]:

Спектральное преобразование Фурье (FFT и IFFT)

В Engee используются функции из пакета FFTW:

  1. fft: прямое быстрое преобразование Фурье.
  2. ifft: обратное преобразование Фурье.
In [ ]:
using FFTW

# Прямое преобразование Фурье
X = fft(x_in)

# Обратное преобразование Фурье
x_e = ifft(X)
Out[0]:
300-element Vector{ComplexF64}:
   0.3979493700936538 - 1.0954200509634878e-16im
  -0.4848600811102097 - 1.556162045853668e-16im
   1.5842814778290957 - 3.5071455278969295e-16im
   1.0508752491852456 - 2.391742695653933e-17im
  -1.2366907113199697 - 4.071259559251789e-17im
 -0.36960235504034455 + 1.4595447252478719e-16im
  -1.1426283543776241 + 4.3707707580700186e-17im
   0.8564914272398407 + 1.2805741139866173e-16im
 -0.12485768932464274 - 5.477902093541194e-17im
   0.3697754344032267 - 1.117313298369714e-16im
   0.5155016257717131 + 1.116622300172914e-16im
   2.2484859491160183 - 3.921415399333804e-17im
   1.4148589776607792 - 6.505444427416594e-17im
                      ⋮
   1.1048524345297837 + 2.0287474369405695e-17im
  0.42662339105753333 + 1.0892700921806983e-16im
  -1.1609732505396098 - 1.8753233893151456e-16im
    0.706873531822772 + 2.324088774315001e-16im
   0.5593066641593584 + 3.0191244662088247e-16im
   1.2912310062932724 - 1.0528583049558412e-16im
   -0.449143888716872 + 1.2244696865724196e-16im
    1.078424926289395 - 1.0546692614961304e-16im
   1.2499823098900609 - 1.7038939789605642e-16im
    0.319974449064981 + 5.9782005894739336e-18im
  -0.6473573556894365 - 1.3769334329958588e-16im
  -0.7737153966819301 + 1.2003344690763118e-16im

В MATLAB имеются встроенные функции для быстрого преобразования Фурье:

  1. fft: прямое быстрое преобразование Фурье.
  2. ifft: обратное быстрое преобразование Фурье.
In [ ]:
using MATLAB

# Прямое преобразование Фурье
mat"""X = fft($(x_in));"""

# Обратное преобразование Фурье
x_m=mat"""ifft(X);"""
Out[0]:
300-element Vector{Float64}:
  0.39794937009365344
 -0.48486008111020956
  1.5842814778290955
  1.0508752491852458
 -1.2366907113199697
 -0.3696023550403446
 -1.1426283543776241
  0.8564914272398406
 -0.1248576893246431
  0.36977543440322674
  0.515501625771713
  2.2484859491160183
  1.4148589776607792
  ⋮
  1.1048524345297834
  0.4266233910575327
 -1.1609732505396104
  0.7068735318227719
  0.5593066641593588
  1.291231006293272
 -0.4491438887168719
  1.078424926289395
  1.2499823098900606
  0.3199744490649812
 -0.6473573556894366
 -0.7737153966819301

Сравним результаты и рассчитаем погрешность.

In [ ]:
plot(x_in)
plot!(real(x_e))
plot!(x_m)
Out[0]:
In [ ]:
error_e = abs.(x_in-real(x_e))
error_m = abs.(x_in-x_m)
println("Средняя ошибка Engee: $(sum(error_e)/length(error_e))")
println("Средняя ошибка MATLAB: $(sum(error_m)/length(error_m))")
Средняя ошибка Engee: 1.6997615699213912e-16
Средняя ошибка MATLAB: 2.294193481022777e-16

В обоих средах результат прямого преобразования Фурье представляет собой комплексные числа, а обратное преобразование возвращает исходный сигнал с небольшой погрешностью, связанной с численным округлением.
Таким образом, вы можете видеть, что синтаксис и логика кода идентичны.

Частотно-временное преобразование Фурье

В Engee под такие преобразования нет стандартных функций, и их необходимо реализовывать вручную.

In [ ]:
using DSP
# Функция для генерации чирп-сигнала
function chirp(t::AbstractVector{T}, f0::T, t1::T, k::T) where T<:Real
    return cos.(2π * (f0 .* t .+ 0.5 * k .* t.^2))
end

# Окно Хэмминга
function hamming(N::Int)
    n = 0:N-1
    return 0.54 .- 0.46 .* cos.(2π * n / (N - 1))
end

x = chirp(0:0.001:1, 0.0, 1.0, 100.0)  # Чирп-сигнал
l=500 # Размер окна
window=vcat(hamming(l),zeros(length(x)-l))

P = DSP.periodogram(x; fs=1000, window=window, nfft=length(x))

# Визуализация спектра
plot(freq(P), 20*log10.(power(P)), xlabel="Frequency (Hz)", ylabel="Power Spectrum (dB)", title="Power Spectrum", linewidth=2)
Out[0]:

MATLAB имеет функцию pspectrum, которая позволяет выполнять частотно-временной анализ сигнала.
pspectrum автоматически выбирает подходящий метод анализа.
Поддерживает различные визуализации (спектрограммы, плотности мощности).

In [ ]:
using Images
mat"""cd($(@__DIR__))"""
mat"""
% Частотно-временной анализ
x = chirp(0:0.001:1, 0, 1, 100);
pspectrum(x, 1000); % Частотно-временной спектр
saveas(gcf, 'frequency_time_spectrum.jpg'); % Сохраняет текущую фигуру как JPG
"""
load( "$(@__DIR__)/frequency_time_spectrum.jpg" )
Out[0]:
No description has been provided for this image

Исходя из графиков можно заметить, что основные тенденции поведения сигнала отражены в Engee. Для более детального построения графиков необходимо дополнительно реализовать более детальный расчёт окна для функции периодограммы.

Также давайте сравним корректность сгенерированных входных данных в MATLAB и Engee.

In [ ]:
plot(mat"chirp(0:0.001:1, 0, 1, 100)"[:])
plot!(chirp(0:0.001:1, 0.0, 1.0, 100.0))
title!("Sum_error: $(sum(mat"chirp(0:0.001:1, 0, 1, 100)"[:]-chirp(0:0.001:1, 0.0, 1.0, 100.0)))")
Out[0]:

Как мы видим, входные сигналы идентичны.

Вывод

В MATLAB есть встроенные функции для выполнения этих операций. В Engee под эту задачу можно использовать сторонние библиотеки, такие, как DSP или FFTW для более сложных операций с сигналами. Также важно учесть, что Engee даёт больше гибкости в том, как можно реализовать преобразования. Чтобы упростить выполнение этих операций для реализации ваших задач в Engee, используйте готовые подходы, описанные в данной демонстрации.

Ещё один аспект в сравнении возможностей двух сред говорит о том, что MATLAB оптимизирован под работу с матрицами и сигналами. Однако Engee в целом быстрее в вычислениях, если код написан с учетом особенностей этой среды, и это даёт существенный прирост по скорости выполнения и тестирования алгоримах в больших системах.