Use ordinary substitution for parameter mapping (#60)

Author: zyn0217

clang/lib/AST/TemplateBase.cpp

@@ -606,11 +606,8 @@ TemplateArgumentLoc::TemplateArgumentLoc(ASTContext &Ctx,
       LocInfo(Ctx, TemplateKWLoc, QualifierLoc, TemplateNameLoc, EllipsisLoc) {
   assert(Argument.getKind() == TemplateArgument::Template ||
          Argument.getKind() == TemplateArgument::TemplateExpansion);
-  // We can't assume QualifierLoc.getNestedNameSpecifier() ==
-  // Argument.getAsTemplateOrTemplatePattern().getQualifier() at this point,
-  // because in template rewriting, we may substitute a DependentTemplateName
-  // (which has a NNSLoc) into a template template parameter (which
-  // doesn't have a NNSLoc).
+  assert(QualifierLoc.getNestedNameSpecifier() ==
+         Argument.getAsTemplateOrTemplatePattern().getQualifier());
 }
 
 NestedNameSpecifierLoc TemplateArgumentLoc::getTemplateQualifierLoc() const {

clang/lib/Sema/SemaConcept.cpp

@@ -2173,9 +2173,6 @@ bool SubstituteParameterMappings::substitute(NormalizedConstraint &N) {
         /*Pattern=*/nullptr,
         /*ForConstraintInstantiation=*/true);
 
-    // Don't build Subst* nodes to model lambda expressions.
-    // The transform of Subst* is oblivious to the lambda type.
-    MLTAL.setKind(TemplateSubstitutionKind::Rewrite);
     return SubstituteParameterMappings(
                SemaRef, &MLTAL, CSE->getTemplateArgsAsWritten(), InFoldExpr)
         .substitute(CC.getNormalizedConstraint());

clang/lib/Sema/SemaTemplateInstantiate.cpp

@@ -2268,15 +2268,6 @@ TemplateInstantiator::TransformTemplateParmRefExpr(DeclRefExpr *E,
     // We're rewriting the template parameter as a reference to another
     // template parameter.
     Arg = getTemplateArgumentPackPatternForRewrite(Arg);
-    if (Arg.getKind() != TemplateArgument::Expression) {
-      assert(SemaRef.inParameterMappingSubstitution() ||
-             SemaRef.inConstraintSubstitution());
-      ExprResult Expr = SemaRef.BuildExpressionFromNonTypeTemplateArgument(
-          Arg, E->getLocation());
-      if (Expr.isInvalid())
-        return E;
-      Arg = TemplateArgument(Expr.get(), /*IsCanonical=*/false);
-    }
     assert(Arg.getKind() == TemplateArgument::Expression &&
            "unexpected nontype template argument kind in template rewrite");
     // FIXME: This can lead to the same subexpression appearing multiple times
@@ -2498,11 +2489,21 @@ ExprResult
 TemplateInstantiator::TransformSubstNonTypeTemplateParmExpr(
                                           SubstNonTypeTemplateParmExpr *E) {
   ExprResult SubstReplacement = E->getReplacement();
+  QualType ParamType = E->getParameterType(getSema().Context);
+  bool WasDependentLambda = false;
+  if (auto *RT = dyn_cast<RecordType>(ParamType);
+      RT && RT->getAsCXXRecordDecl())
+    WasDependentLambda = RT->getAsCXXRecordDecl()->isDependentLambda();
   if (!isa<ConstantExpr>(SubstReplacement.get()))
     SubstReplacement = TransformExpr(E->getReplacement());
   if (SubstReplacement.isInvalid())
     return true;
-  QualType SubstType = TransformType(E->getParameterType(getSema().Context));
+  // FIXME: This transform cannot find the instantiated lambda declaration
+  // because lambdas are instantiated in a unique scope.
+  QualType SubstType =
+      WasDependentLambda
+          ? SubstReplacement.get()->getType().getUnqualifiedType()
+          : TransformType(ParamType);
   if (SubstType.isNull())
     return true;
   // The type may have been previously dependent and not now, which means we

clang/test/SemaTemplate/concepts.cpp

@@ -1308,4 +1308,12 @@ struct __completion_domain_or_none_ : __mdefer_<__mtransform<>> {};
 
 }
 
+namespace case2 {
+
+template<auto& Q, class P> concept C = Q.template operator()<P>();
+template<class P> concept E = C<[]<class Ty>{ return false; }, P>;
+static_assert(!E<int>);
+
+}
+
 }