Add ForceEnergy

src/SimplexCellLists.jl

@@ -46,4 +46,47 @@ let T = Int64, F = Float32
     precompile(cell_line_seg_add!, (LineSegCellList{T,F}, SVector{3,F}, SVector{3,F}, T))
 end
 
-end
+include("force-energy.jl")
+export ForceEnergy
+export ForceEnergyT
+export ForceEnergyFloat64
+export ForceEnergyFloat32
+export ForceEnergyFixedPoint
+export ForceNoEnergyFixedPoint
+export DebugForceEnergy
+export NullForceEnergy
+export add_bead_force!
+export zero_bead_force!
+export add_energy!
+export zero_energy!
+export get_nbeads
+export zero_force_energy!
+export get_force!
+export get_bead_force
+export get_energy
+export get_force
+export combine_force_energy!
+
+# Other fixed point fractional bit choices must be specialized on the
+# caller's parameters.
+for (FE, T) in (
+        (ForceEnergyFloat64, Float64),
+        (ForceEnergyFixedPoint{30, 30}, Float64),
+        (ForceNoEnergyFixedPoint{30}, Float64),
+    )
+    precompile(FE, (Int,))
+    precompile(zero_force_energy!, (FE,))
+    precompile(add_bead_force!, (FE, Int, SVector{3,T}))
+    precompile(zero_bead_force!, (FE, Int))
+    precompile(add_energy!, (FE, T))
+    precompile(zero_energy!, (FE,))
+    precompile(get_force!, (FE, Vector{SVector{3,T}}))
+    precompile(get_bead_force, (FE, Int))
+    precompile(get_force, (FE,))
+    precompile(combine_force_energy!, (FE, FE))
+end
+# ForceNoEnergyFixedPoint has no get_energy method.
+precompile(get_energy, (ForceEnergyFloat64,))
+precompile(get_energy, (ForceEnergyFixedPoint{30, 30},))
+
+end