2 // linq.cs: support for query expressions
4 // Authors: Marek Safar (marek.safar@gmail.com)
6 // Dual licensed under the terms of the MIT X11 or GNU GPL
8 // Copyright 2007-2008 Novell, Inc
12 using System.Collections.Generic;
14 namespace Mono.CSharp.Linq
16 public class QueryExpression : AQueryClause
18 public QueryExpression (AQueryClause start)
19 : base (null, null, start.Location)
24 public override Expression BuildQueryClause (ResolveContext ec, Expression lSide, Parameter parentParameter)
26 return next.BuildQueryClause (ec, lSide, parentParameter);
29 protected override Expression DoResolve (ResolveContext ec)
31 int counter = QueryBlock.TransparentParameter.Counter;
33 Expression e = BuildQueryClause (ec, null, null);
38 // Reset counter in probing mode to ensure that all transparent
39 // identifier anonymous types are created only once
41 if (ec.IsInProbingMode)
42 QueryBlock.TransparentParameter.Counter = counter;
47 protected override string MethodName {
48 get { throw new NotSupportedException (); }
52 public abstract class AQueryClause : ShimExpression
54 protected class QueryExpressionAccess : MemberAccess
56 public QueryExpressionAccess (Expression expr, string methodName, Location loc)
57 : base (expr, methodName, loc)
61 public QueryExpressionAccess (Expression expr, string methodName, TypeArguments typeArguments, Location loc)
62 : base (expr, methodName, typeArguments, loc)
66 protected override void Error_TypeDoesNotContainDefinition (ResolveContext ec, TypeSpec type, string name)
68 ec.Report.Error (1935, loc, "An implementation of `{0}' query expression pattern could not be found. " +
69 "Are you missing `System.Linq' using directive or `System.Core.dll' assembly reference?",
74 protected class QueryExpressionInvocation : Invocation, OverloadResolver.IErrorHandler
76 public QueryExpressionInvocation (QueryExpressionAccess expr, Arguments arguments)
77 : base (expr, arguments)
81 protected override MethodGroupExpr DoResolveOverload (ResolveContext ec)
83 MethodGroupExpr rmg = mg.OverloadResolve (ec, ref arguments, this, OverloadResolver.Restrictions.None);
87 #region IErrorHandler Members
89 bool OverloadResolver.IErrorHandler.AmbiguousCandidates (ResolveContext ec, MemberSpec best, MemberSpec ambiguous)
91 ec.Report.SymbolRelatedToPreviousError (best);
92 ec.Report.SymbolRelatedToPreviousError (ambiguous);
93 ec.Report.Error (1940, loc, "Ambiguous implementation of the query pattern `{0}' for source type `{1}'",
94 best.Name, mg.InstanceExpression.GetSignatureForError ());
98 bool OverloadResolver.IErrorHandler.ArgumentMismatch (ResolveContext rc, MemberSpec best, Argument arg, int index)
103 bool OverloadResolver.IErrorHandler.NoArgumentMatch (ResolveContext rc, MemberSpec best)
108 bool OverloadResolver.IErrorHandler.TypeInferenceFailed (ResolveContext rc, MemberSpec best)
110 var ms = (MethodSpec) best;
111 TypeSpec source_type = ms.Parameters.ExtensionMethodType;
112 if (source_type != null) {
113 Argument a = arguments[0];
115 if (TypeManager.IsGenericType (source_type) && InflatedTypeSpec.ContainsTypeParameter (source_type)) {
116 TypeInferenceContext tic = new TypeInferenceContext (source_type.TypeArguments);
117 tic.OutputTypeInference (rc, a.Expr, source_type);
118 if (tic.FixAllTypes (rc)) {
119 source_type = source_type.GetDefinition ().MakeGenericType (rc, tic.InferredTypeArguments);
123 if (!Convert.ImplicitConversionExists (rc, a.Expr, source_type)) {
124 rc.Report.Error (1936, loc, "An implementation of `{0}' query expression pattern for source type `{1}' could not be found",
125 best.Name, TypeManager.CSharpName (a.Type));
130 if (best.Name == "SelectMany") {
131 rc.Report.Error (1943, loc,
132 "An expression type is incorrect in a subsequent `from' clause in a query expression with source type `{0}'",
133 arguments[0].GetSignatureForError ());
135 rc.Report.Error (1942, loc,
136 "An expression type in `{0}' clause is incorrect. Type inference failed in the call to `{1}'",
137 best.Name.ToLowerInvariant (), best.Name);
146 public AQueryClause next;
147 public QueryBlock block;
149 protected AQueryClause (QueryBlock block, Expression expr, Location loc)
156 protected override void CloneTo (CloneContext clonectx, Expression target)
158 base.CloneTo (clonectx, target);
160 AQueryClause t = (AQueryClause) target;
163 t.block = (QueryBlock) clonectx.LookupBlock (block);
166 t.next = (AQueryClause) next.Clone (clonectx);
169 protected override Expression DoResolve (ResolveContext ec)
171 return expr.Resolve (ec);
174 public virtual Expression BuildQueryClause (ResolveContext ec, Expression lSide, Parameter parameter)
176 Arguments args = null;
177 CreateArguments (ec, parameter, ref args);
178 lSide = CreateQueryExpression (lSide, args);
180 parameter = CreateChildrenParameters (parameter);
182 Select s = next as Select;
183 if (s == null || s.IsRequired (parameter))
184 return next.BuildQueryClause (ec, lSide, parameter);
186 // Skip transparent select clause if any clause follows
187 if (next.next != null)
188 return next.next.BuildQueryClause (ec, lSide, parameter);
194 protected virtual Parameter CreateChildrenParameters (Parameter parameter)
196 // Have to clone the parameter for any children use, it carries block sensitive data
197 return parameter.Clone ();
200 protected virtual void CreateArguments (ResolveContext ec, Parameter parameter, ref Arguments args)
202 args = new Arguments (2);
204 LambdaExpression selector = new LambdaExpression (loc);
206 block.SetParameter (parameter);
207 selector.Block = block;
208 selector.Block.AddStatement (new ContextualReturn (expr));
210 args.Add (new Argument (selector));
213 protected Invocation CreateQueryExpression (Expression lSide, Arguments arguments)
215 return new QueryExpressionInvocation (
216 new QueryExpressionAccess (lSide, MethodName, loc), arguments);
219 protected abstract string MethodName { get; }
221 public AQueryClause Next {
227 public AQueryClause Tail {
229 return next == null ? this : next.Tail;
235 // A query clause with an identifier (range variable)
237 public abstract class ARangeVariableQueryClause : AQueryClause
239 sealed class RangeAnonymousTypeParameter : AnonymousTypeParameter
241 public RangeAnonymousTypeParameter (Expression initializer, RangeVariable parameter)
242 : base (initializer, parameter.Name, parameter.Location)
246 protected override void Error_InvalidInitializer (ResolveContext ec, string initializer)
248 ec.Report.Error (1932, loc, "A range variable `{0}' cannot be initialized with `{1}'",
253 class RangeParameterReference : ParameterReference
257 public RangeParameterReference (Parameter p)
258 : base (null, p.Location)
263 protected override Expression DoResolve (ResolveContext ec)
265 pi = ec.CurrentBlock.ParametersBlock.GetParameterInfo (parameter);
266 return base.DoResolve (ec);
270 protected RangeVariable identifier;
272 protected ARangeVariableQueryClause (QueryBlock block, RangeVariable identifier, Expression expr, Location loc)
273 : base (block, expr, loc)
275 this.identifier = identifier;
278 public FullNamedExpression IdentifierType { get; set; }
280 protected Invocation CreateCastExpression (Expression lSide)
282 return new QueryExpressionInvocation (
283 new QueryExpressionAccess (lSide, "Cast", new TypeArguments (IdentifierType), loc), null);
286 protected override Parameter CreateChildrenParameters (Parameter parameter)
288 return new QueryBlock.TransparentParameter (parameter.Clone (), GetIntoVariable ());
291 protected static Expression CreateRangeVariableType (ResolveContext rc, Parameter parameter, RangeVariable name, Expression init)
293 var args = new List<AnonymousTypeParameter> (2);
296 // The first argument is the reference to the parameter
298 args.Add (new AnonymousTypeParameter (new RangeParameterReference (parameter), parameter.Name, parameter.Location));
301 // The second argument is the linq expression
303 args.Add (new RangeAnonymousTypeParameter (init, name));
306 // Create unique anonymous type
308 return new NewAnonymousType (args, rc.MemberContext.CurrentMemberDefinition.Parent, name.Location);
311 protected virtual RangeVariable GetIntoVariable ()
317 public sealed class RangeVariable : INamedBlockVariable
321 public RangeVariable (string name, Location loc)
338 public bool IsDeclared {
344 public bool IsParameter {
350 public Location Location { get; private set; }
352 public string Name { get; private set; }
356 public Expression CreateReferenceExpression (ResolveContext rc, Location loc)
359 // We know the variable name is somewhere in the scope. This generates
360 // an access expression from current block
362 var pb = rc.CurrentBlock.ParametersBlock;
364 if (pb is QueryBlock) {
365 for (int i = pb.Parameters.Count - 1; i >= 0; --i) {
366 var p = pb.Parameters[i];
368 return pb.GetParameterReference (i, loc);
370 Expression expr = null;
371 var tp = p as QueryBlock.TransparentParameter;
374 expr = pb.GetParameterReference (i, loc);
376 expr = new TransparentMemberAccess (expr, tp.Name);
378 if (tp.Identifier == Name)
379 return new TransparentMemberAccess (expr, Name);
381 if (tp.Parent.Name == Name)
382 return new TransparentMemberAccess (expr, Name);
384 tp = tp.Parent as QueryBlock.TransparentParameter;
392 pb = pb.Parent.ParametersBlock;
397 class QueryStartClause : ARangeVariableQueryClause
399 public QueryStartClause (QueryBlock block, Expression expr, RangeVariable identifier, Location loc)
400 : base (block, identifier, expr, loc)
402 block.AddRangeVariable (identifier);
405 public override Expression BuildQueryClause (ResolveContext ec, Expression lSide, Parameter parameter)
408 expr = expr.Resolve (ec);
412 if (expr.Type == InternalType.Dynamic || expr.Type == TypeManager.void_type) {
413 ec.Report.Error (1979, expr.Location,
414 "Query expression with a source or join sequence of type `{0}' is not allowed",
415 TypeManager.CSharpName (expr.Type));
420 if (IdentifierType != null)
421 expr = CreateCastExpression (expr);
423 if (parameter == null)
426 return next.BuildQueryClause (ec, lSide, new ImplicitLambdaParameter (identifier.Name, identifier.Location));
429 protected override Expression DoResolve (ResolveContext ec)
431 Expression e = BuildQueryClause (ec, null, null);
432 return e.Resolve (ec);
435 protected override string MethodName {
436 get { throw new NotSupportedException (); }
440 public class GroupBy : AQueryClause
442 Expression element_selector;
443 QueryBlock element_block;
445 public GroupBy (QueryBlock block, Expression elementSelector, QueryBlock elementBlock, Expression keySelector, Location loc)
446 : base (block, keySelector, loc)
449 // Optimizes clauses like `group A by A'
451 if (!elementSelector.Equals (keySelector)) {
452 this.element_selector = elementSelector;
453 this.element_block = elementBlock;
457 protected override void CreateArguments (ResolveContext ec, Parameter parameter, ref Arguments args)
459 base.CreateArguments (ec, parameter, ref args);
461 if (element_selector != null) {
462 LambdaExpression lambda = new LambdaExpression (element_selector.Location);
464 element_block.SetParameter (parameter.Clone ());
465 lambda.Block = element_block;
466 lambda.Block.AddStatement (new ContextualReturn (element_selector));
467 args.Add (new Argument (lambda));
471 protected override void CloneTo (CloneContext clonectx, Expression target)
473 GroupBy t = (GroupBy) target;
474 if (element_selector != null) {
475 t.element_selector = element_selector.Clone (clonectx);
476 t.element_block = (QueryBlock) element_block.Clone (clonectx);
479 base.CloneTo (clonectx, t);
482 protected override string MethodName {
483 get { return "GroupBy"; }
487 public class Join : SelectMany
489 QueryBlock inner_selector, outer_selector;
491 public Join (QueryBlock block, RangeVariable lt, Expression inner, QueryBlock outerSelector, QueryBlock innerSelector, Location loc)
492 : base (block, lt, inner, loc)
494 this.outer_selector = outerSelector;
495 this.inner_selector = innerSelector;
498 protected override void CreateArguments (ResolveContext ec, Parameter parameter, ref Arguments args)
500 args = new Arguments (4);
502 if (IdentifierType != null)
503 expr = CreateCastExpression (expr);
505 args.Add (new Argument (expr));
507 outer_selector.SetParameter (parameter.Clone ());
508 var lambda = new LambdaExpression (outer_selector.StartLocation);
509 lambda.Block = outer_selector;
510 args.Add (new Argument (lambda));
512 inner_selector.SetParameter (new ImplicitLambdaParameter (identifier.Name, identifier.Location));
513 lambda = new LambdaExpression (inner_selector.StartLocation);
514 lambda.Block = inner_selector;
515 args.Add (new Argument (lambda));
517 base.CreateArguments (ec, parameter, ref args);
520 protected override void CloneTo (CloneContext clonectx, Expression target)
522 Join t = (Join) target;
523 t.inner_selector = (QueryBlock) inner_selector.Clone (clonectx);
524 t.outer_selector = (QueryBlock) outer_selector.Clone (clonectx);
525 base.CloneTo (clonectx, t);
528 protected override string MethodName {
529 get { return "Join"; }
533 public class GroupJoin : Join
535 readonly RangeVariable into;
537 public GroupJoin (QueryBlock block, RangeVariable lt, Expression inner,
538 QueryBlock outerSelector, QueryBlock innerSelector, RangeVariable into, Location loc)
539 : base (block, lt, inner, outerSelector, innerSelector, loc)
544 protected override RangeVariable GetIntoVariable ()
549 protected override string MethodName {
550 get { return "GroupJoin"; }
554 public class Let : ARangeVariableQueryClause
556 public Let (QueryBlock block, RangeVariable identifier, Expression expr, Location loc)
557 : base (block, identifier, expr, loc)
561 protected override void CreateArguments (ResolveContext ec, Parameter parameter, ref Arguments args)
563 expr = CreateRangeVariableType (ec, parameter, identifier, expr);
564 base.CreateArguments (ec, parameter, ref args);
567 protected override string MethodName {
568 get { return "Select"; }
572 public class Select : AQueryClause
574 public Select (QueryBlock block, Expression expr, Location loc)
575 : base (block, expr, loc)
580 // For queries like `from a orderby a select a'
581 // the projection is transparent and select clause can be safely removed
583 public bool IsRequired (Parameter parameter)
585 SimpleName sn = expr as SimpleName;
589 return sn.Name != parameter.Name;
592 protected override string MethodName {
593 get { return "Select"; }
597 public class SelectMany : ARangeVariableQueryClause
599 public SelectMany (QueryBlock block, RangeVariable identifier, Expression expr, Location loc)
600 : base (block, identifier, expr, loc)
604 protected override void CreateArguments (ResolveContext ec, Parameter parameter, ref Arguments args)
607 if (IdentifierType != null)
608 expr = CreateCastExpression (expr);
610 base.CreateArguments (ec, parameter.Clone (), ref args);
613 Expression result_selector_expr;
614 QueryBlock result_block;
616 var target = GetIntoVariable ();
617 var target_param = new ImplicitLambdaParameter (target.Name, target.Location);
620 // When select follows use it as a result selector
622 if (next is Select) {
623 result_selector_expr = next.Expr;
625 result_block = next.block;
626 result_block.SetParameters (parameter, target_param);
630 result_selector_expr = CreateRangeVariableType (ec, parameter, target, new SimpleName (target.Name, target.Location));
632 result_block = new QueryBlock (block.Parent, block.StartLocation);
633 result_block.SetParameters (parameter, target_param);
636 LambdaExpression result_selector = new LambdaExpression (Location);
637 result_selector.Block = result_block;
638 result_selector.Block.AddStatement (new ContextualReturn (result_selector_expr));
640 args.Add (new Argument (result_selector));
643 protected override string MethodName {
644 get { return "SelectMany"; }
648 public class Where : AQueryClause
650 public Where (QueryBlock block, Expression expr, Location loc)
651 : base (block, expr, loc)
655 protected override string MethodName {
656 get { return "Where"; }
660 public class OrderByAscending : AQueryClause
662 public OrderByAscending (QueryBlock block, Expression expr)
663 : base (block, expr, expr.Location)
667 protected override string MethodName {
668 get { return "OrderBy"; }
672 public class OrderByDescending : AQueryClause
674 public OrderByDescending (QueryBlock block, Expression expr)
675 : base (block, expr, expr.Location)
679 protected override string MethodName {
680 get { return "OrderByDescending"; }
684 public class ThenByAscending : OrderByAscending
686 public ThenByAscending (QueryBlock block, Expression expr)
691 protected override string MethodName {
692 get { return "ThenBy"; }
696 public class ThenByDescending : OrderByDescending
698 public ThenByDescending (QueryBlock block, Expression expr)
703 protected override string MethodName {
704 get { return "ThenByDescending"; }
709 // Implicit query block
711 public class QueryBlock : ParametersBlock
714 // Transparent parameters are used to package up the intermediate results
715 // and pass them onto next clause
717 public sealed class TransparentParameter : ImplicitLambdaParameter
719 public static int Counter;
720 const string ParameterNamePrefix = "<>__TranspIdent";
722 public readonly Parameter Parent;
723 public readonly string Identifier;
725 public TransparentParameter (Parameter parent, RangeVariable identifier)
726 : base (ParameterNamePrefix + Counter++, identifier.Location)
729 Identifier = identifier.Name;
732 public static void Reset ()
738 public QueryBlock (Block parent, Location start)
739 : base (parent, ParametersCompiled.EmptyReadOnlyParameters, start)
741 flags |= Flags.CompilerGenerated;
744 public void AddRangeVariable (RangeVariable variable)
746 variable.Block = this;
747 AddLocalName (variable.Name, variable);
750 public override void Error_AlreadyDeclared (string name, INamedBlockVariable variable, string reason)
752 TopBlock.Report.Error (1931, variable.Location,
753 "A range variable `{0}' conflicts with a previous declaration of `{0}'",
757 public override void Error_AlreadyDeclared (string name, INamedBlockVariable variable)
759 TopBlock.Report.Error (1930, variable.Location,
760 "A range variable `{0}' has already been declared in this scope",
764 public override void Error_AlreadyDeclaredTypeParameter (string name, Location loc)
766 TopBlock.Report.Error (1948, loc,
767 "A range variable `{0}' conflicts with a method type parameter",
771 public void SetParameter (Parameter parameter)
773 base.parameters = new ParametersCompiled (parameter);
774 base.parameter_info = new ParameterInfo[] {
775 new ParameterInfo (this, 0)
779 public void SetParameters (Parameter first, Parameter second)
781 base.parameters = new ParametersCompiled (first, second);
782 base.parameter_info = new ParameterInfo[] {
783 new ParameterInfo (this, 0),
784 new ParameterInfo (this, 1)
789 sealed class TransparentMemberAccess : MemberAccess
791 public TransparentMemberAccess (Expression expr, string name)
796 public override Expression DoResolveLValue (ResolveContext rc, Expression right_side)
798 rc.Report.Error (1947, loc,
799 "A range variable `{0}' cannot be assigned to. Consider using `let' clause to store the value",