def linearize_arg_types(fn, args):
"""
Given a function object which might be one of:
(1) a closure type
(2) the name of an untyped function
(3) an untyped fn object
and some argument types which might
(1) a list
(2) a tuple
(3) an ActualArgs object
linearize the argument types with respect to
the untyped function's argument order and return
both the untyped function and list of arguments
"""
untyped_fundef, closure_args = unpack_closure(fn)
if isinstance(args, (list, tuple)):
args = ActualArgs(args)
if not isinstance(untyped_fundef.args, FormalArgs):
assert False, "Expected function %s to have FormalArgs" % untyped_fundef.name
#formals = FormalArgs()
#for arg in untyped_fundef.args:
# formals.add_positional(arg)
else:
formals = untyped_fundef.args
if len(closure_args) > 0:
args = args.prepend_positional(closure_args)
def keyword_fn(_, v):
return type_conv.typeof(v)
try:
linear_args, extra = formals.linearize_values(args,
keyword_fn=keyword_fn)
except UnexpectedKeyword as e:
e.fn_name = untyped_fundef.name
raise
return untyped_fundef, tuple(linear_args + extra)
def linearize_arg_types(fn, args):
"""
Given a function object which might be one of:
(1) a closure type
(2) the name of an untyped function
(3) an untyped fn object
and some argument types which might
(1) a list
(2) a tuple
(3) an ActualArgs object
linearize the argument types with respect to
the untyped function's argument order and return
both the untyped function and list of arguments
"""
untyped_fundef, closure_args = unpack_closure(fn)
if isinstance(args, (list, tuple)):
args = ActualArgs(args)
if not isinstance(untyped_fundef.args, FormalArgs):
assert False, "Expected function %s to have FormalArgs" % untyped_fundef.name
#formals = FormalArgs()
#for arg in untyped_fundef.args:
# formals.add_positional(arg)
else:
formals = untyped_fundef.args
if len(closure_args) > 0:
args = args.prepend_positional(closure_args)
def keyword_fn(_, v):
return type_conv.typeof(v)
try:
linear_args, extra = formals.linearize_values(args, keyword_fn = keyword_fn)
except UnexpectedKeyword as e:
e.fn_name = untyped_fundef.name
raise
return untyped_fundef, tuple(linear_args + extra)
def linearize_actual_args(fn, args):
untyped_fn, closure_args = unpack_closure(fn)
if isinstance(args, (list, tuple)):
args = ActualArgs(args)
args = args.prepend_positional(closure_args)
arg_types = args.transform(get_type)
# Drop arguments that are assigned defaults,
# since we're assuming those are set in the body
# of the function
if isinstance(untyped_fn.args, FormalArgs):
try:
combined_args = untyped_fn.args.linearize_without_defaults(args, tuple_elts)
except UnexpectedKeyword as e:
e.fn_name = fn.name
raise
else:
combined_args = list(args)
return untyped_fn, combined_args, arg_types
def linearize_actual_args(fn, args):
untyped_fn, closure_args = unpack_closure(fn)
if isinstance(args, (list, tuple)):
args = ActualArgs(args)
args = args.prepend_positional(closure_args)
arg_types = args.transform(get_type)
# Drop arguments that are assigned defaults,
# since we're assuming those are set in the body
# of the function
if isinstance(untyped_fn.args, FormalArgs):
try:
combined_args = untyped_fn.args.linearize_without_defaults(
args, tuple_elts)
except UnexpectedKeyword as e:
e.fn_name = fn.name
raise
else:
combined_args = list(args)
return untyped_fn, combined_args, arg_types
