Replace trunc with round to fix small precision problems

src/ExprEvaluation.jl

@@ -54,7 +54,7 @@ function convert_expr(expr, custom_operations, primitives_with_arity, gradient_f
         if get(primitives_with_arity, expr, 1) == 0
             return :(vars[$(QuoteNode(expr))])
         elseif get(primitives_with_arity, expr, 1) == -1
-            return :(vars[$(QuoteNode(expr))][(vars[:x] + 0.5) * (width-1) + 1 |> trunc |> Int, (vars[:y] + 0.5) * (height-1) + 1 |> trunc |> Int])
+            return :(vars[$(QuoteNode(expr))][(vars[:x] + 0.5) * (width-1) + 1 |> round |> Int, (vars[:y] + 0.5) * (height-1) + 1 |> round |> Int])
         else
             return expr
         end

src/MathOperations.jl

@@ -81,7 +81,7 @@ function x_grad(func, vars, width, height; Δx = 1)
     x_val = vars[:x]
     Δx_scaled = Δx / (width - 1)  # scale Δx to be proportional to the image width
 
-    idx_x = (x_val + 0.5) * (width - 1) + 1 |> trunc |> Int
+    idx_x = (x_val + 0.5) * (width - 1) + 1 |> round |> Int
 
     # Evaluate function at x
     center_val = func(merge(vars, Dict(:x => x_val)))
@@ -99,7 +99,7 @@ function y_grad(func, vars, width, height; Δy = 1)
     y_val = vars[:y]
     Δy_scaled = Δy / (height - 1)  # scale Δy to be proportional to the image height
 
-    idx_y = (y_val + 0.5) * (height - 1) + 1 |> trunc |> Int
+    idx_y = (y_val + 0.5) * (height - 1) + 1 |> round |> Int
 
     # Evaluate function at y
     center_val = func(merge(vars, Dict(:y => y_val)))
@@ -133,8 +133,8 @@ function laplacian(func, vars, width, height; Δx = 1, Δy = 1)
     Δx_scaled = Δx / (width - 1)  # scale Δx to be proportional to the image width
     Δy_scaled = Δy / (height - 1)  # scale Δy to be proportional to the image height
 
-    idx_x = (x_val + 0.5) * (width - 1) + 1 |> trunc |> Int
-    idx_y = (y_val + 0.5) * (height - 1) + 1 |> trunc |> Int
+    idx_x = (x_val + 0.5) * (width - 1) + 1 |> round |> Int
+    idx_y = (y_val + 0.5) * (height - 1) + 1 |> round |> Int
 
     center_val = func(merge(vars, Dict(:x => x_val, :y => y_val)))
 
@@ -190,8 +190,8 @@ function neighbor_min(func, vars, width, height; Δx = 1, Δy = 1)
 
 
     if any([isa(v, Matrix) for v in values(vars)])
-        idx_x = (x_val + 0.5) * (width - 1) + 1 |> trunc |> Int
-        idx_y = (y_val + 0.5) * (height - 1) + 1 |> trunc |> Int
+        idx_x = (x_val + 0.5) * (width - 1) + 1 |> round |> Int
+        idx_y = (y_val + 0.5) * (height - 1) + 1 |> round |> Int
 
         # Filter the iterations that are not in the matrix
         range_x = filter(x -> 1 <= idx_x + x <= width, range_x)
@@ -232,8 +232,8 @@ function neighbor_max(func, vars, width, height; Δx = 1, Δy = 1)
     range_y = -Δy:Δy
 
     if any([isa(v, Matrix) for v in values(vars)])
-        idx_x = (x_val + 0.5) * (width - 1) + 1 |> trunc |> Int
-        idx_y = (y_val + 0.5) * (height - 1) + 1 |> trunc |> Int
+        idx_x = (x_val + 0.5) * (width - 1) + 1 |> round |> Int
+        idx_y = (y_val + 0.5) * (height - 1) + 1 |> round |> Int
 
         # Filter the iterations that are not in the matrix
         range_x = filter(x -> 1 <= idx_x + x <= width, range_x)
@@ -275,8 +275,8 @@ function neighbor_ave(func, vars, width, height; Δx = 1, Δy = 1)
     range_y = -Δy:Δy
 
     if any([isa(v, Matrix) for v in values(vars)])
-        idx_x = (x_val + 0.5) * (width - 1) + 1 |> trunc |> Int
-        idx_y = (y_val + 0.5) * (height - 1) + 1 |> trunc |> Int
+        idx_x = (x_val + 0.5) * (width - 1) + 1 |> round |> Int
+        idx_y = (y_val + 0.5) * (height - 1) + 1 |> round |> Int
 
         # Filter the iterations that are not in the matrix
         range_x = filter(x -> 1 <= idx_x + x <= width, range_x)