1
0
Fork 0

Move the buffer adapter to its own module

This commit is contained in:
Robbert van der Helm 2022-02-02 15:01:41 +01:00
parent 69db1acce8
commit f44597df7c
6 changed files with 141 additions and 121 deletions

View file

@ -18,10 +18,11 @@
extern crate nih_plug; extern crate nih_plug;
use nih_plug::{ use nih_plug::{
buffer::Buffer,
context::ProcessContext, context::ProcessContext,
formatters, formatters,
param::{BoolParam, FloatParam, Param, Params, Range}, param::{BoolParam, FloatParam, Param, Params, Range},
plugin::{Buffer, BufferConfig, BusConfig, Plugin, ProcessStatus, Vst3Plugin}, plugin::{BufferConfig, BusConfig, Plugin, ProcessStatus, Vst3Plugin},
util, util,
}; };
use parking_lot::RwLock; use parking_lot::RwLock;

View file

@ -18,10 +18,11 @@
extern crate nih_plug; extern crate nih_plug;
use nih_plug::{ use nih_plug::{
buffer::Buffer,
context::ProcessContext, context::ProcessContext,
formatters, formatters,
param::{FloatParam, Param, Params, Range}, param::{FloatParam, Param, Params, Range},
plugin::{Buffer, BufferConfig, BusConfig, Plugin, ProcessStatus, Vst3Plugin}, plugin::{BufferConfig, BusConfig, Plugin, ProcessStatus, Vst3Plugin},
util, util,
}; };
use std::f32::consts; use std::f32::consts;

133
src/buffer.rs Normal file
View file

@ -0,0 +1,133 @@
// nih-plug: plugins, but rewritten in Rust
// Copyright (C) 2022 Robbert van der Helm
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
/// The audio buffers used during processing. This contains the output audio output buffers with the
/// inputs already copied to the outputs. You can either use the iterator adapters to conveniently
/// and efficiently iterate over the samples, or you can do your own thing using the raw audio
/// buffers.
pub struct Buffer<'a> {
/// Contains slices for the plugin's outputs. You can't directly create a nested slice form
/// apointer to pointers, so this needs to be preallocated in the setup call and kept around
/// between process calls. And because storing a reference to this means a) that you need a lot
/// of lifetime annotations everywhere and b) that at some point you need unsound lifetime casts
/// because this `Buffers` either cannot have the same lifetime as the separately stored output
/// buffers, and it also cannot be stored in a field next to it because that would mean
/// containing mutable references to data stored in a mutex.
output_slices: Vec<&'a mut [f32]>,
}
impl<'a> Buffer<'a> {
/// Construct a new buffer adapter based on a set of audio buffers.
pub fn new() -> Self {
Self {
output_slices: Vec::new(),
}
}
/// Returns true if this buffer does not contain any samples.
pub fn is_empty(&self) -> bool {
self.output_slices.is_empty() || self.output_slices[0].is_empty()
}
/// Obtain the raw audio buffers.
pub fn as_raw(&mut self) -> &mut [&'a mut [f32]] {
&mut self.output_slices
}
/// Iterate over the samples, returning a channel iterator for each sample.
pub fn iter_mut(&mut self) -> Samples<'_, 'a> {
Samples {
buffers: &mut self.output_slices,
current_sample: 0,
}
}
/// Access the raw output slice vector. This neds to be resized to match the number of output
/// channels during the plugin's initialization. Then during audio processing, these slices
/// should be updated to point to the plugin's audio buffers.
///
/// # Safety
///
/// The stored slices must point to live data when this object is passed to the plugins' process
/// function. The rest of this object also assumes all channel lengths are equal. Panics will
/// likely occur if this is not the case.
pub unsafe fn as_raw_vec(&mut self) -> &mut Vec<&'a mut [f32]> {
&mut self.output_slices
}
}
/// An iterator over all samples in the buffer, yielding iterators over each channel for every
/// sample. This iteration order offers good cache locality for per-sample access.
pub struct Samples<'outer, 'inner> {
/// The raw output buffers.
pub(self) buffers: &'outer mut [&'inner mut [f32]],
pub(self) current_sample: usize,
}
impl<'outer, 'inner> Iterator for Samples<'outer, 'inner> {
type Item = Channels<'outer, 'inner>;
fn next(&mut self) -> Option<Self::Item> {
if self.current_sample < self.buffers[0].len() {
// SAFETY: We guarantee that each sample is only mutably borrowed once in the channels
// iterator
let buffers: &'outer mut _ = unsafe { &mut *(self.buffers as *mut _) };
let channels = Channels {
buffers,
current_sample: self.current_sample,
current_channel: 0,
};
self.current_sample += 1;
Some(channels)
} else {
None
}
}
}
/// An iterator over the channel data for a sample, yielded by [Samples].
pub struct Channels<'outer, 'inner> {
/// The raw output buffers.
pub(self) buffers: &'outer mut [&'inner mut [f32]],
pub(self) current_sample: usize,
pub(self) current_channel: usize,
}
impl<'outer, 'inner> Iterator for Channels<'outer, 'inner> {
type Item = &'inner mut f32;
fn next(&mut self) -> Option<Self::Item> {
if self.current_channel < self.buffers.len() {
// SAFETY: These bounds have already been checked
let sample = unsafe {
self.buffers
.get_unchecked_mut(self.current_channel)
.get_unchecked_mut(self.current_sample)
};
// SAFETY: It is not possible to have multiple mutable references to the same sample at
// the same time
let sample: &'inner mut f32 = unsafe { &mut *(sample as *mut f32) };
self.current_channel += 1;
Some(sample)
} else {
None
}
}
}

View file

@ -16,6 +16,7 @@
// TODO: Once everything is more fleshed out, document the basic usage of this library // TODO: Once everything is more fleshed out, document the basic usage of this library
pub mod buffer;
pub mod context; pub mod context;
#[macro_use] #[macro_use]
pub mod debug; pub mod debug;

View file

@ -16,6 +16,7 @@
use std::pin::Pin; use std::pin::Pin;
use crate::buffer::Buffer;
use crate::context::ProcessContext; use crate::context::ProcessContext;
use crate::param::Params; use crate::param::Params;
@ -140,121 +141,3 @@ pub enum ProcessStatus {
/// infite tail. /// infite tail.
KeepAlive, KeepAlive,
} }
/// The audio buffers used during processing. This contains the output audio output buffers with the
/// inputs already copied to the outputs. You can either use the iterator adapters to conveniently
/// and efficiently iterate over the samples, or you can do your own thing using the raw audio
/// buffers.
pub struct Buffer<'a> {
/// Contains slices for the plugin's outputs. You can't directly create a nested slice form
/// apointer to pointers, so this needs to be preallocated in the setup call and kept around
/// between process calls. And because storing a reference to this means a) that you need a lot
/// of lifetime annotations everywhere and b) that at some point you need unsound lifetime casts
/// because this `Buffers` either cannot have the same lifetime as the separately stored output
/// buffers, and it also cannot be stored in a field next to it because that would mean
/// containing mutable references to data stored in a mutex.
output_slices: Vec<&'a mut [f32]>,
}
impl<'a> Buffer<'a> {
/// Construct a new buffer adapter based on a set of audio buffers.
pub fn new() -> Self {
Self {
output_slices: Vec::new(),
}
}
/// Returns true if this buffer does not contain any samples.
pub fn is_empty(&self) -> bool {
self.output_slices.is_empty() || self.output_slices[0].is_empty()
}
/// Obtain the raw audio buffers.
pub fn as_raw(&mut self) -> &mut [&'a mut [f32]] {
&mut self.output_slices
}
/// Iterate over the samples, returning a channel iterator for each sample.
pub fn iter_mut(&mut self) -> Samples<'_, 'a> {
Samples {
buffers: &mut self.output_slices,
current_sample: 0,
}
}
/// Access the raw output slice vector. This neds to be resized to match the number of output
/// channels during the plugin's initialization. Then during audio processing, these slices
/// should be updated to point to the plugin's audio buffers.
///
/// # Safety
///
/// The stored slices must point to live data when this object is passed to the plugins' process
/// function. The rest of this object also assumes all channel lengths are equal. Panics will
/// likely occur if this is not the case.
pub unsafe fn as_raw_vec(&mut self) -> &mut Vec<&'a mut [f32]> {
&mut self.output_slices
}
}
/// An iterator over all samples in the buffer, yielding iterators over each channel for every
/// sample. This iteration order offers good cache locality for per-sample access.
pub struct Samples<'outer, 'inner> {
/// The raw output buffers.
pub(self) buffers: &'outer mut [&'inner mut [f32]],
pub(self) current_sample: usize,
}
impl<'outer, 'inner> Iterator for Samples<'outer, 'inner> {
type Item = Channels<'outer, 'inner>;
fn next(&mut self) -> Option<Self::Item> {
if self.current_sample < self.buffers[0].len() {
// SAFETY: We guarantee that each sample is only mutably borrowed once in the channels
// iterator
let buffers: &'outer mut _ = unsafe { &mut *(self.buffers as *mut _) };
let channels = Channels {
buffers,
current_sample: self.current_sample,
current_channel: 0,
};
self.current_sample += 1;
Some(channels)
} else {
None
}
}
}
/// An iterator over the channel data for a sample, yielded by [Samples].
pub struct Channels<'outer, 'inner> {
/// The raw output buffers.
pub(self) buffers: &'outer mut [&'inner mut [f32]],
pub(self) current_sample: usize,
pub(self) current_channel: usize,
}
impl<'outer, 'inner> Iterator for Channels<'outer, 'inner> {
type Item = &'inner mut f32;
fn next(&mut self) -> Option<Self::Item> {
if self.current_channel < self.buffers.len() {
// SAFETY: These bounds have already been checked
let sample = unsafe {
self.buffers
.get_unchecked_mut(self.current_channel)
.get_unchecked_mut(self.current_sample)
};
// SAFETY: It is not possible to have multiple mutable references to the same sample at
// the same time
let sample: &'inner mut f32 = unsafe { &mut *(sample as *mut f32) };
self.current_channel += 1;
Some(sample)
} else {
None
}
}
}

View file

@ -40,10 +40,11 @@ use vst3_sys::vst::{
use vst3_sys::VST3; use vst3_sys::VST3;
use widestring::U16CStr; use widestring::U16CStr;
use crate::buffer::Buffer;
use crate::context::{EventLoop, MainThreadExecutor, OsEventLoop, ProcessContext}; use crate::context::{EventLoop, MainThreadExecutor, OsEventLoop, ProcessContext};
use crate::param::internals::ParamPtr; use crate::param::internals::ParamPtr;
use crate::param::Param; use crate::param::Param;
use crate::plugin::{Buffer, BufferConfig, BusConfig, Plugin, ProcessStatus, Vst3Plugin}; use crate::plugin::{BufferConfig, BusConfig, Plugin, ProcessStatus, Vst3Plugin};
use crate::wrapper::state::{ParamValue, State}; use crate::wrapper::state::{ParamValue, State};
use crate::wrapper::util::{hash_param_id, strlcpy, u16strlcpy}; use crate::wrapper::util::{hash_param_id, strlcpy, u16strlcpy};