audio_pool.h

// SPDX-FileCopyrightText: © 2019-2026 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<std::filesystem::path (bool backup)>;
  using SampleRateGetter = std::function<units::sample_rate_t ()>;
 
  AudioPool (
    utils::IObjectRegistry &registry,
    ProjectPoolPathGetter   path_getter,
    SampleRateGetter        sr_getter);
 
public:
  void init_loaded ();

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

  [[nodiscard]] std::filesystem::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);
 
  void
  for_each_clip (std::function<void (dsp::FileAudioSource &)> visitor) const;
 
private:
  friend void init_from (
    AudioPool             &obj,
    const AudioPool       &other,
    utils::ObjectCloneType clone_type);
 
  friend void to_json (nlohmann::json &j, const AudioPool &pool);
  friend void from_json (const nlohmann::json &j, AudioPool &pool);
 
private:
  SampleRateGetter      sample_rate_getter_;
  ProjectPoolPathGetter project_pool_path_getter_;

  utils::IObjectRegistry &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";
    pool.for_each_clip ([&] (const auto &clip) {
      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);
  }
};