如何在Julia中为自己的复合类型定义“：”范围运算符？

Question

我有以下类型定义：

type TransitionMatrix
  A :: SparseMatrixCSC{Float64,Int64}
  d :: Array{Float64,1}
  I :: Array{Int64}
end

并希望允许以下操作：如果我们有T :: TransitionMatrix，我希望可以使用范围功能（即IE，P [：，1]，P [：，3]等） TA我该怎么做？ 我做了类似的事情来定义过渡矩阵的“大小”，如下所示，但我不知道如何让冒号运算符发挥作用。 谢谢！

function Base.size(T :: TransitionMatrix, dim)
  return size(T.A, dim)
end

Answer 1

您将要为TransitionMatrix编写getindex和size版本。 例如：

import Base: getindex, size
getindex(A::TransitionMatrix, indx1, indx2) = @show (indx1, indx2)
size(A::TransitionMatrix) = size(A.A)
size(A::TransitionMatrix, d::Integer) = size(A.A,d)

然后

julia> A[:,2:3]
(indx1,indx2) = (1:5,2:3)
(1:5,2:3)

解析器扩展为:变成1:size(A,d) 。