#include "gc/shenandoah/shenandoahRootProcessor.inline.hpp"

#include "gc/shenandoah/shenandoahHeuristics.hpp"

ShenandoahSerialRoot::ShenandoahSerialRoot(ShenandoahSerialRoot::OopsDo oops_do, ShenandoahPhaseTimings::GCParPhases phase) :

  _claimed(false), _oops_do(oops_do), _phase(phase) {

}

void ShenandoahSerialRoot::oops_do(OopClosure* cl, uint worker_id) {

  if (!_claimed && Atomic::cmpxchg(true, &_claimed, false) == false) {

    ShenandoahWorkerTimings* worker_times = ShenandoahHeap::heap()->phase_timings()->worker_times();

    ShenandoahWorkerTimingsTracker timer(worker_times, _phase, worker_id);

    _oops_do(cl);

  }

}

ShenandoahSerialRoots::ShenandoahSerialRoots() :

  _universe_root(&Universe::oops_do, ShenandoahPhaseTimings::UniverseRoots),

  _object_synchronizer_root(&ObjectSynchronizer::oops_do, ShenandoahPhaseTimings::ObjectSynchronizerRoots),

  _management_root(&Management::oops_do, ShenandoahPhaseTimings::ManagementRoots),

  _system_dictionary_root(&SystemDictionary::oops_do, ShenandoahPhaseTimings::SystemDictionaryRoots),

  _jvmti_root(&JvmtiExport::oops_do, ShenandoahPhaseTimings::JVMTIRoots),

  _jni_handle_root(&JNIHandles::oops_do, ShenandoahPhaseTimings::JNIRoots) {

}

void ShenandoahSerialRoots::oops_do(OopClosure* cl, uint worker_id) {

  _universe_root.oops_do(cl, worker_id);

  _object_synchronizer_root.oops_do(cl, worker_id);

  _management_root.oops_do(cl, worker_id);

  _system_dictionary_root.oops_do(cl, worker_id);

  _jvmti_root.oops_do(cl, worker_id);

  _jni_handle_root.oops_do(cl, worker_id);

}

ShenandoahThreadRoots::ShenandoahThreadRoots(bool is_par) : _is_par(is_par) {

  Threads::change_thread_claim_token();

}

void ShenandoahThreadRoots::oops_do(OopClosure* oops_cl, CodeBlobClosure* code_cl, uint worker_id) {

  ShenandoahWorkerTimings* worker_times = ShenandoahHeap::heap()->phase_timings()->worker_times();

  ShenandoahWorkerTimingsTracker timer(worker_times, ShenandoahPhaseTimings::ThreadRoots, worker_id);

  ResourceMark rm;

  Threads::possibly_parallel_oops_do(_is_par, oops_cl, code_cl);

}

void ShenandoahThreadRoots::threads_do(ThreadClosure* tc, uint worker_id) {

  ShenandoahWorkerTimings* worker_times = ShenandoahHeap::heap()->phase_timings()->worker_times();

  ShenandoahWorkerTimingsTracker timer(worker_times, ShenandoahPhaseTimings::ThreadRoots, worker_id);

  ResourceMark rm;

  Threads::possibly_parallel_threads_do(_is_par, tc);

}

ShenandoahThreadRoots::~ShenandoahThreadRoots() {

  Threads::assert_all_threads_claimed();

}

ShenandoahWeakRoots::ShenandoahWeakRoots(uint n_workers) :

  _process_timings(n_workers),

  _task(&_process_timings, n_workers) {

}

ShenandoahWeakRoots::~ShenandoahWeakRoots() {

  ShenandoahWorkerTimings* worker_times = ShenandoahHeap::heap()->phase_timings()->worker_times();

  ShenandoahTimingConverter::weak_processing_timing_to_shenandoah_timing(&_process_timings,

                                                                         worker_times);

}

ShenandoahStringDedupRoots::ShenandoahStringDedupRoots() {

ShenandoahStringDedupRoots::~ShenandoahStringDedupRoots() {

}

void ShenandoahStringDedupRoots::oops_do(BoolObjectClosure* is_alive, OopClosure* keep_alive, uint worker_id) {

  if (ShenandoahStringDedup::is_enabled()) {

    ShenandoahStringDedup::parallel_oops_do(is_alive, keep_alive, worker_id);

ShenandoahClassLoaderDataRoots::ShenandoahClassLoaderDataRoots() {

  ClassLoaderDataGraph::clear_claimed_marks();

void ShenandoahClassLoaderDataRoots::clds_do(CLDClosure* strong_clds, CLDClosure* weak_clds, uint worker_id) {

  ShenandoahWorkerTimingsTracker timer(worker_times, ShenandoahPhaseTimings::CLDGRoots, worker_id);

  ClassLoaderDataGraph::roots_cld_do(strong_clds, weak_clds);

ShenandoahRootProcessor::ShenandoahRootProcessor(ShenandoahPhaseTimings::Phase phase) :

  _heap(ShenandoahHeap::heap()),

  _phase(phase) {

  assert(SafepointSynchronize::is_at_safepoint(), "Must at safepoint");

  _heap->phase_timings()->record_workers_start(_phase);

}

ShenandoahRootProcessor::~ShenandoahRootProcessor() {

  assert(SafepointSynchronize::is_at_safepoint(), "Must at safepoint");

  _heap->phase_timings()->record_workers_end(_phase);

}

ShenandoahRootScanner::ShenandoahRootScanner(uint n_workers, ShenandoahPhaseTimings::Phase phase) :

  ShenandoahRootProcessor(phase),

  _thread_roots(n_workers > 1) {

void ShenandoahRootScanner::roots_do(uint worker_id, OopClosure* oops) {

  CLDToOopClosure clds_cl(oops, ClassLoaderData::_claim_strong);

  MarkingCodeBlobClosure blobs_cl(oops, !CodeBlobToOopClosure::FixRelocations);

  roots_do(worker_id, oops, &clds_cl, &blobs_cl);

}

void ShenandoahRootScanner::strong_roots_do(uint worker_id, OopClosure* oops) {

  CLDToOopClosure clds_cl(oops, ClassLoaderData::_claim_strong);

  MarkingCodeBlobClosure blobs_cl(oops, !CodeBlobToOopClosure::FixRelocations);

  strong_roots_do(worker_id, oops, &clds_cl, &blobs_cl);

void ShenandoahRootScanner::roots_do(uint worker_id, OopClosure* oops, CLDClosure* clds, CodeBlobClosure* code, ThreadClosure *tc) {

  assert(!ShenandoahHeap::heap()->unload_classes(), "Should be used during class unloading");

  ShenandoahParallelOopsDoThreadClosure tc_cl(oops, code, tc);

  ResourceMark rm;

  _serial_roots.oops_do(oops, worker_id);

  _cld_roots.clds_do(clds, clds, worker_id);

  _thread_roots.threads_do(&tc_cl, worker_id);

  // With ShenandoahConcurrentScanCodeRoots, we avoid scanning the entire code cache here,

  // and instead do that in concurrent phase under the relevant lock. This saves init mark

  // pause time.

  if (code != NULL && !ShenandoahConcurrentScanCodeRoots) {

    _code_roots.code_blobs_do(code, worker_id);

void ShenandoahRootScanner::strong_roots_do(uint worker_id, OopClosure* oops, CLDClosure* clds, CodeBlobClosure* code, ThreadClosure* tc) {

  assert(ShenandoahHeap::heap()->unload_classes(), "Should be used during class unloading");

  ShenandoahParallelOopsDoThreadClosure tc_cl(oops, code, tc);

  ResourceMark rm;

  _serial_roots.oops_do(oops, worker_id);

  _cld_roots.clds_do(clds, NULL, worker_id);

  _thread_roots.threads_do(&tc_cl, worker_id);

ShenandoahRootEvacuator::ShenandoahRootEvacuator(uint n_workers, ShenandoahPhaseTimings::Phase phase) :

  ShenandoahRootProcessor(phase),

  _thread_roots(n_workers > 1),

  _weak_roots(n_workers) {

}

void ShenandoahRootEvacuator::roots_do(uint worker_id, OopClosure* oops) {

  MarkingCodeBlobClosure blobsCl(oops, CodeBlobToOopClosure::FixRelocations);

  CLDToOopClosure clds(oops, ClassLoaderData::_claim_strong);

  CLDToOopClosure* weak_clds = ShenandoahHeap::heap()->unload_classes() ? NULL : &clds;

  _serial_roots.oops_do(oops, worker_id);

  _thread_roots.oops_do(oops, NULL, worker_id);

  _cld_roots.clds_do(&clds, &clds, worker_id);

  _code_roots.code_blobs_do(&blobsCl, worker_id);

  _weak_roots.oops_do<AlwaysTrueClosure, OopClosure>(&always_true, oops, worker_id);

  _dedup_roots.oops_do(&always_true, oops, worker_id);

}

ShenandoahRootUpdater::ShenandoahRootUpdater(uint n_workers, ShenandoahPhaseTimings::Phase phase, bool update_code_cache) :

  ShenandoahRootProcessor(phase),

  _thread_roots(n_workers > 1),

  _weak_roots(n_workers),

  _update_code_cache(update_code_cache) {

ShenandoahRootAdjuster::ShenandoahRootAdjuster(uint n_workers, ShenandoahPhaseTimings::Phase phase) :

  ShenandoahRootProcessor(phase),

  _thread_roots(n_workers > 1),

  _weak_roots(n_workers) {

  assert(ShenandoahHeap::heap()->is_full_gc_in_progress(), "Full GC only");

void ShenandoahRootAdjuster::roots_do(uint worker_id, OopClosure* oops) {

  CodeBlobToOopClosure adjust_code_closure(oops, CodeBlobToOopClosure::FixRelocations);

  CLDToOopClosure adjust_cld_closure(oops, ClassLoaderData::_claim_strong);

  AlwaysTrueClosure always_true;

  _serial_roots.oops_do(oops, worker_id);

  _thread_roots.oops_do(oops, NULL, worker_id);

  _cld_roots.clds_do(&adjust_cld_closure, NULL, worker_id);

  _code_roots.code_blobs_do(&adjust_code_closure, worker_id);

  _weak_roots.oops_do<AlwaysTrueClosure, OopClosure>(&always_true, oops, worker_id);

  _dedup_roots.oops_do(&always_true, oops, worker_id);