mirror of https://github.com/wolfpld/tracy.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.6 KiB
C++
91 lines
2.6 KiB
C++
#pragma once
|
|
|
|
#include <pybind11/pybind11.h>
|
|
namespace py = pybind11;
|
|
|
|
#include "NameBuffer.hpp"
|
|
#include "tracy/Tracy.hpp"
|
|
|
|
using OptionalString = std::optional<std::string>;
|
|
using OptionalInt = std::optional<int>;
|
|
|
|
#ifdef TRACY_ENABLE
|
|
template <typename Type = uint64_t>
|
|
OptionalNumber MemoryAllocate(const Type &type, std::size_t size,
|
|
const OptionalString &name = std::nullopt,
|
|
const OptionalNumber &id = std::nullopt,
|
|
OptionalInt depth = std::nullopt) {
|
|
if (!name && !id) {
|
|
if (!depth)
|
|
TracyAlloc(reinterpret_cast<void *>(type), size);
|
|
else
|
|
TracyAllocS(reinterpret_cast<void *>(type), size, *depth);
|
|
return std::nullopt;
|
|
}
|
|
|
|
BufferEntry entry;
|
|
|
|
if (id) {
|
|
entry.second = NameBuffer::Get(*id);
|
|
if (!entry.second) return std::nullopt;
|
|
} else {
|
|
entry = NameBuffer::Add(*name);
|
|
if (!entry.first) return std::nullopt;
|
|
}
|
|
|
|
if (!depth)
|
|
TracyAllocN(reinterpret_cast<void *>(type), size, entry.second);
|
|
else
|
|
TracyAllocNS(reinterpret_cast<void *>(type), size, *depth, entry.second);
|
|
return entry.first;
|
|
}
|
|
|
|
template <typename Type = uint64_t>
|
|
bool MemoryFree(const Type &type, const OptionalNumber &id = std::nullopt,
|
|
OptionalInt depth = std::nullopt) {
|
|
if (!id) {
|
|
if (!depth)
|
|
TracyFree(reinterpret_cast<void *>(type));
|
|
else
|
|
TracyFreeS(reinterpret_cast<void *>(type), *depth);
|
|
return true;
|
|
}
|
|
|
|
auto ptr = NameBuffer::Get(*id);
|
|
if (!ptr) return false;
|
|
|
|
if (!depth)
|
|
TracyFreeN(reinterpret_cast<void *>(type), ptr);
|
|
else
|
|
TracyFreeNS(reinterpret_cast<void *>(type), *depth, ptr);
|
|
return true;
|
|
}
|
|
#else
|
|
|
|
template <typename Type = uint64_t>
|
|
OptionalNumber MemoryAllocate(const Type &, std::size_t, const OptionalString &,
|
|
const OptionalNumber &, OptionalInt) {
|
|
return 0ul;
|
|
}
|
|
|
|
template <typename Type = uint64_t>
|
|
bool MemoryFree(const Type &, const OptionalNumber &, OptionalInt) {
|
|
return true;
|
|
}
|
|
#endif
|
|
|
|
template <>
|
|
OptionalNumber MemoryAllocate(const py::object &object, std::size_t size,
|
|
const OptionalString &name,
|
|
const OptionalNumber &id, OptionalInt depth) {
|
|
return MemoryAllocate<uint64_t>(reinterpret_cast<uint64_t>(object.ptr()),
|
|
size, name, id, depth);
|
|
}
|
|
|
|
template <>
|
|
bool MemoryFree(const py::object &object, const OptionalNumber &id,
|
|
OptionalInt depth) {
|
|
return MemoryFree<uint64_t>(reinterpret_cast<uint64_t>(object.ptr()), id,
|
|
depth);
|
|
}
|