Add some methods for CxxMatrix

TimSiebert1 · Jun 21, 2024 · f464977 · f464977
1 parent 0a485cb
commit f464977
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/array_types.jl b/src/array_types.jl
@@ -57,7 +57,7 @@ mutable struct CxxMatrix{T} <: AbstractMatrix{T}
         x = new{T}(alloc_mat(T, n, m), n, m)
         return finalizer(cxx_mat_finalizer, x)
     end
-    function CxxMatrix{T}(Y::Matrix{T}) where {T<:Real}
+    function CxxMatrix{T}(Y::AbstractMatrix{T}) where {T<:Real}
         check_type_mat(T)
         n, m = size(Y)
         X = new{T}(alloc_mat(T, n, m), n, m)
@@ -70,6 +70,10 @@ mutable struct CxxMatrix{T} <: AbstractMatrix{T}
     end
 end
 
+Base.size(A::CxxMatrix)=(A.n, A.m)
+Base.setindex!(A::CxxMatrix{T}, v, i,j) where T= setindex!(A.data,T(v),i,j)
+Base.getindex(A::CxxMatrix,i,j) = getindex(A.data,i,j)
+
 function check_type_mat(::Type{T}) where {T<:Real}
     if !(T <: Union{Cdouble})
         throw("Not implemented for type $T")