Fix lifetimes on buffer adapter
This commit is contained in:
parent
73fe012751
commit
a734dd09f3
|
@ -28,7 +28,7 @@ impl<'a> Buffer<'a> {
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Iterate over the samples, returning a channel iterator for each sample.
|
/// Iterate over the samples, returning a channel iterator for each sample.
|
||||||
pub fn iter_mut(&mut self) -> Samples<'a, 'a> {
|
pub fn iter_mut<'slice>(&'slice mut self) -> Samples<'slice, 'a> {
|
||||||
Samples {
|
Samples {
|
||||||
buffers: self.output_slices.as_mut_slice(),
|
buffers: self.output_slices.as_mut_slice(),
|
||||||
current_sample: 0,
|
current_sample: 0,
|
||||||
|
@ -52,15 +52,15 @@ impl<'a> Buffer<'a> {
|
||||||
|
|
||||||
/// An iterator over all samples in the buffer, yielding iterators over each channel for every
|
/// 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.
|
/// sample. This iteration order offers good cache locality for per-sample access.
|
||||||
pub struct Samples<'buffer: 'sample, 'sample> {
|
pub struct Samples<'slice, 'sample: 'slice> {
|
||||||
/// The raw output buffers.
|
/// The raw output buffers.
|
||||||
pub(self) buffers: *mut [&'sample mut [f32]],
|
pub(self) buffers: *mut [&'sample mut [f32]],
|
||||||
pub(self) current_sample: usize,
|
pub(self) current_sample: usize,
|
||||||
pub(self) _marker: PhantomData<&'buffer mut [&'sample mut [f32]]>,
|
pub(self) _marker: PhantomData<&'slice mut [&'sample mut [f32]]>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'buffer, 'sample> Iterator for Samples<'buffer, 'sample> {
|
impl<'slice, 'sample> Iterator for Samples<'slice, 'sample> {
|
||||||
type Item = Channels<'buffer, 'sample>;
|
type Item = Channels<'slice, 'sample>;
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
if self.current_sample < unsafe { (*self.buffers)[0].len() } {
|
if self.current_sample < unsafe { (*self.buffers)[0].len() } {
|
||||||
|
@ -88,25 +88,25 @@ impl ExactSizeIterator for Samples<'_, '_> {}
|
||||||
|
|
||||||
/// Can construct iterators over actual iterator over the channel data for a sample, yielded by
|
/// Can construct iterators over actual iterator over the channel data for a sample, yielded by
|
||||||
/// [Samples].
|
/// [Samples].
|
||||||
pub struct Channels<'buffer: 'sample, 'sample> {
|
pub struct Channels<'slice, 'sample: 'slice> {
|
||||||
/// The raw output buffers.
|
/// The raw output buffers.
|
||||||
pub(self) buffers: *mut [&'sample mut [f32]],
|
pub(self) buffers: *mut [&'sample mut [f32]],
|
||||||
pub(self) current_sample: usize,
|
pub(self) current_sample: usize,
|
||||||
pub(self) _marker: PhantomData<&'buffer mut [&'sample mut [f32]]>,
|
pub(self) _marker: PhantomData<&'slice mut [&'sample mut [f32]]>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The actual iterator over the channel data for a sample, yielded by [Channels].
|
/// The actual iterator over the channel data for a sample, yielded by [Channels].
|
||||||
pub struct ChannelsIter<'buffer: 'sample, 'sample> {
|
pub struct ChannelsIter<'slice, 'sample: 'slice> {
|
||||||
/// The raw output buffers.
|
/// The raw output buffers.
|
||||||
pub(self) buffers: *mut [&'sample mut [f32]],
|
pub(self) buffers: *mut [&'sample mut [f32]],
|
||||||
pub(self) current_sample: usize,
|
pub(self) current_sample: usize,
|
||||||
pub(self) current_channel: usize,
|
pub(self) current_channel: usize,
|
||||||
pub(self) _marker: PhantomData<&'buffer mut [&'sample mut [f32]]>,
|
pub(self) _marker: PhantomData<&'slice mut [&'sample mut [f32]]>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'buffer, 'sample> IntoIterator for Channels<'buffer, 'sample> {
|
impl<'slice, 'sample> IntoIterator for Channels<'slice, 'sample> {
|
||||||
type Item = &'sample mut f32;
|
type Item = &'sample mut f32;
|
||||||
type IntoIter = ChannelsIter<'buffer, 'sample>;
|
type IntoIter = ChannelsIter<'slice, 'sample>;
|
||||||
|
|
||||||
fn into_iter(self) -> Self::IntoIter {
|
fn into_iter(self) -> Self::IntoIter {
|
||||||
ChannelsIter {
|
ChannelsIter {
|
||||||
|
@ -118,20 +118,19 @@ impl<'buffer, 'sample> IntoIterator for Channels<'buffer, 'sample> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<'buffer, 'sample> Iterator for ChannelsIter<'buffer, 'sample> {
|
impl<'slice, 'sample> Iterator for ChannelsIter<'slice, 'sample> {
|
||||||
type Item = &'sample mut f32;
|
type Item = &'sample mut f32;
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
if self.current_channel < unsafe { (*self.buffers).len() } {
|
if self.current_channel < unsafe { (*self.buffers).len() } {
|
||||||
// SAFETY: These bounds have already been checked
|
// SAFETY: These bounds have already been checked
|
||||||
|
// SAFETY: It is also not possible to have multiple mutable references to the same
|
||||||
|
// sample at the same time
|
||||||
let sample = unsafe {
|
let sample = unsafe {
|
||||||
(*self.buffers)
|
(*self.buffers)
|
||||||
.get_unchecked_mut(self.current_channel)
|
.get_unchecked_mut(self.current_channel)
|
||||||
.get_unchecked_mut(self.current_sample)
|
.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: &'sample mut f32 = unsafe { &mut *(sample as *mut f32) };
|
|
||||||
|
|
||||||
self.current_channel += 1;
|
self.current_channel += 1;
|
||||||
|
|
||||||
|
@ -149,7 +148,7 @@ impl<'buffer, 'sample> Iterator for ChannelsIter<'buffer, 'sample> {
|
||||||
|
|
||||||
impl ExactSizeIterator for ChannelsIter<'_, '_> {}
|
impl ExactSizeIterator for ChannelsIter<'_, '_> {}
|
||||||
|
|
||||||
impl<'buffer, 'sample> Channels<'buffer, 'sample> {
|
impl<'slice, 'sample> Channels<'slice, 'sample> {
|
||||||
/// Get the number of channels.
|
/// Get the number of channels.
|
||||||
pub fn len(&self) -> usize {
|
pub fn len(&self) -> usize {
|
||||||
unsafe { (*self.buffers).len() }
|
unsafe { (*self.buffers).len() }
|
||||||
|
@ -157,12 +156,9 @@ impl<'buffer, 'sample> Channels<'buffer, 'sample> {
|
||||||
|
|
||||||
/// A resetting iterator. This lets you iterate over the same channels multiple times. Otherwise
|
/// A resetting iterator. This lets you iterate over the same channels multiple times. Otherwise
|
||||||
/// you don't need to use this function as [Channels] already implements [Iterator].
|
/// you don't need to use this function as [Channels] already implements [Iterator].
|
||||||
pub fn iter_mut(&mut self) -> ChannelsIter<'buffer, 'sample> {
|
pub fn iter_mut(&mut self) -> ChannelsIter<'slice, 'sample> {
|
||||||
// SAFETY: No two [ChannelIters] can exist at a time
|
|
||||||
let buffers: *mut [&'sample mut [f32]] = unsafe { std::mem::transmute(self.buffers) };
|
|
||||||
|
|
||||||
ChannelsIter {
|
ChannelsIter {
|
||||||
buffers,
|
buffers: self.buffers,
|
||||||
current_sample: self.current_sample,
|
current_sample: self.current_sample,
|
||||||
current_channel: 0,
|
current_channel: 0,
|
||||||
_marker: PhantomData,
|
_marker: PhantomData,
|
||||||
|
|
Loading…
Reference in a new issue