allocationtracker.h

Go to the documentation of this file.

#ifndef ION_BASE_ALLOCATIONTRACKER_H_
#define ION_BASE_ALLOCATIONTRACKER_H_

#include "ion/base/shareable.h"
#include "ion/base/sharedptr.h"

namespace ion {
namespace base {

class Allocator;
class AllocationSizeTracker;
typedef SharedPtr<AllocationSizeTracker> AllocationSizeTrackerPtr;

class AllocationTracker : public Shareable {
 public:
  virtual void TrackAllocation(const Allocator& allocator,
                               size_t requested_size, const void* memory) = 0;

  virtual void TrackDeallocation(const Allocator& allocator,
                                 const void* memory) = 0;

  virtual size_t GetAllocationCount() = 0;
  virtual size_t GetDeallocationCount() = 0;

  virtual size_t GetAllocatedBytesCount() = 0;
  virtual size_t GetDeallocatedBytesCount() = 0;

  virtual size_t GetActiveAllocationCount() = 0;
  virtual size_t GetActiveAllocationBytesCount() = 0;

  virtual void SetGpuTracker(const AllocationSizeTrackerPtr& gpu_tracker) = 0;
  virtual AllocationSizeTrackerPtr GetGpuTracker() = 0;
};

typedef SharedPtr<AllocationTracker> AllocationTrackerPtr;

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_ALLOCATIONTRACKER_H_