historical/toontown-classic.git/panda/include/memoryUsage.h
2024-01-16 11:20:27 -06:00

216 lines
6.9 KiB
C++

/**
* PANDA 3D SOFTWARE
* Copyright (c) Carnegie Mellon University. All rights reserved.
*
* All use of this software is subject to the terms of the revised BSD
* license. You should have received a copy of this license along
* with this source code in a file named "LICENSE."
*
* @file memoryUsage.h
* @author drose
* @date 2000-05-25
*/
#ifndef MEMORYUSAGE_H
#define MEMORYUSAGE_H
#include "pandabase.h"
#include "typedObject.h"
#include "memoryInfo.h"
#include "memoryUsagePointerCounts.h"
#include "pmap.h"
#include "memoryHook.h"
class ReferenceCount;
class MemoryUsagePointers;
/**
* This class is used strictly for debugging purposes, specifically for
* tracking memory leaks of reference-counted objects: it keeps a record of
* every such object currently allocated.
*
* When compiled with NDEBUG set, this entire class does nothing and compiles
* to a stub.
*/
class EXPCL_PANDA_EXPRESS MemoryUsage : public MemoryHook {
public:
ALWAYS_INLINE static bool get_track_memory_usage();
INLINE static void record_pointer(ReferenceCount *ptr);
INLINE static void record_pointer(void *ptr, TypeHandle type);
INLINE static void update_type(ReferenceCount *ptr, TypeHandle type);
INLINE static void update_type(ReferenceCount *ptr, TypedObject *typed_ptr);
INLINE static void update_type(void *ptr, TypeHandle type);
INLINE static void remove_pointer(ReferenceCount *ptr);
protected:
// These are not marked public, but they can be accessed via the MemoryHook
// base class.
virtual void *heap_alloc_single(size_t size);
virtual void heap_free_single(void *ptr);
virtual void *heap_alloc_array(size_t size);
virtual void *heap_realloc_array(void *ptr, size_t size);
virtual void heap_free_array(void *ptr);
virtual void mark_pointer(void *ptr, size_t orig_size, ReferenceCount *ref_ptr);
#if (defined(WIN32_VC) || defined(WIN64_VC)) && defined(_DEBUG)
static int win32_malloc_hook(int alloc_type, void *ptr,
size_t size, int block_use, long request,
const unsigned char *filename, int line);
#endif
PUBLISHED:
INLINE static bool is_tracking();
INLINE static bool is_counting();
INLINE static size_t get_current_cpp_size();
INLINE static size_t get_total_cpp_size();
INLINE static size_t get_panda_heap_single_size();
INLINE static size_t get_panda_heap_array_size();
INLINE static size_t get_panda_heap_overhead();
INLINE static size_t get_panda_mmap_size();
INLINE static size_t get_external_size();
INLINE static size_t get_total_size();
INLINE static int get_num_pointers();
INLINE static void get_pointers(MemoryUsagePointers &result);
INLINE static void get_pointers_of_type(MemoryUsagePointers &result,
TypeHandle type);
INLINE static void get_pointers_of_age(MemoryUsagePointers &result,
double from, double to);
INLINE static void get_pointers_with_zero_count(MemoryUsagePointers &result);
INLINE static void freeze();
INLINE static void show_current_types();
INLINE static void show_trend_types();
INLINE static void show_current_ages();
INLINE static void show_trend_ages();
PUBLISHED:
MAKE_PROPERTY(tracking, is_tracking);
MAKE_PROPERTY(counting, is_counting);
MAKE_PROPERTY(current_cpp_size, get_current_cpp_size);
MAKE_PROPERTY(total_cpp_size, get_total_cpp_size);
MAKE_PROPERTY(panda_heap_single_size, get_panda_heap_single_size);
MAKE_PROPERTY(panda_heap_array_size, get_panda_heap_array_size);
MAKE_PROPERTY(panda_heap_overhead, get_panda_heap_overhead);
MAKE_PROPERTY(panda_mmap_size, get_panda_mmap_size);
MAKE_PROPERTY(external_size, get_external_size);
MAKE_PROPERTY(total_size, get_total_size);
protected:
virtual void overflow_heap_size();
private:
MemoryUsage(const MemoryHook &copy);
INLINE static MemoryUsage *get_global_ptr();
static void init_memory_usage();
void ns_record_pointer(ReferenceCount *ptr);
void ns_record_pointer(void *ptr, TypeHandle type);
void ns_update_type(void *ptr, TypeHandle type);
void ns_update_type(void *ptr, TypedObject *typed_ptr);
void ns_remove_pointer(ReferenceCount *ptr);
void ns_record_void_pointer(void *ptr, size_t size);
void ns_remove_void_pointer(void *ptr);
size_t ns_get_total_size();
int ns_get_num_pointers();
void ns_get_pointers(MemoryUsagePointers &result);
void ns_get_pointers_of_type(MemoryUsagePointers &result,
TypeHandle type);
void ns_get_pointers_of_age(MemoryUsagePointers &result,
double from, double to);
void ns_get_pointers_with_zero_count(MemoryUsagePointers &result);
void ns_freeze();
void ns_show_current_types();
void ns_show_trend_types();
void ns_show_current_ages();
void ns_show_trend_ages();
#ifdef DO_MEMORY_USAGE
void consolidate_void_ptr(MemoryInfo *info);
void refresh_info_set();
#endif
static MemoryUsage *_global_ptr;
// We shouldn't use a pmap, since that would be recursive! Actually, it
// turns out that it doesn't matter, since somehow the pallocator gets used
// even though we don't specify it here, so we have to make special code
// that handles the recursion anyway.
/*
* This table stores up to two entiries for each MemoryInfo object: one for
* the void pointer (the pointer to the beginning of the allocated memory
* block), and one for the ReferenceCount pointer. For a particular object,
* these two pointers may be the same or they may be different. Some objects
* may be stored under both pointers, while others may be stored under only
* one pointer or the other. We don't store an entry for an object's
* TypedObject pointer.
*/
typedef std::map<void *, MemoryInfo *> Table;
Table _table;
// This table indexes the individual MemoryInfo objects, for unique
// iteration.
typedef std::set<MemoryInfo *> InfoSet;
InfoSet _info_set;
bool _info_set_dirty;
int _freeze_index;
int _count;
size_t _current_cpp_size;
size_t _total_cpp_size;
size_t _total_size;
class TypeHistogram {
public:
void add_info(TypeHandle type, MemoryInfo *info);
void show() const;
void clear();
private:
// Cannot use a pmap, since that would be recursive!
typedef std::map<TypeHandle, MemoryUsagePointerCounts> Counts;
Counts _counts;
};
TypeHistogram _trend_types;
class AgeHistogram {
public:
AgeHistogram();
void add_info(double age, MemoryInfo *info);
void show() const;
void clear();
private:
int choose_bucket(double age) const;
enum { num_buckets = 5 };
MemoryUsagePointerCounts _counts[num_buckets];
static double _cutoff[num_buckets];
};
AgeHistogram _trend_ages;
bool _track_memory_usage;
bool _startup_track_memory_usage;
bool _count_memory_usage;
bool _report_memory_usage;
double _report_memory_interval;
double _last_report_time;
static bool _recursion_protect;
};
#include "memoryUsage.I"
#endif