Add Backbone view method

MurrellGroup · Dec 28, 2023 · ab24913 · ab24913
1 parent 344b4e3
commit ab24913
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/backbone.jl b/src/backbone.jl
@@ -80,6 +80,7 @@ end
 
 @inline Base.getindex(backbone::Backbone, i::Integer) = backbone.coords[:, i]
 @inline Base.getindex(backbone::Backbone, r::AbstractVector{<:Integer}) = Backbone(backbone.coords[:, r])
+@inline Base.view(backbone::Backbone, r::AbstractVector{<:Integer}) = Backbone(view(backbone.coords, :, r))
 
 @inline Base.setindex!(backbone::Backbone, coords::AbstractVector, i::Integer) = (backbone[i] .= coords)
-@inline Base.setindex!(backbone::Backbone, coords::AbstractMatrix, r::AbstractVector{<:Integer}) = (backbone[r].coords .= coords)
+@inline Base.setindex!(backbone::Backbone, coords::AbstractMatrix, r::AbstractVector{<:Integer}) = (backbone[r].coords .= coords)