2#include <ossia/audio/audio_parameter.hpp>
3#include <ossia/audio/drwav_handle.hpp>
4#include <ossia/dataflow/audio_stretch_mode.hpp>
5#include <ossia/dataflow/graph_node.hpp>
6#include <ossia/dataflow/nodes/media.hpp>
7#include <ossia/dataflow/nodes/sound.hpp>
8#include <ossia/dataflow/port.hpp>
9#include <ossia/detail/pod_vector.hpp>
16class sound_mmap final :
public ossia::sound_node
19 sound_mmap() { m_outlets.push_back(&audio_out); }
21 ~sound_mmap() =
default;
23 std::string label() const noexcept
override {
return "sound_mmap"; }
25 void set_start(std::size_t v) { start = v; }
27 void set_upmix(std::size_t v) { upmix = v; }
29 void set_sound(drwav_handle hdl)
32 m_handle = std::move(hdl);
35 switch(m_handle.translatedFormatTag())
37 case DR_WAVE_FORMAT_PCM: {
38 switch(m_handle.bitsPerSample())
41 m_converter = read_u8;
44 m_converter = read_s16;
47 m_converter = read_s24;
50 m_converter = read_s32;
55 case DR_WAVE_FORMAT_IEEE_FLOAT: {
56 switch(m_handle.bitsPerSample())
59 m_converter = read_f32;
62 m_converter = read_f64;
68 m_converter =
nullptr;
74 void transport(time_value date)
override
77 m_resampler.transport(to_sample(date, m_handle.sampleRate()));
81 int64_t start, int64_t samples_to_write,
double** audio_array_base)
noexcept
83 const int channels = this->channels();
84 const int file_duration = this->duration();
86 m_resampleBuffer.resize(channels);
87 for(
auto& buf : m_resampleBuffer)
88 buf.resize(samples_to_write);
90 float** audio_array = (
float**)alloca(
sizeof(
float*) * channels);
91 for(
int i = 0; i < channels; i++)
93 m_resampleBuffer[i].resize(samples_to_write);
94 audio_array[i] = m_resampleBuffer[i].data();
97 ossia::mutable_audio_span<float> source(channels);
100 if(samples_to_write * channels > 10000)
102 m_safetyBuffer.resize(samples_to_write * channels);
103 frame_data = m_safetyBuffer.data();
109 frame_data = (
double*)alloca(
sizeof(
double) * samples_to_write * channels);
114 for(
int k = 0; k < samples_to_write; k++)
117 int pos = this->m_start_offset_samples
118 + ((start + k) % this->m_loop_duration_samples);
119 if(pos >= file_duration)
121 for(
int i = 0; i < channels; i++)
122 audio_array[i][k] = 0;
126 const bool ok = this->m_handle.seek_to_pcm_frame(pos);
129 for(
int i = 0; i < channels; i++)
130 audio_array[i][k] = 0;
135 const auto count = this->m_handle.read_pcm_frames(
max, frame_data);
138 for(
int i = 0; i < channels; i++)
139 source[i] = std::span(audio_array[i] + k, count);
140 m_converter(source, frame_data, count);
144 for(
int i = 0; i < channels; i++)
145 audio_array[i][k] = 0;
151 for(
int i = 0; i < channels; i++)
153 source[i] = std::span(audio_array[i], samples_to_write);
156 bool ok = start + m_start_offset_samples < file_duration;
158 ok = ok && this->m_handle.seek_to_pcm_frame(start + m_start_offset_samples);
162 const auto count = this->m_handle.read_pcm_frames(samples_to_write, frame_data);
163 m_converter(source, frame_data, count);
164 for(
int i = 0; i < channels; i++)
165 for(
int k = count; k < samples_to_write; k++)
166 audio_array[i][k] = 0;
170 for(
int i = 0; i < channels; i++)
171 for(
int k = 0; k < samples_to_write; k++)
172 audio_array[i][k] = 0;
177 for(
int i = 0; i < channels; i++)
178 std::copy_n(audio_array[i], samples_to_write, audio_array_base[i]);
181 void fetch_audio(int64_t start, int64_t samples_to_write,
float** audio_array)
noexcept
183 const int channels = this->channels();
184 const int file_duration = this->duration();
186 ossia::mutable_audio_span<float> source(channels);
188 double* frame_data{};
189 if(samples_to_write * channels > 10000)
191 m_safetyBuffer.resize(samples_to_write * channels);
192 frame_data = m_safetyBuffer.data();
198 frame_data = (
double*)alloca(
sizeof(
double) * samples_to_write * channels);
203 for(
int k = 0; k < samples_to_write; k++)
206 int pos = this->m_start_offset_samples
207 + ((start + k) % this->m_loop_duration_samples);
208 if(pos >= file_duration)
210 for(
int i = 0; i < channels; i++)
211 audio_array[i][k] = 0;
215 const bool ok = this->m_handle.seek_to_pcm_frame(pos);
218 for(
int i = 0; i < channels; i++)
219 audio_array[i][k] = 0;
224 const auto count = this->m_handle.read_pcm_frames(
max, frame_data);
227 for(
int i = 0; i < channels; i++)
228 source[i] = std::span(audio_array[i] + k, count);
229 m_converter(source, frame_data, count);
233 for(
int i = 0; i < channels; i++)
234 audio_array[i][k] = 0;
240 for(
int i = 0; i < channels; i++)
242 source[i] = std::span(audio_array[i], samples_to_write);
245 const bool ok = this->m_handle.seek_to_pcm_frame(start + m_start_offset_samples);
248 for(
int i = 0; i < channels; i++)
249 for(
int k = 0; k < samples_to_write; k++)
250 audio_array[i][k] = 0;
254 const auto count = this->m_handle.read_pcm_frames(samples_to_write, frame_data);
255 m_converter(source, frame_data, count);
256 for(
int i = 0; i < channels; i++)
257 for(
int k = count; k < samples_to_write; k++)
258 audio_array[i][k] = 0;
262 void run(
const ossia::token_request& t, ossia::exec_state_facade e)
noexcept override
264 if(!m_handle || !m_converter)
271 const auto channels = m_handle.channels();
272 const auto len = m_handle.totalPCMFrameCount();
274 ossia::audio_port& ap = *audio_out;
275 ap.set_channels(std::max((std::size_t)upmix, (std::size_t)channels));
277 const auto [samples_to_read, samples_to_write]
278 = snd::sample_info(e.bufferSize(), e.modelToSamples(), t);
279 if(samples_to_write <= 0)
282 assert(samples_to_write > 0);
284 const auto samples_offset = t.physical_start(e.modelToSamples());
287 if(t.prev_date < m_prev_date)
292 if(t.prev_date != 0_tv)
294 transport(t.prev_date);
304 transport(t.prev_date);
308 for(std::size_t chan = 0; chan < channels; chan++)
310 ap.channel(chan).resize(e.bufferSize());
313 double stretch_ratio = update_stretch(t, e);
317 *
this, t, e, stretch_ratio, channels, len, samples_to_read, samples_to_write,
320 for(std::size_t chan = 0; chan < channels; chan++)
324 t.start_discontinuous, t.end_discontinuous, ap.channel(chan), samples_offset,
328 ossia::snd::perform_upmix(this->upmix, channels, ap);
329 ossia::snd::perform_start_offset(this->start, ap);
331 m_prev_date = t.date;
339 [[nodiscard]] std::size_t channels()
const
341 return m_handle ? m_handle.channels() : 0;
343 [[nodiscard]] std::size_t duration()
const
345 return m_handle ? m_handle.totalPCMFrameCount() : 0;
349 drwav_handle m_handle{};
351 ossia::audio_outlet audio_out;
357 = void (*)(ossia::mutable_audio_span<float>& ap,
void* data, int64_t samples);
358 read_fn_t m_converter{};
359 std::vector<double> m_safetyBuffer;
360 std::vector<std::vector<float>> m_resampleBuffer;
OSSIA_INLINE constexpr auto max(const T a, const U b) noexcept -> typename std::conditional<(sizeof(T) > sizeof(U)), T, U >::type
max function tailored for values
Definition math.hpp:96
The time_value class.
Definition ossia/editor/scenario/time_value.hpp:30