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]>

Author: Tokazama

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"

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()