Add support for integer arrays as subscripts

Author: ZedThree

grammar.js

@@ -1860,7 +1860,9 @@ module.exports = grammar({
           $.extent_specifier,
           $.assumed_size,
           $.assumed_rank,
-          $._expression
+          $._expression,
+          $.multiple_subscript,
+          $.multiple_subscript_triplet,
         )),
         ')'
       )
@@ -1875,12 +1877,16 @@ module.exports = grammar({
       field("value",choice($._expression, $.assumed_size, $.assumed_shape))
     )),
 
-    extent_specifier: $ => seq(
+    _extent_specifier: $ => seq(
       optional($._expression), // start
       ':',
       optional(choice($._expression, $.assumed_size)), // stop
       optional(seq(':', $._expression)) // stride
     ),
+    extent_specifier: $=> $._extent_specifier,
+
+    multiple_subscript: $ => seq('@', $._expression),
+    multiple_subscript_triplet: $ => seq('@', $._extent_specifier),
 
     assumed_size: $ => '*',
 

src/grammar.json

@@ -18354,6 +18354,14 @@
                       {
                         "type": "SYMBOL",
                         "name": "_expression"
+                      },
+                      {
+                        "type": "SYMBOL",
+                        "name": "multiple_subscript"
+                      },
+                      {
+                        "type": "SYMBOL",
+                        "name": "multiple_subscript_triplet"
                       }
                     ]
                   },
@@ -18388,6 +18396,14 @@
                             {
                               "type": "SYMBOL",
                               "name": "_expression"
+                            },
+                            {
+                              "type": "SYMBOL",
+                              "name": "multiple_subscript"
+                            },
+                            {
+                              "type": "SYMBOL",
+                              "name": "multiple_subscript_triplet"
                             }
                           ]
                         }
@@ -18454,7 +18470,7 @@
         ]
       }
     },
-    "extent_specifier": {
+    "_extent_specifier": {
       "type": "SEQ",
       "members": [
         {
@@ -18517,6 +18533,36 @@
         }
       ]
     },
+    "extent_specifier": {
+      "type": "SYMBOL",
+      "name": "_extent_specifier"
+    },
+    "multiple_subscript": {
+      "type": "SEQ",
+      "members": [
+        {
+          "type": "STRING",
+          "value": "@"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "_expression"
+        }
+      ]
+    },
+    "multiple_subscript_triplet": {
+      "type": "SEQ",
+      "members": [
+        {
+          "type": "STRING",
+          "value": "@"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "_extent_specifier"
+        }
+      ]
+    },
     "assumed_size": {
       "type": "STRING",
       "value": "*"