StaticInt support for ntuple (#65)

* StaticInt support for ntuple * Update src/Static.jl Co-authored-by: Chris Elrod <[email protected]> Co-authored-by: Chris Elrod <[email protected]>
SciML · Jun 16, 2022 · f319fb6 · f319fb6 · Tokazama · Jun 16, 2022
1 parent eb4ba62
commit f319fb6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Static"
 uuid = "aedffcd0-7271-4cad-89d0-dc628f76c6d3"
 authors = ["chriselrod", "ChrisRackauckas", "Tokazama"]
-version = "0.7"
+version = "0.7.1"
 
 [deps]
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"

diff --git a/src/Static.jl b/src/Static.jl
@@ -481,8 +481,19 @@ Base.:(^)(x::Integer, y::True) = x
 Base.:(^)(x::BigInt, y::False) = one(x)
 Base.:(^)(x::BigInt, y::True) = x
 
+@inline function Base.ntuple(f::F, ::StaticInt{N}) where {F,N}
+    (N >= 0) || throw(ArgumentError(string("tuple length should be ≥ 0, got ", N)))
+    if @generated
+        quote
+            Base.Cartesian.@ntuple $N i -> f(i)
+        end
+    else
+        Tuple(f(i) for i = 1:N)
+    end
+end
+
 @inline function invariant_permutation(@nospecialize(x::Tuple), @nospecialize(y::Tuple))
-    if y === x === nstatic(Val(nfields(x)))
+    if y === x === ntuple(static, StaticInt(nfields(x)))
         return True()
     else
         return False()