initial commit

Author: tknopp

Project.toml

@@ -3,8 +3,21 @@ uuid = "a4a2c56f-fead-462a-a3ab-85921a5f2575"
 authors = ["Tobias Knopp <[email protected]> and contributors"]
 version = "1.0.0-DEV"
 
+[deps]
+FFTW = "7a1cc6ca-52ef-59f5-83cd-3a7055c09341"
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+LinearOperators = "5c8ed15e-5a4c-59e4-a42b-c7e8811fb125"
+Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
+SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
+Wavelets = "29a6e085-ba6d-5f35-a997-948ac2efa89a"
+
 [compat]
-julia = "1"
+julia = "1.6"
+FFTW = "0.2, 1.0"
+LinearOperators = "2.3.3"
+Reexport = "0.2, 1.0"
+Wavelets = "0.8, 0.9"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

src/LinearOperatorCollection.jl

@@ -1,5 +1,91 @@
 module LinearOperatorCollection
 
-# Write your package code here.
+import Base: length, iterate, \
+using LinearAlgebra
+import LinearAlgebra.BLAS: gemv, gemv!
+import LinearAlgebra: BlasFloat, normalize!, norm, rmul!, lmul!
+using SparseArrays
+using Random
+#using CUDA
+
+using Reexport
+@reexport using Reexport
+@reexport using LinearOperators
+@reexport using FFTW
+@reexport using Wavelets
+
+LinearOperators.use_prod5!(op::opEye) = false
+LinearOperators.has_args5(op::opEye) = false
+
+
+const Trafo = Union{AbstractMatrix, AbstractLinearOperator, Nothing}
+const FuncOrNothing = Union{Function, Nothing}
+
+# Helper function to wrap a prod into a 5-args mul
+function wrapProd(prod::Function)
+  λ = (res, x, α, β) -> begin
+    if β == zero(β)
+      res .= prod(x) .* α
+    else
+      res .= prod(x) .* α .+ β .* res
+    end
+  end
+  return λ
+end
+
+include("FFTOp.jl")
+include("DCTOp.jl")
+include("DSTOp.jl")
+include("WaveletOp.jl")
+include("GradientOp.jl")
+include("SamplingOp.jl")
+include("WeightingOp.jl")
+include("NormalOp.jl")
+
+export linearOperator, linearOperatorList
+
+linearOperator(op::Nothing,shape,T::Type=ComplexF32) = nothing
+
+"""
+  returns a list of currently implemented `LinearOperator`s
+"""
+function linearOperatorList()
+  return ["DCT-II", "DCT-IV", "FFT", "DST", "Wavelet", "Gradient"]
+end
+
+"""
+    linearOperator(op::AbstractString, shape)
+
+returns the `LinearOperator` with name `op`.
+
+# valid names
+* `"FFT"`
+* `"DCT-II"`
+* `"DCT-IV"`
+* `"DST"`
+* `"Wavelet"`
+* `"Gradient"`
+"""
+function linearOperator(op::AbstractString, shape, T::Type=ComplexF32)
+  shape_ = tuple(shape...)
+  if op == "FFT"
+    trafo = FFTOp(T, shape_, false) #FFTOperator(shape)
+  elseif op == "DCT-II"
+    shape_ = tuple(shape[shape .!= 1]...)
+    trafo = DCTOp(T, shape_, 2)
+  elseif op == "DCT-IV"
+    shape_ = tuple(shape[shape .!= 1]...)
+    trafo = DCTOp(T, shape_, 4)
+  elseif op == "DST"
+    trafo = DSTOp(T, shape_)
+  elseif op == "Wavelet"
+    trafo = WaveletOp(T,shape_)
+  elseif op=="Gradient"
+    trafo = GradientOp(T,shape_)
+  else
+    error("Unknown transformation")
+  end
+  trafo
+end
 
 end

test/runtests.jl

@@ -1,6 +1,8 @@
-using LinearOperatorCollection
+using SparsityOperators
 using Test
+using Random
+using LinearAlgebra
+using FFTW
 
-@testset "LinearOperatorCollection.jl" begin
-    # Write your tests here.
-end
+include("testNormalOp.jl")
+include("testOperators.jl")