Утечка памяти WPF CreateBitmapSourceFromHBitmap()

Мне нужно рисовать пиксель за пикселем и отображать его внутри WPF. Я пытаюсь сделать это, используя System.Drawing.Bitmap, затем используя CreateBitmapSourceFromHBitmap(), чтобы создать BitmapSource для управления изображением WPF. У меня есть утечка памяти где-то, потому что, когда вызов CreateBitmapSourceFromBitmap() вызывается многократно, использование памяти увеличивается и не уходит, пока приложение не закончится. Если я не вызываю CreateBitmapSourceFromBitmap(), заметного изменения в использовании памяти не наблюдается.

for (int i = 0; i < 100; i++)
{
    var bmp = new System.Drawing.Bitmap(1000, 1000);
    var source = System.Windows.Interop.Imaging.CreateBitmapSourceFromHBitmap(
        bmp.GetHbitmap(), IntPtr.Zero, Int32Rect.Empty,
        System.Windows.Media.Imaging.BitmapSizeOptions.FromEmptyOptions());
    source = null;
    bmp.Dispose();
    bmp = null;
}

Что я могу сделать, чтобы освободить память BitmapSource?

Ответ 1

MSDN заявляет, что для Bitmap.GetHbitmap(): вы несете ответственность за вызов метода GDI DeleteObject для освобождения памяти, используемой объектом GDI bitmap. Поэтому используйте следующий код:

// at class level
[System.Runtime.InteropServices.DllImport("gdi32.dll")]
public static extern bool DeleteObject(IntPtr hObject);

// your code
using (System.Drawing.Bitmap bmp = new System.Drawing.Bitmap(1000, 1000)) 
{
    IntPtr hBitmap = bmp.GetHbitmap(); 

    try 
    {
        var source = System.Windows.Interop.Imaging.CreateBitmapSourceFromHBitmap(hBitmap, IntPtr.Zero, Int32Rect.Empty, System.Windows.Media.Imaging.BitmapSizeOptions.FromEmptyOptions());
    }
    finally 
    {
        DeleteObject(hBitmap);
    }
}

Я также заменил ваш вызов Dispose() оператором using.

Ответ 2

Всякий раз, когда вы работаете с неуправляемыми ручками, может быть хорошей идеей использовать обертки с "безопасным дескриптором":

public class SafeHBitmapHandle : SafeHandleZeroOrMinusOneIsInvalid
{
    [SecurityCritical]
    public SafeHBitmapHandle(IntPtr preexistingHandle, bool ownsHandle)
        : base(ownsHandle)
    {
        SetHandle(preexistingHandle);
    }

    protected override bool ReleaseHandle()
    {
        return GdiNative.DeleteObject(handle) > 0;
    }
}

Постройте так, как только вы поместите дескриптор (в идеале ваши API никогда не будут выставлять IntPtr, они всегда будут возвращать безопасные дескрипторы):

IntPtr hbitmap = bitmap.GetHbitmap();
var handle = new SafeHBitmapHandle(hbitmap , true);

И используйте его так:

using (handle)
{
  ... Imaging.CreateBitmapSourceFromHBitmap(handle.DangerousGetHandle(), ...)
}

База SafeHandle дает вам автоматический шаблон одноразового/финализатора, все, что вам нужно сделать, это переопределить метод ReleaseHandle.

Ответ 3

У меня было такое же требование и проблема (утечка памяти). Я реализовал то же самое решение, что и как ответ. Но несмотря на то, что решение работает, это вызвало недопустимый удар производительности. Запустив i7, мое тестовое приложение увидело устойчивый 30-40% процессор, 200-400 МБ ОЗУ увеличилось, а сборщик мусора работал почти каждые миллисекунды.

Поскольку я занимаюсь обработкой видео, мне нужна намного лучшая производительность. Я придумал следующее, так что думал, что буду делиться.

Многоразовые глобальные объекты

//set up your Bitmap and WritableBitmap as you see fit
Bitmap colorBitmap = new Bitmap(..);
WriteableBitmap colorWB = new WriteableBitmap(..);

//choose appropriate bytes as per your pixel format, I'll cheat here an just pick 4
int bytesPerPixel = 4;

//rectangles will be used to identify what bits change
Rectangle colorBitmapRectangle = new Rectangle(0, 0, colorBitmap.Width, colorBitmap.Height);
Int32Rect colorBitmapInt32Rect = new Int32Rect(0, 0, colorWB.PixelWidth, colorWB.PixelHeight);

Код конверсии

private void ConvertBitmapToWritableBitmap()
{
    BitmapData data = colorBitmap.LockBits(colorBitmapRectangle, ImageLockMode.WriteOnly, colorBitmap.PixelFormat);

    colorWB.WritePixels(colorBitmapInt32Rect, data.Scan0, data.Width * data.Height * bytesPerPixel, data.Stride);

    colorBitmap.UnlockBits(data); 
}

Пример реализации

//do stuff to your bitmap
ConvertBitmapToWritableBitmap();
Image.Source = colorWB;

В результате получается стабильный 10-13% процессор, 70-150 МБ оперативной памяти, а сборщик мусора работает только дважды за 6 минут.

Ответ 4

Это отличная (!!) почта, хотя со всеми комментариями и предложениями мне понадобился час, чтобы разобраться в деталях. Итак, вот призыв получить BitMapSource с помощью SafeHandles, а затем использовать его пример для создания файла изображения.PNG. В самом низу "употребления" и некоторые ссылки. Конечно, ни один из моих заслуг - я просто писец.

private static BitmapSource CopyScreen()
{
    var left = Screen.AllScreens.Min(screen => screen.Bounds.X);
    var top = Screen.AllScreens.Min(screen => screen.Bounds.Y);
    var right = Screen.AllScreens.Max(screen => screen.Bounds.X + screen.Bounds.Width);
    var bottom = Screen.AllScreens.Max(screen => screen.Bounds.Y + screen.Bounds.Height);
    var width = right - left;
    var height = bottom - top;

    using (var screenBmp = new Bitmap(width, height, System.Drawing.Imaging.PixelFormat.Format32bppArgb))
    {
        BitmapSource bms = null;

        using (var bmpGraphics = Graphics.FromImage(screenBmp))
        {
            IntPtr hBitmap = new IntPtr();
            var handleBitmap = new SafeHBitmapHandle(hBitmap, true);

            try
            {
                bmpGraphics.CopyFromScreen(left, top, 0, 0, new System.Drawing.Size(width, height));

                hBitmap = screenBmp.GetHbitmap();

                using (handleBitmap)
                {
                    bms = Imaging.CreateBitmapSourceFromHBitmap(
                        hBitmap,
                        IntPtr.Zero,
                        Int32Rect.Empty,
                        BitmapSizeOptions.FromEmptyOptions());

                } // using

                return bms;
            }
            catch (Exception ex)
            {
                throw new ApplicationException($"Cannot CopyFromScreen. Err={ex}");
            }

        } // using bmpGraphics
    }   // using screen bitmap
} // method CopyScreen

Вот использование, а также класс "Безопасная ручка":

private void buttonTestScreenCapture_Click(object sender, EventArgs e)
{
    try
    {
        BitmapSource bms = CopyScreen();
        BitmapFrame bmf = BitmapFrame.Create(bms);

        PngBitmapEncoder encoder = new PngBitmapEncoder();
        encoder.Frames.Add(bmf);

        string filepath = @"e:\(test)\test.png";
        using (Stream stm = File.Create(filepath))
        {
            encoder.Save(stm);
        }
    }
    catch (Exception ex)
    {
        MessageBox.Show($"Err={ex}");
    }
}

public class SafeHBitmapHandle : SafeHandleZeroOrMinusOneIsInvalid
{
    [System.Runtime.InteropServices.DllImport("gdi32.dll")]
    public static extern int DeleteObject(IntPtr hObject);

    [SecurityCritical]
    public SafeHBitmapHandle(IntPtr preexistingHandle, bool ownsHandle)
        : base(ownsHandle)
    {
        SetHandle(preexistingHandle);
    }

    protected override bool ReleaseHandle()
    {
        return DeleteObject(handle) > 0;
    }
}

И, наконец, взгляните на мои "игры":

using System;
using System.Linq;
using System.Drawing;
using System.Windows.Forms;
using System.Windows.Media.Imaging;
using System.Windows.Interop;
using System.Windows;
using System.IO;
using Microsoft.Win32.SafeHandles;
using System.Security;

В список DLL, на который были включены ссылки: * PresentationCore * System.Core * System.Deployment * System.Drawing * WindowsBase