Merge pull request #69 from vpuri3/update

method ambiguities

Author: ChrisRackauckas

src/basic.jl

@@ -512,6 +512,22 @@ Base.:*(ops::AbstractSciMLOperator...) = ComposedOperator(ops...)
 Base.:*(A::AbstractSciMLOperator, B::AbstractSciMLOperator) = ∘(A, B)
 Base.:*(A::ComposedOperator, B::AbstractSciMLOperator) = ∘(A.ops[1:end-1]..., A.ops[end] * B)
 Base.:*(A::AbstractSciMLOperator, B::ComposedOperator) = ∘(A * B.ops[1], B.ops[2:end]...)
+Base.:*(A::ComposedOperator, B::ComposedOperator) = ComposedOperator(A.ops..., B.ops...)
+
+for op in (
+           :*, :∘,
+          )
+    # identity
+    @eval function Base.$op(::IdentityOperator{N}, A::ComposedOperator) where{N}
+        @assert size(A, 1) == N
+        A
+    end
+
+    @eval function Base.$op(A::ComposedOperator, ::IdentityOperator{N}) where{N}
+        @assert size(A, 2) == N
+        A
+    end
+end
 
 Base.convert(::Type{AbstractMatrix}, L::ComposedOperator) = prod(op -> convert(AbstractMatrix, op), L.ops)
 SparseArrays.sparse(L::ComposedOperator) = prod(_sparse, L.ops)

src/matrix.jl

@@ -184,6 +184,11 @@ has_mul!(L::AffineOperator) = has_mul!(L.A)
 has_ldiv(L::AffineOperator) = has_ldiv(L.A)
 has_ldiv!(L::AffineOperator) = has_ldiv!(L.A)
 
+function cache_internals(L::AffineOperator, u::AbstractVecOrMat)
+    @set! L.A = cache_operator(L.A, u)
+    @set! L.b = cache_operator(L.b, u)
+    L
+end
 
 Base.:*(L::AffineOperator, u::AbstractVecOrMat) = L.A * u + L.b
 Base.:\(L::AffineOperator, u::AbstractVecOrMat) = L.A \ (u - L.b)