Adjust grid_size when resolution or geometry_lattice are changed

libpympb/pympb.cpp

@@ -1620,6 +1620,12 @@ std::vector<int> mode_solver::get_dims() {
   return dims;
 }
 
+void mode_solver::set_grid_size(vector3 gs) {
+  grid_size.x = gs.x;
+  grid_size.y = gs.y;
+  grid_size.z = gs.z;
+}
+
 int mode_solver::get_libctl_dimensions() {
   return dimensions;
 }

libpympb/pympb.hpp

@@ -156,6 +156,7 @@ struct mode_solver {
   void set_curfield_type(char t);
   std::string get_parity_string();
   std::vector<int> get_dims();
+  void set_grid_size(vector3 gs);
   int get_libctl_dimensions();
   void set_libctl_dimensions(int val);
   bool get_libctl_ensure_periodicity();

python/solver.py

@@ -120,10 +120,7 @@ def __init__(self,
         self.k_split_index = 0
         self.eigensolver_iters = []
 
-        grid_size = self._get_grid_size()
-
-        if self.optimize_grid_size:
-            grid_size = self._optimize_grid_size(grid_size)
+        grid_size = self._adjust_grid_size()
 
         if type(self.default_material) is not mp.Medium and callable(self.default_material):
             self.default_material.eps = False
@@ -175,6 +172,8 @@ def resolution(self, val):
 
         if self.mode_solver:
             self.mode_solver.resolution = self._resolution
+            grid_size = self._adjust_grid_size()
+            self.mode_solver.set_grid_size(grid_size)
 
     @property
     def geometry_lattice(self):
@@ -185,6 +184,8 @@ def geometry_lattice(self, val):
         self._geometry_lattice = val
         if self.mode_solver:
             self.mode_solver.set_libctl_geometry_lattice(val)
+            grid_size = self._adjust_grid_size()
+            self.mode_solver.set_grid_size(grid_size)
 
     @property
     def tolerance(self):
@@ -306,6 +307,14 @@ def eigensolver_block_size(self):
     def eigensolver_block_size(self, val):
         self.mode_solver.eigensolver_block_size = val
 
+    def _adjust_grid_size(self):
+        grid_size = self._get_grid_size()
+
+        if self.optimize_grid_size:
+            grid_size = self._optimize_grid_size(grid_size)
+
+        return grid_size
+
     def allow_negative_epsilon(self):
         self.is_negative_epsilon_ok = True
         self.target_freq = 1 / mp.inf