audio_pool.h

// SPDX-FileCopyrightText: © 2019-2025 Alexandros Theodotou <alex@zrythm.org>
// SPDX-License-Identifier: LicenseRef-ZrythmLicense
 
#pragma once
 
#include "dsp/file_audio_source.h"
#include "utils/hash.h"
#include "utils/units.h"
 
#include <boost/unordered/concurrent_flat_map.hpp>
 
namespace zrythm::dsp
{

struct AudioPool final
{
public:
  using ProjectPoolPathGetter = std::function<fs::path (bool backup)>;
  using SampleRateGetter = std::function<units::sample_rate_t ()>;
 
  AudioPool (
    dsp::FileAudioSourceRegistry &file_audio_source_registry,
    ProjectPoolPathGetter         path_getter,
    SampleRateGetter              sr_getter);
 
public:
  void init_loaded ();

  auto duplicate_clip (const FileAudioSource::Uuid &clip_id, bool write_file)
    -> FileAudioSourceUuidReference;

  [[nodiscard]] fs::path
  get_clip_path (const dsp::FileAudioSource::Uuid &id, bool is_backup) const;

  void write_clip (const FileAudioSource * clip, bool parts, bool backup);

  void remove_unused (bool backup);

  void reload_clip_frame_bufs ();

  void write_to_disk (bool is_backup);
 
#if 0
  std::unique_ptr<juce::PositionableAudioSource>
  generate_audio_source (const FileAudioSource::Uuid &id) const;
#endif
 
  auto get_clip_ptrs () const
  {
    return std::views::transform (
      clip_registry_.get_hash_map () | std::views::values,
      [] (const auto &clip) { return std::get<dsp::FileAudioSource *> (clip); });
  }
 
private:
  friend void init_from (
    AudioPool             &obj,
    const AudioPool       &other,
    utils::ObjectCloneType clone_type);
 
  static constexpr auto kClipsKey = "clips"sv;
  static constexpr auto kLastKnownFileHashesKey = "lastKnownFileHashes"sv;
  friend void           to_json (nlohmann::json &j, const AudioPool &pool)
  {
    j[AudioPool::kLastKnownFileHashesKey] = pool.last_known_file_hashes_;
  }
  friend void from_json (const nlohmann::json &j, AudioPool &pool)
  {
    j.at (AudioPool::kLastKnownFileHashesKey)
      .get_to (pool.last_known_file_hashes_);
  }
 
private:
  SampleRateGetter      sample_rate_getter_;
  ProjectPoolPathGetter project_pool_path_getter_;

  FileAudioSourceRegistry &clip_registry_;

  boost::unordered::concurrent_flat_map<FileAudioSource::Uuid, utils::hash::HashT>
    last_known_file_hashes_;
};
} // namespace zrythm::dsp
 
// Formatter for AudioPool
template <>
struct fmt::formatter<zrythm::dsp::AudioPool> : fmt::formatter<std::string_view>
{
  template <typename FormatContext>
  auto format (const zrythm::dsp::AudioPool &pool, FormatContext &ctx) const
  {
    std::stringstream ss;
    ss << "\nAudio Pool:\n";
    for (const auto &clip : pool.get_clip_ptrs ())
      {
        auto pool_path = pool.get_clip_path (clip->get_uuid (), false);
        ss << fmt::format (
          "[Clip {}] {}: {}\n", clip->get_uuid (), clip->get_name (), pool_path);
      }
 
    return fmt::formatter<std::string_view>::format (
      fmt::format ("{}", ss.str ()), ctx);
  }
};