Move methods from QuantumInterface to here which access a .data field (#199)

Author: akirakyle

Project.toml

@@ -25,7 +25,7 @@ FastGaussQuadrature = "0.5, 1"
 FillArrays = "0.13, 1"
 LRUCache = "1"
 LinearAlgebra = "1"
-QuantumInterface = "0.3.9"
+QuantumInterface = "0.4.0"
 Random = "1"
 RecursiveArrayTools = "3"
 SparseArrays = "1"

src/states.jl

@@ -30,10 +30,11 @@ mutable struct Ket{B,T} <: AbstractKet{B,T}
     end
 end
 
-Base.zero(x::Bra) = Bra(x.basis, zero(x.data))
-Base.zero(x::Ket) = Ket(x.basis, zero(x.data))
-eltype(::Type{K}) where {K <: Ket{B,V}} where {B,V} = eltype(V)
-eltype(::Type{K}) where {K <: Bra{B,V}} where {B,V} = eltype(V)
+Base.zero(x::T) where {T<:Union{Bra,Ket}} = T(x.basis, zero(x.data))
+Base.eltype(::Type{K}) where {K <: Ket{B,V}} where {B,V} = eltype(V)
+Base.eltype(::Type{K}) where {K <: Bra{B,V}} where {B,V} = eltype(V)
+Base.size(x::Union{Bra,Ket}) = size(x.data)
+@inline Base.axes(x::Union{Bra,Ket}) = axes(x.data)
 
 Bra{B}(b::B, data::T) where {B,T} = Bra{B,T}(b, data)
 Ket{B}(b::B, data::T) where {B,T} = Ket{B,T}(b, data)
@@ -51,6 +52,8 @@ Ket{B}(b::B) where B = Ket{B}(ComplexF64, b)
 Bra(b::Basis) = Bra(ComplexF64, b)
 Ket(b::Basis) = Ket(ComplexF64, b)
 
+Base.copy(a::T) where {T<:Union{Bra,Ket}} = T(a.basis, copy(a.data))
+
 ==(x::Ket{B}, y::Ket{B}) where {B} = (samebases(x, y) && x.data==y.data)
 ==(x::Bra{B}, y::Bra{B}) where {B} = (samebases(x, y) && x.data==y.data)
 ==(x::Ket, y::Ket) = false
@@ -80,6 +83,9 @@ Base.isapprox(x::Bra, y::Bra; kwargs...) = false
 /(a::Ket, b::Number) = Ket(a.basis, a.data ./ b)
 /(a::Bra, b::Number) = Bra(a.basis, a.data ./ b)
 
+-(a::T) where {T<:Union{Ket,Bra}} = T(a.basis, -a.data)
+norm(x::Union{Ket,Bra}) = norm(x.data)
+normalize!(x::Union{Ket,Bra}) = (normalize!(x.data); x)
 
 """
     dagger(x)
@@ -278,4 +284,4 @@ RecursiveArrayTools.recursivecopy!(dest::Ket{B,A},src::Ket{B,A}) where {B,A} = c
 RecursiveArrayTools.recursivecopy!(dest::Bra{B,A},src::Bra{B,A}) where {B,A} = copyto!(dest, src)
 RecursiveArrayTools.recursivecopy(x::T) where {T<:Union{Ket, Bra}} = copy(x)
 RecursiveArrayTools.recursivecopy(x::AbstractArray{T}) where {T<:Union{Ket, Bra}} = copy(x)
-RecursiveArrayTools.recursivefill!(x::T, a) where {T<:Union{Ket, Bra}} = fill!(x, a)
+RecursiveArrayTools.recursivefill!(x::T, a) where {T<:Union{Ket, Bra}} = fill!(x, a)