50 lines
1.3 KiB
C++
50 lines
1.3 KiB
C++
/* SPDX-FileCopyrightText: 2024 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup bli
|
|
*/
|
|
|
|
#include "BLI_memory_counter.hh"
|
|
|
|
namespace blender::memory_counter {
|
|
|
|
MemoryCounter::MemoryCounter(MemoryCount &count) : count_(count) {}
|
|
|
|
void MemoryCounter::add(const int64_t bytes)
|
|
{
|
|
count_.total_bytes += bytes;
|
|
}
|
|
|
|
void MemoryCounter::add_shared(const ImplicitSharingInfo *sharing_info,
|
|
const FunctionRef<void(MemoryCounter &memory)> count_fn)
|
|
{
|
|
if (!sharing_info) {
|
|
/* Data is not actually shared. */
|
|
count_fn(*this);
|
|
return;
|
|
}
|
|
if (!count_.handled_shared_data.add_as(sharing_info)) {
|
|
/* Data was counted before, avoid counting it again. */
|
|
return;
|
|
}
|
|
sharing_info->add_weak_user();
|
|
/* Count into the `this` for now. In the future we could pass in a separate #MemoryCounter here
|
|
* if we needed to know the amount of memory used by each shared data. */
|
|
count_fn(*this);
|
|
}
|
|
|
|
void MemoryCounter::add_shared(const ImplicitSharingInfo *sharing_info, const int64_t bytes)
|
|
{
|
|
this->add_shared(sharing_info, [&](MemoryCounter &shared_memory) { shared_memory.add(bytes); });
|
|
}
|
|
|
|
void MemoryCount::reset()
|
|
{
|
|
std::destroy_at(this);
|
|
new (this) MemoryCount();
|
|
}
|
|
|
|
} // namespace blender::memory_counter
|