#include "include/cppgc/allocation.h"

#include "include/cppgc/garbage-collected.h"

#include "include/cppgc/heap-consistency.h"

#include "src/base/macros.h"

#include "src/heap/cppgc/globals.h"

#include "src/heap/cppgc/heap.h"

#include "test/benchmarks/cpp/cppgc/utils.h"

#include "third_party/google_benchmark/src/include/benchmark/benchmark.h"

namespace cppgc {

namespace internal {

namespace {

using Allocate = testing::BenchmarkWithHeap;

class TinyObject final : public cppgc::GarbageCollected<TinyObject> {

public:

void Trace(cppgc::Visitor*) const {}

};

BENCHMARK_F(Allocate, Tiny)(benchmark::State& st) {

subtle::NoGarbageCollectionScope no_gc(*Heap::From(&heap()));

for (auto _ : st) {

USE(_);

benchmark::DoNotOptimize(

cppgc::MakeGarbageCollected<TinyObject>(heap().GetAllocationHandle()));

}

st.SetBytesProcessed(st.iterations() * sizeof(TinyObject));

}

class LargeObject final : public GarbageCollected<LargeObject> {

};

BENCHMARK_F(Allocate, Large)(benchmark::State& st) {

subtle::NoGarbageCollectionScope no_gc(*Heap::From(&heap()));

for (auto _ : st) {

USE(_);

benchmark::DoNotOptimize(

cppgc::MakeGarbageCollected<LargeObject>(heap().GetAllocationHandle()));

}

st.SetBytesProcessed(st.iterations() * sizeof(LargeObject));

}

} // namespace

} // namespace internal

} // namespace cppgc