fixed slicing for images with the condition totalPixels/threadCount != Integer

Co-authored-by: darkress <30271678+DarkressX@users.noreply.github.com>
Reviewed-on: darkress/PixelFloodClient#2

de/darkress/pixelfood/Main.java

@@ -57,7 +57,7 @@ class Main
         return null;
     }
 
-    private static String[] prepareArray(BufferedImage image, int xOffset, int yOffset) {
+    private static ArrayList<String> prepareArray(BufferedImage image, int xOffset, int yOffset) {
 
         int width = image.getWidth();
         int height = image.getHeight();
@@ -79,13 +79,28 @@ class Main
                 }
             }
         }
-        String[] payloadArray = new String[pixelArray.size()];
+        /*String[] payloadArray = new String[pixelArray.size()];
         for(int i = 0; i < payloadArray.length; i++)
         {
             payloadArray[i] = pixelArray.get(i);
         }
 
-        return payloadArray;
+        return payloadArray;*/
+        return pixelArray;
+    }
+
+    public static String[] prepareThreadArray(ArrayList<String> fullArray, int threadNumber, int threadCount)
+    {
+        int count = 0;
+        int arrayStart = (threadNumber*fullArray.size())/threadCount;
+        int arrayEnd = ((threadNumber+1)*fullArray.size())/threadCount;
+        String[] threadArray = new String[arrayEnd-arrayStart];
+        for(int i = arrayStart; i < arrayEnd; i++)
+        {
+            threadArray[count] = fullArray.get(i);
+            count++;
+        }
+        return threadArray;
     }
 
     //Syntax blabla.jar x y Hostname Port Image.png
@@ -94,41 +109,23 @@ class Main
         final String HOSTNAME = args[2];
         final int xOffset = Integer.parseInt(args[0]);
         final int yOffset = Integer.parseInt(args[1]);
-        final String imageName = args[4];
+        int threadCount = Integer.parseInt(args[4]);
+        final String imageName = args[5];
 
         // Read the PNG file
         BufferedImage image = processImage(imageName);
-        String[] pixelArray = prepareArray(image, xOffset, yOffset);
+        ArrayList<String> fullArray = prepareArray(image, xOffset, yOffset);
+        ArrayList<Worker> workerList = new ArrayList<>();
 
-        Worker worker0 = new Worker(HOSTNAME, PORT, pixelArray);
+        for(int i = 0; i < threadCount; i++)
-        Worker worker1 = new Worker(HOSTNAME, PORT, pixelArray);
+        {
-        Worker worker2 = new Worker(HOSTNAME, PORT, pixelArray);
+            String[] threadArray = prepareThreadArray(fullArray, i, threadCount);
-        Worker worker3 = new Worker(HOSTNAME, PORT, pixelArray);
+            workerList.add(new Worker(HOSTNAME, PORT, threadArray));
-        Worker worker4 = new Worker(HOSTNAME, PORT, pixelArray);
+        }
-        Worker worker5 = new Worker(HOSTNAME, PORT, pixelArray);
-        Worker worker6 = new Worker(HOSTNAME, PORT, pixelArray);
-        Worker worker7 = new Worker(HOSTNAME, PORT, pixelArray);
 
-
+        for(int i = 0; i < threadCount; i++)
-        worker0.start();
+        {
-        worker1.start();
+            workerList.get(i).start();
-        worker2.start();
-        worker3.start();
-        worker4.start();
-        worker5.start();
-        worker6.start();
-        worker7.start();
-
-        /*try {
-            Socket socket = new Socket(HOSTNAME, PORT);
-            while(true) {
-                for(int i = 0; i < pixelArray.length; i++) {
-                    byte[] payload = pixelArray[i].getBytes();
-                    socket.getOutputStream().write(payload);
         }
     }
-        } catch (IOException e) {
-            e.printStackTrace();
-        }*/
-    }
 }