mirror of
https://github.com/yse/easy_profiler.git
synced 2024-12-26 16:11:02 +08:00
892 lines
27 KiB
C++
892 lines
27 KiB
C++
/**
|
|
Lightweight profiler library for c++
|
|
Copyright(C) 2016-2017 Sergey Yagovtsev, Victor Zarubkin
|
|
|
|
Licensed under either of
|
|
* MIT license (LICENSE.MIT or http://opensource.org/licenses/MIT)
|
|
* Apache License, Version 2.0, (LICENSE.APACHE or http://www.apache.org/licenses/LICENSE-2.0)
|
|
at your option.
|
|
|
|
The MIT License
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
|
of the Software, and to permit persons to whom the Software is furnished
|
|
to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
|
|
INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
|
|
PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
|
USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
|
|
|
|
The Apache License, Version 2.0 (the "License");
|
|
You may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
|
|
**/
|
|
|
|
#ifndef EASY_PROFILER_MANAGER_H
|
|
#define EASY_PROFILER_MANAGER_H
|
|
|
|
#include <easy/profiler.h>
|
|
#include <easy/easy_socket.h>
|
|
|
|
#include "spin_lock.h"
|
|
#include "outstream.h"
|
|
#include "hashed_cstr.h"
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
#include <unordered_map>
|
|
#include <thread>
|
|
#include <atomic>
|
|
#include <list>
|
|
#include <type_traits>
|
|
#include <cstring>
|
|
#include <cstddef>
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifdef _WIN32
|
|
#include <Windows.h>
|
|
#elif defined(__APPLE__)
|
|
#include <pthread.h>
|
|
#include <Availability.h>
|
|
#else
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <sys/syscall.h>
|
|
#include <chrono>
|
|
#include <time.h>
|
|
#include <malloc.h>
|
|
#endif
|
|
|
|
#ifdef max
|
|
#undef max
|
|
#endif
|
|
|
|
inline profiler::thread_id_t getCurrentThreadId()
|
|
{
|
|
#ifdef _WIN32
|
|
return (profiler::thread_id_t)::GetCurrentThreadId();
|
|
#elif defined(__APPLE__)
|
|
# if (defined(__MAC_OS_X_VERSION_MIN_REQUIRED) && __MAC_OS_X_VERSION_MIN_REQUIRED >= __MAC_10_6) || \
|
|
(defined(__IPHONE_OS_VERSION_MIN_REQUIRED) && __IPHONE_OS_VERSION_MIN_REQUIRED >= __IPHONE_8_0)
|
|
EASY_THREAD_LOCAL static uint64_t _id = 0;
|
|
if (!_id)
|
|
pthread_threadid_np(NULL, &_id);
|
|
return (profiler::thread_id_t)_id;
|
|
# else
|
|
return (profiler::thread_id_t)pthread_self();
|
|
# endif
|
|
#else
|
|
EASY_THREAD_LOCAL static const profiler::thread_id_t _id = (profiler::thread_id_t)syscall(__NR_gettid);
|
|
return _id;
|
|
#endif
|
|
}
|
|
|
|
namespace profiler {
|
|
|
|
class SerializedBlock;
|
|
|
|
struct do_not_calc_hash {
|
|
template <class T> inline size_t operator()(T _value) const {
|
|
return static_cast<size_t>(_value);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef EASY_ENABLE_BLOCK_STATUS
|
|
# define EASY_ENABLE_BLOCK_STATUS 1
|
|
#endif
|
|
|
|
#ifndef EASY_ENABLE_ALIGNMENT
|
|
# define EASY_ENABLE_ALIGNMENT 0
|
|
#endif
|
|
|
|
#ifndef EASY_ALIGNMENT_SIZE
|
|
# define EASY_ALIGNMENT_SIZE alignof(std::max_align_t)
|
|
#endif
|
|
|
|
#if EASY_ENABLE_ALIGNMENT == 0
|
|
# define EASY_ALIGNED(TYPE, VAR, A) TYPE VAR
|
|
# define EASY_MALLOC(MEMSIZE, A) malloc(MEMSIZE)
|
|
# define EASY_FREE(MEMPTR) free(MEMPTR)
|
|
#else
|
|
# if defined(_MSC_VER)
|
|
# define EASY_ALIGNED(TYPE, VAR, A) __declspec(align(A)) TYPE VAR
|
|
# define EASY_MALLOC(MEMSIZE, A) _aligned_malloc(MEMSIZE, A)
|
|
# define EASY_FREE(MEMPTR) _aligned_free(MEMPTR)
|
|
# elif defined(__GNUC__)
|
|
# define EASY_ALIGNED(TYPE, VAR, A) TYPE VAR __attribute__((aligned(A)))
|
|
# define EASY_MALLOC(MEMSIZE, A) memalign(A, MEMSIZE)
|
|
# define EASY_FREE(MEMPTR) free(MEMPTR)
|
|
# else
|
|
# define EASY_ALIGNED(TYPE, VAR, A) TYPE VAR
|
|
# define EASY_MALLOC(MEMSIZE, A) malloc(MEMSIZE)
|
|
# define EASY_FREE(MEMPTR) free(MEMPTR)
|
|
# endif
|
|
#endif
|
|
|
|
//! Checks if a pointer is aligned.
|
|
//! \param ptr The pointer to check.
|
|
//! \param alignment The alignement (must be a power of 2)
|
|
//! \returns true if the memory is aligned.
|
|
//!
|
|
template <uint32_t ALIGNMENT>
|
|
EASY_FORCE_INLINE bool is_aligned(void* ptr)
|
|
{
|
|
static_assert(ALIGNMENT % 2 == 0, "Alignment must be a power of two.");
|
|
return ((uintptr_t)ptr & (ALIGNMENT-1)) == 0;
|
|
}
|
|
|
|
EASY_FORCE_INLINE void unaligned_zero16(void* ptr)
|
|
{
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*(uint16_t*)ptr = 0;
|
|
#else
|
|
((char*)ptr)[0] = 0;
|
|
((char*)ptr)[1] = 0;
|
|
#endif
|
|
}
|
|
|
|
EASY_FORCE_INLINE void unaligned_zero32(void* ptr)
|
|
{
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*(uint32_t*)ptr = 0;
|
|
#else
|
|
((char*)ptr)[0] = 0;
|
|
((char*)ptr)[1] = 0;
|
|
((char*)ptr)[2] = 0;
|
|
((char*)ptr)[3] = 0;
|
|
#endif
|
|
}
|
|
|
|
EASY_FORCE_INLINE void unaligned_zero64(void* ptr)
|
|
{
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*(uint64_t*)ptr = 0;
|
|
#else
|
|
// Assume unaligned is more common.
|
|
if (!is_aligned<alignof(uint64_t)>(ptr)) {
|
|
((char*)ptr)[0] = 0;
|
|
((char*)ptr)[1] = 0;
|
|
((char*)ptr)[2] = 0;
|
|
((char*)ptr)[3] = 0;
|
|
((char*)ptr)[4] = 0;
|
|
((char*)ptr)[5] = 0;
|
|
((char*)ptr)[6] = 0;
|
|
((char*)ptr)[7] = 0;
|
|
}
|
|
else {
|
|
*(uint64_t*)ptr = 0;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE void unaligned_store16(void* ptr, T val)
|
|
{
|
|
static_assert(sizeof(T) == 2, "16 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*(T*)ptr = val;
|
|
#else
|
|
const char* const temp = (char*)&val;
|
|
((char*)ptr)[0] = temp[0];
|
|
((char*)ptr)[1] = temp[1];
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE void unaligned_store32(void* ptr, T val)
|
|
{
|
|
static_assert(sizeof(T) == 4, "32 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*(T*)ptr = val;
|
|
#else
|
|
const char* const temp = (char*)&val;
|
|
((char*)ptr)[0] = temp[0];
|
|
((char*)ptr)[1] = temp[1];
|
|
((char*)ptr)[2] = temp[2];
|
|
((char*)ptr)[3] = temp[3];
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE void unaligned_store64(void* ptr, T val)
|
|
{
|
|
static_assert(sizeof(T) == 8, "64 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*(T*)ptr = val;
|
|
#else
|
|
const char* const temp = (char*)&val;
|
|
// Assume unaligned is more common.
|
|
if (!is_aligned<alignof(T)>(ptr)) {
|
|
((char*)ptr)[0] = temp[0];
|
|
((char*)ptr)[1] = temp[1];
|
|
((char*)ptr)[2] = temp[2];
|
|
((char*)ptr)[3] = temp[3];
|
|
((char*)ptr)[4] = temp[4];
|
|
((char*)ptr)[5] = temp[5];
|
|
((char*)ptr)[6] = temp[6];
|
|
((char*)ptr)[7] = temp[7];
|
|
}
|
|
else {
|
|
*(T*)ptr = val;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE T unaligned_load16(const void* ptr)
|
|
{
|
|
static_assert(sizeof(T) == 2, "16 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
return *(T*)ptr;
|
|
#else
|
|
T value;
|
|
((char*)&value)[0] = ((char*)ptr)[0];
|
|
((char*)&value)[1] = ((char*)ptr)[1];
|
|
return value;
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE T unaligned_load16(const void* ptr, T* val)
|
|
{
|
|
static_assert(sizeof(T) == 2, "16 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*val = *(T*)ptr;
|
|
return *val;
|
|
#else
|
|
((char*)val)[0] = ((char*)ptr)[0];
|
|
((char*)val)[1] = ((char*)ptr)[1];
|
|
return *val;
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE T unaligned_load32(const void* ptr)
|
|
{
|
|
static_assert(sizeof(T) == 4, "32 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
return *(T*)ptr;
|
|
#else
|
|
T value;
|
|
((char*)&value)[0] = ((char*)ptr)[0];
|
|
((char*)&value)[1] = ((char*)ptr)[1];
|
|
((char*)&value)[2] = ((char*)ptr)[2];
|
|
((char*)&value)[3] = ((char*)ptr)[3];
|
|
return value;
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE T unaligned_load32(const void* ptr, T* val)
|
|
{
|
|
static_assert(sizeof(T) == 4, "32 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*val = *(T*)ptr;
|
|
#else
|
|
((char*)&val)[0] = ((char*)ptr)[0];
|
|
((char*)&val)[1] = ((char*)ptr)[1];
|
|
((char*)&val)[2] = ((char*)ptr)[2];
|
|
((char*)&val)[3] = ((char*)ptr)[3];
|
|
return *val;
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE T unaligned_load64(const void* ptr)
|
|
{
|
|
static_assert(sizeof(T) == 8, "64 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
return *(T*)ptr;
|
|
#else
|
|
if (!is_aligned<alignof(T)>(ptr)) {
|
|
T value;
|
|
((char*)&value)[0] = ((char*)ptr)[0];
|
|
((char*)&value)[1] = ((char*)ptr)[1];
|
|
((char*)&value)[2] = ((char*)ptr)[2];
|
|
((char*)&value)[3] = ((char*)ptr)[3];
|
|
((char*)&value)[4] = ((char*)ptr)[4];
|
|
((char*)&value)[5] = ((char*)ptr)[5];
|
|
((char*)&value)[6] = ((char*)ptr)[6];
|
|
((char*)&value)[7] = ((char*)ptr)[7];
|
|
return value;
|
|
}
|
|
else {
|
|
return *(T*)ptr;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
template <typename T>
|
|
EASY_FORCE_INLINE T unaligned_load64(const void* ptr, T* val)
|
|
{
|
|
static_assert(sizeof(T) == 8, "64 bit type required.");
|
|
#ifndef EASY_ENABLE_STRICT_ALIGNMENT
|
|
*val = *(T*)ptr;
|
|
#else
|
|
if (!is_aligned<alignof(T)>(ptr)) {
|
|
((char*)&val)[0] = ((char*)ptr)[0];
|
|
((char*)&val)[1] = ((char*)ptr)[1];
|
|
((char*)&val)[2] = ((char*)ptr)[2];
|
|
((char*)&val)[3] = ((char*)ptr)[3];
|
|
((char*)&val)[4] = ((char*)ptr)[4];
|
|
((char*)&val)[5] = ((char*)ptr)[5];
|
|
((char*)&val)[6] = ((char*)ptr)[6];
|
|
((char*)&val)[7] = ((char*)ptr)[7];
|
|
return *val;
|
|
}
|
|
else {
|
|
*val = *(T*)ptr;
|
|
return *val;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
|
|
template <uint16_t N>
|
|
class chunk_allocator
|
|
{
|
|
struct chunk { EASY_ALIGNED(char, data[N], EASY_ALIGNMENT_SIZE); chunk* prev = nullptr; };
|
|
|
|
struct chunk_list
|
|
{
|
|
chunk* last;
|
|
|
|
chunk_list() : last(nullptr)
|
|
{
|
|
static_assert(sizeof(char) == 1, "easy_profiler logic error: sizeof(char) != 1 for this platform! Please, contact easy_profiler authors to resolve your problem.");
|
|
emplace_back();
|
|
}
|
|
|
|
~chunk_list()
|
|
{
|
|
do free_last(); while (last != nullptr);
|
|
}
|
|
|
|
void clear_all_except_last()
|
|
{
|
|
while (last->prev != nullptr)
|
|
free_last();
|
|
zero_last_chunk_size();
|
|
}
|
|
|
|
void emplace_back()
|
|
{
|
|
auto prev = last;
|
|
last = ::new (EASY_MALLOC(sizeof(chunk), EASY_ALIGNMENT_SIZE)) chunk();
|
|
last->prev = prev;
|
|
zero_last_chunk_size();
|
|
}
|
|
|
|
/** Invert current chunks list to enable to iterate over chunks list in direct order.
|
|
|
|
This method is used by serialize().
|
|
*/
|
|
void invert()
|
|
{
|
|
chunk* next = nullptr;
|
|
|
|
while (last->prev != nullptr) {
|
|
auto p = last->prev;
|
|
last->prev = next;
|
|
next = last;
|
|
last = p;
|
|
}
|
|
|
|
last->prev = next;
|
|
}
|
|
|
|
private:
|
|
|
|
chunk_list(const chunk_list&) = delete;
|
|
chunk_list(chunk_list&&) = delete;
|
|
|
|
void free_last()
|
|
{
|
|
auto p = last;
|
|
last = last->prev;
|
|
EASY_FREE(p);
|
|
}
|
|
|
|
void zero_last_chunk_size()
|
|
{
|
|
// Although there is no need for unaligned access stuff b/c a new chunk will
|
|
// usually be at least 8 byte aligned (and we only need 2 byte alignment),
|
|
// this is the only way I have been able to get rid of the GCC strict-aliasing warning
|
|
// without using std::memset. It's an extra line, but is just as fast as *(uint16_t*)last->data = 0;
|
|
char* const data = last->data;
|
|
*(uint16_t*)data = (uint16_t)0;
|
|
}
|
|
};
|
|
|
|
// Used in serialize(): workaround for no constexpr support in MSVC 2013.
|
|
static const int_fast32_t MAX_CHUNK_OFFSET = N - sizeof(uint16_t);
|
|
static const uint16_t N_MINUS_ONE = N - 1;
|
|
|
|
chunk_list m_chunks; ///< List of chunks.
|
|
uint32_t m_size; ///< Number of elements stored(# of times allocate() has been called.)
|
|
uint16_t m_chunkOffset; ///< Number of bytes used in the current chunk.
|
|
|
|
public:
|
|
|
|
chunk_allocator() : m_size(0), m_chunkOffset(0)
|
|
{
|
|
}
|
|
|
|
/** Allocate n bytes.
|
|
|
|
Automatically checks if there is enough preserved memory to store additional n bytes
|
|
and allocates additional buffer if needed.
|
|
*/
|
|
void* allocate(uint16_t n)
|
|
{
|
|
++m_size;
|
|
|
|
if (!need_expand(n))
|
|
{
|
|
// Temp to avoid extra load due to this* aliasing.
|
|
uint16_t chunkOffset = m_chunkOffset;
|
|
char* data = m_chunks.last->data + chunkOffset;
|
|
chunkOffset += n + sizeof(uint16_t);
|
|
m_chunkOffset = chunkOffset;
|
|
|
|
unaligned_store16(data, n);
|
|
data += sizeof(uint16_t);
|
|
|
|
// If there is enough space for at least another payload size,
|
|
// set it to zero.
|
|
if (chunkOffset < N_MINUS_ONE)
|
|
unaligned_zero16(data + n);
|
|
|
|
return data;
|
|
}
|
|
|
|
m_chunkOffset = n + sizeof(uint16_t);
|
|
m_chunks.emplace_back();
|
|
|
|
char* data = m_chunks.last->data;
|
|
unaligned_store16(data, n);
|
|
data += sizeof(uint16_t);
|
|
|
|
// We assume here that it takes more than one element to fill a chunk.
|
|
unaligned_zero16(data + n);
|
|
|
|
return data;
|
|
}
|
|
|
|
/** Check if current storage is not enough to store additional n bytes.
|
|
*/
|
|
bool need_expand(uint16_t n) const
|
|
{
|
|
return (m_chunkOffset + n + sizeof(uint16_t)) > N;
|
|
}
|
|
|
|
uint32_t size() const
|
|
{
|
|
return m_size;
|
|
}
|
|
|
|
bool empty() const
|
|
{
|
|
return m_size == 0;
|
|
}
|
|
|
|
void clear()
|
|
{
|
|
m_size = 0;
|
|
m_chunkOffset = 0;
|
|
m_chunks.clear_all_except_last(); // There is always at least one chunk
|
|
}
|
|
|
|
/** Serialize data to stream.
|
|
|
|
\warning Data will be cleared after serialization.
|
|
*/
|
|
void serialize(profiler::OStream& _outputStream)
|
|
{
|
|
// Chunks are stored in reversed order (stack).
|
|
// To be able to iterate them in direct order we have to invert the chunks list.
|
|
m_chunks.invert();
|
|
|
|
// Each chunk is an array of N bytes that can hold between
|
|
// 1(if the list isn't empty) and however many elements can fit in a chunk,
|
|
// where an element consists of a payload size + a payload as follows:
|
|
// elementStart[0..1]: size as a uint16_t
|
|
// elementStart[2..size-1]: payload.
|
|
|
|
// The maximum chunk offset is N-sizeof(uint16_t) b/c, if we hit that (or go past),
|
|
// there is either no space left, 1 byte left, or 2 bytes left, all of which are
|
|
// too small to cary more than a zero-sized element.
|
|
|
|
chunk* current = m_chunks.last;
|
|
do {
|
|
const char* data = current->data;
|
|
int_fast32_t chunkOffset = 0; // signed int so overflow is not checked.
|
|
uint16_t payloadSize = unaligned_load16<uint16_t>(data);
|
|
while (chunkOffset < MAX_CHUNK_OFFSET && payloadSize != 0) {
|
|
const uint16_t chunkSize = sizeof(uint16_t) + payloadSize;
|
|
_outputStream.write(data, chunkSize);
|
|
data += chunkSize;
|
|
chunkOffset += chunkSize;
|
|
unaligned_load16(data, &payloadSize);
|
|
}
|
|
|
|
current = current->prev;
|
|
} while (current != nullptr);
|
|
|
|
clear();
|
|
}
|
|
|
|
private:
|
|
|
|
chunk_allocator(const chunk_allocator&) = delete;
|
|
chunk_allocator(chunk_allocator&&) = delete;
|
|
|
|
}; // END of class chunk_allocator.
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
class NonscopedBlock : public profiler::Block
|
|
{
|
|
char* m_runtimeName; ///< a copy of _runtimeName to make it safe to begin block in one function and end it in another
|
|
|
|
NonscopedBlock() = delete;
|
|
NonscopedBlock(const NonscopedBlock&) = delete;
|
|
NonscopedBlock(NonscopedBlock&&) = delete;
|
|
NonscopedBlock& operator = (const NonscopedBlock&) = delete;
|
|
NonscopedBlock& operator = (NonscopedBlock&&) = delete;
|
|
|
|
public:
|
|
|
|
NonscopedBlock(const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName, bool = false);
|
|
~NonscopedBlock();
|
|
|
|
/** Copy string from m_name to m_runtimeName to make it safe to end block in another function.
|
|
|
|
Performs any work if block is ON and m_name != ""
|
|
*/
|
|
void copyname();
|
|
|
|
void destroy();
|
|
|
|
}; // END of class NonscopedBlock.
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
template <class T>
|
|
inline void destroy_elem(T*)
|
|
{
|
|
|
|
}
|
|
|
|
inline void destroy_elem(NonscopedBlock* _elem)
|
|
{
|
|
_elem->destroy();
|
|
}
|
|
|
|
template <class T>
|
|
class StackBuffer
|
|
{
|
|
struct chunk { int8_t data[sizeof(T)]; };
|
|
|
|
std::list<chunk> m_overflow; ///< List of additional stack elements if current capacity of buffer is not enough
|
|
T* m_buffer; ///< Contiguous buffer used for stack
|
|
uint32_t m_size; ///< Current size of stack
|
|
uint32_t m_capacity; ///< Current capacity of m_buffer
|
|
uint32_t m_maxcapacity; ///< Maximum used capacity including m_buffer and m_overflow
|
|
|
|
public:
|
|
|
|
StackBuffer(uint32_t N) : m_buffer(static_cast<T*>(malloc(N * sizeof(T)))), m_size(0), m_capacity(N), m_maxcapacity(N)
|
|
{
|
|
}
|
|
|
|
~StackBuffer()
|
|
{
|
|
for (uint32_t i = 0; i < m_size; ++i)
|
|
destroy_elem(m_buffer + i);
|
|
|
|
free(m_buffer);
|
|
|
|
for (auto& elem : m_overflow)
|
|
destroy_elem(reinterpret_cast<T*>(elem.data + 0));
|
|
}
|
|
|
|
template <class ... TArgs>
|
|
T& push(TArgs ... _args)
|
|
{
|
|
if (m_size < m_capacity)
|
|
return *(::new (m_buffer + m_size++) T(_args...));
|
|
|
|
m_overflow.emplace_back();
|
|
const uint32_t cap = m_capacity + static_cast<uint32_t>(m_overflow.size());
|
|
if (m_maxcapacity < cap)
|
|
m_maxcapacity = cap;
|
|
|
|
return *(::new (m_overflow.back().data + 0) T(_args...));
|
|
}
|
|
|
|
void pop()
|
|
{
|
|
if (m_overflow.empty())
|
|
{
|
|
// m_size should not be equal to 0 here because ProfileManager behavior does not allow such situation
|
|
destroy_elem(m_buffer + --m_size);
|
|
|
|
if (m_size == 0 && m_maxcapacity > m_capacity)
|
|
{
|
|
// When stack gone empty we can resize buffer to use enough space in the future
|
|
free(m_buffer);
|
|
m_maxcapacity = m_capacity = std::max(m_maxcapacity, m_capacity << 1);
|
|
m_buffer = static_cast<T*>(malloc(m_capacity * sizeof(T)));
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
destroy_elem(reinterpret_cast<T*>(m_overflow.back().data + 0));
|
|
m_overflow.pop_back();
|
|
}
|
|
|
|
private:
|
|
|
|
StackBuffer(const StackBuffer&) = delete;
|
|
StackBuffer(StackBuffer&&) = delete;
|
|
|
|
}; // END of class StackBuffer.
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
template <class T, const uint16_t N>
|
|
struct BlocksList
|
|
{
|
|
BlocksList() = default;
|
|
|
|
std::vector<T> openedList;
|
|
chunk_allocator<N> closedList;
|
|
uint64_t usedMemorySize = 0;
|
|
|
|
void clearClosed() {
|
|
//closedList.clear();
|
|
usedMemorySize = 0;
|
|
}
|
|
|
|
private:
|
|
|
|
BlocksList(const BlocksList&) = delete;
|
|
BlocksList(BlocksList&&) = delete;
|
|
|
|
}; // END of struct BlocksList.
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
class CSwitchBlock : public profiler::CSwitchEvent
|
|
{
|
|
const char* m_name;
|
|
|
|
public:
|
|
|
|
CSwitchBlock(profiler::timestamp_t _begin_time, profiler::thread_id_t _tid, const char* _runtimeName);
|
|
inline const char* name() const { return m_name; }
|
|
};
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
const uint16_t SIZEOF_BLOCK = sizeof(profiler::BaseBlockData) + 1 + sizeof(uint16_t); // SerializedBlock stores BaseBlockData + at least 1 character for name ('\0') + 2 bytes for size of serialized data
|
|
const uint16_t SIZEOF_CSWITCH = sizeof(profiler::CSwitchEvent) + 1 + sizeof(uint16_t); // SerializedCSwitch also stores additional 4 bytes to be able to save 64-bit thread_id
|
|
|
|
struct ThreadStorage
|
|
{
|
|
StackBuffer<NonscopedBlock> nonscopedBlocks;
|
|
BlocksList<std::reference_wrapper<profiler::Block>, SIZEOF_BLOCK * (uint16_t)128U> blocks;
|
|
BlocksList<CSwitchBlock, SIZEOF_CSWITCH * (uint16_t)128U> sync;
|
|
|
|
std::string name; ///< Thread name
|
|
|
|
#ifndef _WIN32
|
|
const pthread_t pthread_id; ///< Thread pointer
|
|
#endif
|
|
|
|
const profiler::thread_id_t id; ///< Thread ID
|
|
std::atomic<char> expired; ///< Is thread expired
|
|
std::atomic_bool frame; ///< Is new frame opened
|
|
bool allowChildren; ///< False if one of previously opened blocks has OFF_RECURSIVE or ON_WITHOUT_CHILDREN status
|
|
bool named; ///< True if thread name was set
|
|
bool guarded; ///< True if thread has been registered using ThreadGuard
|
|
|
|
void storeBlock(const profiler::Block& _block);
|
|
void storeCSwitch(const CSwitchBlock& _block);
|
|
void clearClosed();
|
|
void popSilent();
|
|
|
|
ThreadStorage();
|
|
|
|
private:
|
|
|
|
ThreadStorage(const ThreadStorage&) = delete;
|
|
ThreadStorage(ThreadStorage&&) = delete;
|
|
|
|
}; // END of struct ThreadStorage.
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
typedef uint64_t processid_t;
|
|
|
|
class BlockDescriptor;
|
|
|
|
class ProfileManager
|
|
{
|
|
#ifndef EASY_MAGIC_STATIC_CPP11
|
|
friend class ProfileManagerInstance;
|
|
#endif
|
|
|
|
ProfileManager();
|
|
ProfileManager(const ProfileManager& p) = delete;
|
|
ProfileManager& operator=(const ProfileManager&) = delete;
|
|
|
|
typedef profiler::guard_lock<profiler::spin_lock> guard_lock_t;
|
|
typedef std::map<profiler::thread_id_t, ThreadStorage> map_of_threads_stacks;
|
|
typedef std::vector<BlockDescriptor*> block_descriptors_t;
|
|
|
|
#ifdef EASY_PROFILER_HASHED_CSTR_DEFINED
|
|
typedef std::unordered_map<profiler::hashed_cstr, profiler::block_id_t> descriptors_map_t;
|
|
#else
|
|
typedef std::unordered_map<profiler::hashed_stdstring, profiler::block_id_t> descriptors_map_t;
|
|
#endif
|
|
|
|
const processid_t m_processId;
|
|
|
|
map_of_threads_stacks m_threads;
|
|
block_descriptors_t m_descriptors;
|
|
descriptors_map_t m_descriptorsMap;
|
|
uint64_t m_usedMemorySize;
|
|
profiler::timestamp_t m_beginTime;
|
|
profiler::timestamp_t m_endTime;
|
|
std::atomic<profiler::timestamp_t> m_frameMax;
|
|
std::atomic<profiler::timestamp_t> m_frameAvg;
|
|
std::atomic<profiler::timestamp_t> m_frameCur;
|
|
profiler::spin_lock m_spin;
|
|
profiler::spin_lock m_storedSpin;
|
|
profiler::spin_lock m_dumpSpin;
|
|
std::atomic<profiler::thread_id_t> m_mainThreadId;
|
|
std::atomic<char> m_profilerStatus;
|
|
std::atomic_bool m_isEventTracingEnabled;
|
|
std::atomic_bool m_isAlreadyListening;
|
|
std::atomic_bool m_frameMaxReset;
|
|
std::atomic_bool m_frameAvgReset;
|
|
|
|
std::string m_csInfoFilename = "/tmp/cs_profiling_info.log";
|
|
|
|
uint32_t dumpBlocksToStream(profiler::OStream& _outputStream, bool _lockSpin);
|
|
void setBlockStatus(profiler::block_id_t _id, profiler::EasyBlockStatus _status);
|
|
|
|
std::thread m_listenThread;
|
|
void listen(uint16_t _port);
|
|
|
|
std::atomic_bool m_stopListen;
|
|
|
|
public:
|
|
|
|
static ProfileManager& instance();
|
|
~ProfileManager();
|
|
|
|
const profiler::BaseBlockDescriptor* addBlockDescriptor(profiler::EasyBlockStatus _defaultStatus,
|
|
const char* _autogenUniqueId,
|
|
const char* _name,
|
|
const char* _filename,
|
|
int _line,
|
|
profiler::block_type_t _block_type,
|
|
profiler::color_t _color,
|
|
bool _copyName = false);
|
|
|
|
bool storeBlock(const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName);
|
|
bool storeBlock(const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName, profiler::timestamp_t _beginTime, profiler::timestamp_t _endTime);
|
|
void beginBlock(profiler::Block& _block);
|
|
void beginNonScopedBlock(const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName);
|
|
void endBlock();
|
|
profiler::timestamp_t maxFrameDuration();
|
|
profiler::timestamp_t avgFrameDuration();
|
|
profiler::timestamp_t curFrameDuration() const;
|
|
void setEnabled(bool isEnable);
|
|
bool isEnabled() const;
|
|
void setEventTracingEnabled(bool _isEnable);
|
|
bool isEventTracingEnabled() const;
|
|
uint32_t dumpBlocksToFile(const char* filename);
|
|
const char* registerThread(const char* name, profiler::ThreadGuard& threadGuard);
|
|
const char* registerThread(const char* name);
|
|
|
|
void setContextSwitchLogFilename(const char* name)
|
|
{
|
|
m_csInfoFilename = name;
|
|
}
|
|
|
|
const char* getContextSwitchLogFilename() const
|
|
{
|
|
return m_csInfoFilename.c_str();
|
|
}
|
|
|
|
void beginContextSwitch(profiler::thread_id_t _thread_id, profiler::timestamp_t _time, profiler::thread_id_t _target_thread_id, const char* _target_process, bool _lockSpin = true);
|
|
void endContextSwitch(profiler::thread_id_t _thread_id, processid_t _process_id, profiler::timestamp_t _endtime, bool _lockSpin = true);
|
|
void startListen(uint16_t _port);
|
|
void stopListen();
|
|
bool isListening() const;
|
|
|
|
private:
|
|
|
|
void registerThread();
|
|
|
|
void beginFrame();
|
|
void endFrame();
|
|
|
|
void enableEventTracer();
|
|
void disableEventTracer();
|
|
|
|
static char checkThreadExpired(ThreadStorage& _registeredThread);
|
|
|
|
void storeBlockForce(const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName, ::profiler::timestamp_t& _timestamp);
|
|
void storeBlockForce2(const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName, ::profiler::timestamp_t _timestamp);
|
|
void storeBlockForce2(ThreadStorage& _registeredThread, const profiler::BaseBlockDescriptor* _desc, const char* _runtimeName, ::profiler::timestamp_t _timestamp);
|
|
|
|
ThreadStorage& _threadStorage(profiler::thread_id_t _thread_id);
|
|
ThreadStorage* _findThreadStorage(profiler::thread_id_t _thread_id);
|
|
|
|
inline ThreadStorage& threadStorage(profiler::thread_id_t _thread_id)
|
|
{
|
|
guard_lock_t lock(m_spin);
|
|
return _threadStorage(_thread_id);
|
|
}
|
|
|
|
inline ThreadStorage* findThreadStorage(profiler::thread_id_t _thread_id)
|
|
{
|
|
guard_lock_t lock(m_spin);
|
|
return _findThreadStorage(_thread_id);
|
|
}
|
|
|
|
}; // END of class ProfileManager.
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
#endif // EASY_PROFILER_MANAGER_H
|