Function ITEST

core/src/itest/itest_windowfunction.cpp:23–85

#include "itest.h"

ITEST(windowfunction)
{
    const auto frameSize = 1024;
    const auto overlapSize = frameSize / 4;
    const auto tukeySize = frameSize + overlapSize;

    float rectangular[frameSize];
    float bartlett[frameSize];
    float hann[frameSize];
    float hamming[frameSize];
    float blackman[frameSize];
    float blackmanHarris[frameSize];
    float tukey[tukeySize];

    WindowFunction::rectangular(frameSize, rectangular);
    WindowFunction::bartlett(frameSize, bartlett);
    WindowFunction::hann(frameSize, hann);
    WindowFunction::hamming(frameSize, hamming);
    WindowFunction::blackman(frameSize, blackman);
    WindowFunction::blackmanHarris(frameSize, blackmanHarris);
    WindowFunction::tukey(frameSize, overlapSize, tukey);

    auto gui = [&]()
    {
        if (ImGui::Begin("Window Functions"))
        {
            const char* types[] = {
                "Rectangular",
                "Bartlett",
                "Hann",
                "Hamming",
                "Blackman",
                "Blackman-Harris",
                "Tukey"
            };

            const float* data[] = {
                rectangular,
                bartlett,
                hann,
                hamming,
                blackman,
                blackmanHarris,
                tukey
            };

            static auto currentType = -1;
            ImGui::Combo("Type", &currentType, types, 7);

            if (0 <= currentType && currentType < 7)
            {
                auto size = (currentType == 6) ? tukeySize : frameSize;
                ImGui::PlotLines(types[currentType], data[currentType], size, 0, nullptr, 0.0f, 1.0f, ImVec2(512, 512));
            }
        }

        ImGui::End();
    };