XnbReader.cs

Go to the documentation of this file.

using System;
using System.IO;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Xna.Framework.Content;
 
namespace ReLogic.Content.Readers;
 
public class XnbReader : IAssetReader, IDisposable
{
    public static class LoadOnMainThread<T>
    {
        public static bool Value;
    }
 
    private class InternalContentManager : ContentManager
    {
        private Stream _stream;
 
        public InternalContentManager(IServiceProvider serviceProvider)
            : base(serviceProvider)
        {
        }
 
        public void SetStream(Stream stream)
        {
            _stream = stream;
        }
 
        public T Load<T>()
        {
            return ((ContentManager)this).ReadAsset<T>("XnaAsset", (Action<IDisposable>)null);
        }
 
        protected override Stream OpenStream(string assetName)
        {
            return _stream;
        }
    }
 
    private readonly ThreadLocal<InternalContentManager> _contentLoader;
 
    private bool _disposedValue;
 
    public XnbReader(IServiceProvider services)
    {
        _contentLoader = new ThreadLocal<InternalContentManager>(() => new InternalContentManager(services));
    }
 
    public async ValueTask<T> FromStream<T>(Stream stream, MainThreadCreationContext mainThreadCtx) where T : class
    {
        InternalContentManager value = _contentLoader.Value;
        if (LoadOnMainThread<T>.Value)
        {
            await mainThreadCtx;
        }
        value.SetStream(stream);
        return value.Load<T>();
    }
 
    protected virtual void Dispose(bool disposing)
    {
        if (!_disposedValue)
        {
            if (disposing)
            {
                _contentLoader.Dispose();
            }
            _disposedValue = true;
        }
    }
 
    public void Dispose()
    {
        Dispose(disposing: true);
    }
}