Puppet2D.System/StreamAdaptor.cs - master - EugenyCh7/Puppet2D.Net

Puppet2D.System/StreamAdaptor.cs
162 строки5 KB
EugenyCh7
comments fix
24 июл 2025, 22:54
24 июл 2025, 22:54421ed1f
using System;
using System.IO;
using System.Runtime.InteropServices;

namespace Puppet2D.System
{
    /// <summary>
    /// Structure that contains InputStream callbacks
    /// (directly maps to a CSFML sfInputStream)
    /// </summary>
    [StructLayout(LayoutKind.Sequential)]
    public struct InputStream
    {
        /// <summary>
        /// Type of callback to read data from the current stream
        /// </summary>
        [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
        public delegate long ReadCallbackType(IntPtr data, long size, IntPtr userData);

        /// <summary>
        /// Type of callback to seek the current stream's position
        /// </summary>
        [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
        public delegate long SeekCallbackType(long position, IntPtr userData);

        /// <summary>
        /// Type of callback to return the current stream's position
        /// </summary>
        [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
        public delegate long TellCallbackType(IntPtr userData);

        /// <summary>
        /// Type of callback to return the current stream's size
        /// </summary>
        [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
        public delegate long GetSizeCallbackType(IntPtr userData);

        /// <summary>
        /// Function that is called to read data from the stream
        /// </summary>
        public ReadCallbackType Read;

        /// <summary>
        /// Function that is called to seek the stream
        /// </summary>
        public SeekCallbackType Seek;

        /// <summary>
        /// Function that is called to return the position
        /// </summary>
        public TellCallbackType Tell;

        /// <summary>
        /// Function that is called to return the size
        /// </summary>
        public GetSizeCallbackType GetSize;
    }

    /// <summary>
    /// Adapts a System.IO.Stream to be usable as a Puppet2D InputStream
    /// </summary>
    public class StreamAdaptor : IDisposable
    {
        /// <summary>
        /// Construct from a System.IO.Stream
        /// </summary>
        /// <param name="stream">Stream to adapt</param>
        public StreamAdaptor(Stream stream)
        {
            myStream = stream;

            myInputStream = new InputStream
            {
                Read = new InputStream.ReadCallbackType(Read),
                Seek = new InputStream.SeekCallbackType(Seek),
                Tell = new InputStream.TellCallbackType(Tell),
                GetSize = new InputStream.GetSizeCallbackType(GetSize)
            };

            myInputStreamPtr = Marshal.AllocHGlobal(Marshal.SizeOf(myInputStream));
            Marshal.StructureToPtr(myInputStream, myInputStreamPtr, false);
        }

        /// <summary>
        /// Dispose the object
        /// </summary>
        ~StreamAdaptor()
        {
            Dispose(false);
        }

        /// <summary>
        /// The pointer to the CSFML InputStream structure
        /// </summary>
        public IntPtr InputStreamPtr => myInputStreamPtr;

        /// <summary>
        /// Explicitly dispose the object
        /// </summary>
        public void Dispose()
        {
            Dispose(true);
            GC.SuppressFinalize(this);
        }

        /// <summary>
        /// Destroy the object
        /// </summary>
        /// <param name="disposing">Is the GC disposing the object, or is it an explicit call ?</param>
        private void Dispose(bool disposing)
        {
            Marshal.FreeHGlobal(myInputStreamPtr);
        }

        /// <summary>
        /// Called to read from the stream
        /// </summary>
        /// <param name="data">Where to copy the read bytes</param>
        /// <param name="size">Size to read, in bytes</param>
        /// <param name="userData">User data -- unused</param>
        /// <returns>Number of bytes read</returns>
        private long Read(IntPtr data, long size, IntPtr userData)
        {
            byte[] buffer = new byte[size];
            int count = myStream.Read(buffer, 0, (int)size);
            Marshal.Copy(buffer, 0, data, count);
            return count;
        }

        /// <summary>
        /// Called to set the read position in the stream
        /// </summary>
        /// <param name="position">New read position</param>
        /// <param name="userData">User data -- unused</param>
        /// <returns>Actual position</returns>
        private long Seek(long position, IntPtr userData)
        {
            return myStream.Seek(position, SeekOrigin.Begin);
        }

        /// <summary>
        /// Get the current read position in the stream
        /// </summary>
        /// <param name="userData">User data -- unused</param>
        /// <returns>Current position in the stream</returns>
        private long Tell(IntPtr userData)
        {
            return myStream.Position;
        }

        /// <summary>
        /// Called to get the total size of the stream
        /// </summary>
        /// <param name="userData">User data -- unused</param>
        /// <returns>Number of bytes in the stream</returns>
        private long GetSize(IntPtr userData) => myStream.Length;

        private Stream myStream;
        private InputStream myInputStream;
        private readonly IntPtr myInputStreamPtr;
    }
}