Implement P0614R1, Range-based for statements with initializer.
authorMarek Polacek <polacek@redhat.com>
Wed, 23 May 2018 16:39:47 +0000 (16:39 +0000)
committerMarek Polacek <mpolacek@gcc.gnu.org>
Wed, 23 May 2018 16:39:47 +0000 (16:39 +0000)
* parser.c (cp_parser_range_based_for_with_init_p): New.
(cp_parser_init_statement): Use it.  Parse the optional init-statement
for a range-based for loop.
(cp_parser_skip_to_closing_parenthesis_1): Handle balancing ?:.

* g++.dg/cpp2a/range-for1.C: New test.
* g++.dg/cpp2a/range-for2.C: New test.
* g++.dg/cpp2a/range-for3.C: New test.
* g++.dg/cpp2a/range-for4.C: New test.
* g++.dg/cpp2a/range-for5.C: New test.
* g++.dg/cpp2a/range-for6.C: New test.
* g++.dg/cpp2a/range-for7.C: New test.

From-SVN: r260619

gcc/cp/ChangeLog
gcc/cp/parser.c
gcc/testsuite/ChangeLog
gcc/testsuite/g++.dg/cpp2a/range-for1.C [new file with mode: 0644]
gcc/testsuite/g++.dg/cpp2a/range-for2.C [new file with mode: 0644]
gcc/testsuite/g++.dg/cpp2a/range-for3.C [new file with mode: 0644]
gcc/testsuite/g++.dg/cpp2a/range-for4.C [new file with mode: 0644]
gcc/testsuite/g++.dg/cpp2a/range-for5.C [new file with mode: 0644]
gcc/testsuite/g++.dg/cpp2a/range-for6.C [new file with mode: 0644]
gcc/testsuite/g++.dg/cpp2a/range-for7.C [new file with mode: 0644]

index ee7361b..1e0854f 100644 (file)
@@ -1,3 +1,11 @@
+2018-05-23  Marek Polacek  <polacek@redhat.com>
+
+       Implement P0614R1, Range-based for statements with initializer.
+       * parser.c (cp_parser_range_based_for_with_init_p): New.
+       (cp_parser_init_statement): Use it.  Parse the optional init-statement
+       for a range-based for loop.
+       (cp_parser_skip_to_closing_parenthesis_1): Handle balancing ?:.
+
 2018-05-22  Jason Merrill  <jason@redhat.com>
 
        PR c++/81420 - not extending temporary lifetime.
index 6f51f03..d3e7348 100644 (file)
@@ -3493,6 +3493,7 @@ cp_parser_skip_to_closing_parenthesis_1 (cp_parser *parser,
   unsigned paren_depth = 0;
   unsigned brace_depth = 0;
   unsigned square_depth = 0;
+  unsigned condop_depth = 0;
 
   if (recovering && or_ttype == CPP_EOF
       && cp_parser_uncommitted_to_tentative_parse_p (parser))
@@ -3504,7 +3505,7 @@ cp_parser_skip_to_closing_parenthesis_1 (cp_parser *parser,
 
       /* Have we found what we're looking for before the closing paren?  */
       if (token->type == or_ttype && or_ttype != CPP_EOF
-         && !brace_depth && !paren_depth && !square_depth)
+         && !brace_depth && !paren_depth && !square_depth && !condop_depth)
        return -1;
 
       switch (token->type)
@@ -3551,6 +3552,16 @@ cp_parser_skip_to_closing_parenthesis_1 (cp_parser *parser,
            }
          break;
 
+       case CPP_QUERY:
+         if (!brace_depth && !paren_depth && !square_depth)
+           ++condop_depth;
+         break;
+
+       case CPP_COLON:
+         if (!brace_depth && !paren_depth && !square_depth && condop_depth > 0)
+           condop_depth--;
+         break;
+
        default:
          break;
        }
@@ -11255,6 +11266,40 @@ cp_parser_statement_seq_opt (cp_parser* parser, tree in_statement_expr)
     }
 }
 
+/* Return true if this is the C++20 version of range-based-for with
+   init-statement.  */
+
+static bool
+cp_parser_range_based_for_with_init_p (cp_parser *parser)
+{
+  bool r = false;
+
+  /* Save tokens so that we can put them back.  */
+  cp_lexer_save_tokens (parser->lexer);
+
+  /* There has to be an unnested ; followed by an unnested :.  */
+  if (cp_parser_skip_to_closing_parenthesis_1 (parser,
+                                              /*recovering=*/false,
+                                              CPP_SEMICOLON,
+                                              /*consume_paren=*/false) != -1)
+    goto out;
+
+  /* We found the semicolon, eat it now.  */
+  cp_lexer_consume_token (parser->lexer);
+
+  /* Now look for ':' that is not nested in () or {}.  */
+  r = (cp_parser_skip_to_closing_parenthesis_1 (parser,
+                                               /*recovering=*/false,
+                                               CPP_COLON,
+                                               /*consume_paren=*/false) == -1);
+
+out:
+  /* Roll back the tokens we skipped.  */
+  cp_lexer_rollback_tokens (parser->lexer);
+
+  return r;
+}
+
 /* Return true if we're looking at (init; cond), false otherwise.  */
 
 static bool
@@ -12299,7 +12344,7 @@ cp_parser_iteration_statement (cp_parser* parser, bool *if_p, bool ivdep,
      simple-declaration  */
 
 static bool
-cp_parser_init_statement (cp_parserparser, tree *decl)
+cp_parser_init_statement (cp_parser *parser, tree *decl)
 {
   /* If the next token is a `;', then we have an empty
      expression-statement.  Grammatically, this is also a
@@ -12312,6 +12357,29 @@ cp_parser_init_statement (cp_parser* parser, tree *decl)
       bool is_range_for = false;
       bool saved_colon_corrects_to_scope_p = parser->colon_corrects_to_scope_p;
 
+      /* Try to parse the init-statement.  */
+      if (cp_parser_range_based_for_with_init_p (parser))
+       {
+         tree dummy;
+         cp_parser_parse_tentatively (parser);
+         /* Parse the declaration.  */
+         cp_parser_simple_declaration (parser,
+                                       /*function_definition_allowed_p=*/false,
+                                       &dummy);
+         cp_parser_require (parser, CPP_SEMICOLON, RT_SEMICOLON);
+         if (!cp_parser_parse_definitely (parser))
+           /* That didn't work, try to parse it as an expression-statement.  */
+           cp_parser_expression_statement (parser, NULL_TREE);
+
+         if (cxx_dialect < cxx2a)
+           {
+             pedwarn (cp_lexer_peek_token (parser->lexer)->location, 0,
+                      "range-based %<for%> loops with initializer only "
+                      "available with -std=c++2a or -std=gnu++2a");
+             *decl = error_mark_node;
+           }
+       }
+
       /* A colon is used in range-based for.  */
       parser->colon_corrects_to_scope_p = false;
 
@@ -12325,7 +12393,7 @@ cp_parser_init_statement (cp_parser* parser, tree *decl)
       parser->colon_corrects_to_scope_p = saved_colon_corrects_to_scope_p;
       if (cp_lexer_next_token_is (parser->lexer, CPP_COLON))
        {
-         /* It is a range-for, consume the ':' */
+         /* It is a range-for, consume the ':' */
          cp_lexer_consume_token (parser->lexer);
          is_range_for = true;
          if (cxx_dialect < cxx11)
@@ -12337,9 +12405,9 @@ cp_parser_init_statement (cp_parser* parser, tree *decl)
            }
        }
       else
-         /* The ';' is not consumed yet because we told
-            cp_parser_simple_declaration not to.  */
-         cp_parser_require (parser, CPP_SEMICOLON, RT_SEMICOLON);
+       /* The ';' is not consumed yet because we told
+          cp_parser_simple_declaration not to.  */
+       cp_parser_require (parser, CPP_SEMICOLON, RT_SEMICOLON);
 
       if (cp_parser_parse_definitely (parser))
        return is_range_for;
index f45de09..db9ae1d 100644 (file)
@@ -1,3 +1,13 @@
+2018-05-23  Marek Polacek  <polacek@redhat.com>
+
+       * g++.dg/cpp2a/range-for1.C: New test.
+       * g++.dg/cpp2a/range-for2.C: New test.
+       * g++.dg/cpp2a/range-for3.C: New test.
+       * g++.dg/cpp2a/range-for4.C: New test.
+       * g++.dg/cpp2a/range-for5.C: New test.
+       * g++.dg/cpp2a/range-for6.C: New test.
+       * g++.dg/cpp2a/range-for7.C: New test.
+
 2018-05-23  Uros Bizjak  <ubizjak@gmail.com>
 
        * gcc.target/i386/cvt-3.c: New test.
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for1.C b/gcc/testsuite/g++.dg/cpp2a/range-for1.C
new file mode 100644 (file)
index 0000000..3a55235
--- /dev/null
@@ -0,0 +1,16 @@
+// P0614R1
+// { dg-do compile { target c++11 } }
+// { dg-options "" }
+
+void
+fn1 ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  for (int i = 0; auto x : a) // { dg-warning "range-based .for. loops with initializer only available with" "" { target c++17_down } }
+    ++i;
+
+  int i;
+  for (i = 0; auto x : a) // { dg-warning "range-based .for. loops with initializer only available with" "" { target c++17_down } }
+    ++i;
+}
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for2.C b/gcc/testsuite/g++.dg/cpp2a/range-for2.C
new file mode 100644 (file)
index 0000000..acb16c5
--- /dev/null
@@ -0,0 +1,16 @@
+// P0614R1
+// { dg-do compile }
+// { dg-options "-std=c++2a" }
+
+void
+fn1 ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  for (int i = 0; auto x : a)
+    ++i;
+
+  int i;
+  for (i = 0; auto x : a)
+    ++i;
+}
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for3.C b/gcc/testsuite/g++.dg/cpp2a/range-for3.C
new file mode 100644 (file)
index 0000000..291e605
--- /dev/null
@@ -0,0 +1,26 @@
+// P0614R1
+// { dg-do compile }
+// { dg-options "-std=c++2a" }
+
+static const int a[] = { 1, 2, 3, 4, 5 };
+extern void foo (int);
+extern void bar (int, int);
+
+constexpr int
+baz ()
+{
+  return 6;
+}
+
+void
+fn1 (int i)
+{
+  for ((i += 2); auto x : a)
+    foo (i);
+
+  for (auto j = 0, k = 0; auto x : a)
+    bar (j + k, x);
+
+  for (constexpr int j = baz (); auto x : a)
+    bar (x, j);
+}
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for4.C b/gcc/testsuite/g++.dg/cpp2a/range-for4.C
new file mode 100644 (file)
index 0000000..6ba783f
--- /dev/null
@@ -0,0 +1,27 @@
+// P0614R1
+// { dg-do run }
+// { dg-options "-std=c++2a" }
+
+int
+main ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  for (int i = 1; auto x : a)
+    if (i++ != x)
+      __builtin_abort ();
+
+  int i;
+  for (i = 1; auto x : a)
+    if (i++ != x)
+      __builtin_abort ();
+
+  i = 0;
+  for (i++; auto x : a)
+    if (i != 1)
+      __builtin_abort ();
+
+  for (int s[] = { 1, 1, 1 }; auto x : s)
+    if (x != 1)
+      __builtin_abort ();
+}
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for5.C b/gcc/testsuite/g++.dg/cpp2a/range-for5.C
new file mode 100644 (file)
index 0000000..62f1c2f
--- /dev/null
@@ -0,0 +1,46 @@
+// P0614R1
+// { dg-do compile }
+// { dg-options "-std=c++2a" }
+
+void
+fn1 ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  for (int i = 0; auto x : a)
+    ++i;
+
+  i = 0; // { dg-error "not declared" }
+
+  for (int i = 0; auto x : a)
+    {
+      for (int j = 0; auto x : a)
+       {
+         for (int k = 0; auto x : a)
+           k++;
+         k++; // { dg-error "not declared" }
+       }
+      j++; // { dg-error "not declared" }
+    }
+}
+
+void
+fn2 ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+  for (int i = 0; auto x : a)
+    int i = 3; // { dg-error "redeclaration" }
+}
+void
+fn3 ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  for (;:) // { dg-error "expected" }
+    {
+    }
+
+  for (;;:) // { dg-error "expected" }
+    {
+    }
+}
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for6.C b/gcc/testsuite/g++.dg/cpp2a/range-for6.C
new file mode 100644 (file)
index 0000000..4cee60a
--- /dev/null
@@ -0,0 +1,17 @@
+// P0614R1
+// { dg-do run }
+// { dg-options "-std=c++2a" }
+
+int
+main ()
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  for (int i = []{ return 3; }(); auto x : a)
+    if (i != 3)
+      __builtin_abort ();
+
+  for (int i = ({ 3; }); auto x : a)
+    if (i != 3)
+      __builtin_abort ();
+}
diff --git a/gcc/testsuite/g++.dg/cpp2a/range-for7.C b/gcc/testsuite/g++.dg/cpp2a/range-for7.C
new file mode 100644 (file)
index 0000000..5a3a89c
--- /dev/null
@@ -0,0 +1,45 @@
+// P0614R1
+// { dg-do compile }
+// { dg-options "-std=c++2a" }
+
+extern void bar (int);
+
+void
+fn0 (int n)
+{
+  int a[] = { 1, 2, 3, 4, 5 };
+
+  /* Don't get confused by the colon here.  */
+  for (int i = 0; n > 0 ? true : false; i++)
+    bar (i);
+
+  for (int i = n ? 3 : 4; auto x : a)
+    bar (x);
+
+  for (int i = n ? ({ a: 3; }) : 4; i < 10; i++)
+    bar (i);
+
+  for (int i = n ? ({ L: 3; }) : 4; auto x : a)
+    bar (x);
+
+  for (int i = n; auto x : a)
+    bar (x);
+
+  for (int i = n ? n ? n : 3 : 3; auto x : a)
+    bar (x);
+
+  for (int i = n ? n ? 3 : n ? 3 : 3 : 3; auto x : a)
+    bar (x);
+
+  for (int i = [=]{ return n ? 1 : 2; }(); auto x : a)
+    bar (x);
+
+  for (int i = [=]{ L2: if (!n) goto L2; else return 2; }(); auto x : a)
+    bar (x);
+
+  for (auto x = n ? 1 : 2 : a) // { dg-error "initializer" }
+    bar (x);
+
+  for (int i = 1; auto x = n ? 1 : 2 : a) // { dg-error "initializer" }
+    bar (x);
+}