Let image processor create shared array

Author: Abogical

lfgen/image_processor.py

@@ -88,6 +88,9 @@ def set_dims_and_get_array(self, x, y, extension):
 
         return self._get_array(image)
 
+    def get_shared_numpy_array(self, array):
+        return np.ndarray((self.buffer_height, self.buffer_width, 3), dtype=np.uint8, buffer=array.buf)
+
     def set_shared_array_from_image(self, x, y, np_img, image):
         array_x_start = x*self.output_width
         array_y_start = (self.max_y-y)*self.output_height
@@ -103,7 +106,7 @@ def set_shared_array(self, x, y, extension, array):
         self.set_shared_array_from_image(
             x,
             y,
-            np.ndarray((self.buffer_height, self.buffer_width, 3), dtype=np.uint8, buffer=array.buf),
+            self.get_shared_numpy_array(array),
             self._get_array(self._get_image(x, y, extension))
         )
 

lfgen/main.py

@@ -126,11 +126,7 @@ def main():
                             shared_array = smm.SharedMemory(
                                 (max_x+1)*img_processor.output_width*(max_y+1)*img_processor.output_height*3
                             )
-                            shared_np_array = np.ndarray((
-                                img_processor.buffer_height,
-                                img_processor.buffer_width,
-                                3
-                            ), buffer=shared_array.buf, dtype=np.uint8)
+                            shared_np_array = img_processor.get_shared_numpy_array(shared_array)
 
                             img_processor.set_shared_array_from_image(x, y, shared_np_array, subimage)