TensorFlow.NET/src/TensorFlowNET.Core/DisposableObject.cs

/*****************************************************************************
   Copyright 2018 The TensorFlow.NET Authors. All Rights Reserved.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
******************************************************************************/

using System;
using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Runtime.CompilerServices;
using System.Text;

namespace Tensorflow
{
    /// <summary>
    /// Abstract class for disposable object allocated in unmanaged runtime.
    /// </summary>
    public abstract class DisposableObject : IDisposable
    {
        protected IntPtr _handle;
        protected bool _disposed;

        [SuppressMessage("ReSharper", "UnusedMember.Global")]
        protected DisposableObject()
        { }

        protected DisposableObject(IntPtr handle)
            => _handle = handle;

        [SuppressMessage("ReSharper", "InvertIf")]
        private void internal_dispose(bool disposing)
        {
            if (_disposed)
                return;

            _disposed = true;

            //first handle managed, they might use the unmanaged resources.
            if (disposing)
                // dispose managed state (managed objects).
                DisposeManagedResources();

            //free unmanaged memory
            if (_handle != IntPtr.Zero)
            {
                DisposeUnmanagedResources(_handle);
                _handle = IntPtr.Zero;
            }
        }

        /// <summary>
        ///     Dispose any managed resources.
        /// </summary>
        /// <remarks>Equivalent to what you would perform inside <see cref="Dispose()"/></remarks>
        protected virtual void DisposeManagedResources()
        { }

        /// <summary>
        ///     Dispose any unmanaged resources related to given <paramref name="handle"/>.
        /// </summary>
        protected abstract void DisposeUnmanagedResources(IntPtr handle);

        ~DisposableObject()
        {
            internal_dispose(false);
        }

        public void Dispose()
        {
            lock(this)
            {
                internal_dispose(true);
                GC.SuppressFinalize(this);
            }
        }

        /// <summary>
        ///     If <see cref="_handle"/> is <see cref="IntPtr.Zero"/> then throws <see cref="ObjectDisposedException"/>
        /// </summary>
        /// <exception cref="ObjectDisposedException">When <see cref="_handle"/> is <see cref="IntPtr.Zero"/></exception>
        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        protected void EnsureNotDisposed()
        {
            if (_disposed)
                throw new ObjectDisposedException($"Unable to access disposed object, Type: {GetType().Name}");
        }
    }
}