KwidgetGUI_OfflineAnalyser.cpp

Go to the documentation of this file.

namespace krotos
{
    KwidgetGUI_OfflineAnalyser::KwidgetGUI_OfflineAnalyser(Kwidget& owner)
        : KwidgetGUI(owner), m_stft(2048, 2048, 0.5f, WindowType::Hann, ShortTimeFourierTransform::fftMode::freqMagOnly,
                                    44100 /*TODO: Find way to pass sampleRate here!!*/)
    {
        setOpaque(true);
        // hide the connection Button
        // getTopBarComponent().m_selectButton.setVisible(false);
 
        addAndMakeVisible(m_openButton);
        m_openButton.setButtonText("Open");
 
        addAndMakeVisible(m_clearButton);
        m_clearButton.setButtonText("Clear");
 
        addAndMakeVisible(m_erbButton);
        m_erbButton.setButtonText("ERB Spectro");
 
        addAndMakeVisible(m_stftButton);
        m_stftButton.setButtonText("STFT Spectro");
 
        m_openButton.onClick = [this] { openButtonClicked(); };
        m_clearButton.onClick = [this] { clearButtonClicked(); };
        m_erbButton.onClick = [this] { erbButtonClicked(); };
        m_stftButton.onClick = [this] { stftButtonClicked(); };
    }
 
    void KwidgetGUI_OfflineAnalyser::openButtonClicked()
    {
        // Set up the File browser window
        m_chooser = std::make_unique<juce::FileChooser>("Select a Wav file to load..", juce::File{}, "*.wav");
 
        auto chooserFlags = juce::FileBrowserComponent::openMode | juce::FileBrowserComponent::canSelectFiles;
 
        // Launch the File browser Async
        m_chooser->launchAsync(chooserFlags, [this](const juce::FileChooser& fc) {
            auto file = fc.getResult();
 
            if (file == juce::File{})
                return;
 
            AudioFormatManager formatManager;
            formatManager.registerBasicFormats();
 
            std::unique_ptr<juce::AudioFormatReader> reader(formatManager.createReaderFor(file));
 
            if (reader.get() != nullptr)
            {
                m_fileBuffer.setSize((int)reader->numChannels, (int)reader->lengthInSamples);
                reader->read(&m_fileBuffer, 0, (int)reader->lengthInSamples, 0, true, true);
 
                m_sampleRate = static_cast<float>(reader->sampleRate);
            }
        });
    }
 
    void KwidgetGUI_OfflineAnalyser::clearButtonClicked()
    {
        // Clear the image and the Buffer
        m_fileBuffer.clear();
 
        m_spectrogramImage.clear(getLocalBounds(), Colour::fromRGB(0, 0, 0));
 
        repaint();
    }
 
    void KwidgetGUI_OfflineAnalyser::erbButtonClicked()
    {
        // m_spectrogramImage.clear(getLocalBounds(), Colour::fromRGB(0, 0, 0));
 
        m_erbSTFT.setSampleRate(m_sampleRate);
        // avergae buffer to mono
        averageToMono(m_fileBuffer);
        // extract data to vector
        auto signalData = extractData(m_fileBuffer);
 
        auto erbSpec = m_erbSTFT.filterSpectrum(signalData);
        m_erbSTFT.drawSpectrogram(m_spectrogramImage, erbSpec);
 
        repaint();
    }
 
    void KwidgetGUI_OfflineAnalyser::stftButtonClicked()
    {
        auto stftMatrix = m_stft.stft(m_fileBuffer);
        m_stft.drawSpectrogram(m_spectrogramImage, stftMatrix);
 
        repaint();
    }
 
    void KwidgetGUI_OfflineAnalyser::paint(Graphics& g)
    {
        KwidgetGUI::paint(g);
 
        g.drawImage(m_spectrogramImage, m_imageBounds.toFloat());
    }
 
    void KwidgetGUI_OfflineAnalyser::resized()
    {
        KwidgetGUI::resized();
 
        auto bounds = m_currentBounds;
 
        auto topBar = bounds.removeFromTop(40);
 
        Rectangle<int> openButtonBounds(m_currentBounds.getX() + k_padding, topBar.getY(), 60, 40);
        Rectangle<int> clearButtonBounds(m_currentBounds.getX() + k_padding + 60, topBar.getY(), 60, 40);
        Rectangle<int> stftButtonBounds(m_currentBounds.getX() + k_padding + 120, topBar.getY(), 60, 40);
        Rectangle<int> erbButtonBounds(m_currentBounds.getX() + k_padding + 180, topBar.getY(), 60, 40);
 
        m_openButton.setBounds(openButtonBounds);
        m_clearButton.setBounds(clearButtonBounds);
        m_erbButton.setBounds(erbButtonBounds);
        m_stftButton.setBounds(stftButtonBounds);
 
        m_imageBounds = bounds.reduced(30, 30);
 
        m_spectrogramImage = Image(Image::RGB, m_imageBounds.getWidth(), m_imageBounds.getHeight(), true);
    }
} // namespace krotos