Loading...
Searching...
No Matches
GpuUtils.hpp
1#pragma once
2
3#include <avnd/introspection/gfx.hpp>
4#if SCORE_PLUGIN_GFX
5#include <Process/ExecutionContext.hpp>
6
7#include <Crousti/File.hpp>
8#include <Crousti/GppCoroutines.hpp>
9#include <Crousti/GppShaders.hpp>
10#include <Crousti/MessageBus.hpp>
11#include <Crousti/TextureConversion.hpp>
12#include <Crousti/TextureFormat.hpp>
13#include <Gfx/GfxExecNode.hpp>
14#include <Gfx/Graph/Node.hpp>
15#include <Gfx/Graph/OutputNode.hpp>
16#include <Gfx/Graph/RenderList.hpp>
17#include <Gfx/Graph/RenderState.hpp>
18
19#include <score/tools/ThreadPool.hpp>
20
21#include <ossia/detail/small_flat_map.hpp>
22
23#include <ossia-qt/invoke.hpp>
24
25#include <QCoreApplication>
26#include <QTimer>
27#include <QtGui/private/qrhi_p.h>
28
29#include <avnd/binding/ossia/metadatas.hpp>
30#include <avnd/binding/ossia/port_run_postprocess.hpp>
31#include <avnd/binding/ossia/port_run_preprocess.hpp>
32#include <avnd/binding/ossia/soundfiles.hpp>
33#include <avnd/concepts/parameter.hpp>
34#include <avnd/introspection/input.hpp>
35#include <avnd/introspection/output.hpp>
36#include <fmt/format.h>
37#include <gpp/layout.hpp>
38
39#include <score_plugin_avnd_export.h>
40
41namespace oscr
42{
43struct GpuWorker
44{
45 template <typename T>
46 void initWorker(this auto& self, std::shared_ptr<T>& state) noexcept
47 {
48 if constexpr(avnd::has_worker<T>)
49 {
50 auto ptr = QPointer{&self};
51 auto& tq = score::TaskPool::instance();
52 using worker_type = decltype(state->worker);
53
54 auto wk_state = std::weak_ptr{state};
55 state->worker.request = [ptr, &tq, wk_state]<typename... Args>(Args&&... f) {
56 using type_of_result = decltype(worker_type::work(std::forward<Args>(f)...));
57 tq.post([... ff = std::forward<Args>(f), wk_state, ptr]() mutable {
58 if constexpr(std::is_void_v<type_of_result>)
59 {
60 worker_type::work(std::forward<decltype(ff)>(ff)...);
61 }
62 else
63 {
64 // If the worker returns a std::function, it
65 // is to be invoked back in the processor DSP thread
66 auto res = worker_type::work(std::forward<decltype(ff)>(ff)...);
67 if(!res || !ptr)
68 return;
69
70 ossia::qt::run_async(
71 QCoreApplication::instance(),
72 [res = std::move(res), wk_state, ptr]() mutable {
73 if(ptr)
74 if(auto state = wk_state.lock())
75 res(*state);
76 });
77 }
78 });
79 };
80 }
81 }
82};
83
84template <typename GpuNodeRenderer, typename Node>
85struct GpuProcessIns
86{
87 GpuNodeRenderer& gpu;
88 Node& state;
89 const score::gfx::Message& prev_mess;
90 const score::gfx::Message& mess;
91 const score::DocumentContext& ctx;
92
93 bool can_process_message(std::size_t N)
94 {
95 if(mess.input.size() <= N)
96 return false;
97
98 if(prev_mess.input.size() == mess.input.size())
99 {
100 auto& prev = prev_mess.input[N];
101 auto& next = mess.input[N];
102 if(prev.index() == 1 && next.index() == 1)
103 {
104 if(ossia::get<ossia::value>(prev) == ossia::get<ossia::value>(next))
105 {
106 return false;
107 }
108 }
109 }
110 return true;
111 }
112
113 void operator()(avnd::parameter_port auto& t, auto field_index)
114 {
115 if(!can_process_message(field_index))
116 return;
117
118 if(auto val = ossia::get_if<ossia::value>(&mess.input[field_index]))
119 {
120 oscr::from_ossia_value(t, *val, t.value);
121 if_possible(t.update(state));
122 }
123 }
124
125#if OSCR_HAS_MMAP_FILE_STORAGE
126 template <avnd::raw_file_port Field, std::size_t NField>
127 void operator()(Field& t, avnd::field_index<NField> field_index)
128 {
129 // FIXME we should be loading a file there
130 using node_type = std::remove_cvref_t<decltype(gpu.node())>;
131 using file_ports = avnd::raw_file_input_introspection<Node>;
132
133 if(!can_process_message(field_index))
134 return;
135
136 auto val = ossia::get_if<ossia::value>(&mess.input[field_index]);
137 if(!val)
138 return;
139
140 static constexpr bool has_text = requires { decltype(Field::file)::text; };
141 static constexpr bool has_mmap = requires { decltype(Field::file)::mmap; };
142
143 // First we can load it directly since execution hasn't started yet
144 if(auto hdl = loadRawfile(*val, ctx, has_text, has_mmap))
145 {
146 static constexpr auto N = file_ports::field_index_to_index(NField);
147 if constexpr(avnd::port_can_process<Field>)
148 {
149 // FIXME also do it when we get a run-time message from the exec engine,
150 // OSC, etc
151 auto func = executePortPreprocess<Field>(*hdl);
152 const_cast<node_type&>(gpu.node())
153 .file_loaded(
154 state, hdl, avnd::predicate_index<N>{}, avnd::field_index<NField>{});
155 if(func)
156 func(state);
157 }
158 else
159 {
160 const_cast<node_type&>(gpu.node())
161 .file_loaded(
162 state, hdl, avnd::predicate_index<N>{}, avnd::field_index<NField>{});
163 }
164 }
165 }
166#endif
167
168 template <avnd::buffer_port Field, std::size_t NField>
169 void operator()(Field& t, avnd::field_index<NField> field_index)
170 {
171 using node_type = std::remove_cvref_t<decltype(gpu.node())>;
172 auto& node = const_cast<node_type&>(gpu.node());
173 auto val = ossia::get_if<ossia::render_target_spec>(&mess.input[field_index]);
174 if(!val)
175 return;
176 node.process(NField, *val);
177 }
178
179 template <avnd::texture_port Field, std::size_t NField>
180 void operator()(Field& t, avnd::field_index<NField> field_index)
181 {
182 using node_type = std::remove_cvref_t<decltype(gpu.node())>;
183 auto& node = const_cast<node_type&>(gpu.node());
184 auto val = ossia::get_if<ossia::render_target_spec>(&mess.input[field_index]);
185 if(!val)
186 return;
187 node.process(NField, *val);
188 }
189
190 template <avnd::geometry_port Field, std::size_t NField>
191 void operator()(Field& t, avnd::field_index<NField> field_index)
192 {
193 using node_type = std::remove_cvref_t<decltype(gpu.node())>;
194 auto& node = const_cast<node_type&>(gpu.node());
195
196 // FIXME
197 }
198
199 void operator()(auto& t, auto field_index) = delete;
200};
201
202struct GpuControlIns
203{
204 template <typename Self, typename Node_T>
205 static void processControlIn(
206 Self& self, Node_T& state, score::gfx::Message& renderer_mess,
207 const score::gfx::Message& mess, const score::DocumentContext& ctx) noexcept
208 {
209 // Apply the controls
210 avnd::input_introspection<Node_T>::for_all_n(
211 avnd::get_inputs<Node_T>(state),
212 GpuProcessIns<Self, Node_T>{self, state, renderer_mess, mess, ctx});
213 renderer_mess = mess;
214 }
215};
216
217struct GpuControlOuts
218{
219 std::weak_ptr<Execution::ExecutionCommandQueue> queue;
220 Gfx::exec_controls control_outs;
221
222 int64_t instance{};
223
224 template <typename Node_T>
225 void processControlOut(Node_T& state) const noexcept
226 {
227 if(!this->control_outs.empty())
228 {
229 auto q = this->queue.lock();
230 if(!q)
231 return;
232 auto& qq = *q;
233 int parm_k = 0;
234 avnd::parameter_output_introspection<Node_T>::for_all(
235 avnd::get_outputs(state), [&]<avnd::parameter_port T>(const T& t) {
236 qq.enqueue([v = oscr::to_ossia_value(t, t.value),
237 port = control_outs[parm_k]]() mutable {
238 std::swap(port->value, v);
239 port->changed = true;
240 });
241
242 parm_k++;
243 });
244 }
245 }
246};
247
248template <typename T>
249struct SCORE_PLUGIN_AVND_EXPORT GpuNodeElements
250{
251 [[no_unique_address]] oscr::soundfile_storage<T> soundfiles;
252
253 [[no_unique_address]] oscr::midifile_storage<T> midifiles;
254
255#if defined(OSCR_HAS_MMAP_FILE_STORAGE)
256 [[no_unique_address]] oscr::raw_file_storage<T> rawfiles;
257#endif
258
259 template <std::size_t N, std::size_t NField>
260 void file_loaded(
261 auto& state, const std::shared_ptr<oscr::raw_file_data>& hdl,
262 avnd::predicate_index<N>, avnd::field_index<NField>)
263 {
264 this->rawfiles.load(
265 state, hdl, avnd::predicate_index<N>{}, avnd::field_index<NField>{});
266 }
267};
268
269struct SCORE_PLUGIN_AVND_EXPORT CustomGfxNodeBase : score::gfx::NodeModel
270{
271 explicit CustomGfxNodeBase(const score::DocumentContext& ctx)
272 : score::gfx::NodeModel{}
273 , m_ctx{ctx}
274 {
275 }
276 virtual ~CustomGfxNodeBase();
277 const score::DocumentContext& m_ctx;
278 score::gfx::Message last_message;
279 void process(score::gfx::Message&& msg) override;
281};
282struct SCORE_PLUGIN_AVND_EXPORT CustomGfxOutputNodeBase : score::gfx::OutputNode
283{
284 virtual ~CustomGfxOutputNodeBase();
285
286 score::gfx::Message last_message;
287 void process(score::gfx::Message&& msg) override;
288};
289struct CustomGpuNodeBase
291 , GpuWorker
292 , GpuControlIns
293 , GpuControlOuts
294{
295 CustomGpuNodeBase(
296 std::weak_ptr<Execution::ExecutionCommandQueue>&& q, Gfx::exec_controls&& ctls,
297 const score::DocumentContext& ctx)
298 : GpuControlOuts{std::move(q), std::move(ctls)}
299 , m_ctx{ctx}
300 {
301 }
302
303 virtual ~CustomGpuNodeBase() = default;
304
305 const score::DocumentContext& m_ctx;
306 QString vertex, fragment, compute;
307 score::gfx::Message last_message;
308 void process(score::gfx::Message&& msg) override;
309};
310
311struct SCORE_PLUGIN_AVND_EXPORT CustomGpuOutputNodeBase
313 , GpuWorker
314 , GpuControlIns
315 , GpuControlOuts
316{
317 CustomGpuOutputNodeBase(
318 std::weak_ptr<Execution::ExecutionCommandQueue> q, Gfx::exec_controls&& ctls,
319 const score::DocumentContext& ctx);
320 virtual ~CustomGpuOutputNodeBase();
321
322 const score::DocumentContext& m_ctx;
323 std::weak_ptr<score::gfx::RenderList> m_renderer{};
324 std::shared_ptr<score::gfx::RenderState> m_renderState{};
325
326 QString vertex, fragment, compute;
327 score::gfx::Message last_message;
328 void process(score::gfx::Message&& msg) override;
330
331 void setRenderer(std::shared_ptr<score::gfx::RenderList>) override;
332 score::gfx::RenderList* renderer() const override;
333
334 void startRendering() override;
335 void render() override;
336 void stopRendering() override;
337 bool canRender() const override;
338 void onRendererChange() override;
339
340 void createOutput(score::gfx::OutputConfiguration) override;
341
342 void destroyOutput() override;
343 std::shared_ptr<score::gfx::RenderState> renderState() const override;
344
345 Configuration configuration() const noexcept override;
346};
347
348template <typename Node_T, typename Node>
349void prepareNewState(std::shared_ptr<Node_T>& eff, const Node& parent)
350{
351 if constexpr(avnd::has_worker<Node_T>)
352 {
353 parent.initWorker(eff);
354 }
355 if constexpr(avnd::has_processor_to_gui_bus<Node_T>)
356 {
357 auto& process = parent.processModel;
358 eff->send_message = [ptr = QPointer{&process}](auto&& b) mutable {
359 // FIXME right now all the rendering is done in the UI thread, which is very MEH
360 // this->in_edit([&process, bb = std::move(b)]() mutable {
361
362 if(ptr && ptr->to_ui)
363 MessageBusSender{ptr->to_ui}(std::move(b));
364 // });
365 };
366
367 // FIXME GUI -> engine. See executor.hpp
368 }
369
370 avnd::init_controls(*eff);
371
372 if constexpr(avnd::can_prepare<Node_T>)
373 {
374 if constexpr(avnd::function_reflection<&Node_T::prepare>::count == 1)
375 {
376 using prepare_type = avnd::first_argument<&Node_T::prepare>;
377 prepare_type t;
378 if_possible(t.instance = parent.instance);
379 eff->prepare(t);
380 }
381 else
382 {
383 eff->prepare();
384 }
385 }
386}
387
388struct port_to_type_enum
389{
390 template <std::size_t I, avnd::buffer_port F>
391 constexpr auto operator()(avnd::field_reflection<I, F> p)
392 {
393 return score::gfx::Types::Buffer;
394 }
395
396 template <std::size_t I, avnd::cpu_texture_port F>
397 constexpr auto operator()(avnd::field_reflection<I, F> p)
398 {
399 using texture_type = std::remove_cvref_t<decltype(F::texture)>;
400 return (avnd::cpu_fixed_format_texture<texture_type> || avnd::cpu_dynamic_format_texture<texture_type>)
401 ? score::gfx::Types::Image
402 : score::gfx::Types::Buffer;
403 }
404
405 template <std::size_t I, avnd::gpu_texture_port F>
406 constexpr auto operator()(avnd::field_reflection<I, F> p)
407 {
408 return score::gfx::Types::Image;
409 }
410
411 template <std::size_t I, avnd::sampler_port F>
412 constexpr auto operator()(avnd::field_reflection<I, F> p)
413 {
414 return score::gfx::Types::Image;
415 }
416 template <std::size_t I, avnd::image_port F>
417 constexpr auto operator()(avnd::field_reflection<I, F> p)
418 {
419 return score::gfx::Types::Image;
420 }
421 template <std::size_t I, avnd::attachment_port F>
422 constexpr auto operator()(avnd::field_reflection<I, F> p)
423 {
424 return score::gfx::Types::Image;
425 }
426
427 template <std::size_t I, avnd::geometry_port F>
428 constexpr auto operator()(avnd::field_reflection<I, F> p)
429 {
430 return score::gfx::Types::Geometry;
431 }
432 template <std::size_t I, avnd::mono_audio_port F>
433 constexpr auto operator()(avnd::field_reflection<I, F> p)
434 {
435 return score::gfx::Types::Audio;
436 }
437 template <std::size_t I, avnd::poly_audio_port F>
438 constexpr auto operator()(avnd::field_reflection<I, F> p)
439 {
440 return score::gfx::Types::Audio;
441 }
442 template <std::size_t I, avnd::int_parameter F>
443 constexpr auto operator()(avnd::field_reflection<I, F> p)
444 {
445 return score::gfx::Types::Int;
446 }
447 template <std::size_t I, avnd::enum_parameter F>
448 constexpr auto operator()(avnd::field_reflection<I, F> p)
449 {
450 return score::gfx::Types::Int;
451 }
452 template <std::size_t I, avnd::float_parameter F>
453 constexpr auto operator()(avnd::field_reflection<I, F> p)
454 {
455 return score::gfx::Types::Float;
456 }
457 template <std::size_t I, avnd::parameter_port F>
458 constexpr auto operator()(avnd::field_reflection<I, F> p)
459 {
460 using value_type = std::remove_cvref_t<decltype(F::value)>;
461
462 if constexpr(std::is_aggregate_v<value_type>)
463 {
464 constexpr int sz = boost::pfr::tuple_size_v<value_type>;
465 if constexpr(sz == 2)
466 {
467 return score::gfx::Types::Vec2;
468 }
469 else if constexpr(sz == 3)
470 {
471 return score::gfx::Types::Vec3;
472 }
473 else if constexpr(sz == 4)
474 {
475 return score::gfx::Types::Vec4;
476 }
477 }
478 return score::gfx::Types::Empty;
479 }
480 template <std::size_t I, typename F>
481 constexpr auto operator()(avnd::field_reflection<I, F> p)
482 {
483 return score::gfx::Types::Empty;
484 }
485};
486
487template <typename Node_T>
488inline void initGfxPorts(auto* self, auto& input, auto& output)
489{
490 avnd::input_introspection<Node_T>::for_all(
491 [self, &input]<typename Field, std::size_t I>(avnd::field_reflection<I, Field> f) {
492 static constexpr auto type = port_to_type_enum{}(f);
493 input.push_back(new score::gfx::Port{self, {}, type, {}});
494 });
495 avnd::output_introspection<Node_T>::for_all(
496 [self,
497 &output]<typename Field, std::size_t I>(avnd::field_reflection<I, Field> f) {
498 static constexpr auto type = port_to_type_enum{}(f);
499 output.push_back(new score::gfx::Port{self, {}, type, {}});
500 });
501}
502
503static score::gfx::BufferView getInputBuffer(
504 score::gfx::RenderList& renderer, const score::gfx::Node& parent, int port_index)
505{
506 const auto& inputs = parent.input;
507 // SCORE_ASSERT(port_index == 0);
508 {
509 score::gfx::Port* p = inputs[port_index];
510 for(auto& edge : p->edges)
511 {
512 auto src_node = edge->source->node;
513 score::gfx::NodeRenderer* src_renderer = src_node->renderedNodes.at(&renderer);
514 if(src_renderer)
515 {
516 return src_renderer->bufferForOutput(*edge->source);
517 }
518 break;
519 }
520 }
521 return {};
522}
523
524
525static void readbackInputBuffer(
526 score::gfx::RenderList& renderer
527 , QRhiResourceUpdateBatch& res
528 , const score::gfx::Node& parent
529 , QRhiBufferReadbackResult& readback
530 , int port_index
531 )
532{
533 // FIXME: instead of doing this we could do the readback in the
534 // producer node and just read its bytearray once...
535 if(auto buf = getInputBuffer(renderer, parent, port_index))
536 {
537 readback = {};
538 res.readBackBuffer(buf.handle, buf.byte_offset, buf.byte_size, &readback);
539 }
540}
541
542static void recreateOutputBuffer(
543 score::gfx::RenderList& renderer, avnd::cpu_buffer auto& cpu_buf,
544 QRhiResourceUpdateBatch& res, score::gfx::BufferView& buf)
545{
546 const auto bytesize = avnd::get_bytesize(cpu_buf);
547 if(!buf.handle)
548 {
549 if(bytesize > 0)
550 {
551 buf.handle = renderer.state.rhi->newBuffer(
552 QRhiBuffer::Static, QRhiBuffer::StorageBuffer | QRhiBuffer::VertexBuffer,
553 bytesize);
554 buf.handle->setName("GpuUtils::recreateOutputBuffer");
555 buf.byte_offset = 0;
556 buf.byte_size = bytesize;
557
558 buf.handle->create();
559 }
560 else
561 {
562 cpu_buf.changed = false;
563 return;
564 }
565 }
566 else if(buf.handle->size() != bytesize)
567 {
568 buf.handle->destroy();
569 buf.handle->setSize(bytesize);
570 buf.handle->create();
571 buf.byte_size = bytesize;
572 }
573}
574
575static void uploadOutputBuffer(
576 score::gfx::RenderList& renderer, avnd::cpu_buffer auto& cpu_buf,
577 QRhiResourceUpdateBatch& res, score::gfx::BufferView& rhi_buf)
578{
579 if(cpu_buf.changed)
580 {
581 const auto bytesize = avnd::get_bytesize(cpu_buf);
582 recreateOutputBuffer(renderer, cpu_buf, res, rhi_buf);
584 &res, rhi_buf.handle, 0, cpu_buf.byte_size,
585 (const char*)avnd::get_bytes(cpu_buf));
586 cpu_buf.changed = false;
587 }
588}
589
590static void uploadOutputBuffer(
591 score::gfx::RenderList& renderer, avnd::gpu_buffer auto& gpu_buf,
592 QRhiResourceUpdateBatch& res, score::gfx::BufferView& rhi_buf)
593{
594 rhi_buf.handle = reinterpret_cast<QRhiBuffer*>(gpu_buf.handle);
595 rhi_buf.byte_size = gpu_buf.byte_size;
596 rhi_buf.byte_offset = gpu_buf.byte_offset;
597}
598
599template <typename T>
600struct geometry_inputs_storage;
601
602struct mesh_input_storage
603{
604 std::vector<QRhiBufferReadbackResult> readbacks;
605 std::vector<QRhiBuffer*> buffers;
606};
607struct geometry_input_storage
608{
609 ossia::geometry_spec spec;
610 std::vector<mesh_input_storage> meshes;
611};
612
613template <typename T>
614 requires(avnd::geometry_input_introspection<T>::size > 0)
615struct geometry_inputs_storage<T>
616{
617 // FIXME in Gfx/Graph/NodeRenderer.hpp
618 static_assert(avnd::geometry_input_introspection<T>::size == 1);
619
620 geometry_input_storage inputs[avnd::geometry_input_introspection<T>::size];
621 ossia::small_vector<QRhiBuffer*, 4> allocated;
622
623 void readInputGeometries(
624 score::gfx::RenderList& renderer, const ossia::geometry_spec& spec, auto& parent,
625 auto& state)
626 {
627 // Copy the readback output inside the structure
628 // TODO it would be much better to do this inside the readback's
629 // "completed" callback.
630 avnd::geometry_input_introspection<T>::for_all_n(
631 avnd::get_inputs<T>(state),
632 [&]<typename Field, std::size_t N>(Field& t, avnd::predicate_index<N> np) {
633 this->inputs[N].spec = spec; // FIXME multiple geometry input ports
634 this->inputs[N].meshes.resize(1); // FIXME
635
636 // Here we fetch the readbacks results
637 auto& meshes = this->inputs[N].meshes[0];
638
639 oscr::meshes_from_ossia(
640 spec.meshes, t.mesh,
641 [&](auto& write_buf, int buffer_index, void* data, int64_t bytesize) {
642 // CPU input geometry, upload was done before
643 SCORE_ASSERT(buffer_index >= 0);
644 if(buffer_index < meshes.readbacks.size())
645 {
646 QRhiBuffer* handle = meshes.buffers[buffer_index];
647 write_buf.handle = handle;
648 write_buf.byte_size = handle->size();
649 }
650 }, [&](auto& write_buf, int buffer_index, void* handle) {
651 // GPU input buffer, CPU output buffer: need to fetch our readback
652 SCORE_ASSERT(buffer_index >= 0);
653 if(buffer_index < meshes.readbacks.size())
654 {
655 // FIXME investigate why runInitialPasses is called before inputAboutToFinish
656 auto& readback = meshes.readbacks[buffer_index].data;
657 write_buf.raw_data = reinterpret_cast<unsigned char*>(readback.data());
658 write_buf.byte_size = readback.size();
659 }
660 });
661 });
662 }
663
664 void inputAboutToFinish(
665 score::gfx::RenderList& renderer, QRhiResourceUpdateBatch*& res,
666 const ossia::geometry_spec& spec, auto& state, auto& parent)
667 {
668 avnd::geometry_input_introspection<T>::for_all_n2(
669 avnd::get_inputs<T>(state),
670 [&]<typename Field, std::size_t N, std::size_t NField>(
671 Field& t, avnd::predicate_index<N> np, avnd::field_index<NField> nf) {
672 this->inputs[N].spec = spec; // FIXME multiple geometry input ports
673 this->inputs[N].meshes.resize(1); // FIXME
674 // Here we request readbacks if necessary
675
676 auto& meshes = this->inputs[N].meshes[0];
677 oscr::meshes_from_ossia(
678 spec.meshes, t.mesh,
679 [&](auto& write_buf, int buffer_index, void* data, int64_t bytesize) {
680 // cpu -> gpu
681 if(meshes.buffers.size() <= buffer_index)
682 {
683 meshes.buffers.resize(buffer_index + 1);
684 meshes.readbacks.resize(buffer_index + 1);
685
686 auto buf = renderer.state.rhi->newBuffer(
687 QRhiBuffer::Static, QRhiBuffer::StorageBuffer | QRhiBuffer::VertexBuffer,
688 bytesize);
689 buf->setName(oscr::getUtf8Name<T>() + "::" + oscr::getUtf8Name(t));
690 buf->create();
691 allocated.push_back(buf);
692 meshes.buffers[buffer_index] = buf;
693 }
694
695 res->uploadStaticBuffer(meshes.buffers[buffer_index], 0, bytesize, data);
696 }, [&](auto& write_buf, int buffer_index, void* handle) {
697 // gpu -> cpu
698 if(meshes.readbacks.size() <= buffer_index)
699 {
700 meshes.buffers.resize(buffer_index + 1);
701 meshes.readbacks.resize(buffer_index + 1);
702 }
703
704 meshes.readbacks[buffer_index] = {};
705 if(auto buf = static_cast<QRhiBuffer*>(handle))
706 {
707 meshes.buffers[buffer_index] = buf;
708 res->readBackBuffer(buf, 0, buf->size(), &meshes.readbacks[buffer_index]);
709 }
710 else
711 {
712 meshes.buffers[buffer_index] = {};
713 meshes.readbacks[buffer_index] = {};
714 }
715 });
716 });
717 }
718
719 void release(score::gfx::RenderList& renderer)
720 {
721 for(auto& buf : allocated)
722 renderer.releaseBuffer(buf);
723 allocated.clear();
724 }
725};
726
727template <typename T>
728 requires(avnd::geometry_input_introspection<T>::size == 0)
729struct geometry_inputs_storage<T>
730{
731 static void readInputBuffers(auto&&...) { }
732
733 static void inputAboutToFinish(auto&&...) { }
734};
735
736template<typename T>
737struct buffer_inputs_storage;
738
739template<typename T>
740 requires (avnd::buffer_input_introspection<T>::size > 0)
741struct buffer_inputs_storage<T>
742{
743 // +1 because of zero-array-size unsupported
744 QRhiBufferReadbackResult
745 m_readbacks[avnd::cpu_buffer_input_introspection<T>::size + 1];
746 score::gfx::BufferView m_gpubufs[avnd::gpu_buffer_input_introspection<T>::size + 1];
747
748 void readInputBuffers(
749 score::gfx::RenderList& renderer, auto& parent, auto& state)
750 {
751 if constexpr(avnd::cpu_buffer_input_introspection<T>::size > 0)
752 {
753 // Copy the readback output inside the structure
754 // TODO it would be much better to do this inside the readback's
755 // "completed" callback.
756 avnd::cpu_buffer_input_introspection<T>::for_all_n(
757 avnd::get_inputs<T>(state),
758 [&]<typename Field, std::size_t N>
759 (Field& t, avnd::predicate_index<N> np)
760 {
761 auto& readback = m_readbacks[N].data;
762 t.buffer.raw_data = reinterpret_cast<unsigned char*>(readback.data());
763 t.buffer.byte_size = readback.size();
764 t.buffer.byte_offset = 0; // FIXME
765 t.buffer.changed = true;
766 });
767 }
768
769 if constexpr(avnd::gpu_buffer_input_introspection<T>::size > 0)
770 {
771 // Copy the readback output inside the structure
772 // TODO it would be much better to do this inside the readback's
773 // "completed" callback.
774 avnd::gpu_buffer_input_introspection<T>::for_all_n2(
775 avnd::get_inputs<T>(state),
776 [&]<typename Field, std::size_t N, std::size_t NField>(
777 Field& t, avnd::predicate_index<N> np, avnd::field_index<NField> nf) {
778 score::gfx::BufferView& buf = m_gpubufs[N];
779 if(!buf)
780 buf = getInputBuffer(renderer, parent, nf);
781 if(!buf)
782 return;
783 t.buffer.handle = buf.handle;
784 t.buffer.byte_size = buf.byte_size;
785 t.buffer.byte_offset = buf.byte_offset;
786 // t.buffer.changed = true; FIXME
787 });
788 }
789 }
790
791 void inputAboutToFinish(
792 score::gfx::RenderList& renderer,
793 QRhiResourceUpdateBatch*& res,
794 auto& state,
795 auto& parent)
796 {
797 avnd::cpu_buffer_input_introspection<T>::for_all_n2(
798 avnd::get_inputs<T>(state),
799 [&]<typename Field, std::size_t N, std::size_t NField>
800 (Field& port, avnd::predicate_index<N> np, avnd::field_index<NField> nf) {
801 readbackInputBuffer(renderer, *res, parent, m_readbacks[N], nf);
802 });
803 avnd::gpu_buffer_input_introspection<T>::for_all_n2(
804 avnd::get_inputs<T>(state),
805 [&]<typename Field, std::size_t N, std::size_t NField>
806 (Field& port, avnd::predicate_index<N> np, avnd::field_index<NField> nf) {
807 m_gpubufs[N] = getInputBuffer(renderer, parent, nf);
808 });
809 }
810};
811
812template<typename T>
813 requires (avnd::buffer_input_introspection<T>::size == 0)
814struct buffer_inputs_storage<T>
815{
816 static void readInputBuffers(auto&&...)
817 {
818
819 }
820
821 static void inputAboutToFinish(auto&&...)
822 {
823
824 }
825};
826
827struct MaybeOwnedBuffer : score::gfx::BufferView
828{
829 bool owned{false};
830};
831
832template<typename T>
833struct buffer_outputs_storage;
834
835template<typename T>
836 requires (avnd::buffer_output_introspection<T>::size > 0)
837struct buffer_outputs_storage<T>
838{
839 std::pair<const score::gfx::Port*, MaybeOwnedBuffer>
840 m_buffers[avnd::buffer_output_introspection<T>::size];
841
842 QRhiResourceUpdateBatch* currentResourceUpdateBatch{};
843
844 template <typename Field, std::size_t N, std::size_t NField>
845 requires avnd::cpu_buffer<std::decay_t<decltype(Field::buffer)>>
846 void createOutput(
847 score::gfx::RenderList& renderer, auto& parent, Field& port,
848 avnd::predicate_index<N> np, avnd::field_index<NField> nf)
849 {
850 auto& [gfx_port, buf] = m_buffers[N];
851 gfx_port = parent.output[nf];
852 buf.handle = renderer.state.rhi->newBuffer(
853 QRhiBuffer::Static, QRhiBuffer::StorageBuffer | QRhiBuffer::VertexBuffer, 1);
854 buf.handle->setName(oscr::getUtf8Name<T>() + "::" + oscr::getUtf8Name(port));
855 buf.byte_offset = 0;
856 buf.byte_size = 1;
857 buf.owned = true;
858
859 buf.handle->create();
860
861 port.buffer.upload
862 = [this, &renderer, &port](const char* data, int64_t offset, int64_t bytesize) {
863 // FIXME is offset and bytesize relative to the input or the output data ?
864 SCORE_ASSERT(currentResourceUpdateBatch);
865 auto& [gfx_port, buf] = m_buffers[N];
866
867 if(!buf.handle)
868 {
869 if(bytesize > 0)
870 {
871 buf.handle = renderer.state.rhi->newBuffer(
872 QRhiBuffer::Static, QRhiBuffer::StorageBuffer | QRhiBuffer::VertexBuffer,
873 bytesize);
874 buf.handle->setName(oscr::getUtf8Name<T>() + "::" + oscr::getUtf8Name(port));
875 buf.byte_offset = 0;
876 buf.byte_size = bytesize;
877 buf.owned = true;
878
879 buf.handle->create();
880 }
881 else
882 {
883 buf.handle = renderer.state.rhi->newBuffer(
884 QRhiBuffer::Static, QRhiBuffer::StorageBuffer | QRhiBuffer::VertexBuffer,
885 1);
886 buf.handle->setName(oscr::getUtf8Name<T>() + "::" + oscr::getUtf8Name(port));
887 buf.byte_offset = 0;
888 buf.byte_size = 1;
889 buf.owned = true;
890
891 buf.handle->create();
892 return;
893 }
894 }
895 else if(buf.handle->size() != bytesize)
896 {
897 buf.handle->destroy();
898 buf.handle->setSize(bytesize);
899 buf.handle->create();
900 buf.byte_size = bytesize;
901 }
902
904 currentResourceUpdateBatch, buf.handle, offset, bytesize, data);
905 };
906 }
907
908 template <typename Field, std::size_t N, std::size_t NField>
909 requires avnd::gpu_buffer<std::decay_t<decltype(Field::buffer)>>
910 void createOutput(
911 score::gfx::RenderList& renderer, auto& parent, Field& port,
912 avnd::predicate_index<N> np, avnd::field_index<NField> nf)
913 {
914 auto& [gfx_port, buf] = m_buffers[N];
915 gfx_port = parent.output[nf];
916 buf.handle = reinterpret_cast<QRhiBuffer*>(port.buffer.handle);
917 buf.byte_size = port.buffer.byte_size;
918 buf.byte_offset = port.buffer.byte_offset;
919 buf.owned = false;
920 }
921
922 void init(score::gfx::RenderList& renderer, auto& state, auto& parent)
923 {
924 // Init buffers for the outputs
925 avnd::buffer_output_introspection<T>::for_all_n2(
926 avnd::get_outputs<T>(state), [&]<typename Field, std::size_t N, std::size_t NField>
927 (Field& port, avnd::predicate_index<N> np, avnd::field_index<NField> nf) {
928 SCORE_ASSERT(parent.output.size() > nf);
929 SCORE_ASSERT(parent.output[nf]->type == score::gfx::Types::Buffer);
930 using buffer_type = std::decay_t<decltype(port.buffer)>;
931
932 if constexpr(avnd::cpu_raw_buffer<buffer_type> && requires {
933 port.buffer.upload(nullptr, 0, 0);
934 })
935 {
936 createOutput(renderer, parent, port, np, nf);
937 }
938 else if constexpr(avnd::gpu_buffer<buffer_type>)
939 {
940 createOutput(renderer, parent, port, np, nf);
941 }
942 else
943 {
944 // m_buffers[N] = createOutput(renderer, *parent.output[nf], port.buffer);
945 static_assert(std::is_same_v<T, void>, "unsupported");
946 }
947 });
948 }
949
950 void prepareUpload(QRhiResourceUpdateBatch& res)
951 {
952 currentResourceUpdateBatch = &res;
953 }
954
955 void upload(score::gfx::RenderList& renderer, auto& state, QRhiResourceUpdateBatch& res)
956 {
957 avnd::buffer_output_introspection<T>::for_all_n(
958 avnd::get_outputs<T>(state), [&]<std::size_t N>(auto& t, avnd::predicate_index<N> idx) {
959 auto& [port, buf] = m_buffers[N];
960 uploadOutputBuffer(renderer, t.buffer, res, buf);
961 });
962 }
963
964 void release(score::gfx::RenderList& renderer)
965 {
966 // Free outputs
967 for(auto& [p, buf] : m_buffers)
968 {
969 if(buf.owned)
970 renderer.releaseBuffer(buf.handle);
971 buf.handle = nullptr;
972 buf.owned = false;
973 }
974 }
975};
976
977template<typename T>
978 requires (avnd::buffer_output_introspection<T>::size == 0)
979struct buffer_outputs_storage<T>
980{
981 static void init(auto&&...)
982 {
983
984 }
985
986 static void prepareUpload(auto&&...)
987 {
988 }
989
990 static void upload(auto&&...)
991 {
992 }
993
994 static void release(auto&&...)
995 {
996 }
997};
998
999
1000template <typename Tex>
1001static auto
1002createOutputTexture(score::gfx::RenderList& renderer, const Tex& texture_spec, QSize size)
1003{
1004 auto& rhi = *renderer.state.rhi;
1005 QRhiTexture* texture = &renderer.emptyTexture();
1006 if(size.width() > 0 && size.height() > 0)
1007 {
1008 texture = rhi.newTexture(
1009 gpp::qrhi::textureFormat(texture_spec), size, 1, QRhiTexture::Flag{});
1010
1011 texture->create();
1012 }
1013
1014 auto sampler = rhi.newSampler(
1015 QRhiSampler::Linear, QRhiSampler::Linear, QRhiSampler::None,
1016 QRhiSampler::ClampToEdge, QRhiSampler::ClampToEdge);
1017
1018 sampler->create();
1019 return score::gfx::Sampler{sampler, texture};
1020}
1021
1022
1023template<typename T>
1024struct texture_inputs_storage;
1025
1026template<typename T>
1027 requires (avnd::texture_input_introspection<T>::size > 0)
1028struct texture_inputs_storage<T>
1029{
1030 ossia::small_flat_map<const score::gfx::Port*, score::gfx::TextureRenderTarget, 2>
1031 m_rts;
1032
1033 QRhiReadbackResult m_readbacks[avnd::texture_input_introspection<T>::size];
1034
1035 template <typename Tex>
1036 QRhiTexture* createInput(
1037 score::gfx::RenderList& renderer, score::gfx::Port* port, Tex& texture_spec,
1039 {
1040 static constexpr auto flags
1041 = QRhiTexture::RenderTarget | QRhiTexture::UsedAsTransferSource;
1042 QRhiTexture::Format fmt{};
1043 if constexpr(requires (Tex tex) { tex.format = {}; } && !requires (Tex tex) { tex.request_format; })
1044 {
1045 // Format freely assignable: we use what the user sets in the GUI
1046 fmt = spec.format;
1047 gpp::qrhi::toTextureFormat(fmt, texture_spec);
1048 }
1049 else
1050 {
1051 fmt = gpp::qrhi::textureFormat(texture_spec);
1052 }
1053
1054 QRhiTexture* texture = renderer.state.rhi->newTexture(
1055 fmt, spec.size, 1, flags);
1056
1057 SCORE_ASSERT(texture->create());
1058 m_rts[port] = score::gfx::createRenderTarget(
1059 renderer.state, texture, renderer.samples(), renderer.requiresDepth(*port));
1060 return texture;
1061 }
1062
1063 void init(auto& self, score::gfx::RenderList& renderer)
1064 {
1065 // Init input render targets
1066 avnd::texture_input_introspection<T>::for_all_n2(
1067 avnd::get_inputs<T>(*self.state),
1068 [&]<typename F, std::size_t K, std::size_t N>(F& t, avnd::predicate_index<K>, avnd::field_index<N>) {
1069 auto& parent = self.node();
1070 auto spec = parent.resolveRenderTargetSpecs(N, renderer);
1071 if constexpr(requires {
1072 t.request_width;
1073 t.request_height;
1074 })
1075 {
1076 spec.size.rwidth() = t.request_width;
1077 spec.size.rheight() = t.request_height;
1078 }
1079
1080 auto tex = createInput(renderer, parent.input[N], t.texture, spec);
1081 if constexpr(avnd::cpu_texture_port<F>)
1082 {
1083 t.texture.width = spec.size.width();
1084 t.texture.height = spec.size.height();
1085 }
1086 else if constexpr(avnd::gpu_texture_port<F>)
1087 {
1088 t.texture.handle = tex;
1089 t.texture.width = spec.size.width();
1090 t.texture.height = spec.size.height();
1091 }
1092 });
1093 }
1094
1095 bool update(auto& self,
1096 score::gfx::RenderList& renderer, QRhiResourceUpdateBatch& res)
1097 {
1098#if 0
1099 bool need_update = false;
1100 avnd::texture_input_introspection<T>::for_all_n2(
1101 avnd::get_inputs<T>(*self.state),
1102 [&]<typename F, std::size_t K, std::size_t N>(F& t, avnd::predicate_index<K>, avnd::field_index<N>) {
1103 if constexpr(requires {
1104 t.request_width;
1105 t.request_height;
1106 })
1107 {
1108 auto& parent = self.node();
1109 auto port = parent.input[N];
1110 const score::gfx::TextureRenderTarget& texture = m_rts[port];
1111 QSizeF sz{};
1112 if(texture.texture)
1113 sz = texture.texture->pixelSize();
1114 if(sz.width() != t.request_width || sz.height() != t.request_height)
1115 {
1116 // FIXME right now this doesn't work because
1117 // the render target spec is stored in the node.
1118 // Also the RenderList just recomputes everything anyways,
1119 // so we should just emit a "need to change" signal and abort as
1120 // long as things aren't more optimized and actually follow the graph
1121
1122 // m_rts[port].release();
1123
1124 // auto spec = parent.resolveRenderTargetSpecs(N, renderer);
1125 // spec.size.rwidth() = t.request_width;
1126 // spec.size.rheight() = t.request_height;
1127
1128 // createInput(renderer, port, t.texture, sz);
1129
1130 // t.texture.width = spec.size.width();
1131 // t.texture.height = spec.size.height();
1132 // need_update = true;
1133 //
1134 }
1135 }
1136 });
1137 return need_update;
1138#endif
1139 return false;
1140 }
1141
1142 void runInitialPasses(auto& self, QRhi& rhi)
1143 {
1144 // Fetch input textures (if any)
1145 // Copy the readback output inside the structure
1146 // TODO it would be much better to do this inside the readback's
1147 // "completed" callback.
1148 if constexpr(avnd::cpu_texture_input_introspection<T>::size > 0)
1149 {
1150 avnd::texture_input_introspection<T>::for_all_n(
1151 avnd::get_inputs<T>(*self.state), [&]<typename F, std::size_t K>(F& t, avnd::predicate_index<K>) {
1152 if constexpr(avnd::cpu_texture_port<F>)
1153 {
1154 oscr::loadInputTexture(rhi, m_readbacks, t.texture, K);
1155 }
1156 });
1157 }
1158 }
1159
1160 void release()
1161 {
1162 // Free inputs
1163 // TODO investigate why reference does not work here:
1164 for(auto [port, rt] : m_rts)
1165 rt.release();
1166 m_rts.clear();
1167 }
1168
1169 void inputAboutToFinish(auto& parent, const score::gfx::Port& p, QRhiResourceUpdateBatch*& res)
1170 {
1171 if constexpr(avnd::cpu_texture_input_introspection<T>::size > 0)
1172 {
1173 const auto& inputs = parent.input;
1174 auto index_of_port = ossia::find(inputs, &p) - inputs.begin();
1175 {
1176 auto tex = m_rts[&p].texture;
1177 auto& readback = m_readbacks[index_of_port];
1178 readback = {};
1179 res->readBackTexture(QRhiReadbackDescription{tex}, &readback);
1180 }
1181 }
1182 }
1183
1184};
1185template<typename T>
1186 requires (avnd::texture_input_introspection<T>::size == 0)
1187struct texture_inputs_storage<T>
1188{
1189 static void init(auto&&...) { }
1190 static void runInitialPasses(auto&&...) { }
1191 static void release(auto&&...) { }
1192 static void inputAboutToFinish(auto&&...) { }
1193};
1194
1195
1196
1197template <avnd::cpu_texture Tex>
1198static QRhiTexture* updateTexture(auto& self, score::gfx::RenderList& renderer, int k, const Tex& cpu_tex)
1199{
1200 auto& [sampler, texture] = self.m_samplers[k];
1201 if(texture)
1202 {
1203 auto sz = texture->pixelSize();
1204 if(cpu_tex.width == sz.width() && cpu_tex.height == sz.height())
1205 return texture;
1206 }
1207
1208 // Check the texture size
1209 if(cpu_tex.width > 0 && cpu_tex.height > 0)
1210 {
1211 QRhiTexture* oldtex = texture;
1212 QRhiTexture* newtex = renderer.state.rhi->newTexture(
1213 gpp::qrhi::textureFormat(cpu_tex), QSize{cpu_tex.width, cpu_tex.height}, 1,
1214 QRhiTexture::Flag{});
1215 newtex->create();
1216 for(auto& [edge, pass] : self.m_p)
1217 if(pass.srb)
1218 score::gfx::replaceTexture(*pass.srb, sampler, newtex);
1219 texture = newtex;
1220
1221 if(oldtex && oldtex != &renderer.emptyTexture())
1222 {
1223 oldtex->deleteLater();
1224 }
1225
1226 return newtex;
1227 }
1228 else
1229 {
1230 for(auto& [edge, pass] : self.m_p)
1231 if(pass.srb)
1232 score::gfx::replaceTexture(*pass.srb, sampler, &renderer.emptyTexture());
1233
1234 return &renderer.emptyTexture();
1235 }
1236}
1237
1238template <avnd::cpu_texture Tex>
1239static void uploadOutputTexture(auto& self,
1240 score::gfx::RenderList& renderer, int k, Tex& cpu_tex,
1241 QRhiResourceUpdateBatch* res)
1242{
1243 if(cpu_tex.changed)
1244 {
1245 if(auto texture = updateTexture(self, renderer, k, cpu_tex))
1246 {
1247 QByteArray buf
1248 = QByteArray::fromRawData((const char*)cpu_tex.bytes, cpu_tex.bytesize());
1249 if constexpr(requires { Tex::RGB; })
1250 {
1251 // RGB -> RGBA
1252 // FIXME other conversions
1253 const QByteArray rgb = buf;
1254 QByteArray rgba;
1255 rgba.resize(cpu_tex.width * cpu_tex.height * 4);
1256 auto src = (const unsigned char*)rgb.constData();
1257 auto dst = (unsigned char*)rgba.data();
1258 for(int rgb_byte = 0, rgba_byte = 0, N = rgb.size(); rgb_byte < N;)
1259 {
1260 dst[rgba_byte + 0] = src[rgb_byte + 0];
1261 dst[rgba_byte + 1] = src[rgb_byte + 1];
1262 dst[rgba_byte + 2] = src[rgb_byte + 2];
1263 dst[rgba_byte + 3] = 255;
1264 rgb_byte += 3;
1265 rgba_byte += 4;
1266 }
1267 buf = rgba;
1268 }
1269
1270 // Upload it (mirroring is done in shader generic_texgen_fs if necessary)
1271 {
1272 QRhiTextureSubresourceUploadDescription sd(buf);
1273 QRhiTextureUploadDescription desc{QRhiTextureUploadEntry{0, 0, sd}};
1274
1275 res->uploadTexture(texture, desc);
1276 }
1277
1278 cpu_tex.changed = false;
1279 }
1280 }
1281}
1282
1283static const constexpr auto generic_texgen_vs = R"_(#version 450
1284layout(location = 0) in vec2 position;
1285layout(location = 1) in vec2 texcoord;
1286
1287layout(binding=3) uniform sampler2D y_tex;
1288layout(location = 0) out vec2 v_texcoord;
1289
1290layout(std140, binding = 0) uniform renderer_t {
1291 mat4 clipSpaceCorrMatrix;
1292 vec2 renderSize;
1293} renderer;
1294
1295out gl_PerVertex { vec4 gl_Position; };
1296
1297void main()
1298{
1299#if defined(QSHADER_SPIRV) || defined(QSHADER_GLSL)
1300 v_texcoord = vec2(texcoord.x, 1. - texcoord.y);
1301#else
1302 v_texcoord = texcoord;
1303#endif
1304 gl_Position = renderer.clipSpaceCorrMatrix * vec4(position.xy, 0.0, 1.);
1305}
1306)_";
1307
1308static const constexpr auto generic_texgen_fs = R"_(#version 450
1309layout(location = 0) in vec2 v_texcoord;
1310layout(location = 0) out vec4 fragColor;
1311
1312layout(std140, binding = 0) uniform renderer_t {
1313mat4 clipSpaceCorrMatrix;
1314vec2 renderSize;
1315} renderer;
1316
1317layout(binding=3) uniform sampler2D y_tex;
1318
1319void main ()
1320{
1321 fragColor = texture(y_tex, v_texcoord);
1322}
1323)_";
1324
1325template<typename T>
1326struct texture_outputs_storage;
1327
1328// If we have texture outs we need the whole rendering infrastructure
1329template<typename T>
1330 requires (avnd::texture_output_introspection<T>::size > 0)
1331struct texture_outputs_storage<T>
1332{
1333 void init(auto& self, score::gfx::RenderList& renderer, QRhiResourceUpdateBatch& res)
1334 {
1335 const auto& mesh = renderer.defaultTriangle();
1336 self.defaultMeshInit(renderer, mesh, res);
1337 self.processUBOInit(renderer);
1338 // Not needed here as we do not have a GPU pass:
1339 // this->m_material.init(renderer, this->node.input, this->m_samplers);
1340
1341 std::tie(self.m_vertexS, self.m_fragmentS)
1342 = score::gfx::makeShaders(renderer.state, generic_texgen_vs, generic_texgen_fs);
1343
1344 avnd::cpu_texture_output_introspection<T>::for_all(
1345 avnd::get_outputs<T>(*self.state), [&](auto& t) {
1346 self.m_samplers.push_back(
1347 createOutputTexture(renderer, t.texture, QSize{t.texture.width, t.texture.height}));
1348 });
1349
1350 self.defaultPassesInit(renderer, mesh);
1351 }
1352
1353 void runInitialPasses(auto& self,
1354 score::gfx::RenderList& renderer,
1355 QRhiResourceUpdateBatch*& res)
1356 {
1357 avnd::cpu_texture_output_introspection<T>::for_all_n(
1358 avnd::get_outputs<T>(*self.state), [&]<std::size_t N>(auto& t, avnd::predicate_index<N>) {
1359 uploadOutputTexture(self, renderer, N, t.texture, res);
1360 });
1361 }
1362
1363 void release(auto& self, score::gfx::RenderList& r)
1364 {
1365 // Free outputs
1366 for(auto& [sampl, texture] : self.m_samplers)
1367 {
1368 if(texture != &r.emptyTexture())
1369 texture->deleteLater();
1370 texture = nullptr;
1371 }
1372 }
1373
1374};
1375
1376template<typename T>
1377 requires (avnd::texture_output_introspection<T>::size == 0)
1378struct texture_outputs_storage<T>
1379{
1380 static void init(auto& self, score::gfx::RenderList& renderer, QRhiResourceUpdateBatch& res)
1381 {
1382 }
1383
1384 static void runInitialPasses(auto& self,
1385 score::gfx::RenderList& renderer,
1386 QRhiResourceUpdateBatch*& res)
1387 {
1388 }
1389
1390 static void release(auto& self, score::gfx::RenderList& r)
1391 {
1392 }
1393};
1394template<typename T>
1395struct geometry_outputs_storage;
1396
1397template<typename T>
1398 requires (avnd::geometry_output_introspection<T>::size > 0)
1399struct geometry_outputs_storage<T>
1400{
1401 ossia::geometry_spec specs[avnd::geometry_output_introspection<T>::size];
1402
1403 template <avnd::geometry_port Field>
1404 void reload_mesh(Field& ctrl, ossia::geometry_spec& spc)
1405 {
1406 spc.meshes = std::make_shared<ossia::mesh_list>();
1407 auto& ossia_meshes = *spc.meshes;
1408 if constexpr(avnd::static_geometry_type<Field> || avnd::dynamic_geometry_type<Field>)
1409 {
1410 ossia_meshes.meshes.resize(1);
1411 load_geometry(ctrl, ossia_meshes.meshes[0]);
1412 }
1413 else if constexpr(
1414 avnd::static_geometry_type<decltype(Field::mesh)>
1415 || avnd::dynamic_geometry_type<decltype(Field::mesh)>)
1416 {
1417 ossia_meshes.meshes.resize(1);
1418 load_geometry(ctrl.mesh, ossia_meshes.meshes[0]);
1419 }
1420 else
1421 {
1422 load_geometry(ctrl, ossia_meshes);
1423 }
1424 }
1425
1426 template <avnd::geometry_port Field, std::size_t N>
1427 void upload(
1428 score::gfx::RenderList& renderer, Field& ctrl, score::gfx::Edge& edge,
1429 avnd::predicate_index<N>)
1430 {
1431 auto edge_sink = edge.sink;
1432 if(auto pnode = edge_sink->node)
1433 {
1434 ossia::geometry_spec& spc = specs[N];
1435
1436 // 1. Reload mesh
1437 {
1438 if(ctrl.dirty_mesh)
1439 {
1440 reload_mesh(ctrl, spc);
1441 }
1442 else
1443 {
1444 if(spc.meshes)
1445 {
1446 auto& ossia_meshes = *spc.meshes;
1447
1448 bool any_need_reload = false;
1449 bool any_need_upload = false;
1450 if constexpr(avnd::static_geometry_type<Field> || avnd::dynamic_geometry_type<Field>)
1451 {
1452 SCORE_ASSERT(ossia_meshes.meshes.size() == 1);
1453 auto [need_reload, need_upload]
1454 = update_geometry(ctrl, ossia_meshes.meshes[0]);
1455 any_need_reload = need_reload;
1456 any_need_upload = need_upload;
1457 }
1458 else if constexpr(
1459 avnd::static_geometry_type<decltype(Field::mesh)>
1460 || avnd::dynamic_geometry_type<decltype(Field::mesh)>)
1461 {
1462 SCORE_ASSERT(ossia_meshes.meshes.size() == 1);
1463 auto [need_reload, need_upload]
1464 = update_geometry(ctrl.mesh, ossia_meshes.meshes[0]);
1465 any_need_reload = need_reload;
1466 any_need_upload = need_upload;
1467 }
1468 else
1469 {
1470 auto [need_reload, need_upload] = update_geometry(ctrl, ossia_meshes);
1471 any_need_reload = need_reload;
1472 any_need_upload = need_upload;
1473 }
1474
1475 if(any_need_reload)
1476 {
1477 reload_mesh(ctrl, spc);
1478 }
1479 }
1480 }
1481 ctrl.dirty_mesh = false;
1482 }
1483
1484 // 2. Push to next node
1485 // FIXME this should be for the renderer of edge, not the node, since
1486 // geometries can have gpu buffers
1487 auto rendered_node = pnode->renderedNodes.find(&renderer);
1488 SCORE_ASSERT(rendered_node != pnode->renderedNodes.end());
1489
1490 auto it = std::find(
1491 edge_sink->node->input.begin(), edge_sink->node->input.end(), edge_sink);
1492 SCORE_ASSERT(it != edge_sink->node->input.end());
1493 int n = it - edge_sink->node->input.begin();
1494
1495 rendered_node->second->process(n, spc);
1496
1497 // 3. Same for transform3d
1498
1499 if constexpr(requires { ctrl.transform; })
1500 {
1501 if(ctrl.dirty_transform)
1502 {
1503 ossia::transform3d transform;
1504 std::copy_n(ctrl.transform, std::ssize(ctrl.transform), transform.matrix);
1505 ctrl.dirty_transform = false;
1506
1507 rendered_node->second->process(n, transform);
1508 if(auto pnode = dynamic_cast<score::gfx::ProcessNode*>(edge_sink->node))
1509 pnode->process(n, transform);
1510 }
1511 }
1512 }
1513 }
1514
1515 void upload(score::gfx::RenderList& renderer, auto& state, score::gfx::Edge& edge)
1516 {
1517 // FIXME we need something such as port_run_{pre,post}process for GPU nodes
1518 avnd::geometry_output_introspection<T>::for_all_n(
1519 avnd::get_outputs(state),
1520 [&](auto& field, auto pred) { this->upload(renderer, field, edge, pred); });
1521 }
1522};
1523
1524
1525template<typename T>
1526 requires (avnd::geometry_output_introspection<T>::size == 0)
1527struct geometry_outputs_storage<T>
1528{
1529 static void upload(auto&&...)
1530 {
1531
1532 }
1533};
1534}
1535
1536#endif
Root data model for visual nodes.
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:74
std::vector< Port * > input
Input ports of that node.
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:103
virtual void process(Message &&msg)
Process a message from the execution engine.
Definition Node.cpp:25
ossia::small_pod_vector< Port *, 1 > output
Output ports of that node.
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:109
Common base class for most single-pass, simple nodes.
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:203
Renderer for a given node.
Definition NodeRenderer.hpp:11
Base class for sink nodes (QWindow, spout, syphon, NDI output, ...)
Definition OutputNode.hpp:31
Common base class for nodes that map to score processes.
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:176
List of nodes to be rendered to an output.
Definition RenderList.hpp:19
bool requiresDepth(score::gfx::Port &p) const noexcept
Whether this list of rendering actions requires depth testing at all.
Definition RenderList.cpp:343
const score::gfx::Mesh & defaultTriangle() const noexcept
A triangle mesh correct for this API.
Definition RenderList.cpp:382
RenderState & state
RenderState corresponding to this RenderList.
Definition RenderList.hpp:94
QRhiTexture & emptyTexture() const noexcept
Texture to use when a texture is missing.
Definition RenderList.hpp:117
TreeNode< DeviceExplorerNode > Node
Definition DeviceNode.hpp:74
Definition Factories.hpp:19
void uploadStaticBufferWithStoredData(QRhiResourceUpdateBatch *ub, QRhiBuffer *buf, int offset, int64_t bytesize, const char *data)
Schedule a Static buffer update when we can guarantee the buffer outlives the frame.
Definition score-plugin-gfx/Gfx/Graph/Utils.hpp:338
std::pair< QShader, QShader > makeShaders(const RenderState &v, QString vert, QString frag)
Get a pair of compiled vertex / fragment shaders from GLSL 4.5 sources.
Definition score-plugin-gfx/Gfx/Graph/Utils.cpp:394
TextureRenderTarget createRenderTarget(const RenderState &state, QRhiTexture *tex, int samples, bool depth)
Create a render target from a texture.
Definition score-plugin-gfx/Gfx/Graph/Utils.cpp:10
Base toolkit upon which the software is built.
Definition Application.cpp:111
STL namespace.
Definition DocumentContext.hpp:18
Definition Mesh.hpp:15
Connection between two score::gfx::Port.
Definition score-plugin-gfx/Gfx/Graph/Utils.hpp:71
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:49
Definition OutputNode.hpp:11
Port of a score::gfx::Node.
Definition score-plugin-gfx/Gfx/Graph/Utils.hpp:53
Definition score-plugin-gfx/Gfx/Graph/Node.hpp:56
Stores a sampler and the texture currently associated with it.
Definition score-plugin-gfx/Gfx/Graph/Utils.hpp:26
Useful abstraction for storing all the data related to a render target.
Definition score-plugin-gfx/Gfx/Graph/Utils.hpp:116