class TypeProf::ISeqMethodDef

# File typeprof-0.21.9/lib/typeprof/method.rb, line 9
def initialize(iseq, cref, outer_ep, pub_meth)
  @iseq = iseq
  raise if iseq.nil?
  @cref = cref
  @outer_ep = outer_ep
  @pub_meth = pub_meth
end

# File typeprof-0.21.9/lib/typeprof/method.rb, line 52
def do_check_send(msig, recv, mid, ep, scratch)
  klass, singleton = recv.method_dispatch_info
  cur_subst = {}
  direct_method = true
  scratch.adjust_substitution(klass, singleton, mid, self, recv.generate_substitution) do |subst, direct|
    direct_method &&= direct
    cur_subst = Type.merge_substitution(cur_subst, subst)
  end

  lead_num = @iseq.fargs_format[:lead_num] || 0
  post_num = @iseq.fargs_format[:post_num] || 0
  rest_start = @iseq.fargs_format[:rest_start]
  opt = @iseq.fargs_format[:opt] || [0]

  # TODO: check keywords
  if rest_start
    # almost ok
  else
    if msig.rest_ty
      scratch.error(ep, "RBS says that a rest argument is accepted, but the method definition does not accept one")
      return
    end
    if msig.lead_tys.size + msig.post_tys.size < lead_num + post_num
      scratch.error(ep, "RBS says that the arity may be %d, but the method definition requires at least %d arguments" % [msig.lead_tys.size + msig.post_tys.size, lead_num + post_num])
      return
    end
    if msig.lead_tys.size + msig.opt_tys.size + msig.post_tys.size > lead_num + opt.size - 1 + post_num
      scratch.error(ep, "RBS says that the arity may be %d, but the method definition requires at most %d arguments" % [msig.lead_tys.size + msig.opt_tys.size + msig.post_tys.size, lead_num + opt.size - 1 + post_num])
      return
    end
  end

  lead_num = @iseq.fargs_format[:lead_num] || 0
  post_start = @iseq.fargs_format[:post_start]
  kw_start = @iseq.fargs_format[:kwbits]
  keyword = @iseq.fargs_format[:keyword]
  kw_start -= keyword.size if kw_start
  kw_rest = @iseq.fargs_format[:kwrest]
  block_start = @iseq.fargs_format[:block_start]

  # XXX: need to check .rbs msig and .rb fargs

  ctx = Context.new(@iseq, @cref, mid)
  callee_ep = ExecutionPoint.new(ctx, 0, @outer_ep)

  locals = [Type.nil] * @iseq.locals.size
  nenv = Env.new(StaticEnv.new(recv, msig.blk_ty, false, true), locals, [], Utils::HashWrapper.new({}))
  alloc_site = AllocationSite.new(callee_ep)
  idx = 0
  msig.lead_tys.each_with_index do |ty, i|
    alloc_site2 = alloc_site.add_id(idx += 1)
    ty = ty.substitute(cur_subst, Config.current.options[:type_depth_limit]).remove_type_vars
    nenv, ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
    nenv = nenv.local_update(i, ty)
  end
  if msig.opt_tys
    msig.opt_tys.each_with_index do |ty, i|
      alloc_site2 = alloc_site.add_id(idx += 1)
      ty = ty.substitute(cur_subst, Config.current.options[:type_depth_limit]).remove_type_vars
      nenv, ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
      nenv = nenv.local_update(lead_num + i, ty)
    end
  end
  if msig.rest_ty
    alloc_site2 = alloc_site.add_id(idx += 1)
    ty = Type::Array.new(Type::Array::Elements.new([], msig.rest_ty), Type::Instance.new(Type::Builtin[:ary]))
    ty = ty.substitute(cur_subst, Config.current.options[:type_depth_limit]).remove_type_vars
    nenv, rest_ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
    # TODO: handle a case where rest_start is not found
    nenv = nenv.local_update(rest_start, rest_ty)
  elsif rest_start
    alloc_site2 = alloc_site.add_id(idx += 1)
    ty = Type::Array.new(Type::Array::Elements.new([], Type.any), Type::Instance.new(Type::Builtin[:ary]))
    nenv, rest_ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
    nenv = nenv.local_update(rest_start, rest_ty)
  end
  if msig.post_tys
    msig.post_tys.each_with_index do |ty, i|
      alloc_site2 = alloc_site.add_id(idx += 1)
      ty = ty.substitute(cur_subst, Config.current.options[:type_depth_limit]).remove_type_vars
      nenv, ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
      nenv = nenv.local_update(post_start + i, ty)
    end
  end
  if msig.kw_tys && keyword # TODO: support the case where RBS writes kw_tys and RB method accepts **kwrest
    msig.kw_tys.each do |_, key, ty|
      i = keyword.index {|callee_key,| callee_key == key }
      unless i
        # warn
        next
      end
      alloc_site2 = alloc_site.add_id(key)
      ty = ty.substitute(cur_subst, Config.current.options[:type_depth_limit]).remove_type_vars
      nenv, ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
      nenv = nenv.local_update(kw_start + i, ty)
    end
  end
  if msig.kw_rest_ty
    ty = msig.kw_rest_ty
    alloc_site2 = alloc_site.add_id(:**)
    ty = ty.substitute(cur_subst, Config.current.options[:type_depth_limit]).remove_type_vars
    nenv, ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
    nenv = nenv.local_update(kw_rest, ty)
  elsif kw_rest
    alloc_site2 = alloc_site.add_id(:**)
    ty = Type.gen_hash {}
    nenv, ty = scratch.localize_type(ty, nenv, callee_ep, alloc_site2)
    nenv = nenv.local_update(kw_rest, ty)
  end
  nenv = nenv.local_update(block_start, msig.blk_ty) if block_start

  opt.each do |start_pc|
    scratch.merge_env(callee_ep.jump(start_pc), nenv)
  end
  scratch.add_executed_iseq(@iseq)

  ctx
end

# File typeprof-0.21.9/lib/typeprof/method.rb, line 19
def do_send(recv, mid, aargs, caller_ep, caller_env, scratch, &ctn)
  recv = recv.base_type while recv.respond_to?(:base_type)
  recv = scratch.globalize_type(recv, caller_env, caller_ep)
  aargs = scratch.globalize_type(aargs, caller_env, caller_ep)

  locals = [Type.nil] * @iseq.locals.size

  blk_ty, start_pcs = aargs.setup_formal_arguments(:method, locals, @iseq.fargs_format)
  if blk_ty.is_a?(String)
    scratch.error(caller_ep, blk_ty)
    ctn[Type.any, caller_ep, caller_env]
    return
  end

  nctx = Context.new(@iseq, @cref, mid)
  callee_ep = ExecutionPoint.new(nctx, 0, @outer_ep)
  nenv = Env.new(StaticEnv.new(recv, blk_ty, false, true), locals, [], Utils::HashWrapper.new({}))
  alloc_site = AllocationSite.new(callee_ep)
  locals.each_with_index do |ty, i|
    alloc_site2 = alloc_site.add_id(i)
    # nenv is top-level, so it is okay to call Type#localize directly
    nenv, ty = ty.localize(nenv, alloc_site2, Config.current.options[:type_depth_limit])
    nenv = nenv.local_update(i, ty)
  end

  start_pcs.each do |start_pc|
    scratch.merge_env(ExecutionPoint.new(nctx, start_pc, @outer_ep), nenv)
  end

  scratch.add_iseq_method_call!(self, nctx)
  scratch.add_callsite!(nctx, caller_ep, caller_env, &ctn)
end