Add system level property definitions removed in AtomsBase 0.5

src/AtomsToolbox.jl

@@ -2,7 +2,8 @@ module AtomsToolbox
 
 using Distances: Distances, pairwise, euclidean, peuclidean,
                  Euclidean, PeriodicEuclidean
-using AtomsBase: AbstractSystem,
+using AtomsBase: AtomsBase,
+                 AbstractSystem,
                  FastSystem,
                  FlexibleSystem,
                  Atom,
@@ -12,13 +13,18 @@ using AtomsBase: AbstractSystem,
                  atomic_symbol,
                  atomic_mass,
                  cell_vectors,
-                 periodicity
+                 periodicity,
+                 species
 using Unitful: Unitful, ustrip, @u_str, unit
 import Graphs
 using LinearAlgebra: LinearAlgebra, ⋅, det, norm, lu, ×
 using StaticArrays: StaticMatrix, Size
 import Base: angle, sort # To extend for AbstractSystem 
 
+# Some type piracy to deal with other packages
+# Should be upstreamed or removed eventually
+include("piracy.jl")
+
 # Constants
 include("constants.jl")
 

src/getters.jl

@@ -1,14 +1,3 @@
-# Workaround for Unitful StaticMatrix not working.
-# See: https://github.com/PainterQubits/Unitful.jl/issues/538
-function Base.inv(x::StaticMatrix{N,M,T}) where {N,M,T <: Unitful.AbstractQuantity}
-    m = inv(ustrip.(x))
-    iq = eltype(m)
-    reinterpret(
-        Unitful.Quantity{iq, inv(Unitful.dimension(T)), typeof(inv(unit(T)))}, 
-        m
-    )
-end 
-
 # Functions to find shortest distance accounting for PBC, from pymatgen.
 include("pbc_utils.jl")
 

src/piracy.jl

@@ -0,0 +1,15 @@
+# Workaround for Unitful StaticMatrix not working.
+# See: https://github.com/PainterQubits/Unitful.jl/issues/538
+function Base.inv(x::StaticMatrix{N,M,T}) where {N,M,T <: Unitful.AbstractQuantity}
+    m = inv(ustrip.(x))
+    iq = eltype(m)
+    reinterpret(
+        Unitful.Quantity{iq, inv(Unitful.dimension(T)), typeof(inv(unit(T)))}, 
+        m
+    )
+end 
+
+# These were removed in the AtomsBase 0.4 update
+Base.position(sys::AbstractSystem) = position.(sys, :)
+AtomsBase.atomic_symbol(sys::AbstractSystem) = atomic_symbol.(species(sys, :))
+AtomsBase.atomic_number(sys::AbstractSystem) = atomic_number.(species(sys, :))