Fix customisation (#78)

Co-authored-by: mped822 <[email protected]>

Author: scorpionknifes

src/quinzical/controllers/CustomiseController.java

@@ -64,8 +64,8 @@ public void initialize() {
 
     @FXML
     private void handleBackButton(final ActionEvent event) {
+        _imageModel.getImageView().setImage(new Image(_images[_index].getFilename()));
         _imageModel.postImage(_images[_index]);
-        _sceneManager.unloadScene();
         _sceneManager.backScene();
     }
 

src/quinzical/controllers/OpeningMenuController.java

@@ -64,6 +64,7 @@ public void initialize() {
         } else {
             _sheep.setImage(new Image(currentSheep.getFilename()));
         }
+        ImageModel.getInstance().setImageView(_sheep);
     }
 
     /**

src/quinzical/util/api/ImageModel.java

@@ -1,5 +1,6 @@
 package quinzical.util.api;
 
+import javafx.scene.image.ImageView;
 import quinzical.controllers.util.Sheep;
 import quinzical.util.models.LoginModel;
 
@@ -13,6 +14,8 @@ public final class ImageModel {
 
     private static ImageModel _instance;
 
+    private ImageView _sheep;
+
     private Image _image = new Image();
 
     private ImageModel() {
@@ -30,6 +33,24 @@ public static ImageModel getInstance() {
         return _instance;
     }
 
+    /**
+     * Used to set sheep ImageView
+     * 
+     * @param sheep
+     */
+    public void setImageView(final ImageView sheep) {
+        _sheep = sheep;
+    }
+
+    /**
+     * Used to get sheep ImageView
+     * 
+     * @return sheep
+     */
+    public ImageView getImageView() {
+        return _sheep;
+    }
+
     /**
      * Used to post sheep image
      *