From 0411dcf3db277ed0d3c9f0201b7554a7d61ed1e8 Mon Sep 17 00:00:00 2001
From: wartech0 <wartech0@hotmail.com>
Date: Tue, 31 Dec 2019 04:20:36 -0600
Subject: Finished chest events

---
 .../StateTracking/Snapshots/LocationSnapshot.cs      | 20 ++++++++++----------
 .../Snapshots/WorldLocationsSnapshot.cs              |  2 +-
 2 files changed, 11 insertions(+), 11 deletions(-)

(limited to 'src/SMAPI/Framework/StateTracking/Snapshots')

diff --git a/src/SMAPI/Framework/StateTracking/Snapshots/LocationSnapshot.cs b/src/SMAPI/Framework/StateTracking/Snapshots/LocationSnapshot.cs
index 4074336b..62a56c84 100644
--- a/src/SMAPI/Framework/StateTracking/Snapshots/LocationSnapshot.cs
+++ b/src/SMAPI/Framework/StateTracking/Snapshots/LocationSnapshot.cs
@@ -1,6 +1,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using Microsoft.Xna.Framework;
+using StardewModdingAPI.Events;
 using StardewValley;
 using StardewValley.Buildings;
 using StardewValley.TerrainFeatures;
@@ -35,7 +36,7 @@ namespace StardewModdingAPI.Framework.StateTracking.Snapshots
         public SnapshotListDiff<KeyValuePair<Vector2, TerrainFeature>> TerrainFeatures { get; } = new SnapshotListDiff<KeyValuePair<Vector2, TerrainFeature>>();
 
         /// <summary>Tracks changed chest inventories.</summary>
-        public IDictionary<Vector2, SnapshotListDiff<Item>> ChestItems { get; } = new Dictionary<Vector2, SnapshotListDiff<Item>>();
+        public IDictionary<StardewValley.Objects.Chest, ItemStackChange[]> ChestItems { get; } = new Dictionary<StardewValley.Objects.Chest, ItemStackChange[]>();
 
 
         /*********
@@ -61,17 +62,16 @@ namespace StardewModdingAPI.Framework.StateTracking.Snapshots
             this.TerrainFeatures.Update(watcher.TerrainFeaturesWatcher);
 
             // chest inventories
-            foreach (Vector2 key in this.ChestItems.Keys.ToArray())
-            {
-                if (!watcher.ChestWatchers.ContainsKey(key))
-                    this.ChestItems.Remove(key);
-            }
             foreach (var pair in watcher.ChestWatchers)
             {
-                if (!this.ChestItems.TryGetValue(pair.Key, out var diff))
-                    this.ChestItems[pair.Key] = diff = new SnapshotListDiff<Item>();
-
-                diff.Update(pair.Value);
+                IEnumerable<ItemStackChange> temp = pair.Value.GetInventoryChanges();
+                if (temp.Any())
+                    if (this.ChestItems.ContainsKey(pair.Value.Chest))
+                        this.ChestItems[pair.Value.Chest] = pair.Value.GetInventoryChanges().ToArray();
+                    else
+                        this.ChestItems.Add(pair.Value.Chest, pair.Value.GetInventoryChanges().ToArray());
+                else
+                    this.ChestItems.Remove(pair.Value.Chest);
             }
         }
     }
diff --git a/src/SMAPI/Framework/StateTracking/Snapshots/WorldLocationsSnapshot.cs b/src/SMAPI/Framework/StateTracking/Snapshots/WorldLocationsSnapshot.cs
index ed8001d6..73ed2d8f 100644
--- a/src/SMAPI/Framework/StateTracking/Snapshots/WorldLocationsSnapshot.cs
+++ b/src/SMAPI/Framework/StateTracking/Snapshots/WorldLocationsSnapshot.cs
@@ -43,7 +43,7 @@ namespace StardewModdingAPI.Framework.StateTracking.Snapshots
             foreach (LocationTracker locationWatcher in watcher.Locations)
             {
                 if (!this.LocationsDict.TryGetValue(locationWatcher.Location, out LocationSnapshot snapshot))
-                    this.LocationsDict[locationWatcher.Location] = snapshot = new LocationSnapshot(locationWatcher);
+                    this.LocationsDict[locationWatcher.Location] = snapshot = new LocationSnapshot(locationWatcher.Location);
 
                 snapshot.Update(locationWatcher);
             }
-- 
cgit