BLI_memory_counter.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2024 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once

 
#include "BLI_function_ref.hh"
#include "BLI_implicit_sharing_ptr.hh"
#include "BLI_memory_counter_fwd.hh"
#include "BLI_set.hh"
 
namespace blender::memory_counter {
 
class MemoryCount {
 public:
  int64_t total_bytes = 0;
 
  Set<WeakImplicitSharingPtr> handled_shared_data;
 
  void reset();
};

class MemoryCounter : NonCopyable, NonMovable {
 private:
  MemoryCount &count_;
 
 public:
  MemoryCounter(MemoryCount &count);

  void add(const int64_t bytes);

  void add_shared(const ImplicitSharingInfo *sharing_info,
                  const FunctionRef<void(MemoryCounter &shared_memory)> count_fn);

  void add_shared(const ImplicitSharingInfo *sharing_info, const int64_t bytes);
};
 
}  // namespace blender::memory_counter