Print int[] as int array. (#13700)

Co-authored-by: Edgar <[email protected]>
Co-authored-by: Don Syme <[email protected]>
Co-authored-by: Vlad Zarytovskii <[email protected]>

Author: 4 people

src/Compiler/Checking/CheckExpressions.fs

@@ -4303,6 +4303,9 @@ and TcTypeOrMeasure kindOpt (cenv: cenv) newOk checkConstraints occ (iwsam: Warn
     | SynType.LongIdent synLongId ->
         TcLongIdentType kindOpt cenv newOk checkConstraints occ iwsam env tpenv synLongId
 
+    | MultiDimensionArrayType (rank, elemTy, m) ->
+        TcElementType cenv newOk checkConstraints occ env tpenv rank elemTy m
+    
     | SynType.App (StripParenTypes (SynType.LongIdent longId), _, args, _, _, postfix, m) ->
         TcLongIdentAppType kindOpt cenv newOk checkConstraints occ iwsam env tpenv longId postfix args m
 

src/Compiler/Checking/NicePrint.fs

@@ -180,7 +180,11 @@ module internal PrintUtilities =
         let isArray = not prefix && isArrayTyconRef denv.g tcref
         let demangled = 
             if isArray then
-                tcref.DisplayNameCore // no backticks for arrays "int[]"
+                let numberOfCommas = tcref.CompiledName |> Seq.filter (fun c -> c = ',') |> Seq.length
+                if numberOfCommas = 0 then
+                    "array"
+                else
+                    $"array{numberOfCommas + 1}d"
             else
                 let name =
                     if denv.includeStaticParametersInTypeNames then 
@@ -198,11 +202,7 @@ module internal PrintUtilities =
             tagEntityRefName denv tcref demangled
             |> mkNav tcref.DefinitionRange
 
-        let tyconTextL =
-            if isArray then
-                tyconTagged |> rightL
-            else
-                tyconTagged |> wordL
+        let tyconTextL = tyconTagged |> wordL
 
         if denv.shortTypeNames then 
             tyconTextL
@@ -873,6 +873,16 @@ module PrintTypes =
             | [arg] -> layoutTypeWithInfoAndPrec denv env 2 arg ^^ tcL
             | args -> bracketIfL (prec <= 1) (bracketL (layoutTypesWithInfoAndPrec denv env 2 (sepL (tagPunctuation ",")) args) --- tcL)
 
+    and layoutTypeForGenericMultidimensionalArrays denv env prec tcref innerT level =
+        let innerLayout = layoutTypeWithInfoAndPrec denv env prec innerT
+
+        let arrayLayout =
+            tagEntityRefName denv tcref $"array{level}d"
+            |> mkNav tcref.DefinitionRange
+            |> wordL
+        
+        innerLayout ^^ arrayLayout
+    
     /// Layout a type, taking precedence into account to insert brackets where needed
     and layoutTypeWithInfoAndPrec denv env prec ty =
         let g = denv.g
@@ -893,6 +903,10 @@ module PrintTypes =
         // Always prefer 'float' to 'float<1>'
         | TType_app (tc, args, _) when tc.IsMeasureableReprTycon && List.forall (isDimensionless g) args ->
           layoutTypeWithInfoAndPrec denv env prec (reduceTyconRefMeasureableOrProvided g tc args)
+        
+        // Special case for nested array<array<'t>> shape
+        | TTypeMultiDimensionalArrayAsGeneric (tcref, innerT, level) ->
+            layoutTypeForGenericMultidimensionalArrays denv env prec tcref innerT level
 
         // Layout a type application
         | TType_ucase (UnionCaseRef(tc, _), args)

src/Compiler/SyntaxTree/SyntaxTreeOps.fs

@@ -1059,3 +1059,19 @@ let getTypeFromTuplePath (path: SynTupleTypeSegment list) : SynType list =
     |> List.choose (function
         | SynTupleTypeSegment.Type t -> Some t
         | _ -> None)
+
+let (|MultiDimensionArrayType|_|) (t: SynType) =
+    match t with
+    | SynType.App (StripParenTypes (SynType.LongIdent (SynLongIdent ([ identifier ], _, _))), _, [ elementType ], _, _, true, m) ->
+        if System.Text.RegularExpressions.Regex.IsMatch(identifier.idText, "^array\d\d?d$") then
+            let rank =
+                identifier.idText
+                |> Seq.filter System.Char.IsDigit
+                |> Seq.toArray
+                |> System.String
+                |> int
+
+            Some(rank, elementType, m)
+        else
+            None
+    | _ -> None

src/Compiler/SyntaxTree/SyntaxTreeOps.fsi

@@ -353,3 +353,5 @@ val normalizeTupleExpr: exprs: SynExpr list -> commas: range list -> SynExpr lis
 val desugarGetSetMembers: memberDefns: SynMemberDefns -> SynMemberDefns
 
 val getTypeFromTuplePath: path: SynTupleTypeSegment list -> SynType list
+
+val (|MultiDimensionArrayType|_|): t: SynType -> (int * SynType * range) option

src/Compiler/TypedTree/TypedTreeOps.fs

@@ -10384,7 +10384,7 @@ let (|EmptyModuleOrNamespaces|_|) (moduleOrNamespaceContents: ModuleOrNamespaceC
                 | ModuleOrNamespaceContents.TMDefRec _
                 | ModuleOrNamespaceContents.TMDefs _ -> true
                 | _ -> false)
-        
+
         let emptyModuleOrNamespaces =
             defs
             |> List.choose (function
@@ -10407,3 +10407,17 @@ let (|EmptyModuleOrNamespaces|_|) (moduleOrNamespaceContents: ModuleOrNamespaceC
         else
             None
     | _ -> None
+
+let (|TTypeMultiDimensionalArrayAsGeneric|_|) (t: TType) =
+    let rec (|Impl|_|) t =
+        match t with
+        | TType_app(tc, [Impl(outerTc, innerT, currentLevel)], _) when tc.DisplayNameCore = "array" ->
+            Some (outerTc, innerT, currentLevel + 1)
+        | TType_app(tc, [arg], _) when tc.DisplayNameCore = "array" ->
+            Some (tc, arg, 1)
+        | _ -> None
+
+    match t with
+    | Impl (tc, arg, level) ->
+        if level > 2 then Some (tc, arg, level) else None
+    | _ -> None

src/Compiler/TypedTree/TypedTreeOps.fsi

@@ -2687,3 +2687,6 @@ type TraitConstraintInfo with
 /// This will match anything that does not have any types or bindings.
 val (|EmptyModuleOrNamespaces|_|):
     moduleOrNamespaceContents: ModuleOrNamespaceContents -> (ModuleOrNamespace list) option
+
+/// Captures an application type with a multi-dimensional array as postfix.
+val (|TTypeMultiDimensionalArrayAsGeneric|_|): t: TType -> (TyconRef * TType * int) option

tests/FSharp.Compiler.ComponentTests/Conformance/DeclarationElements/CustomAttributes/Basic/Basic.fs

@@ -103,7 +103,7 @@ module Basic =
         |> verifyCompile
         |> shouldFail
         |> withDiagnostics [
-            (Error 1, Line 10, Col 3, Line 10, Col 59, "This expression was expected to have type\n    'int[]'    \nbut here has type\n    'unit'    ")
+            (Error 1, Line 10, Col 3, Line 10, Col 59, "This expression was expected to have type\n    'int array'    \nbut here has type\n    'unit'    ")
             (Error 267, Line 10, Col 3, Line 10, Col 59, "This is not a valid constant expression or custom attribute value")
             (Error 850, Line 10, Col 3, Line 10, Col 59, "This attribute cannot be used in this version of F#")
             (Error 850, Line 13, Col 3, Line 13, Col 52, "This attribute cannot be used in this version of F#")

tests/FSharp.Compiler.ComponentTests/Conformance/DeclarationElements/MemberDefinitions/MethodsAndProperties/MethodsAndProperties.fs

@@ -97,10 +97,10 @@ module MethodsAndProperties =
         |> verifyCompile
         |> shouldFail
         |> withDiagnostics [
-            (Error 1, Line 11, Col 24, Line 11, Col 33, "This expression was expected to have type\n    ''a[,]'    \nbut here has type\n    'int[]'    ")
-            (Error 1, Line 12, Col 25, Line 12, Col 32, "This expression was expected to have type\n    ''a[]'    \nbut here has type\n    'int[,]'    ")
-            (Error 1, Line 13, Col 26, Line 13, Col 37, "This expression was expected to have type\n    ''a[]'    \nbut here has type\n    'int[,,]'    ")
-            (Error 1, Line 14, Col 27, Line 14, Col 38, "This expression was expected to have type\n    ''a[]'    \nbut here has type\n    'int[,,,]'    ")
+            (Error 1, Line 11, Col 24, Line 11, Col 33, "This expression was expected to have type\n    ''a array2d'    \nbut here has type\n    'int array'    ")
+            (Error 1, Line 12, Col 25, Line 12, Col 32, "This expression was expected to have type\n    ''a array'    \nbut here has type\n    'int array2d'    ")
+            (Error 1, Line 13, Col 26, Line 13, Col 37, "This expression was expected to have type\n    ''a array'    \nbut here has type\n    'int array3d'    ")
+            (Error 1, Line 14, Col 27, Line 14, Col 38, "This expression was expected to have type\n    ''a array'    \nbut here has type\n    'int array4d'    ")
         ]
 
     // SOURCE=E_IndexerArityMismatch02.fs     SCFLAGS="--test:ErrorRanges --flaterrors"	# E_IndexerArityMismatch02.fs
@@ -110,10 +110,10 @@ module MethodsAndProperties =
         |> verifyCompile
         |> shouldFail
         |> withDiagnostics [
-            (Error 1, Line 11, Col 24, Line 11, Col 35, "This expression was expected to have type\n    ''a[,,]'    \nbut here has type\n    'int[]'    ")
-            (Error 1, Line 12, Col 25, Line 12, Col 32, "This expression was expected to have type\n    ''a[]'    \nbut here has type\n    'int[,]'    ")
-            (Error 1, Line 13, Col 27, Line 13, Col 38, "This expression was expected to have type\n    ''a[,,]'    \nbut here has type\n    'int[,,,]'    ")
-            (Error 1, Line 14, Col 27, Line 14, Col 36, "This expression was expected to have type\n    ''a[,]'    \nbut here has type\n    'int[,,,]'    ")
+            (Error 1, Line 11, Col 24, Line 11, Col 35, "This expression was expected to have type\n    ''a array3d'    \nbut here has type\n    'int array'    ")
+            (Error 1, Line 12, Col 25, Line 12, Col 32, "This expression was expected to have type\n    ''a array'    \nbut here has type\n    'int array2d'    ")
+            (Error 1, Line 13, Col 27, Line 13, Col 38, "This expression was expected to have type\n    ''a array3d'    \nbut here has type\n    'int array4d'    ")
+            (Error 1, Line 14, Col 27, Line 14, Col 36, "This expression was expected to have type\n    ''a array2d'    \nbut here has type\n    'int array4d'    ")
         ]
 
     // SOURCE=E_IndexerIndeterminateType01.fs SCFLAGS=--test:ErrorRanges			# E_IndexerIndeterminateType01.fs

tests/FSharp.Compiler.ComponentTests/Conformance/RecordTypes/RecordTypes.fs

@@ -118,7 +118,7 @@ module RecordTypes =
         |> verifyTypeCheck
         |> shouldFail
         |> withDiagnostics [
-            (Error 1, Line 7, Col 17, Line 7, Col 47, "This expression was expected to have type\n    'int[]'    \nbut here has type\n    'RecType'    ")
+            (Error 1, Line 7, Col 17, Line 7, Col 47, "This expression was expected to have type\n    'int array'    \nbut here has type\n    'RecType'    ")
         ]
 
     // SOURCE=E_RecordsNotNull01.fs                                                             # E_RecordsNotNull01.fs

tests/FSharp.Compiler.ComponentTests/FSharp.Compiler.ComponentTests.fsproj

@@ -199,6 +199,7 @@
     <Compile Include="Signatures\TestHelpers.fs" />
     <Compile Include="Signatures\ModuleOrNamespaceTests.fs" />
     <Compile Include="Signatures\RecordTests.fs" />
+    <Compile Include="Signatures\ArrayTests.fs" />
   </ItemGroup>
   <ItemGroup>
     <Content Include="resources\**" CopyToOutputDirectory="Never" CopyToPublishDirectory="PreserveNewest" />