ImageSharp/src/ImageSharp/Memory/DiscontinuousProto/UniformMemoryGroup{T}.Allocated.cs

using System;
using System.Buffers;
using System.Collections.Generic;
using System.Linq;

namespace SixLabors.ImageSharp.Memory.DiscontinuousProto
{
    public abstract partial class UniformMemoryGroup<T>
    {
        private class Allocated : UniformMemoryGroup<T>
        {
            private IMemoryOwner<T>[] memoryOwners;

            public Allocated(IMemoryOwner<T>[] memoryOwners)
            {
                this.memoryOwners = memoryOwners;
            }

            public override IEnumerator<Memory<T>> GetEnumerator()
            {
                this.EnsureNotDisposed();
                return this.memoryOwners.Select(mo => mo.Memory).GetEnumerator();
            }


            public override int Count
            {
                get
                {
                    this.EnsureNotDisposed();
                    return this.memoryOwners.Length;
                }
            }

            public override Memory<T> this[int index]
            {
                get
                {
                    this.EnsureNotDisposed();
                    return this.memoryOwners[index].Memory;
                }
            }

            public override void Dispose()
            {
                if (this.memoryOwners == null)
                {
                    return;
                }

                foreach (IMemoryOwner<T> memoryOwner in this.memoryOwners)
                {
                    memoryOwner.Dispose();
                }

                this.memoryOwners = null;
                this.IsValid = false;
            }

            private void EnsureNotDisposed()
            {
                if (this.memoryOwners == null)
                {
                    throw new ObjectDisposedException(nameof(UniformMemoryGroup<T>));
                }
            }
        }
    }
}