From 7f493f682503f5186308de7b8f74b5b49233cfe4 Mon Sep 17 00:00:00 2001
From: chai <215380520@qq.com>
Date: Thu, 2 Nov 2023 11:51:31 +0800
Subject: +init

---
 GameCode/Mill_ImprovementExplosiveTrap.cs | 64 +++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)
 create mode 100644 GameCode/Mill_ImprovementExplosiveTrap.cs

(limited to 'GameCode/Mill_ImprovementExplosiveTrap.cs')

diff --git a/GameCode/Mill_ImprovementExplosiveTrap.cs b/GameCode/Mill_ImprovementExplosiveTrap.cs
new file mode 100644
index 0000000..2b087fd
--- /dev/null
+++ b/GameCode/Mill_ImprovementExplosiveTrap.cs
@@ -0,0 +1,64 @@
+using UnityEngine;
+
+public class Mill_ImprovementExplosiveTrap : MonoBehaviour, DayNightCycle.IDaytimeSensitive
+{
+	[SerializeField]
+	private Weapon weaponLvl1;
+
+	[SerializeField]
+	private Weapon weaponLvl2;
+
+	[SerializeField]
+	private Weapon weaponLvl3;
+
+	[SerializeField]
+	private Hp hpOfBuilding;
+
+	[SerializeField]
+	private BuildSlot buildSlot;
+
+	public Mesh lvl2Mesh;
+
+	public Mesh lvl3Mesh;
+
+	private void Start()
+	{
+		DayNightCycle.Instance.RegisterDaytimeSensitiveObject(this);
+	}
+
+	public void OnDawn_AfterSunrise()
+	{
+	}
+
+	public void OnDawn_BeforeSunrise()
+	{
+	}
+
+	public void OnDusk()
+	{
+		SetCorrectWeapon();
+	}
+
+	private void OnEnable()
+	{
+		buildSlot.upgrades[1].upgradeBranches[0].replacementMesh = lvl2Mesh;
+		buildSlot.upgrades[2].upgradeBranches[0].replacementMesh = lvl3Mesh;
+		SetCorrectWeapon();
+	}
+
+	private void SetCorrectWeapon()
+	{
+		if (buildSlot.Level == 1)
+		{
+			hpOfBuilding.SpwanAttackOnDeath = weaponLvl1;
+		}
+		else if (buildSlot.Level == 2)
+		{
+			hpOfBuilding.SpwanAttackOnDeath = weaponLvl2;
+		}
+		else if (buildSlot.Level == 3)
+		{
+			hpOfBuilding.SpwanAttackOnDeath = weaponLvl3;
+		}
+	}
+}
-- 
cgit v1.1-26-g67d0