1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149
use rotary_core::{Buf, BufMut, Channel, ChannelMut, ExactSizeBuf, WriteBuf}; /// Make a mutable buffer into a write adapter that implements /// [WriteBuf]. /// /// # Examples /// /// ```rust /// use rotary::{Buf as _, BufMut as _, ReadBuf as _, WriteBuf as _}; /// use rotary::io::{Read, Write, copy_remaining}; /// /// let from = rotary::interleaved![[1.0f32, 2.0f32, 3.0f32, 4.0f32]; 2]; /// let to = rotary::interleaved![[0.0f32; 4]; 2]; /// let mut to = Write::new(to); /// let mut from = Read::new(from.skip(2)); /// /// assert_eq!(to.remaining_mut(), 4); /// copy_remaining(from, &mut to); /// assert_eq!(to.remaining_mut(), 2); /// /// assert_eq! { /// to.as_ref().as_slice(), /// &[3.0, 3.0, 4.0, 4.0, 0.0, 0.0, 0.0, 0.0], /// }; /// ``` pub struct Write<B> { buf: B, available: usize, } impl<B> Write<B> where B: ExactSizeBuf, { /// Construct a new write adapter. pub fn new(buf: B) -> Self { let available = buf.frames(); Self { buf, available } } /// Access the underlying buffer. /// /// # Examples /// /// ```rust /// use rotary::Buf as _; /// use rotary::io; /// /// let buffer: rotary::Interleaved<i16> = rotary::interleaved![[1, 2, 3, 4]; 4]; /// let mut buffer = io::Write::new(buffer); /// /// io::copy_remaining(rotary::wrap::interleaved(&[0i16; 16][..], 4), &mut buffer); /// /// assert_eq!(buffer.as_ref().channels(), 4); /// ``` pub fn as_ref(&self) -> &B { &self.buf } /// Access the underlying buffer mutably. /// /// # Examples /// /// ```rust /// use rotary::Buf as _; /// use rotary::io; /// /// let buffer: rotary::Interleaved<i16> = rotary::interleaved![[1, 2, 3, 4]; 4]; /// let mut buffer = io::Write::new(buffer); /// /// io::copy_remaining(rotary::wrap::interleaved(&[0i16; 16][..], 4), &mut buffer); /// /// buffer.as_mut().resize_channels(2); /// /// assert_eq!(buffer.channels(), 2); /// ``` pub fn as_mut(&mut self) -> &mut B { &mut self.buf } /// Convert into the underlying buffer. /// /// # Examples /// /// ```rust /// use rotary::Buf as _; /// use rotary::io; /// /// let buffer: rotary::Interleaved<i16> = rotary::interleaved![[1, 2, 3, 4]; 4]; /// let mut buffer = io::Write::new(buffer); /// /// io::copy_remaining(rotary::wrap::interleaved(&[0i16; 16][..], 4), &mut buffer); /// /// let buffer = buffer.into_inner(); /// /// assert_eq!(buffer.channels(), 4); /// ``` pub fn into_inner(self) -> B { self.buf } } impl<B> WriteBuf for Write<B> { /// Remaining number of frames available. #[inline] fn remaining_mut(&self) -> usize { self.available } #[inline] fn advance_mut(&mut self, n: usize) { self.available = self.available.saturating_sub(n); } } impl<B> ExactSizeBuf for Write<B> where B: ExactSizeBuf, { fn frames(&self) -> usize { self.buf.frames() } } impl<B, T> Buf<T> for Write<B> where B: Buf<T>, { fn frames_hint(&self) -> Option<usize> { self.buf.frames_hint() } fn channels(&self) -> usize { self.buf.channels() } fn channel(&self, channel: usize) -> Channel<'_, T> { self.buf.channel(channel) } } impl<B, T> BufMut<T> for Write<B> where B: BufMut<T>, { fn channel_mut(&mut self, channel: usize) -> ChannelMut<'_, T> { self.buf.channel_mut(channel).tail(self.available) } }