rhythmbullet/RhythmBullet/Zer01HD/Utilities/ContentSystem/ContentSystem.cs

using Microsoft.Xna.Framework.Content;
using Microsoft.Xna.Framework.Graphics;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace RhythmBullet.Zer01HD.Utilities.ContentSystem
{
    public class ContentSystem
    {
        volatile bool queued;
        Thread thread;
        readonly ContentManager contentManager;
        readonly Queue<LoadableContent> queue;
        Dictionary<string, IDisposable> assets;
        public readonly Dictionary<Type, IContentResolver> contentResolver;

        public ContentSystem(ContentManager contentManager)
        {
            this.contentManager = contentManager;
            assets = new Dictionary<string, IDisposable>();
            queue = new Queue<LoadableContent>();
            contentResolver = new Dictionary<Type, IContentResolver>();
        }

        private void Load(string assetName, Type type)
        {
            IContentResolver handler = contentResolver[type];
            string path = handler.Load(assetName);
            assets.Add(assetName, contentManager.Load<IDisposable>(path));
        }

        private void LoadBatch()
        {
            while (queue.Count != 0)
            {
                lock (queue)
                {
                    LoadableContent content = queue.Dequeue();
                    Load(content.assetName, content.type);
                }
                queued = false;
            }
        }

        public T Get<T>(string assetName)
        {
            lock(queue)
            {
                return (T)assets[assetName];
            }
        }

        public void Queue(string assetName, Type type)
        {
            queued = true;
            lock (queue)
            {
                if (!assets.ContainsKey(assetName))
                {
                    queue.Enqueue(new LoadableContent(assetName, type));
                }
            }
        }

        /// <summary>
        /// Called whenever a batch of assets should be loaded from the queue. Safe to call once every frame.
        /// </summary>
        public void Update()
        {
            if (queue.Count > 0)
            {
                if (thread == null || !thread.IsAlive)
                {
                    ThreadStart threadStart = new ThreadStart(LoadBatch);
                    thread = new Thread(threadStart);
                }
            }
        }

        /// <summary>
        /// Removes the asset from the list of assets in the system.
        /// Cannot remove from queue.
        /// </summary>
        /// <param name="name">the string name used to load the asset</param>
        public void Remove(string name)
        {
            if (assets.ContainsKey(name))
            {
                assets[name].Dispose();
                assets.Remove(name);
            }
        }

        /// <summary>
        /// Clears the queue.
        /// </summary>
        public void ClearQueue()
        {
            lock(queue)
            {
            queue.Clear();
            }
        }

        /// <summary>
        /// Unloads everything from both queue and loaded list while properly disposing of the assets loaded.
        /// </summary>
        public void UnloadAll()
        {
            foreach (KeyValuePair<string, IDisposable> asset in assets)
            {
                asset.Value.Dispose();
                assets.Remove(asset.Key);
            }
            ClearQueue();
        }

        public bool Done()
        {
            return queued;
        }
    }
}