def test_ndarray_s16p_align_8(self): frame = AudioFrame(format="s16p", layout="stereo", samples=159, align=8) array = frame.to_ndarray() self.assertEqual(array.dtype, "i2") self.assertEqual(array.shape, (2, 159))
def test_ndarray_s16p_align_8(self): frame = AudioFrame(format='s16p', layout='stereo', samples=159, align=8) array = frame.to_ndarray() self.assertEqual(array.dtype, '<i2') self.assertEqual(array.shape, (2, 159))
def process_audio(frame: av.AudioFrame) -> av.AudioFrame: raw_samples = frame.to_ndarray() sound = pydub.AudioSegment( data=raw_samples.tobytes(), sample_width=frame.format.bytes, frame_rate=frame.sample_rate, channels=len(frame.layout.channels), ) sound = sound.apply_gain(gain) # Ref: https://github.com/jiaaro/pydub/blob/master/API.markdown#audiosegmentget_array_of_samples # noqa channel_sounds = sound.split_to_mono() channel_samples = [s.get_array_of_samples() for s in channel_sounds] new_samples: np.ndarray = np.array(channel_samples).T new_samples = new_samples.reshape(raw_samples.shape) new_frame = av.AudioFrame.from_ndarray(new_samples, layout=frame.layout.name) new_frame.sample_rate = frame.sample_rate return new_frame
def test_basic_to_ndarray(self): frame = AudioFrame(format="s16p", layout="stereo", samples=160) array = frame.to_ndarray() self.assertEqual(array.dtype, "i2") self.assertEqual(array.shape, (2, 160))
def test_basic_to_ndarray(self): frame = AudioFrame(format='s16p', layout='stereo', samples=160) array = frame.to_ndarray() self.assertEqual(array.dtype, '<i2') self.assertEqual(array.shape, (2, 160))
def test_basic_to_ndarray(self): frame = AudioFrame(format='s16p', layout='stereo', samples=160) array = frame.to_ndarray() self.assertEqual(array.dtype, '<i2') self.assertEqual(array.shape, (2, 160))
def test_ndarray_s16p_align_8(self): frame = AudioFrame(format='s16p', layout='stereo', samples=159, align=8) array = frame.to_ndarray() self.assertEqual(array.dtype, '<i2') self.assertEqual(array.shape, (2, 159))