From 6f4cb3c0ad70f721440e74f47ba50b307205c3c8 Mon Sep 17 00:00:00 2001
From: FICTURE7 <ficture7@gmail.com>
Date: Sun, 11 Apr 2021 13:54:33 +0400
Subject: [PATCH] Make Counter<T> disposable

---
 ARMeilleure/Common/Counter.cs | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/ARMeilleure/Common/Counter.cs b/ARMeilleure/Common/Counter.cs
index f8c5d2732..8fd73f114 100644
--- a/ARMeilleure/Common/Counter.cs
+++ b/ARMeilleure/Common/Counter.cs
@@ -6,8 +6,9 @@ namespace ARMeilleure.Common
     /// Represents a numeric counter.
     /// </summary>
     /// <typeparam name="T">Type of the counter</typeparam>
-    class Counter<T> where T : unmanaged
+    class Counter<T> : IDisposable where T : unmanaged
     {
+        private bool _disposed;
         private readonly int _index;
         private readonly EntryTable<T> _countTable;
 
@@ -26,7 +27,18 @@ namespace ARMeilleure.Common
         /// <summary>
         /// Gets a reference to the value of the counter.
         /// </summary>
-        public ref T Value => ref _countTable.GetValue(_index);
+        public ref T Value 
+        {
+            get
+            {
+                if (_disposed)
+                {
+                    throw new ObjectDisposedException(null);
+                }
+
+                return ref _countTable.GetValue(_index); 
+            }
+        }
 
         /// <summary>
         /// Tries to create a <see cref="Counter"/> instance from the specified <see cref="EntryTable{byte}"/> instance.
@@ -64,5 +76,18 @@ namespace ARMeilleure.Common
 
             return false;
         }
+
+        /// <summary>
+        /// Releases all resources used by the <see cref="Counter{T}"/> instance.
+        /// </summary>
+        public void Dispose()
+        {
+            if (!_disposed)
+            {
+                _countTable.Free(_index);
+
+                _disposed = true;
+            }
+        }
     }
 }
\ No newline at end of file