IntervalExecutionHelpers.hpp

#pragma once
#include <Process/Dataflow/Port.hpp>
#include <Process/ExecutionContext.hpp>
#include <Process/ExecutionSetup.hpp>
#include <Process/ExecutionTransaction.hpp>
#include <Process/Process.hpp>
 
#include <Curve/CurveConversion.hpp>
 
#include <Scenario/Document/Interval/IntervalModel.hpp>
#include <Scenario/Document/ScenarioDocument/ScenarioDocumentModel.hpp>
#include <Scenario/Document/Tempo/TempoProcess.hpp>
 
#include <ossia/dataflow/graph/graph_interface.hpp>
#include <ossia/dataflow/graph_edge.hpp>
#include <ossia/dataflow/graph_node.hpp>
#include <ossia/dataflow/port.hpp>
#include <ossia/detail/pod_vector.hpp>
#include <ossia/editor/curve/curve.hpp>
#include <ossia/editor/scenario/time_interval.hpp>
#include <ossia/editor/scenario/time_process.hpp>
 
namespace Execution
{
 
inline std::pair<std::optional<ossia::tempo_curve>, Scenario::TempoProcess*>
tempoCurve(const Scenario::IntervalModel& itv, const Execution::Context& ctx)
{
  OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
  // TODO
  if(auto proc = itv.tempoCurve())
  {
    auto& curve = proc->curve();
    // TODO recompute whenever tempo changes
    const auto defaultdur = itv.duration.defaultDuration().msec();
    auto scale_x = [&ctx, defaultdur](double val) -> int64_t {
      return ctx.time(TimeVal::fromMsecs(val * defaultdur)).impl;
    };
    auto scale_y = [=](double val) -> double {
      using namespace Scenario;
      return val * (TempoProcess::max - TempoProcess::min) + TempoProcess::min;
    };
 
    ossia::tempo_curve t;
 
    auto segt_data = curve.sortedSegments();
    if(segt_data.size() != 0)
    {
      t = std::move(*Engine::score_to_ossia::curve<int64_t, double>(
          scale_x, scale_y, segt_data, {}));
    }
 
    return {std::optional<ossia::tempo_curve>{std::move(t)}, proc};
  }
  else
  {
    return {};
  }
}
inline ossia::time_signature_map
timeSignatureMap(const Scenario::IntervalModel& itv, const Execution::Context& ctx)
{
  OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
  ossia::time_signature_map ret;
  for(const auto& [time, sig] : itv.timeSignatureMap())
  {
    ret[ctx.time(time)] = sig;
  }
  return ret;
}
 
inline auto propagatedOutlets(const Process::Outlets& outlets) noexcept
{
  OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
  ossia::pod_vector<std::size_t> propagated_outlets;
  for(std::size_t i = 0; i < outlets.size(); i++)
  {
    if(auto o = qobject_cast<Process::AudioOutlet*>(outlets[i]))
      if(o->propagate())
        propagated_outlets.push_back(i);
  }
  return propagated_outlets;
}
 
inline void connectPropagated(
    const ossia::node_ptr& process_node, const ossia::node_ptr& interval_node,
    ossia::graph_interface& g,
    const ossia::pod_vector<std::size_t>& propagated_outlets) noexcept
{
  OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Audio);
  const auto& outs = process_node->root_outputs();
  for(std::size_t propagated : propagated_outlets)
  {
    if(propagated >= outs.size())
      continue;
 
    if(outs[propagated]->which() == ossia::audio_port::which)
    {
      auto cable = g.allocate_edge(
          ossia::immediate_glutton_connection{}, outs[propagated],
          interval_node->root_inputs()[0], process_node, interval_node);
      g.connect(cable);
    }
  }
}
 
inline void updatePropagated(
    const ossia::node_ptr& process_node, const ossia::node_ptr& interval_node,
    ossia::graph_interface& g, std::size_t port_idx, bool is_propagated) noexcept
{
  OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Audio);
  const auto& outs = process_node->root_outputs();
 
  if(port_idx >= outs.size())
    return;
 
  const ossia::outlet& outlet = *outs[port_idx];
 
  if(!outlet.target<ossia::audio_port>())
    return;
 
  // Remove cables if depropagated, add cables if repropagated
  if(is_propagated)
  {
    for(const ossia::graph_edge* edge : outlet.targets)
    {
      if(edge->in_node == interval_node)
        return;
    }
 
    auto cable = g.allocate_edge(
        ossia::immediate_glutton_connection{}, outs[port_idx],
        interval_node->root_inputs()[0], process_node, interval_node);
    g.connect(cable);
  }
  else
  {
    for(ossia::graph_edge* edge : outlet.targets)
    {
      if(edge->in_node == interval_node)
      {
        g.disconnect(edge);
        return;
      }
    }
  }
}
 
struct AddProcess
{
  // Note : this looks a bit wtf, it is to handle RawPtrExecution
  std::shared_ptr<ossia::time_interval> cst;
  ossia::time_interval* cst_ptr{};
  std::weak_ptr<ossia::time_process> oproc_weak;
  std::weak_ptr<ossia::graph_interface> g_weak;
  ossia::pod_vector<std::size_t> propagated_outlets;
 
  void operator()() const noexcept
  {
    OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Audio);
    auto oproc = oproc_weak.lock();
    if(!oproc)
      return;
 
    auto g = g_weak.lock();
    if(!g)
      return;
 
    cst_ptr->add_time_process(oproc);
    if(!oproc->node)
      return;
 
    connectPropagated(oproc->node, cst_ptr->node, *g, propagated_outlets);
  }
};
 
struct RecomputePropagate
{
  const Execution::Context& system;
  Process::ProcessModel& proc;
  std::weak_ptr<ossia::graph_node> cst_node_weak;
  std::weak_ptr<ossia::time_process> oproc_weak;
  std::weak_ptr<ossia::graph_interface> g_weak;
  Process::Outlet* outlet{};
 
  void operator()(bool propagate) const noexcept
  {
    OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
 
    // TODO find a better way !
    auto port_index
        = std::distance(proc.outlets().begin(), ossia::find(proc.outlets(), outlet));
 
    system.executionQueue.enqueue(
        [cst_node_weak = this->cst_node_weak, g_weak = this->g_weak,
         oproc_weak = this->oproc_weak, port_index, propagate] {
      const auto g = g_weak.lock();
      if(!g)
        return;
 
      const auto cst_node = cst_node_weak.lock();
      if(!cst_node)
        return;
 
      const auto oproc = oproc_weak.lock();
      if(!oproc)
        return;
 
      const auto& proc_node = oproc->node;
      if(!oproc->node)
        return;
 
      updatePropagated(proc_node, cst_node, *g, port_index, propagate);
    });
  }
};
 
template <typename T>
struct ReconnectOutlets
{
  T& component;
  std::weak_ptr<ossia::graph_node> fw_node;
 
  Process::ProcessModel& proc;
  std::weak_ptr<ossia::time_process> oproc_weak;
 
  std::weak_ptr<ossia::graph_interface> g_weak;
 
  void operator()() const noexcept
  {
    OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
    for(Process::Outlet* outlet : proc.outlets())
    {
      if(auto o = qobject_cast<Process::AudioOutlet*>(outlet))
      {
        QObject::disconnect(
            o, &Process::AudioOutlet::propagateChanged, &component, nullptr);
        QObject::connect(
            o, &Process::AudioOutlet::propagateChanged, &component,
            RecomputePropagate{
                component.system(), proc, fw_node, oproc_weak, g_weak, outlet});
      }
    }
  }
};
 
struct HandlePortsReplaced
{
  std::weak_ptr<ossia::graph_node> cst_node_weak;
  std::weak_ptr<ossia::time_process> oproc_weak;
  std::weak_ptr<ossia::graph_interface> g_weak;
  Process::ProcessModel& proc;
 
  void operator()(Execution::Transaction* commands) const noexcept
  {
    OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
    commands->push_back([cst_node_weak = this->cst_node_weak,
                         oproc_weak = this->oproc_weak, g_weak = this->g_weak,
                         propagated = propagatedOutlets(proc.outlets())] {
      OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Audio);
      auto cst_node = cst_node_weak.lock();
      if(!cst_node)
        return;
      auto g = g_weak.lock();
      if(!g)
        return;
      auto oproc = oproc_weak.lock();
      if(!oproc || !oproc->node)
        return;
 
      const std::size_t n = oproc->node->root_outputs().size();
      for(std::size_t i = 0; i < n; i++)
        updatePropagated(oproc->node, cst_node, *g, i, ossia::contains(propagated, i));
    });
  }
};
 
struct HandleNodeChange
{
  std::weak_ptr<ossia::graph_node> cst_node_weak;
  std::weak_ptr<ossia::time_process> oproc_weak;
  std::weak_ptr<ossia::time_interval> itv_weak;
  std::weak_ptr<ossia::graph_interface> g_weak;
  Process::ProcessModel& proc;
 
  void operator()(
      const ossia::node_ptr& old_node, const ossia::node_ptr& new_node,
      Execution::Transaction* commands) const noexcept
  {
    OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Ui);
 
    commands->push_back([cst_node_weak = this->cst_node_weak,
                         oproc_weak = this->oproc_weak,
                         itv_weak = this->itv_weak, g_weak = this->g_weak,
                         propagated = propagatedOutlets(proc.outlets()), old_node,
                         new_node] {
      OSSIA_ENSURE_CURRENT_THREAD_KIND(ossia::thread_type::Audio);
      auto cst_node = cst_node_weak.lock();
      if(!cst_node)
        return;
      auto g = g_weak.lock();
      if(!g)
        return;
 
      // Re-seek the resampler to the current model time so a file swap
      // on a running interval lands in phase.
      if(auto itv = itv_weak.lock())
      {
        if(auto oproc = oproc_weak.lock())
        {
          const auto date = itv->get_date();
          if(date.impl != 0)
          {
            oproc->transport(date, itv->current_transport_info());
          }
        }
      }
 
      // In-place reload: skip the rewire to avoid a momentary audio break.
      if(old_node == new_node)
        return;
 
      // Remove propagate edges from the old node
      if(old_node)
      {
        ossia::graph_node& n = *old_node;
        for(auto& outlet : n.root_outputs())
        {
          auto targets = outlet->targets;
          for(auto e : targets)
          {
            if(e->in_node.get() == cst_node.get())
            {
              g->disconnect(e);
            }
          }
        }
      }
 
      // Add edges to the new node
      if(new_node)
      {
        connectPropagated(new_node, cst_node, *g, propagated);
      }
    });
  }
};
 
}