// Copyright 2020 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#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-internal/globals.h"
#include "src/heap/cppgc-internal/heap.h"
#include "test/benchmarks/cpp/cppgc/benchmark_utils.h"
#include "third_party/google_benchmark_chrome/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(_);
    TinyObject* result =
        cppgc::MakeGarbageCollected<TinyObject>(heap().GetAllocationHandle());
    benchmark::DoNotOptimize(result);
  }
  st.SetBytesProcessed(st.iterations() * sizeof(TinyObject));
}

class LargeObject final : public GarbageCollected<LargeObject> {
 public:
  void Trace(cppgc::Visitor*) const {}
  char padding[kLargeObjectSizeThreshold + 1];
};

BENCHMARK_F(Allocate, Large)(benchmark::State& st) {
  subtle::NoGarbageCollectionScope no_gc(*Heap::From(&heap()));
  for (auto _ : st) {
    USE(_);
    LargeObject* result =
        cppgc::MakeGarbageCollected<LargeObject>(heap().GetAllocationHandle());
    benchmark::DoNotOptimize(result);
  }
  st.SetBytesProcessed(st.iterations() * sizeof(LargeObject));
}

}  // namespace
}  // namespace internal
}  // namespace cppgc