fullallocationtracker.h

Go to the documentation of this file.

#ifndef ION_BASE_FULLALLOCATIONTRACKER_H_
#define ION_BASE_FULLALLOCATIONTRACKER_H_

#include <iostream>  // NOLINT
#include <memory>

#include "ion/base/allocationtracker.h"

namespace ion {
namespace base {

class ION_API FullAllocationTracker : public AllocationTracker {
 public:
  FullAllocationTracker();

  void SetTracingStream(std::ostream* s) { tracing_ostream_ = s; }
  std::ostream* GetTracingStream() const { return tracing_ostream_; }

  void TrackAllocation(const Allocator& allocator,
                       size_t requested_size, const void* memory) override;
  void TrackDeallocation(const Allocator& allocator,
                         const void* memory) override;
  size_t GetAllocationCount() override;
  size_t GetDeallocationCount() override;
  size_t GetAllocatedBytesCount() override;
  size_t GetDeallocatedBytesCount() override;
  size_t GetActiveAllocationCount() override;
  size_t GetActiveAllocationBytesCount() override;

  void SetGpuTracker(const AllocationSizeTrackerPtr& gpu_tracker) override;
  AllocationSizeTrackerPtr GetGpuTracker() override;

 protected:
  ~FullAllocationTracker() override;

 private:
  class Helper;
  std::unique_ptr<Helper> helper_;

  std::ostream* tracing_ostream_;
};

typedef SharedPtr<FullAllocationTracker> FullAllocationTrackerPtr;

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_FULLALLOCATIONTRACKER_H_