/* SPDX-License-Identifier: GPL-2.0-or-later */

#include <memory>

#include "BLI_math_vec_types.hh"

#include "COM_morphological_distance_feather_weights.hh"
#include "COM_symmetric_blur_weights.hh"
#include "COM_symmetric_separable_blur_weights.hh"

#include "COM_static_cache_manager.hh"

namespace blender::realtime_compositor {

/* --------------------------------------------------------------------
 * Static Cache Manager.
 */

void StaticCacheManager::reset()
{
  /* First, delete all resources that are no longer needed. */
  symmetric_blur_weights_.remove_if([](auto item) { return !item.value->needed; });
  symmetric_separable_blur_weights_.remove_if([](auto item) { return !item.value->needed; });
  morphological_distance_feather_weights_.remove_if([](auto item) { return !item.value->needed; });

  /* Second, reset the needed status of the remaining resources to false to ready them to track
   * their needed status for the next evaluation. */
  for (auto &value : symmetric_blur_weights_.values()) {
    value->needed = false;
  }
  for (auto &value : symmetric_separable_blur_weights_.values()) {
    value->needed = false;
  }
  for (auto &value : morphological_distance_feather_weights_.values()) {
    value->needed = false;
  }
}

SymmetricBlurWeights &StaticCacheManager::get_symmetric_blur_weights(int type, float2 radius)
{
  const SymmetricBlurWeightsKey key(type, radius);

  auto &weights = *symmetric_blur_weights_.lookup_or_add_cb(
      key, [&]() { return std::make_unique<SymmetricBlurWeights>(type, radius); });

  weights.needed = true;
  return weights;
}

SymmetricSeparableBlurWeights &StaticCacheManager::get_symmetric_separable_blur_weights(
    int type, float radius)
{
  const SymmetricSeparableBlurWeightsKey key(type, radius);

  auto &weights = *symmetric_separable_blur_weights_.lookup_or_add_cb(
      key, [&]() { return std::make_unique<SymmetricSeparableBlurWeights>(type, radius); });

  weights.needed = true;
  return weights;
}

MorphologicalDistanceFeatherWeights &StaticCacheManager::
    get_morphological_distance_feather_weights(int type, int radius)
{
  const MorphologicalDistanceFeatherWeightsKey key(type, radius);

  auto &weights = *morphological_distance_feather_weights_.lookup_or_add_cb(
      key, [&]() { return std::make_unique<MorphologicalDistanceFeatherWeights>(type, radius); });

  weights.needed = true;
  return weights;
}

}  // namespace blender::realtime_compositor