Prepare split of AbstractQ and AbstractMatrix

slimgroup · Feb 12, 2023 · 2ab3cd7 · 2ab3cd7
1 parent 9c2477f
commit 2ab3cd7
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 7 deletions.
diff --git a/src/JOLI.jl b/src/JOLI.jl
@@ -74,7 +74,7 @@ import Base.getindex
 
 
 # what's imported from LinearAlgebra
-import LinearAlgebra.norm
+import LinearAlgebra.norm, LinearAlgebra.AbstractQ
 import LinearAlgebra.issymmetric, LinearAlgebra.ishermitian
 import LinearAlgebra.mul!, LinearAlgebra.ldiv!
 

diff --git a/src/joAbstractDAparallelLinearOperator/constructors.jl b/src/joAbstractDAparallelLinearOperator/constructors.jl
@@ -35,7 +35,7 @@ Create a linear operator working on 2D DArray in multi-vector (over 2nd dimensio
 
 """
 function joDAdistributedLinOp(A::joAbstractLinearOperator{DDT,RDT},psin::joPAsetup,
-        fclean::Bool=false,rclean::Bool=false) where {DDT<:Number,RDT<:Number,INT<:Integer}
+        fclean::Bool=false,rclean::Bool=false) where {DDT<:Number,RDT<:Number}
 
     psin.chunks[1]==1 || throw(joDAdistributedLinearOperatorException("joDAdistributedLinearOperator: invalid joPAsetup - must not be distributed in 1st dimension"))
     A.n==psin.dims[1]  || throw(joDAdistributedLinearOperatorException("joDAdistributedLinearOperator: invalid joPAsetup - 1st dimension off joPAsetup does not match size(A,2)"))

diff --git a/src/joAbstractLinearOperator/constructors.jl b/src/joAbstractLinearOperator/constructors.jl
@@ -8,7 +8,7 @@
 """
 joMatrix outer constructor
 
-    joMatrix(array::AbstractMatrix;
+    joMatrix(array::Union{AbstractMatrix,AbstractQ};
              DDT::DataType=eltype(array),
              RDT::DataType=promote_type(eltype(array),DDT),
              name::String="joMatrix")
@@ -26,7 +26,7 @@ Look up argument names in help to joMatrix type.
 - if RDT:<Real for complex matrix then imaginary part will be neglected for forward/conjugate operator
 
 """
-function joMatrix(array::AbstractMatrix{EDT};
+function joMatrix(array::Union{AbstractMatrix{EDT},AbstractQ{EDT}};
     DDT::DataType=EDT,RDT::DataType=promote_type(EDT,DDT),name::String="joMatrix") where {EDT}
 
         (typeof(array)<:DArray || typeof(array)<:SharedArray) && @warn "Creating joMatrix from non-local array like $(typeof(array)) is likely going to have adverse impact on JOLI's health. Please, avoid it."

diff --git a/src/joAbstractSAparallelLinearOperator/constructors.jl b/src/joAbstractSAparallelLinearOperator/constructors.jl
@@ -35,7 +35,7 @@ Create a linear operator working on 2D SharedArray in multi-vector (over 2nd dim
 
 """
 function joSAdistributedLinOp(A::joAbstractLinearOperator{DDT,RDT},psin::joPAsetup,
-        fclean::Bool=false,rclean::Bool=false) where {DDT<:Number,RDT<:Number,INT<:Integer}
+        fclean::Bool=false,rclean::Bool=false) where {DDT<:Number,RDT<:Number}
 
     psin.chunks[1]==1 || throw(joSAdistributedLinearOperatorException("joSAdistributedLinearOperator: invalid joPAsetup - must not be distributed in 1st dimension"))
     A.n==psin.dims[1]  || throw(joSAdistributedLinearOperatorException("joSAdistributedLinearOperator: invalid joPAsetup - 1st dimension off joPAsetup does not match size(A,2)"))

diff --git a/src/joMiscTypesMethods/joPAsetup.jl b/src/joMiscTypesMethods/joPAsetup.jl
@@ -146,8 +146,7 @@ Creates joPAsetup type - basic 1D distribution
 """
 function joPAsetup(wpool::WorkerPool,n::Integer;
         DT::DataType=joFloat,
-        name::String="joPAsetup",
-        ) where INT<:Integer
+        name::String="joPAsetup")
     n>0 || throw(joPAsetupException("joPAsetup: invalid lenght of the vector: $n"))
     dims=(n,)
     chunks=joPAsetup_etc.default_chunks(dims,sort(workers(wpool)))