vctr_juce_helpers.h

/*
  ==============================================================================
    DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 
    Copyright 2023 by sonible GmbH.
 
    This file is part of VCTR - Versatile Container Templates Reconceptualized.
 
    VCTR is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License version 3
    only, as published by the Free Software Foundation.
 
    VCTR is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License version 3 for more details.
 
    You should have received a copy of the GNU Lesser General Public License
    version 3 along with VCTR.  If not, see <https://www.gnu.org/licenses/>.
  ==============================================================================
*/
 
#pragma once
 
#include "../vctr/vctr.h"
#include <juce_dsp/juce_dsp.h>
 
namespace vctr::juce_helpers
{
 
template <is::anyVctr Src>
auto toSIMDRegister (const Src& src, size_t i = 0)
{
    using SIMDRegister = juce::dsp::SIMDRegister<typename Src::value_type>;
 
    VCTR_ASSERT (src.size() == SIMDRegister::SIMDNumElements);
 
    if constexpr (Config::archARM)
    {
        return SIMDRegister { src.getNeon (i).value };
    }
    else
    {
        constexpr size_t registerSize = SIMDRegister::SIMDRegisterSize;
 
        if constexpr (registerSize == 32)
        {
            return SIMDRegister { src.getAVX (i).value };
        }
 
        return SIMDRegister { src.getSSE (i).value };
    }
}
 
template <class SampleType>
auto blockChannelAsSpan (const juce::dsp::AudioBlock<SampleType>& block, size_t channelIdx)
{
    auto numSamples = block.getNumSamples();
    return Span { block.getChannelPointer (channelIdx), numSamples };
}
 
template <class SampleType>
auto blockChannelAsSpanSIMDAligned (const juce::dsp::AudioBlock<SampleType>& block, size_t channelIdx)
{
    auto numSamples = block.getNumSamples();
    return Span { block.getChannelPointer (channelIdx), numSamples, simdAlignedSpanStorageInfo<SampleType>() };
}
 
template <class SampleType>
auto audioBufferChannelAsSpan (juce::AudioBuffer<SampleType>& buffer, size_t channelIdx)
{
    auto numSamples = size_t (buffer.getNumSamples());
    return Span { buffer.getWritePointer (int (channelIdx)), numSamples };
}
 
template <class SampleType>
auto audioBufferChannelAsSpan (const juce::AudioBuffer<SampleType>& buffer, size_t channelIdx)
{
    auto numSamples = size_t (buffer.getNumSamples());
    return Span { buffer.getReadPointer (int (channelIdx)), numSamples };
}
 
}
 
namespace  vctr::expressions
{
 
template <std::floating_point SampleType, is::reductionExpressionChainBuilderForSource<Span<const SampleType>> ChannelReduction>
class AudioBlockChannelReduction : ExpressionTemplateBase
{
public:
    using value_type = ValueType<ExpressionTypeForSourceType<ChannelReduction, Span<const SampleType>>>;
 
    template <class Src>
    constexpr AudioBlockChannelReduction (const Src& s, ChannelReduction& re)
        : src (s),
          channelReduction (re)
    {}
 
    constexpr const auto& getStorageInfo() const
    {
        return storageInfo;
    }
 
    constexpr size_t size() const
    {
        return src.getNumChannels();
    }
 
    constexpr bool isNotAliased (const void*) const
    {
        return false;
    }
 
    template <size_t i, class RuntimeArgs>
    constexpr void iterateOverRuntimeArgChain (const RuntimeArgs& rtArgs)
    {
        tryApplyingRuntimeArgsToSrc<i + 1> (rtArgs, src);
    }
 
    VCTR_FORCEDINLINE constexpr auto operator[] (size_t i) const
    {
        return channelReduction << juce_helpers::blockChannelAsSpan (src, i);
    }
 
private:
    juce::dsp::AudioBlock<const SampleType> src;
    ChannelReduction& channelReduction;
 
    static constexpr StaticStorageInfo<false, false, alignof (SampleType)> storageInfo;
};
}
 
namespace vctr::juce_helpers
{
 
template <std::floating_point SampleType, is::reductionExpressionChainBuilderForSource<Span<const SampleType>> ChannelReduction>
auto transformChannelsByExpression (const juce::dsp::AudioBlock<SampleType>& block, ChannelReduction& channelReduction)
{
    return expressions::AudioBlockChannelReduction<SampleType, ChannelReduction> (block, channelReduction);
}
}