/*

 * Copyright (c) 2000, 2011, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * 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.

 *

 */

#include "precompiled.hpp"

#include "ci/bcEscapeAnalyzer.hpp"

#include "ci/ciCPCache.hpp"

#include "ci/ciMethodHandle.hpp"

#include "classfile/javaClasses.hpp"

#include "compiler/compileLog.hpp"

#include "opto/addnode.hpp"

#include "opto/callGenerator.hpp"

#include "opto/callnode.hpp"

#include "opto/cfgnode.hpp"

#include "opto/connode.hpp"

#include "opto/parse.hpp"

#include "opto/rootnode.hpp"

#include "opto/runtime.hpp"

#include "opto/subnode.hpp"

CallGenerator::CallGenerator(ciMethod* method) {

  _method = method;

}

// Utility function.

const TypeFunc* CallGenerator::tf() const {

  return TypeFunc::make(method());

}

//-----------------------------ParseGenerator---------------------------------

// Internal class which handles all direct bytecode traversal.

class ParseGenerator : public InlineCallGenerator {

private:

  bool  _is_osr;

  float _expected_uses;

public:

  ParseGenerator(ciMethod* method, float expected_uses, bool is_osr = false)

    : InlineCallGenerator(method)

  {

    _is_osr        = is_osr;

    _expected_uses = expected_uses;

    assert(can_parse(method, is_osr), "parse must be possible");

  }

  // Can we build either an OSR or a regular parser for this method?

  static bool can_parse(ciMethod* method, int is_osr = false);

  virtual bool      is_parse() const           { return true; }

  virtual JVMState* generate(JVMState* jvms);

  int is_osr() { return _is_osr; }

};

JVMState* ParseGenerator::generate(JVMState* jvms) {

  Compile* C = Compile::current();

  if (is_osr()) {

    // The JVMS for a OSR has a single argument (see its TypeFunc).

    assert(jvms->depth() == 1, "no inline OSR");

  }

  if (C->failing()) {

    return NULL;  // bailing out of the compile; do not try to parse

  }

  Parse parser(jvms, method(), _expected_uses);

  // Grab signature for matching/allocation

#ifdef ASSERT

  if (parser.tf() != (parser.depth() == 1 ? C->tf() : tf())) {

    MutexLockerEx ml(Compile_lock, Mutex::_no_safepoint_check_flag);

    assert(C->env()->system_dictionary_modification_counter_changed(),

           "Must invalidate if TypeFuncs differ");

  }

#endif

  GraphKit& exits = parser.exits();

  if (C->failing()) {

    while (exits.pop_exception_state() != NULL) ;

    return NULL;

  }

  assert(exits.jvms()->same_calls_as(jvms), "sanity");

  // Simply return the exit state of the parser,

  // augmented by any exceptional states.

  return exits.transfer_exceptions_into_jvms();

}

//---------------------------DirectCallGenerator------------------------------

// Internal class which handles all out-of-line calls w/o receiver type checks.

class DirectCallGenerator : public CallGenerator {

 private:

  CallStaticJavaNode* _call_node;

  // Force separate memory and I/O projections for the exceptional

  // paths to facilitate late inlinig.

  bool                _separate_io_proj;

 public:

  DirectCallGenerator(ciMethod* method, bool separate_io_proj)

    : CallGenerator(method),

      _separate_io_proj(separate_io_proj)

  {

  }

  virtual JVMState* generate(JVMState* jvms);

  CallStaticJavaNode* call_node() const { return _call_node; }

};

JVMState* DirectCallGenerator::generate(JVMState* jvms) {

  GraphKit kit(jvms);

  bool is_static = method()->is_static();

  address target = is_static ? SharedRuntime::get_resolve_static_call_stub()

                             : SharedRuntime::get_resolve_opt_virtual_call_stub();

  if (kit.C->log() != NULL) {

    kit.C->log()->elem("direct_call bci='%d'", jvms->bci());

  }

  CallStaticJavaNode *call = new (kit.C, tf()->domain()->cnt()) CallStaticJavaNode(tf(), target, method(), kit.bci());

  if (!is_static) {

    // Make an explicit receiver null_check as part of this call.

    // Since we share a map with the caller, his JVMS gets adjusted.

    kit.null_check_receiver(method());

    if (kit.stopped()) {

      // And dump it back to the caller, decorated with any exceptions:

      return kit.transfer_exceptions_into_jvms();

    }

    // Mark the call node as virtual, sort of:

    call->set_optimized_virtual(true);

    if (method()->is_method_handle_invoke()) {

      call->set_method_handle_invoke(true);

      kit.C->set_has_method_handle_invokes(true);

    }

  }

  kit.set_arguments_for_java_call(call);

  kit.set_edges_for_java_call(call, false, _separate_io_proj);

  Node* ret = kit.set_results_for_java_call(call, _separate_io_proj);

  kit.push_node(method()->return_type()->basic_type(), ret);

  _call_node = call;  // Save the call node in case we need it later

  return kit.transfer_exceptions_into_jvms();

}

//---------------------------DynamicCallGenerator-----------------------------

// Internal class which handles all out-of-line invokedynamic calls.

class DynamicCallGenerator : public CallGenerator {

public:

  DynamicCallGenerator(ciMethod* method)

    : CallGenerator(method)

  {

  }

  virtual JVMState* generate(JVMState* jvms);

};

JVMState* DynamicCallGenerator::generate(JVMState* jvms) {

  GraphKit kit(jvms);

  if (kit.C->log() != NULL) {

    kit.C->log()->elem("dynamic_call bci='%d'", jvms->bci());

  }

  // Get the constant pool cache from the caller class.

  ciMethod* caller_method = jvms->method();

  ciBytecodeStream str(caller_method);

  str.force_bci(jvms->bci());  // Set the stream to the invokedynamic bci.

  assert(str.cur_bc() == Bytecodes::_invokedynamic, "wrong place to issue a dynamic call!");

  ciCPCache* cpcache = str.get_cpcache();

  // Get the offset of the CallSite from the constant pool cache

  // pointer.

  int index = str.get_method_index();

  size_t call_site_offset = cpcache->get_f1_offset(index);

  // Load the CallSite object from the constant pool cache.

  const TypeOopPtr* cpcache_ptr = TypeOopPtr::make_from_constant(cpcache);

  Node* cpcache_adr = kit.makecon(cpcache_ptr);

  Node* call_site_adr = kit.basic_plus_adr(cpcache_adr, cpcache_adr, call_site_offset);

  Node* call_site = kit.make_load(kit.control(), call_site_adr, TypeInstPtr::BOTTOM, T_OBJECT, Compile::AliasIdxRaw);

  // Load the target MethodHandle from the CallSite object.

  Node* target_mh_adr = kit.basic_plus_adr(call_site, call_site, java_lang_invoke_CallSite::target_offset_in_bytes());

  Node* target_mh = kit.make_load(kit.control(), target_mh_adr, TypeInstPtr::BOTTOM, T_OBJECT);

  address resolve_stub = SharedRuntime::get_resolve_opt_virtual_call_stub();

  CallStaticJavaNode *call = new (kit.C, tf()->domain()->cnt()) CallStaticJavaNode(tf(), resolve_stub, method(), kit.bci());

  // invokedynamic is treated as an optimized invokevirtual.

  call->set_optimized_virtual(true);

  // Take extra care (in the presence of argument motion) not to trash the SP:

  call->set_method_handle_invoke(true);

  kit.C->set_has_method_handle_invokes(true);

  // Pass the target MethodHandle as first argument and shift the

  // other arguments.

  call->init_req(0 + TypeFunc::Parms, target_mh);

  uint nargs = call->method()->arg_size();

  for (uint i = 1; i < nargs; i++) {

    Node* arg = kit.argument(i - 1);

    call->init_req(i + TypeFunc::Parms, arg);

  }

  kit.set_edges_for_java_call(call);

  Node* ret = kit.set_results_for_java_call(call);

  kit.push_node(method()->return_type()->basic_type(), ret);

  return kit.transfer_exceptions_into_jvms();

}

//--------------------------VirtualCallGenerator------------------------------

// Internal class which handles all out-of-line calls checking receiver type.

class VirtualCallGenerator : public CallGenerator {

private:

  int _vtable_index;

public:

  VirtualCallGenerator(ciMethod* method, int vtable_index)

    : CallGenerator(method), _vtable_index(vtable_index)

  {

    assert(vtable_index == methodOopDesc::invalid_vtable_index ||

           vtable_index >= 0, "either invalid or usable");

  }

  virtual bool      is_virtual() const          { return true; }

  virtual JVMState* generate(JVMState* jvms);

};

JVMState* VirtualCallGenerator::generate(JVMState* jvms) {

  GraphKit kit(jvms);

  Node* receiver = kit.argument(0);

  if (kit.C->log() != NULL) {

    kit.C->log()->elem("virtual_call bci='%d'", jvms->bci());

  }

  // If the receiver is a constant null, do not torture the system

  // by attempting to call through it.  The compile will proceed

  // correctly, but may bail out in final_graph_reshaping, because

  // the call instruction will have a seemingly deficient out-count.

  // (The bailout says something misleading about an "infinite loop".)

  if (kit.gvn().type(receiver)->higher_equal(TypePtr::NULL_PTR)) {

    kit.inc_sp(method()->arg_size());  // restore arguments

    kit.uncommon_trap(Deoptimization::Reason_null_check,

                      Deoptimization::Action_none,

                      NULL, "null receiver");

    return kit.transfer_exceptions_into_jvms();

  }

  // Ideally we would unconditionally do a null check here and let it

  // be converted to an implicit check based on profile information.

  // However currently the conversion to implicit null checks in

  // Block::implicit_null_check() only looks for loads and stores, not calls.

  ciMethod *caller = kit.method();

  ciMethodData *caller_md = (caller == NULL) ? NULL : caller->method_data();

  if (!UseInlineCaches || !ImplicitNullChecks ||

       ((ImplicitNullCheckThreshold > 0) && caller_md &&

       (caller_md->trap_count(Deoptimization::Reason_null_check)

       >= (uint)ImplicitNullCheckThreshold))) {

    // Make an explicit receiver null_check as part of this call.

    // Since we share a map with the caller, his JVMS gets adjusted.

    receiver = kit.null_check_receiver(method());

    if (kit.stopped()) {

      // And dump it back to the caller, decorated with any exceptions:

      return kit.transfer_exceptions_into_jvms();

    }

  }

  assert(!method()->is_static(), "virtual call must not be to static");

  assert(!method()->is_final(), "virtual call should not be to final");

  assert(!method()->is_private(), "virtual call should not be to private");

  assert(_vtable_index == methodOopDesc::invalid_vtable_index || !UseInlineCaches,

         "no vtable calls if +UseInlineCaches ");

  address target = SharedRuntime::get_resolve_virtual_call_stub();

  // Normal inline cache used for call

  CallDynamicJavaNode *call = new (kit.C, tf()->domain()->cnt()) CallDynamicJavaNode(tf(), target, method(), _vtable_index, kit.bci());

  kit.set_arguments_for_java_call(call);

  kit.set_edges_for_java_call(call);

  Node* ret = kit.set_results_for_java_call(call);

  kit.push_node(method()->return_type()->basic_type(), ret);

  // Represent the effect of an implicit receiver null_check

  // as part of this call.  Since we share a map with the caller,

  // his JVMS gets adjusted.

  kit.cast_not_null(receiver);

  return kit.transfer_exceptions_into_jvms();

}

bool ParseGenerator::can_parse(ciMethod* m, int entry_bci) {

  // Certain methods cannot be parsed at all:

  if (!m->can_be_compiled())              return false;

  if (!m->has_balanced_monitors())        return false;

  if (m->get_flow_analysis()->failing())  return false;

  // (Methods may bail out for other reasons, after the parser is run.

  // We try to avoid this, but if forced, we must return (Node*)NULL.

  // The user of the CallGenerator must check for this condition.)

  return true;

}

CallGenerator* CallGenerator::for_inline(ciMethod* m, float expected_uses) {

  if (!ParseGenerator::can_parse(m))  return NULL;

  return new ParseGenerator(m, expected_uses);

}

// As a special case, the JVMS passed to this CallGenerator is

// for the method execution already in progress, not just the JVMS

// of the caller.  Thus, this CallGenerator cannot be mixed with others!

CallGenerator* CallGenerator::for_osr(ciMethod* m, int osr_bci) {

  if (!ParseGenerator::can_parse(m, true))  return NULL;

  float past_uses = m->interpreter_invocation_count();

  float expected_uses = past_uses;

  return new ParseGenerator(m, expected_uses, true);

}

CallGenerator* CallGenerator::for_direct_call(ciMethod* m, bool separate_io_proj) {

  assert(!m->is_abstract(), "for_direct_call mismatch");

  return new DirectCallGenerator(m, separate_io_proj);

}

CallGenerator* CallGenerator::for_dynamic_call(ciMethod* m) {

  assert(m->is_method_handle_invoke(), "for_dynamic_call mismatch");

  return new DynamicCallGenerator(m);

}

CallGenerator* CallGenerator::for_virtual_call(ciMethod* m, int vtable_index) {

  assert(!m->is_static(), "for_virtual_call mismatch");

  assert(!m->is_method_handle_invoke(), "should be a direct call");

  return new VirtualCallGenerator(m, vtable_index);

}

// Allow inlining decisions to be delayed

class LateInlineCallGenerator : public DirectCallGenerator {

  CallGenerator* _inline_cg;

 public:

  LateInlineCallGenerator(ciMethod* method, CallGenerator* inline_cg) :

    DirectCallGenerator(method, true), _inline_cg(inline_cg) {}

  virtual bool      is_late_inline() const { return true; }

  // Convert the CallStaticJava into an inline

  virtual void do_late_inline();

  JVMState* generate(JVMState* jvms) {

    // Record that this call site should be revisited once the main

    // parse is finished.

    Compile::current()->add_late_inline(this);

    // Emit the CallStaticJava and request separate projections so

    // that the late inlining logic can distinguish between fall

    // through and exceptional uses of the memory and io projections

    // as is done for allocations and macro expansion.

    return DirectCallGenerator::generate(jvms);

  }

};

void LateInlineCallGenerator::do_late_inline() {

  // Can't inline it

  if (call_node() == NULL || call_node()->outcnt() == 0 ||

      call_node()->in(0) == NULL || call_node()->in(0)->is_top())

    return;

  CallStaticJavaNode* call = call_node();

  // Make a clone of the JVMState that appropriate to use for driving a parse

  Compile* C = Compile::current();

  JVMState* jvms     = call->jvms()->clone_shallow(C);

  uint size = call->req();

  SafePointNode* map = new (C, size) SafePointNode(size, jvms);

  for (uint i1 = 0; i1 < size; i1++) {

    map->init_req(i1, call->in(i1));

  }

  // Make sure the state is a MergeMem for parsing.

  if (!map->in(TypeFunc::Memory)->is_MergeMem()) {

    map->set_req(TypeFunc::Memory, MergeMemNode::make(C, map->in(TypeFunc::Memory)));

  }

  // Make enough space for the expression stack and transfer the incoming arguments

  int nargs    = method()->arg_size();

  jvms->set_map(map);

  map->ensure_stack(jvms, jvms->method()->max_stack());

  if (nargs > 0) {

    for (int i1 = 0; i1 < nargs; i1++) {

      map->set_req(i1 + jvms->argoff(), call->in(TypeFunc::Parms + i1));

    }

  }

  CompileLog* log = C->log();

  if (log != NULL) {

    log->head("late_inline method='%d'", log->identify(method()));

    JVMState* p = jvms;

    while (p != NULL) {

      log->elem("jvms bci='%d' method='%d'", p->bci(), log->identify(p->method()));

      p = p->caller();

    }

    log->tail("late_inline");

  }

  // Setup default node notes to be picked up by the inlining

  Node_Notes* old_nn = C->default_node_notes();

  if (old_nn != NULL) {

    Node_Notes* entry_nn = old_nn->clone(C);

    entry_nn->set_jvms(jvms);

    C->set_default_node_notes(entry_nn);

  }

  // Now perform the inling using the synthesized JVMState

  JVMState* new_jvms = _inline_cg->generate(jvms);

  if (new_jvms == NULL)  return;  // no change

  if (C->failing())      return;

  // Capture any exceptional control flow

  GraphKit kit(new_jvms);

  // Find the result object

  Node* result = C->top();

  int   result_size = method()->return_type()->size();

  if (result_size != 0 && !kit.stopped()) {

    result = (result_size == 1) ? kit.pop() : kit.pop_pair();

  }

  kit.replace_call(call, result);

}

CallGenerator* CallGenerator::for_late_inline(ciMethod* method, CallGenerator* inline_cg) {

  return new LateInlineCallGenerator(method, inline_cg);

}

//---------------------------WarmCallGenerator--------------------------------

// Internal class which handles initial deferral of inlining decisions.

class WarmCallGenerator : public CallGenerator {

  WarmCallInfo*   _call_info;

  CallGenerator*  _if_cold;

  CallGenerator*  _if_hot;

  bool            _is_virtual;   // caches virtuality of if_cold

  bool            _is_inline;    // caches inline-ness of if_hot

public:

  WarmCallGenerator(WarmCallInfo* ci,

                    CallGenerator* if_cold,

                    CallGenerator* if_hot)

    : CallGenerator(if_cold->method())

  {

    assert(method() == if_hot->method(), "consistent choices");

    _call_info  = ci;

    _if_cold    = if_cold;

    _if_hot     = if_hot;

    _is_virtual = if_cold->is_virtual();

    _is_inline  = if_hot->is_inline();

  }

  virtual bool      is_inline() const           { return _is_inline; }

  virtual bool      is_virtual() const          { return _is_virtual; }

  virtual bool      is_deferred() const         { return true; }

  virtual JVMState* generate(JVMState* jvms);

};

CallGenerator* CallGenerator::for_warm_call(WarmCallInfo* ci,

                                            CallGenerator* if_cold,

                                            CallGenerator* if_hot) {

  return new WarmCallGenerator(ci, if_cold, if_hot);

}

JVMState* WarmCallGenerator::generate(JVMState* jvms) {

  Compile* C = Compile::current();

  if (C->log() != NULL) {

    C->log()->elem("warm_call bci='%d'", jvms->bci());

  }

  jvms = _if_cold->generate(jvms);

  if (jvms != NULL) {

    Node* m = jvms->map()->control();

    if (m->is_CatchProj()) m = m->in(0);  else m = C->top();

    if (m->is_Catch())     m = m->in(0);  else m = C->top();

    if (m->is_Proj())      m = m->in(0);  else m = C->top();

    if (m->is_CallJava()) {

      _call_info->set_call(m->as_Call());

      _call_info->set_hot_cg(_if_hot);

#ifndef PRODUCT

      if (PrintOpto || PrintOptoInlining) {

        tty->print_cr("Queueing for warm inlining at bci %d:", jvms->bci());

        tty->print("WCI: ");

        _call_info->print();

      }

#endif

      _call_info->set_heat(_call_info->compute_heat());

      C->set_warm_calls(_call_info->insert_into(C->warm_calls()));