From b380707462d11e5c550fa1ad0d980853726ea4cf Mon Sep 17 00:00:00 2001 From: cosmonaut Date: Thu, 18 Aug 2022 20:45:34 +0000 Subject: [PATCH] Video Optimization (#22) - Videos are now shoved into memory when created to avoid disk latency issues - Added VideoPlayer class to avoid redundant texture creation on videos - Most Video functions are now on VideoPlayer Reviewed-on: https://gitea.moonside.games/MoonsideGames/MoonWorks/pulls/22 --- src/Video/Video.cs | 291 ++++------------------------------ src/Video/VideoPlayer.cs | 335 +++++++++++++++++++++++++++++++++++++++ 2 files changed, 364 insertions(+), 262 deletions(-) create mode 100644 src/Video/VideoPlayer.cs diff --git a/src/Video/Video.cs b/src/Video/Video.cs index 7a5c8283..204fb584 100644 --- a/src/Video/Video.cs +++ b/src/Video/Video.cs @@ -17,62 +17,35 @@ namespace MoonWorks.Video public unsafe class Video : IDisposable { internal IntPtr Handle; + private IntPtr rwData; + private void* videoData; - public bool Loop { get; private set; } - public float Volume { - get => volume; - set - { - volume = value; - if (audioStream != null) - { - audioStream.Volume = value; - } - } - } - public float PlaybackSpeed { get; set; } = 1; public double FramesPerSecond => fps; - private VideoState State = VideoState.Stopped; + public int Width => yWidth; + public int Height => yHeight; + public int UVWidth { get; } + public int UVHeight { get; } private double fps; private int yWidth; private int yHeight; - private int uvWidth; - private int uvHeight; - - private void* yuvData = null; - private int yuvDataLength; - private int currentFrame; - - private GraphicsDevice GraphicsDevice; - private Texture RenderTexture = null; - private Texture yTexture = null; - private Texture uTexture = null; - private Texture vTexture = null; - private Sampler LinearSampler; - - private AudioDevice AudioDevice = null; - private StreamingSoundTheora audioStream = null; - private float volume = 1.0f; - - private Stopwatch timer; - private double lastTimestamp; - private double timeElapsed; private bool disposed; /* TODO: is there some way for us to load the data into memory? */ - public Video(GraphicsDevice graphicsDevice, AudioDevice audioDevice, string filename) + public Video(string filename) { - GraphicsDevice = graphicsDevice; - AudioDevice = audioDevice; - if (!System.IO.File.Exists(filename)) { throw new ArgumentException("Video file not found!"); } - if (Theorafile.tf_fopen(filename, out Handle) < 0) + var bytes = System.IO.File.ReadAllBytes(filename); + videoData = NativeMemory.Alloc((nuint) bytes.Length); + Marshal.Copy(bytes, 0, (IntPtr) videoData, bytes.Length); + rwData = SDL2.SDL.SDL_RWFromMem((IntPtr) videoData, bytes.Length); + + if (Theorafile.tf_open_callbacks(rwData, out Handle, callbacks) < 0) { throw new ArgumentException("Invalid video file!"); } @@ -88,237 +61,35 @@ namespace MoonWorks.Video if (format == Theorafile.th_pixel_fmt.TH_PF_420) { - uvWidth = yWidth / 2; - uvHeight = yHeight / 2; + UVWidth = Width / 2; + UVHeight = Height / 2; } else if (format == Theorafile.th_pixel_fmt.TH_PF_422) { - uvWidth = yWidth / 2; - uvHeight = yHeight; + UVWidth = Width / 2; + UVHeight = Height; } else if (format == Theorafile.th_pixel_fmt.TH_PF_444) { - uvWidth = yWidth; - uvHeight = yHeight; + UVWidth = Width; + UVHeight = Height; } else { throw new NotSupportedException("Unrecognized YUV format!"); } - - yuvDataLength = ( - (yWidth * yHeight) + - (uvWidth * uvHeight * 2) - ); - - yuvData = NativeMemory.Alloc((nuint) yuvDataLength); - - InitializeTheoraStream(); - - if (Theorafile.tf_hasvideo(Handle) == 1) - { - RenderTexture = Texture.CreateTexture2D( - GraphicsDevice, - (uint) yWidth, - (uint) yHeight, - TextureFormat.R8G8B8A8, - TextureUsageFlags.ColorTarget | TextureUsageFlags.Sampler - ); - - yTexture = Texture.CreateTexture2D( - GraphicsDevice, - (uint) yWidth, - (uint) yHeight, - TextureFormat.R8, - TextureUsageFlags.Sampler - ); - - uTexture = Texture.CreateTexture2D( - GraphicsDevice, - (uint) uvWidth, - (uint) uvHeight, - TextureFormat.R8, - TextureUsageFlags.Sampler - ); - - vTexture = Texture.CreateTexture2D( - GraphicsDevice, - (uint) uvWidth, - (uint) uvHeight, - TextureFormat.R8, - TextureUsageFlags.Sampler - ); - - LinearSampler = new Sampler(GraphicsDevice, SamplerCreateInfo.LinearClamp); - } - - timer = new Stopwatch(); } - public void Play(bool loop = false) + private static IntPtr Read(IntPtr ptr, IntPtr size, IntPtr nmemb, IntPtr datasource) => (IntPtr) SDL2.SDL.SDL_RWread(datasource, ptr, size, nmemb); + private static int Seek(IntPtr datasource, long offset, Theorafile.SeekWhence whence) => (int) SDL2.SDL.SDL_RWseek(datasource, offset, (int) whence); + private static int Close(IntPtr datasource) => (int) SDL2.SDL.SDL_RWclose(datasource); + + private static Theorafile.tf_callbacks callbacks = new Theorafile.tf_callbacks { - if (State == VideoState.Playing) - { - return; - } - - Loop = loop; - timer.Start(); - - if (audioStream != null) - { - audioStream.Play(); - } - - State = VideoState.Playing; - } - - public void Pause() - { - if (State != VideoState.Playing) - { - return; - } - - timer.Stop(); - - if (audioStream != null) - { - audioStream.Pause(); - } - - State = VideoState.Paused; - } - - public void Stop() - { - if (State == VideoState.Stopped) - { - return; - } - - timer.Stop(); - timer.Reset(); - - Theorafile.tf_reset(Handle); - lastTimestamp = 0; - timeElapsed = 0; - - if (audioStream != null) - { - audioStream.StopImmediate(); - audioStream.Dispose(); - audioStream = null; - } - - State = VideoState.Stopped; - } - - public Texture GetTexture() - { - if (RenderTexture == null) - { - throw new InvalidOperationException(); - } - - if (State == VideoState.Stopped) - { - return RenderTexture; - } - - timeElapsed += (timer.Elapsed.TotalMilliseconds - lastTimestamp) * PlaybackSpeed; - lastTimestamp = timer.Elapsed.TotalMilliseconds; - - int thisFrame = ((int) (timeElapsed / (1000.0 / FramesPerSecond))); - if (thisFrame > currentFrame) - { - if (Theorafile.tf_readvideo( - Handle, - (IntPtr) yuvData, - thisFrame - currentFrame - ) == 1 || currentFrame == -1) { - UpdateTexture(); - } - - currentFrame = thisFrame; - } - - bool ended = Theorafile.tf_eos(Handle) == 1; - if (ended) - { - timer.Stop(); - timer.Reset(); - - if (audioStream != null) - { - audioStream.Stop(); - audioStream.Dispose(); - audioStream = null; - } - - Theorafile.tf_reset(Handle); - - if (Loop) - { - // Start over! - InitializeTheoraStream(); - - timer.Start(); - } - else - { - State = VideoState.Stopped; - } - } - - return RenderTexture; - } - - private void UpdateTexture() - { - var commandBuffer = GraphicsDevice.AcquireCommandBuffer(); - - commandBuffer.SetTextureDataYUV( - yTexture, - uTexture, - vTexture, - (IntPtr) yuvData, - (uint) yuvDataLength - ); - - commandBuffer.BeginRenderPass( - new ColorAttachmentInfo(RenderTexture, Color.Black) - ); - - commandBuffer.BindGraphicsPipeline(GraphicsDevice.VideoPipeline); - commandBuffer.BindFragmentSamplers( - new TextureSamplerBinding(yTexture, LinearSampler), - new TextureSamplerBinding(uTexture, LinearSampler), - new TextureSamplerBinding(vTexture, LinearSampler) - ); - - commandBuffer.DrawPrimitives(0, 1, 0, 0); - - commandBuffer.EndRenderPass(); - - GraphicsDevice.Submit(commandBuffer); - } - - private void InitializeTheoraStream() - { - // Grab the first video frame ASAP. - while (Theorafile.tf_readvideo(Handle, (IntPtr) yuvData, 1) == 0); - - // Grab the first bit of audio. We're trying to start the decoding ASAP. - if (AudioDevice != null && Theorafile.tf_hasaudio(Handle) == 1) - { - int channels, sampleRate; - Theorafile.tf_audioinfo(Handle, out channels, out sampleRate); - audioStream = new StreamingSoundTheora(AudioDevice, Handle, channels, (uint) sampleRate); - } - - currentFrame = -1; - } + read_func = Read, + seek_func = Seek, + close_func = Close + }; protected virtual void Dispose(bool disposing) { @@ -327,15 +98,11 @@ namespace MoonWorks.Video if (disposing) { // dispose managed state (managed objects) - RenderTexture.Dispose(); - yTexture.Dispose(); - uTexture.Dispose(); - vTexture.Dispose(); } // free unmanaged resources (unmanaged objects) Theorafile.tf_close(ref Handle); - NativeMemory.Free(yuvData); + NativeMemory.Free(videoData); disposed = true; } diff --git a/src/Video/VideoPlayer.cs b/src/Video/VideoPlayer.cs new file mode 100644 index 00000000..987ef736 --- /dev/null +++ b/src/Video/VideoPlayer.cs @@ -0,0 +1,335 @@ +using System; +using System.Diagnostics; +using System.Runtime.InteropServices; +using MoonWorks.Audio; +using MoonWorks.Graphics; + +namespace MoonWorks.Video +{ + public unsafe class VideoPlayer : IDisposable + { + public Texture RenderTexture { get; private set; } = null; + public VideoState State { get; private set; } = VideoState.Stopped; + public bool Loop { get; set; } + public float Volume { + get => volume; + set + { + volume = value; + if (audioStream != null) + { + audioStream.Volume = value; + } + } + } + public float PlaybackSpeed { get; set; } = 1; + + private Video Video = null; + + private GraphicsDevice GraphicsDevice; + private Texture yTexture = null; + private Texture uTexture = null; + private Texture vTexture = null; + private Sampler LinearSampler; + + private void* yuvData = null; + private int yuvDataLength = 0; + + private int currentFrame; + + private AudioDevice AudioDevice; + private StreamingSoundTheora audioStream = null; + private float volume = 1.0f; + + private Stopwatch timer; + private double lastTimestamp; + private double timeElapsed; + + private bool disposed; + + public VideoPlayer(GraphicsDevice graphicsDevice, AudioDevice audioDevice) + { + GraphicsDevice = graphicsDevice; + AudioDevice = audioDevice; + LinearSampler = new Sampler(graphicsDevice, SamplerCreateInfo.LinearClamp); + + timer = new Stopwatch(); + } + + public void Load(Video video) + { + Stop(); + + if (RenderTexture == null) + { + RenderTexture = CreateRenderTexture(GraphicsDevice, video.Width, video.Height); + } + + if (yTexture == null) + { + yTexture = CreateSubTexture(GraphicsDevice, video.Width, video.Height); + } + + if (uTexture == null) + { + uTexture = CreateSubTexture(GraphicsDevice, video.UVWidth, video.UVHeight); + } + + if (vTexture == null) + { + vTexture = CreateSubTexture(GraphicsDevice, video.UVWidth, video.UVHeight); + } + + if (video.Width != RenderTexture.Width || video.Height != RenderTexture.Height) + { + RenderTexture.Dispose(); + RenderTexture = CreateRenderTexture(GraphicsDevice, video.Width, video.Height); + } + + if (video.Width != yTexture.Width || video.Height != yTexture.Height) + { + yTexture.Dispose(); + yTexture = CreateSubTexture(GraphicsDevice, video.Width, video.Height); + } + + if (video.UVWidth != uTexture.Width || video.UVHeight != uTexture.Height) + { + uTexture.Dispose(); + uTexture = CreateSubTexture(GraphicsDevice, video.UVWidth, video.UVHeight); + } + + if (video.UVWidth != vTexture.Width || video.UVHeight != vTexture.Height) + { + vTexture.Dispose(); + vTexture = CreateSubTexture(GraphicsDevice, video.UVWidth, video.UVHeight); + } + + var newDataLength = ( + (video.Width * video.Height) + + (video.UVWidth * video.UVHeight * 2) + ); + + if (newDataLength != yuvDataLength) + { + yuvData = NativeMemory.Realloc(yuvData, (nuint) newDataLength); + yuvDataLength = newDataLength; + } + + Video = video; + + InitializeTheoraStream(); + } + + public void Play() + { + if (State == VideoState.Playing) + { + return; + } + + timer.Start(); + + if (audioStream != null) + { + audioStream.Play(); + } + + State = VideoState.Playing; + } + + public void Pause() + { + if (State != VideoState.Playing) + { + return; + } + + timer.Stop(); + + if (audioStream != null) + { + audioStream.Pause(); + } + + State = VideoState.Paused; + } + + public void Stop() + { + if (State == VideoState.Stopped) + { + return; + } + + timer.Stop(); + timer.Reset(); + + Theorafile.tf_reset(Video.Handle); + lastTimestamp = 0; + timeElapsed = 0; + + if (audioStream != null) + { + audioStream.StopImmediate(); + audioStream.Dispose(); + audioStream = null; + } + + State = VideoState.Stopped; + } + + public void Render() + { + if (Video == null || State == VideoState.Stopped) + { + return; + } + + timeElapsed += (timer.Elapsed.TotalMilliseconds - lastTimestamp) * PlaybackSpeed; + lastTimestamp = timer.Elapsed.TotalMilliseconds; + + int thisFrame = ((int) (timeElapsed / (1000.0 / Video.FramesPerSecond))); + if (thisFrame > currentFrame) + { + if (Theorafile.tf_readvideo( + Video.Handle, + (IntPtr) yuvData, + thisFrame - currentFrame + ) == 1 || currentFrame == -1) { + UpdateRenderTexture(); + } + + currentFrame = thisFrame; + } + + bool ended = Theorafile.tf_eos(Video.Handle) == 1; + if (ended) + { + timer.Stop(); + timer.Reset(); + + if (audioStream != null) + { + audioStream.Stop(); + audioStream.Dispose(); + audioStream = null; + } + + Theorafile.tf_reset(Video.Handle); + + if (Loop) + { + // Start over! + InitializeTheoraStream(); + + timer.Start(); + } + else + { + State = VideoState.Stopped; + } + } + } + + private void UpdateRenderTexture() + { + var commandBuffer = GraphicsDevice.AcquireCommandBuffer(); + + commandBuffer.SetTextureDataYUV( + yTexture, + uTexture, + vTexture, + (IntPtr) yuvData, + (uint) yuvDataLength + ); + + commandBuffer.BeginRenderPass( + new ColorAttachmentInfo(RenderTexture, Color.Black) + ); + + commandBuffer.BindGraphicsPipeline(GraphicsDevice.VideoPipeline); + commandBuffer.BindFragmentSamplers( + new TextureSamplerBinding(yTexture, LinearSampler), + new TextureSamplerBinding(uTexture, LinearSampler), + new TextureSamplerBinding(vTexture, LinearSampler) + ); + + commandBuffer.DrawPrimitives(0, 1, 0, 0); + + commandBuffer.EndRenderPass(); + + GraphicsDevice.Submit(commandBuffer); + } + + private static Texture CreateRenderTexture(GraphicsDevice graphicsDevice, int width, int height) + { + return Texture.CreateTexture2D( + graphicsDevice, + (uint) width, + (uint) height, + TextureFormat.R8G8B8A8, + TextureUsageFlags.ColorTarget | TextureUsageFlags.Sampler + ); + } + + private static Texture CreateSubTexture(GraphicsDevice graphicsDevice, int width, int height) + { + return Texture.CreateTexture2D( + graphicsDevice, + (uint) width, + (uint) height, + TextureFormat.R8, + TextureUsageFlags.Sampler + ); + } + + private void InitializeTheoraStream() + { + // Grab the first video frame ASAP. + while (Theorafile.tf_readvideo(Video.Handle, (IntPtr) yuvData, 1) == 0); + + // Grab the first bit of audio. We're trying to start the decoding ASAP. + if (AudioDevice != null && Theorafile.tf_hasaudio(Video.Handle) == 1) + { + int channels, sampleRate; + Theorafile.tf_audioinfo(Video.Handle, out channels, out sampleRate); + audioStream = new StreamingSoundTheora(AudioDevice, Video.Handle, channels, (uint) sampleRate); + } + + currentFrame = -1; + } + + protected virtual void Dispose(bool disposing) + { + if (!disposed) + { + if (disposing) + { + // dispose managed state (managed objects) + RenderTexture.Dispose(); + yTexture.Dispose(); + uTexture.Dispose(); + vTexture.Dispose(); + } + + // free unmanaged resources (unmanaged objects) and override finalizer + NativeMemory.Free(yuvData); + + disposed = true; + } + } + + ~VideoPlayer() + { + // Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method + Dispose(disposing: false); + } + + public void Dispose() + { + // Do not change this code. Put cleanup code in 'Dispose(bool disposing)' method + Dispose(disposing: true); + GC.SuppressFinalize(this); + } + } +}