NBTExplorer/SubstrateCS/Source/RegionManager.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;
using Substrate.Core;

namespace Substrate
{
    public class BetaRegionManager : RegionManager
    {
        public BetaRegionManager (string regionDir, ChunkCache cache)
            : base(regionDir, cache)
        {
        }

        protected override IRegion CreateRegionCore (int rx, int rz)
        {
            return new BetaRegion(this, _chunkCache, rx, rz);
        }

        protected override RegionFile CreateRegionFileCore (int rx, int rz)
        {
            string fp = "r." + rx + "." + rz + ".mcr";
            return new RegionFile(Path.Combine(_regionPath, fp));
        }

        protected override void DeleteRegionCore (IRegion region)
        {
            BetaRegion r = region as BetaRegion;
            if (r != null) {
                r.Dispose();
            }
        }

        public override IRegion GetRegion (string filename)
        {
            int rx, rz;
            if (!BetaRegion.ParseFileName(filename, out rx, out rz)) {
                throw new ArgumentException("Malformed region file name: " + filename, "filename");
            }

            return GetRegion(rx, rz);
        }
    }

    public class AnvilRegionManager : RegionManager
    {
        public AnvilRegionManager (string regionDir, ChunkCache cache)
            : base(regionDir, cache)
        {
        }

        protected override IRegion CreateRegionCore (int rx, int rz)
        {
            return new AnvilRegion(this, _chunkCache, rx, rz);
        }

        protected override RegionFile CreateRegionFileCore (int rx, int rz)
        {
            string fp = "r." + rx + "." + rz + ".mca";
            return new RegionFile(Path.Combine(_regionPath, fp));
        }

        protected override void DeleteRegionCore (IRegion region)
        {
            AnvilRegion r = region as AnvilRegion;
            if (r != null) {
                r.Dispose();
            }
        }

        public override IRegion GetRegion (string filename)
        {
            int rx, rz;
            if (!AnvilRegion.ParseFileName(filename, out rx, out rz)) {
                throw new ArgumentException("Malformed region file name: " + filename, "filename");
            }

            return GetRegion(rx, rz);
        }
    }

    /// <summary>
    /// Manages the regions of a Beta-compatible world.
    /// </summary>
    public abstract class RegionManager : IRegionManager
    {
        protected string _regionPath;

        protected Dictionary<RegionKey, IRegion> _cache;

        protected ChunkCache _chunkCache;


        protected abstract IRegion CreateRegionCore (int rx, int rz);

        protected abstract RegionFile CreateRegionFileCore (int rx, int rz);

        protected abstract void DeleteRegionCore (IRegion region);

        public abstract IRegion GetRegion (string filename);

        /// <summary>
        /// Creates a new instance of a <see cref="RegionManager"/> for the given region directory and chunk cache.
        /// </summary>
        /// <param name="regionDir">The path to a directory containing region files.</param>
        /// <param name="cache">The shared chunk cache to hold chunk data in.</param>
        public RegionManager (string regionDir, ChunkCache cache)
        {
            _regionPath = regionDir;
            _chunkCache = cache;
            _cache = new Dictionary<RegionKey, IRegion>();
        }

        /// <summary>
        /// Gets a <see cref="Region"/> at the given coordinates.
        /// </summary>
        /// <param name="rx">The global X-coordinate of a region.</param>
        /// <param name="rz">The global Z-coordinate of a region.</param>
        /// <returns>A <see cref="Region"/> representing a region at the given coordinates, or null if the region does not exist.</returns>
        public IRegion GetRegion (int rx, int rz)
        {
            RegionKey k = new RegionKey(rx, rz);
            IRegion r;

            try {
                if (_cache.TryGetValue(k, out r) == false) {
                    r = CreateRegionCore(rz, rz);
                    _cache.Add(k, r);
                }
                return r;
            }
            catch (FileNotFoundException) {
                _cache.Add(k, null);
                return null;
            }
        }

        /// <inherits />
        public bool RegionExists (int rx, int rz)
        {
            IRegion r = GetRegion(rx, rz);
            return r != null;
        }

        /// <inherits />
        public IRegion CreateRegion (int rx, int rz)
        {
            IRegion r = GetRegion(rx, rz);
            if (r == null) {
                string fp = "r." + rx + "." + rz + ".mca";
                using (RegionFile rf = CreateRegionFileCore(rx, rz)) {
                    
                }

                r = CreateRegionCore(rx, rz);

                RegionKey k = new RegionKey(rx, rz);
                _cache[k] = r;
            }

            return r;
        }

        /// <summary>
        /// Get the current region directory path.
        /// </summary>
        /// <returns>The path to the region directory.</returns>
        public string GetRegionPath ()
        {
            return _regionPath;
        }

        // XXX: Exceptions
        /// <inherits />
        public bool DeleteRegion (int rx, int rz)
        {
            IRegion r = GetRegion(rx, rz);
            if (r == null) {
                return false;
            }

            RegionKey k = new RegionKey(rx, rz);
            _cache.Remove(k);

            DeleteRegionCore(r);

            try {
                File.Delete(r.GetFilePath());
            }
            catch (Exception e) {
                Console.WriteLine("NOTICE: " + e.Message);
                return false;
            }

            return true;
        }

        #region IEnumerable<IRegion> Members

        /// <summary>
        /// Returns an enumerator that iterates over all of the regions in the underlying dimension.
        /// </summary>
        /// <returns>An enumerator instance.</returns>
        public IEnumerator<IRegion> GetEnumerator ()
        {
            return new Enumerator(this);
        }

        #endregion

        #region IEnumerable Members

        /// <summary>
        /// Returns an enumerator that iterates over all of the regions in the underlying dimension.
        /// </summary>
        /// <returns>An enumerator instance.</returns>
        IEnumerator IEnumerable.GetEnumerator ()
        {
            return new Enumerator(this);
        }

        #endregion


        private struct Enumerator : IEnumerator<IRegion>
        {
            private List<IRegion> _regions;
            private int _pos;

            public Enumerator (RegionManager rm)
            {
                _regions = new List<IRegion>();
                _pos = -1;

                if (!Directory.Exists(rm.GetRegionPath())) {
                    throw new DirectoryNotFoundException();
                }

                string[] files = Directory.GetFiles(rm.GetRegionPath());
                _regions.Capacity = files.Length;

                foreach (string file in files) {
                    try {
                        IRegion r = rm.GetRegion(file);
                        _regions.Add(r);
                    }
                    catch (ArgumentException) {
                        continue;
                    }
                }
            }

            public bool MoveNext ()
            {
                _pos++;
                return (_pos < _regions.Count);
            }

            public void Reset ()
            {
                _pos = -1;
            }

            void IDisposable.Dispose () { }

            object IEnumerator.Current
            {
                get
                {
                    return Current;
                }
            }

            IRegion IEnumerator<IRegion>.Current
            {
                get
                {
                    return Current;
                }
            }

            public IRegion Current
            {
                get
                {
                    try {
                        return _regions[_pos];
                    }
                    catch (IndexOutOfRangeException) {
                        throw new InvalidOperationException();
                    }
                }
            }
        }

    }
}