More elegant way for diagonaloperator to accept different types

qojulia · Jan 12, 2017 · b64fce2 · b64fce2
1 parent 7645e23
commit b64fce2
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/operators_sparse.jl b/src/operators_sparse.jl
@@ -93,13 +93,9 @@ end
 """
 Diagonal operator.
 """
-function diagonaloperator(b::Basis, diag::Vector{Complex128})
+function diagonaloperator{T <: Number}(b::Basis, diag::Vector{T})
   @assert 1 <= length(diag) <= b.shape[1]
-  SparseOperator(b, spdiagm(diag))
+  SparseOperator(b, spdiagm(complex(float(diag))))
 end
 
-diagonaloperator(b::Basis, diag::Vector{Float64}) = diagonaloperator(b, complex(diag))
-diagonaloperator(b::Basis, diag::Vector{Int}) = diagonaloperator(b, complex(float(diag)))
-diagonaloperator(b::Basis, diag::Vector{Complex{Int}}) = diagonaloperator(b, float(diag))
-
 end # module