Add the Visualization instance to VisualizationEvent (#945)

src/main/java/me/ryanhamshire/GriefPrevention/events/VisualizationEvent.java

@@ -1,6 +1,7 @@
 package me.ryanhamshire.GriefPrevention.events;
 
 import me.ryanhamshire.GriefPrevention.Claim;
+import me.ryanhamshire.GriefPrevention.Visualization;
 import org.bukkit.entity.Player;
 import org.bukkit.event.HandlerList;
 import org.bukkit.event.player.PlayerEvent;
@@ -14,6 +15,7 @@ import java.util.Collections;
 public class VisualizationEvent extends PlayerEvent
 {
     private static final HandlerList handlers = new HandlerList();
+    private final Visualization visualization;
     private final Collection<Claim> claims;
     private final boolean showSubdivides;
     private final boolean visualizingNearbyClaims;
@@ -24,9 +26,10 @@ public class VisualizationEvent extends PlayerEvent
      * @param player Player receiving visuals
      * @param claim The claim being visualized (with subdivides), or null if visuals being removed
      */
-    public VisualizationEvent(Player player, Claim claim)
+    public VisualizationEvent(Player player, Visualization visualization, Claim claim)
     {
         super(player);
+        this.visualization = visualization;
         this.claims = Collections.singleton(claim);
         this.showSubdivides = true;
         this.visualizingNearbyClaims = false;
@@ -38,9 +41,9 @@ public class VisualizationEvent extends PlayerEvent
      * @param player Player receiving visuals
      * @param claims Claims being visualized (without subdivides)
      */
-    public VisualizationEvent(Player player, Collection<Claim> claims)
+    public VisualizationEvent(Player player, Visualization visualization, Collection<Claim> claims)
     {
-        this(player, claims, false);
+        this(player, visualization, claims, false);
     }
 
     /**
@@ -50,16 +53,27 @@ public class VisualizationEvent extends PlayerEvent
      * @param claims Claims being visualized (without subdivides)
      * @param visualizingNearbyClaims If the event is called on nearby claims (shift inspecting)
      */
-    public VisualizationEvent(Player player, Collection<Claim> claims, boolean visualizingNearbyClaims)
+    public VisualizationEvent(Player player, Visualization visualization, Collection<Claim> claims, boolean visualizingNearbyClaims)
     {
         super(player);
+        this.visualization = visualization;
         this.claims = claims;
         this.showSubdivides = false;
         this.visualizingNearbyClaims = visualizingNearbyClaims;
     }
 
     /**
-     * Get the claims being visualized, or null if visualization being removed
+     * Get the visualization, or null if visualization being removed
+     *
+     * @return The visualization object
+     */
+    public Visualization getVisualization()
+    {
+        return visualization;
+    }
+
+    /**
+     * Get the claims being visualized, or an empty list if visualization being removed
      *
      * @return Claims being visualized
      */