ImageSurface: Support memory pressure

gircore · Aug 18, 2024 · 4ca5f91 · 4ca5f91
1 parent 3f6a62b
commit 4ca5f91
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 7 deletions.
diff --git a/src/Libs/cairo-1.0/Internal/SurfaceHandle.cs b/src/Libs/cairo-1.0/Internal/SurfaceHandle.cs
@@ -1,9 +1,30 @@
-namespace Cairo.Internal;
+using System;
+
+namespace Cairo.Internal;
 
 public partial class SurfaceOwnedHandle
 {
+    private long _memoryPressure;
+
+    internal void AddMemoryPressure(long memoryPressure)
+    {
+        if (memoryPressure > 0)
+        {
+            _memoryPressure = memoryPressure;
+            GC.AddMemoryPressure(_memoryPressure);   
+        }
+    }
+
+    private void RemoveMemoryPressure()
+    {
+        if(_memoryPressure > 0)
+            GC.RemoveMemoryPressure(_memoryPressure);
+    }
+
     protected override bool ReleaseHandle()
     {
+        RemoveMemoryPressure();
+
         Surface.Destroy(handle);
         return true;
     }

diff --git a/src/Libs/cairo-1.0/Public/ImageSurface.cs b/src/Libs/cairo-1.0/Public/ImageSurface.cs
@@ -7,20 +7,27 @@ public class ImageSurface : Surface
     public ImageSurface(Format format, int width, int height)
         : base(Internal.ImageSurface.Create(format, width, height))
     {
+        Handle.AddMemoryPressure(GetSizeInBytes());
     }
 
     public Format Format => Internal.ImageSurface.GetFormat(Handle);
     public int Height => Internal.ImageSurface.GetHeight(Handle);
     public int Width => Internal.ImageSurface.GetWidth(Handle);
+
+    /// <summary>
+    /// Number of bytes per row.
+    /// </summary>
     public int Stride => Internal.ImageSurface.GetStride(Handle);
 
+    private int GetSizeInBytes() => Stride * Height;
+
     public Span<byte> GetData()
     {
-        IntPtr data = Internal.ImageSurface.GetData(Handle);
-        int len = Stride * Height; // Stride is the number of bytes per row.
+        var data = Internal.ImageSurface.GetData(Handle);
+
         unsafe
         {
-            return new Span<byte>(data.ToPointer(), len);
+            return new Span<byte>(data.ToPointer(), GetSizeInBytes());
         }
     }
 }

diff --git a/src/Libs/cairo-1.0/Public/Surface.cs b/src/Libs/cairo-1.0/Public/Surface.cs
@@ -44,13 +44,28 @@ public void GetFontOptions(FontOptions options)
         => Internal.Surface.GetFontOptions(Handle, options.Handle);
 
     public Surface CreateSimilar(Content content, int width, int height)
-        => new Surface(Internal.Surface.CreateSimilar(Handle, content, width, height));
+    {
+        var handle = Internal.Surface.CreateSimilar(Handle, content, width, height);
+        handle.AddMemoryPressure(width * height);
+
+        return new Surface(handle);
+    }
 
     public Surface CreateSimilarImage(Format format, int width, int height)
-        => new Surface(Internal.Surface.CreateSimilarImage(Handle, format, width, height));
+    {
+        var handle = Internal.Surface.CreateSimilarImage(Handle, format, width, height);
+        handle.AddMemoryPressure(width * height);
+
+        return new Surface(handle);
+    }
 
     public Surface CreateForRectangle(double x, double y, double width, double height)
-        => new Surface(Internal.Surface.CreateForRectangle(Handle, x, y, width, height));
+    {
+        var handle = Internal.Surface.CreateForRectangle(Handle, x, y, width, height);
+        handle.AddMemoryPressure((long) (width * height));
+
+        return new Surface(handle);
+    }
 
     public void Finish() => Internal.Surface.Finish(Handle);