internal override void RewriteBody(MSAst.ExpressionVisitor visitor)
{
_dlrBody = null; // clear the cached body if we've been reduced
MSAst.Expression funcCode = GlobalParent.Constant(GetOrMakeFunctionCode());
FuncCodeExpr = funcCode;
Body = new RewrittenBodyStatement(Body, visitor.Visit(Body));
}
/// <summary>
/// Creates the LambdaExpression which is the actual function body.
/// </summary>
private LightLambdaExpression EnsureFunctionLambda()
{
if (_dlrBody == null)
{
PerfTrack.NoteEvent(PerfTrack.Categories.Compiler, "Creating FunctionBody");
_dlrBody = CreateFunctionLambda();
}
return(_dlrBody);
}
internal override LightLambdaExpression GetLambda()
{
if (_dlrBody == null)
{
PerfTrack.NoteEvent(PerfTrack.Categories.Compiler, "Creating FunctionBody");
_dlrBody = MakeClassBody();
}
return(_dlrBody);
}
/// <summary>
/// Returns an expression which creates the function object.
/// </summary>
internal MSAst.Expression MakeFunctionExpression()
{
var defaults = new List <MSAst.Expression>();
var kwdefaults = new List <MSAst.Expression>();
var annotations = new List <MSAst.Expression>();
if (ReturnAnnotation != null)
{
// value needs to come before key in the array
annotations.Add(AstUtils.Convert(ReturnAnnotation, typeof(object)));
annotations.Add(Ast.Constant("return", typeof(string)));
}
foreach (var param in _parameters)
{
if (param.Kind == ParameterKind.Normal && param.DefaultValue != null)
{
defaults.Add(AstUtils.Convert(param.DefaultValue, typeof(object)));
}
if (param.Kind == ParameterKind.KeywordOnly && param.DefaultValue != null)
{
// value needs to come before key in the array
kwdefaults.Add(AstUtils.Convert(param.DefaultValue, typeof(object)));
kwdefaults.Add(Ast.Constant(param.Name, typeof(string)));
}
if (param.Annotation != null)
{
// value needs to come before key in the array
annotations.Add(AstUtils.Convert(param.Annotation, typeof(object)));
annotations.Add(Ast.Constant(param.Name, typeof(string)));
}
}
MSAst.Expression funcCode = GlobalParent.Constant(GetOrMakeFunctionCode());
FuncCodeExpr = funcCode;
MSAst.Expression ret;
if (EmitDebugFunction())
{
LightLambdaExpression code = CreateFunctionLambda();
// we need to compile all of the debuggable code together at once otherwise mdbg gets confused. If we're
// in tracing mode we'll still compile things one off though just to keep things simple. The code will still
// be debuggable but naive debuggers like mdbg will have more issues.
ret = Ast.Call(
AstMethods.MakeFunctionDebug, // method
Parent.LocalContext, // 1. Emit CodeContext
FuncCodeExpr, // 2. FunctionCode
((IPythonGlobalExpression)GetVariableExpression(_nameVariable)).RawValue(), // 3. module name
defaults.Count == 0 ? // 4. default values
AstUtils.Constant(null, typeof(object[])) :
(MSAst.Expression)Ast.NewArrayInit(typeof(object), defaults),
kwdefaults.Count == 0 ? AstUtils.Constant(null, typeof(PythonDictionary)) :
(MSAst.Expression)Ast.Call( // 5. kwdefaults
AstMethods.MakeDictFromItems,
Ast.NewArrayInit(
typeof(object),
kwdefaults
)
),
annotations.Count == 0 ? AstUtils.Constant(null, typeof(PythonDictionary)) :
(MSAst.Expression)Ast.Call( // 6. annotations
AstMethods.MakeDictFromItems,
Ast.NewArrayInit(
typeof(object),
annotations
)
),
IsGenerator ?
(MSAst.Expression) new PythonGeneratorExpression(code, GlobalParent.PyContext.Options.CompilationThreshold) :
(MSAst.Expression)code
);
}
else
{
ret = Ast.Call(
AstMethods.MakeFunction, // method
Parent.LocalContext, // 1. Emit CodeContext
FuncCodeExpr, // 2. FunctionCode
((IPythonGlobalExpression)GetVariableExpression(_nameVariable)).RawValue(), // 3. module name
defaults.Count == 0 ? // 4. default values
AstUtils.Constant(null, typeof(object[])) :
(MSAst.Expression)Ast.NewArrayInit(typeof(object), defaults),
kwdefaults.Count == 0 ? AstUtils.Constant(null, typeof(PythonDictionary)) :
(MSAst.Expression)Ast.Call( // 5. kwdefaults
AstMethods.MakeDictFromItems,
Ast.NewArrayInit(
typeof(object),
kwdefaults
)
),
annotations.Count == 0 ? AstUtils.Constant(null, typeof(PythonDictionary)) :
(MSAst.Expression)Ast.Call( // 6. annotations
AstMethods.MakeDictFromItems,
Ast.NewArrayInit(
typeof(object),
annotations
)
)
);
}
return(AddDecorators(ret, Decorators));
}
internal override void RewriteBody(MSAst.ExpressionVisitor visitor)
{
_dlrBody = null;
_body = new RewrittenBodyStatement(Body, visitor.Visit(Body));
}
