Forbidden plants

addons/pvzadventure/scripts/seedpacket-editor/ForbiddableSeedpacket.cs

@@ -0,0 +1,28 @@
+using Godot;
+using Newlon.Resources;
+
+[Tool]
+public partial class ForbiddableSeedpacket : TextureButton
+{
+	[Signal] public delegate void SaveCallbackEventHandler();
+	[Export] private TextureRect preview;
+	[Export] private TextureRect forrbidTexture;
+	public PlantResource HeldResource;
+	public bool forbidden = false;
+	public override void _Pressed()
+	{
+		forbidden = !forbidden;
+		Update();
+	}
+
+	private void Update()
+	{
+		RefreshTexture();
+		EmitSignal(SignalName.SaveCallback);
+	}
+	public void RefreshTexture()
+	{
+		preview.Texture = HeldResource.Preview;
+		forrbidTexture.Visible = forbidden;
+	}
+}

addons/pvzadventure/scripts/seedpacket-editor/ForbiddableSeedpacket.cs.uid

@@ -0,0 +1 @@
+uid://cy5t2lk5g75x7

addons/pvzadventure/scripts/seedpacket-editor/SeedpacketEditor.cs

@@ -1,5 +1,7 @@
+using System.Collections.Generic;
 using Godot;
 using Godot.Collections;
+using Newlon.Components.GUI;
 using Newlon.Resources;
 
 [Tool]
@@ -9,15 +11,17 @@ public partial class SeedpacketEditor : BaseEditor
     public override void SetEditedData(AdventureLevelResource data)
     {
         base.SetEditedData(data);
-        SetPrepickedPlants();
+        SetupPrepickedPlants();
+        SetupForbiddablePlants();
     }
 
     public override void Save()
     {
         CalculatePrepickedPlants();
+        CalculateForbiddenPlants();
         base.Save();
     }
-    private void SetPrepickedPlants()
+    private void SetupPrepickedPlants()
     {
         for (int i = 0; i < editedResource.prepickedPlants.Count; i++)
         {
@@ -28,6 +32,21 @@ public partial class SeedpacketEditor : BaseEditor
             }
         }
     }
+    private void SetupForbiddablePlants()
+    {
+        var plants = GetPlants();
+        var packed = ResourceLoader.Load<PackedScene>("uid://dymyownbt688c");
+        for (int i = 0; i < plants.Count; i++)
+        {
+            var seedpacket = packed.Instantiate<ForbiddableSeedpacket>();
+            seedpacket.HeldResource = plants[i];
+            seedpacket.forbidden = editedResource.forbiddenPlants.Contains(plants[i].GetInternalID());
+            seedpacket.RefreshTexture();
+            seedpacket.SaveCallback += Save;
+
+            GetNode("%FPlantsContainer").AddChild(seedpacket);
+        }
+    }
     private void CalculatePrepickedPlants()
     {
         Array<string> prepicked = new();
@@ -40,6 +59,20 @@ public partial class SeedpacketEditor : BaseEditor
         }
         editedResource.prepickedPlants = prepicked;
     }
+    private void CalculateForbiddenPlants()
+    {
+        Array<string> forbidden = new();
+
+        foreach (var child in GetNode("%FPlantsContainer").GetChildren())
+        {
+            if (child is ForbiddableSeedpacket packet && packet.forbidden)
+            {
+                forbidden.Add(packet.HeldResource.GetInternalID());
+            }
+        }
+
+        editedResource.forbiddenPlants = forbidden;
+    }
     private PlantResource TryGetPlant(string plantName)
     {
         foreach (var file in ResourceLoader.ListDirectory(PLANTS_DIRECTORY))
@@ -49,5 +82,20 @@ public partial class SeedpacketEditor : BaseEditor
         }
         return null;
     }
+    private List<PlantResource> GetPlants()
+    {
+        var result = new List<PlantResource>();
+
+        foreach (var file in ResourceLoader.ListDirectory(PLANTS_DIRECTORY))
+        {
+            result.Add(ResourceLoader.Load<PlantResource>(PLANTS_DIRECTORY + file));
+        }
+        result.Sort((a, b) =>
+        {
+            return a.Order - b.Order;
+        });
+
+        return result;
+    }
 
 }

addons/pvzadventure/scripts/seedpacket-editor/editor_seedpacket.gd

@@ -1,28 +0,0 @@
-@tool
-extends TextureButton
-
-signal save_callback
-
-var held_data : PlantResource = null
-@export var preview : TextureRect
-
-func _can_drop_data(at_position: Vector2, data: Variant) -> bool:
-	if typeof(data) == TYPE_DICTIONARY:
-		if data.type == "files":
-			return load(data.files[0]) is PlantResource
-	return false
-
-func _drop_data(at_position: Vector2, data: Variant) -> void:
-	held_data = load(data.files[0]) as PlantResource
-	update()
-
-func update():
-	if held_data:
-		preview.texture = held_data.Preview
-	else:
-		preview.texture = null
-	save_callback.emit()
-
-func _pressed() -> void:
-	held_data = null
-	update()

addons/pvzadventure/scripts/seedpacket-editor/editor_seedpacket.gd.uid

@@ -1 +0,0 @@
-uid://donugf6vnchij