Add StaticArrays support for conv

JuliaDSP · Oct 28, 2024 · 0405374 · 0405374
1 parent 61a2c91
commit 0405374
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/Project.toml b/Project.toml
@@ -16,9 +16,11 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [weakdeps]
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
+StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
 
 [extensions]
 OffsetArraysExt = "OffsetArrays"
+StaticArraysExt = "StaticArrays"
 
 [compat]
 Bessels = "0.2"

diff --git a/ext/StaticArraysExt.jl b/ext/StaticArraysExt.jl
@@ -0,0 +1,10 @@
+module StaticArraysExt
+import DSP
+import StaticArrays
+
+DSP.conv_with_offset(::StaticArrays.SOneTo) = false
+
+DSP.conv_output_axis(::StaticArrays.SOneTo{M}, ::StaticArrays.SOneTo{N}) where {M, N} =
+    StaticArrays.SOneTo{M + N - 1}()
+
+end