Merge pull request #127 from MichielStock/copy

Copy

Project.toml

@@ -1,7 +1,7 @@
 name = "Kronecker"
 uuid = "2c470bb0-bcc8-11e8-3dad-c9649493f05e"
 authors = ["MichielStock <[email protected]>"]
-version = "0.5.4"
+version = "0.5.5"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

src/base.jl

@@ -150,6 +150,11 @@ Returns a the size of an `GeneralizedKroneckerProduct` instance.
 """
 size(K::GeneralizedKroneckerProduct, dim::Integer) = size(K)[dim]
 
+Base.copy(K::AbstractKroneckerProduct) = kronecker(copy(K.A), copy(K.B))
+Base.deepcopy(K::AbstractKroneckerProduct) = kronecker(deepcopy(K.A), deepcopy(K.B))
+
+Base.similar(K::AbstractKroneckerProduct) = kronecker(similar(K.A), similar(K.B))
+
 # CHECKS
 
 """
@@ -406,7 +411,6 @@ end
     return _kron!(C, As, Bs, f)
 end
 
-
 """
     Matrix(K::GeneralizedKroneckerProduct)
 

src/kroneckerpowers.jl

@@ -44,6 +44,10 @@ getmatrices(K::KroneckerPower) = (K.pow == 2 ? K.A : KroneckerPower(K.A, K.pow -
 
 lastmatrix(K::KroneckerPower) = K.A
 
+Base.copy(K::KroneckerPower) = kronecker(copy(K.A), K.pow)
+Base.deepcopy(K::KroneckerPower) = kronecker(deepcopy(K.A), K.pow)
+Base.similar(K::KroneckerPower) = kronecker(similar(K.A), K.pow)
+
 order(K::KroneckerPower) = K.pow
 Base.size(K::KroneckerPower) = size(K.A) .^ K.pow
 issquare(K::KroneckerPower) = issquare(K.A)

src/kroneckersum.jl

@@ -11,6 +11,10 @@ struct KroneckerSum{T<:Any,TA<:AbstractMatrix,TB<:AbstractMatrix} <: AbstractKro
     end
 end
 
+Base.copy(K::KroneckerSum) = kroneckersum(copy(K.A), copy(K.B))
+Base.deepcopy(K::KroneckerSum) = kroneckersum(deepcopy(K.A), deepcopy(K.B))
+Base.similar(K::KroneckerSum) = kroneckersum(similar(K.A), similar(K.B))
+
 order(M::AbstractKroneckerSum) = order(M.A) + order(M.B)
 issquare(M::AbstractKroneckerSum) = true
 

test/testbase.jl

@@ -42,6 +42,14 @@
         @test K isa AbstractKroneckerProduct{Float64}
 
         @test K isa AbstractMatrix{Float64}
+
+        @test copy(K) isa AbstractKroneckerProduct
+
+        Kcopy = deepcopy(K)
+        @test Kcopy ≈ K
+        @test Kcopy isa AbstractKroneckerProduct
+
+        @test similar(K) isa AbstractKroneckerProduct
     end
 
     @testset "Using vectors" begin

test/testkroneckerpowers.jl

@@ -40,6 +40,14 @@
 
         @test sum(K1) ≈ sum(K1dense)
 
+        @test copy(K2) isa KroneckerPower
+
+        Kcopy = deepcopy(K2)
+        @test Kcopy ≈ K2
+        @test Kcopy isa KroneckerPower
+
+        @test similar(K2) isa KroneckerPower
+
     end
 
     @testset "Inplace scaling" begin

test/testkroneckersum.jl

@@ -24,6 +24,14 @@
 
         @test issquare(KS)
 
+        @test copy(KS) isa KroneckerSum 
+
+        KScopy = deepcopy(KS)
+        @test KScopy ≈ KS
+        @test KScopy isa KroneckerSum
+
+        @test similar(KS) isa KroneckerSum
+
         IC = oneunit(C)
         KS3 = A ⊕ B ⊕ C
         KS3AB = (A ⊕ B) ⊕ C