Change scaling of coefficients for FD solver

fresnel/finite_differences.py

@@ -27,13 +27,13 @@ class Solver2d:
         self.f = self._compute_f(F0)
 
     def _compute_rz(self, Az):
-        return Az * 2 * self.dx ** 2 / self.dz * (1 + 0j)
+        return 2 * Az / self.dz * (1 + 0j)
 
     def _compute_rxx(self, Axx):
-        return -Axx * (1 + 0j)
+        return -Axx / self.dx ** 2 * (1 + 0j)
 
     def _compute_f(self, F):
-        return F * self.dx ** 2 * self._ones
+        return F * self._ones
 
     def step(self, F, boundary):
 
@@ -78,16 +78,16 @@ class Solver2dSym:
         self.f = self._compute_f(F0)
 
     def _compute_rz(self, Az):
-        return Az * 2 * self.dx ** 2 / self.dz * (1 + 0j)
+        return 2 * Az / self.dz * (1 + 0j)
 
     def _compute_rxx(self, Axx):
-        return -Axx * (1 + 0j)
+        return -Axx / self.dx ** 2 * (1 + 0j)
 
     def _compute_rx(self, Ax):
-        return -Ax * (1 + 0j)
+        return -Ax / self.dx ** 2 * (1 + 0j)
 
     def _compute_f(self, F):
-        return F * self.dx ** 2 * self._ones
+        return F * self._ones
 
     def step(self, F, boundary):
 
@@ -135,16 +135,16 @@ class Solver3d:
         self.f = self._compute_f(F0)
 
     def _compute_rz(self, Az):
-        return Az * 2 * self.dx ** 2 / self.dz * (1 + 0j)
+        return 2 * Az * 2 / self.dz * (1 + 0j)
 
     def _compute_rxx(self, Axx):
-        return -Axx * (1 + 0j)
+        return -Axx / self.dx ** 2 * (1 + 0j)
 
     def _compute_ryy(self, Ayy):
-        return -Ayy * self.dx ** 2 / self.dy ** 2 * (1 + 0j)
+        return -Ayy / self.dy ** 2 * (1 + 0j)
 
     def _compute_f(self, F):
-        return F * self.dx ** 2 / 2 * self._ones
+        return F / 2 * self._ones
 
     def step(self, F, boundary):