/*

 * Copyright (c) 2019, Red Hat, Inc. All rights reserved.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 *

 */

#ifndef SHARE_GC_SHENANDOAH_SHENANDOAHROOTPROCESSOR_INLINE_HPP

#define SHARE_GC_SHENANDOAH_SHENANDOAHROOTPROCESSOR_INLINE_HPP

#include "classfile/classLoaderDataGraph.hpp"

#include "classfile/stringTable.hpp"

#include "classfile/systemDictionary.hpp"

#include "gc/shared/oopStorageParState.inline.hpp"

#include "gc/shenandoah/shenandoahHeuristics.hpp"

#include "gc/shenandoah/shenandoahRootProcessor.hpp"

#include "gc/shenandoah/shenandoahTimingTracker.hpp"

#include "gc/shenandoah/shenandoahUtils.hpp"

#include "memory/resourceArea.hpp"

#include "prims/resolvedMethodTable.hpp"

#include "runtime/safepoint.hpp"

template <bool CONCURRENT>

inline ShenandoahWeakRoot<CONCURRENT>::ShenandoahWeakRoot(OopStorage* storage, ShenandoahPhaseTimings::GCParPhases phase) :

  _itr(storage), _phase(phase) {

}

template <bool CONCURRENT>

template <typename Closure>

inline void ShenandoahWeakRoot<CONCURRENT>::oops_do(Closure* cl, uint worker_id) {

  if (CONCURRENT) {

    _itr.oops_do(cl);

  } else {

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

    _itr.oops_do(cl);

  }

}

inline ShenandoahWeakRoot<false>::ShenandoahWeakRoot(OopStorage* storage, ShenandoahPhaseTimings::GCParPhases phase) :

  _itr(storage), _phase(phase) {

}

template <typename IsAliveClosure, typename KeepAliveClosure>

void ShenandoahWeakRoot<false /* concurrent */>::weak_oops_do(IsAliveClosure* is_alive, KeepAliveClosure* keep_alive, uint worker_id) {

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

  ShenandoahWorkerTimingsTracker timer(worker_times, _phase, worker_id);

  _itr.weak_oops_do(is_alive, keep_alive);

}

template <bool CONCURRENT>

ShenandoahWeakRoots<CONCURRENT>::ShenandoahWeakRoots() :

  _jni_roots(JNIHandles::weak_global_handles(), ShenandoahPhaseTimings::JNIWeakRoots),

  _string_table_roots(StringTable::weak_storage(), ShenandoahPhaseTimings::StringTableRoots),

  _resolved_method_table_roots(ResolvedMethodTable::weak_storage(), ShenandoahPhaseTimings::ResolvedMethodTableRoots),

  _vm_roots(SystemDictionary::vm_weak_oop_storage(), ShenandoahPhaseTimings::VMWeakRoots) {

}

template <bool CONCURRENT>

template <typename Closure>

void ShenandoahWeakRoots<CONCURRENT>::oops_do(Closure* cl, uint worker_id) {

  _jni_roots.oops_do(cl, worker_id);

  _string_table_roots.oops_do(cl, worker_id);

  _resolved_method_table_roots.oops_do(cl, worker_id);

  _vm_roots.oops_do(cl, worker_id);

}

inline ShenandoahWeakRoots<false /* concurrent */>::ShenandoahWeakRoots() :

  _jni_roots(JNIHandles::weak_global_handles(), ShenandoahPhaseTimings::JNIWeakRoots),

  _string_table_roots(StringTable::weak_storage(), ShenandoahPhaseTimings::StringTableRoots),

  _resolved_method_table_roots(ResolvedMethodTable::weak_storage(), ShenandoahPhaseTimings::ResolvedMethodTableRoots),

  _vm_roots(SystemDictionary::vm_weak_oop_storage(), ShenandoahPhaseTimings::VMWeakRoots) {

}

template <typename IsAliveClosure, typename KeepAliveClosure>

void ShenandoahWeakRoots<false /* concurrent*/>::weak_oops_do(IsAliveClosure* is_alive, KeepAliveClosure* keep_alive, uint worker_id) {

  _jni_roots.weak_oops_do(is_alive, keep_alive, worker_id);

  _string_table_roots.weak_oops_do(is_alive, keep_alive, worker_id);

  _resolved_method_table_roots.weak_oops_do(is_alive, keep_alive, worker_id);

  _vm_roots.weak_oops_do(is_alive, keep_alive, worker_id);

}

template <typename Closure>

void ShenandoahWeakRoots<false /* concurrent */>::oops_do(Closure* cl, uint worker_id) {

  AlwaysTrueClosure always_true;

  weak_oops_do<AlwaysTrueClosure, Closure>(&always_true, cl, worker_id);

}

template <bool CONCURRENT>

ShenandoahJNIHandleRoots<CONCURRENT>::ShenandoahJNIHandleRoots() :

  _itr(JNIHandles::global_handles()) {

}

template <bool CONCURRENT>

template <typename T>

void ShenandoahJNIHandleRoots<CONCURRENT>::oops_do(T* cl, uint worker_id) {

  if (CONCURRENT) {

    _itr.oops_do(cl);

  } else {

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

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

    _itr.oops_do(cl);

  }

}

template <bool CONCURRENT, bool SINGLE_THREADED>

ShenandoahClassLoaderDataRoots<CONCURRENT, SINGLE_THREADED>::ShenandoahClassLoaderDataRoots() {

  if (!SINGLE_THREADED) {

    ClassLoaderDataGraph::clear_claimed_marks();

  }

  if (CONCURRENT) {

    ClassLoaderDataGraph_lock->lock();

  }

}

template <bool CONCURRENT, bool SINGLE_THREADED>

ShenandoahClassLoaderDataRoots<CONCURRENT, SINGLE_THREADED>::~ShenandoahClassLoaderDataRoots() {

  if (CONCURRENT) {

    ClassLoaderDataGraph_lock->unlock();

  }

}

template <bool CONCURRENT, bool SINGLE_THREADED>

void ShenandoahClassLoaderDataRoots<CONCURRENT, SINGLE_THREADED>::always_strong_cld_do(CLDClosure* clds, uint worker_id) {

  if (SINGLE_THREADED) {

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

    assert(Thread::current()->is_VM_thread(), "Single threaded CLDG iteration can only be done by VM thread");

    ClassLoaderDataGraph::always_strong_cld_do(clds);

  } else if (CONCURRENT) {

     ClassLoaderDataGraph::always_strong_cld_do(clds);

  } else {

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

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

   ClassLoaderDataGraph::always_strong_cld_do(clds);

  }

}

template <bool CONCURRENT, bool SINGLE_THREADED>

void ShenandoahClassLoaderDataRoots<CONCURRENT, SINGLE_THREADED>::cld_do(CLDClosure* clds, uint worker_id) {

  if (SINGLE_THREADED) {

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

    assert(Thread::current()->is_VM_thread(), "Single threaded CLDG iteration can only be done by VM thread");

    ClassLoaderDataGraph::cld_do(clds);

  } else if (CONCURRENT) {

    ClassLoaderDataGraph::cld_do(clds);

  }  else {

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

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

    ClassLoaderDataGraph::cld_do(clds);

  }

}

template <typename ITR>

ShenandoahCodeCacheRoots<ITR>::ShenandoahCodeCacheRoots() {

  nmethod::oops_do_marking_prologue();

}

template <typename ITR>

void ShenandoahCodeCacheRoots<ITR>::code_blobs_do(CodeBlobClosure* blob_cl, uint worker_id) {

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

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

  _coderoots_iterator.possibly_parallel_blobs_do(blob_cl);

}

template <typename ITR>

ShenandoahCodeCacheRoots<ITR>::~ShenandoahCodeCacheRoots() {

  nmethod::oops_do_marking_epilogue();

}

class ShenandoahParallelOopsDoThreadClosure : public ThreadClosure {

private:

  OopClosure* _f;

  CodeBlobClosure* _cf;

  ThreadClosure* _thread_cl;

public:

  ShenandoahParallelOopsDoThreadClosure(OopClosure* f, CodeBlobClosure* cf, ThreadClosure* thread_cl) :

    _f(f), _cf(cf), _thread_cl(thread_cl) {}

  void do_thread(Thread* t) {

    if (_thread_cl != NULL) {

      _thread_cl->do_thread(t);

    }

    t->oops_do(_f, _cf);

  }

};

template <typename ITR>

ShenandoahRootScanner<ITR>::ShenandoahRootScanner(uint n_workers, ShenandoahPhaseTimings::Phase phase) :

  ShenandoahRootProcessor(phase),

  _thread_roots(n_workers > 1) {

}

template <typename ITR>

void ShenandoahRootScanner<ITR>::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);

}

template <typename ITR>

void ShenandoahRootScanner<ITR>::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);

}

template <typename ITR>

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

  assert(!ShenandoahSafepoint::is_at_shenandoah_safepoint() ||

         !ShenandoahHeap::heap()->unload_classes() ||

          ShenandoahHeap::heap()->is_traversal_mode(),

          "Expect class unloading or traversal when Shenandoah cycle is running");

  ShenandoahParallelOopsDoThreadClosure tc_cl(oops, code, tc);

  ResourceMark rm;

  _serial_roots.oops_do(oops, worker_id);

  _jni_roots.oops_do(oops, worker_id);

  if (clds != NULL) {

    _cld_roots.cld_do(clds, worker_id);

  } else {

    assert(ShenandoahHeap::heap()->is_concurrent_traversal_in_progress(), "Only possible with traversal GC");

  }

  _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);

  }

}

template <typename ITR>

void ShenandoahRootScanner<ITR>::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);

  _jni_roots.oops_do(oops, worker_id);

  _cld_roots.always_strong_cld_do(clds, worker_id);

  _thread_roots.threads_do(&tc_cl, worker_id);

}

template <typename IsAlive, typename KeepAlive>

void ShenandoahRootUpdater::roots_do(uint worker_id, IsAlive* is_alive, KeepAlive* keep_alive) {

  CodeBlobToOopClosure update_blobs(keep_alive, CodeBlobToOopClosure::FixRelocations);

  CLDToOopClosure clds(keep_alive, ClassLoaderData::_claim_strong);

  _serial_roots.oops_do(keep_alive, worker_id);

  _jni_roots.oops_do(keep_alive, worker_id);

  _thread_roots.oops_do(keep_alive, NULL, worker_id);

  _cld_roots.cld_do(&clds, worker_id);

  if(_update_code_cache) {

    _code_roots.code_blobs_do(&update_blobs, worker_id);

  }

  _serial_weak_roots.weak_oops_do(is_alive, keep_alive, worker_id);

  _weak_roots.weak_oops_do(is_alive, keep_alive, worker_id);

  _dedup_roots.oops_do(is_alive, keep_alive, worker_id);

}

#endif // SHARE_GC_SHENANDOAH_SHENANDOAHROOTPROCESSOR_INLINE_HPP