engine.h

// SPDX-FileCopyrightText: © 2018-2025 Alexandros Theodotou <alex@zrythm.org>
// SPDX-License-Identifier: LicenseRef-ZrythmLicense
 
#pragma once
 
#include "dsp/audio_callback.h"
#include "dsp/audio_port.h"
#include "dsp/graph_dispatcher.h"
#include "dsp/midi_panic_processor.h"
#include "dsp/midi_port.h"
#include "dsp/transport.h"
#include "utils/concurrency.h"
#include "utils/types.h"
 
namespace zrythm::dsp
{

class AudioEngine : public QObject
{
  Q_OBJECT
  Q_PROPERTY (int sampleRate READ sampleRate NOTIFY sampleRateChanged)
  QML_ELEMENT
  QML_UNCREATABLE ("")
 
public:
  enum class State : std::uint8_t
  {
    Uninitialized,
    Initialized,
    Active,
  };
 
public:
  AudioEngine (
    dsp::Transport                           &transport,
    std::shared_ptr<juce::AudioDeviceManager> device_mgr,
    dsp::DspGraphDispatcher                  &graph_dispatcher,
    QObject *                                 parent = nullptr);

  ~AudioEngine () override;
 
  // =========================================================
  // QML interface
  // =========================================================
 
  Q_INVOKABLE int xRunCount () const { return load_measurer_.getXRunCount (); }
  Q_INVOKABLE double loadPercentage () const
  {
    return load_measurer_.getLoadAsPercentage ();
  }
 
  int sampleRate () const { return get_sample_rate ().in (units::sample_rate); }
 
  Q_SIGNAL void sampleRateChanged (int sampleRate);
 
  // =========================================================

  void wait_for_pause (EngineState &state, bool force_pause, bool with_fadeout);
 
  void resume (const EngineState &state);
 
  void activate ();
  void deactivate ();

  [[gnu::hot]] bool process_prepare (
    dsp::Transport::TransportSnapshot               &transport_snapshot,
    nframes_t                                        nframes,
    SemaphoreRAII<moodycamel::LightweightSemaphore> &sem) noexcept
    [[clang::nonblocking]];
 
  enum class ProcessReturnStatus : std::uint8_t
  {
    // Process completed normally
    ProcessCompleted,
    // Process skipped (e.g., when recalculating the graph)
    ProcessSkipped,
    // Process failed for some reason
    ProcessFailed,
  };

  [[gnu::hot]] auto process (
    const dsp::PlayheadProcessingGuard &playhead_guard,
    nframes_t total_frames_to_process) noexcept [[clang::nonblocking]]
  -> ProcessReturnStatus;

  [[gnu::hot]] void advance_playhead_after_processing (
    dsp::Transport::TransportSnapshot  &transport_snapshot,
    const dsp::PlayheadProcessingGuard &playhead_guard,
    nframes_t                           roll_nframes,
    nframes_t nframes) noexcept [[clang::nonblocking]];
 
  auto &get_monitor_out_port () { return monitor_out_; }
 
  auto * midi_panic_processor () const { return midi_panic_processor_.get (); }

  void panic_all ();
 
  auto get_device_manager () const { return device_manager_; }
 
  bool  activated () const { return state_ == State::Active; }
  bool  running () const { return run_.load (); }
  void  set_running (bool run) { run_.store (run); }
  auto &graph_dispatcher () { return graph_dispatcher_; }
 
  bool exporting () const { return exporting_; }
  void set_exporting (bool exporting) { exporting_.store (exporting); }
 
  auto get_processing_lock () [[clang::blocking]]
  {
    return SemaphoreRAII (process_lock_, true);
  }
 
  nframes_t get_block_length () const
  {
    auto * dev = device_manager_->getCurrentAudioDevice ();
    assert (dev != nullptr);
    return dev->getCurrentBufferSizeSamples ();
  }
 
  units::sample_rate_t get_sample_rate () const
  {
    auto * dev = device_manager_->getCurrentAudioDevice ();
    assert (dev != nullptr);
    return units::sample_rate (static_cast<int> (dev->getCurrentSampleRate ()));
  }

  void execute_function_with_paused_processing_synchronously (
    const std::function<void ()> &func,
    bool                          recalculate_graph);
 
private:
  void activate_impl (bool activate);
 
private:
  dsp::PortRegistry               port_registry_;
  dsp::ProcessorParameterRegistry param_registry_{ port_registry_ };
 
  dsp::Transport &transport_;

  dsp::DspGraphDispatcher &graph_dispatcher_;
 
  std::shared_ptr<juce::AudioDeviceManager> device_manager_;

  std::atomic_uint64_t cycle_{ 0 };

  dsp::AudioPort monitor_out_;

  std::unique_ptr<dsp::MidiPort> midi_in_;

  moodycamel::LightweightSemaphore process_lock_{ 1 };

  std::atomic_bool run_{ false };

  std::atomic_bool exporting_{ false };
 
  juce::AudioProcessLoadMeasurer load_measurer_;

  nframes_t remaining_latency_preroll_{};

  std::atomic_bool capture_cc_{ false };

  std::array<midi_byte_t, 3> last_cc_captured_{};
 
  std::atomic<State> state_{ State::Uninitialized };
  static_assert (decltype (state_)::is_always_lock_free);
 
  utils::QObjectUniquePtr<dsp::MidiPanicProcessor> midi_panic_processor_;
 
  std::unique_ptr<AudioCallback> audio_callback_;

  bool callback_running_{};
};
}