Add scaled_position for AtomView

rashidrafeek · Dec 7, 2024 · b2866c4 · b2866c4
1 parent a429789
commit b2866c4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/AtomsToolbox.jl b/src/AtomsToolbox.jl
@@ -6,6 +6,7 @@ using AtomsBase: AbstractSystem,
                  FastSystem,
                  FlexibleSystem,
                  Atom,
+                 AtomView,
                  position,
                  atomic_number,
                  atomic_symbol,

diff --git a/src/getters.jl b/src/getters.jl
@@ -294,7 +294,7 @@ function scaled_position(sys::AbstractSystem)
     Vector.(eachrow(frpos))
 end
 scaled_position(sys::AbstractSystem, index) = scaled_position(sys[index], cell_matrix(sys))
-function scaled_position(atom::Atom, cellmat)
+function scaled_position(atom::Union{Atom,AtomView}, cellmat)
     frpos = position(atom)' * inv(cellmat')
 
     frpos'