DarthAffe/StableDiffusion.NET
1
0
Fork 0
mirror of https://github.com/DarthAffe/StableDiffusion.NET.git synced 2025-12-12 21:38:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added BitmapToImage-Helper Method to example project

Browse Source
This commit is contained in:
Darth Affe 2024-04-27 17:21:09 +02:00
parent 09355fcd6e
commit 8f99e38f99
1 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
Examples/ImageCreationUI/Extensions/ImageExtension.cs
View File

@ -1,8 +1,10 @@
using System.Drawing;
using System.Buffers;
using System.Drawing;
using System.Drawing.Imaging;
using StableDiffusion.NET.Helper.Images.Colors;
using StableDiffusion.NET.Helper.Images;
using System.IO;
using System.Runtime.InteropServices;
namespace ImageCreationUI;
@ -42,4 +44,34 @@ public static class ImageExtension
return ms.ToArray();
}
public static unsafe Image<ColorRGB> ToImage(this Bitmap bitmap)
{
int width = bitmap.Width;
int height = bitmap.Height;
byte[] buffer = new byte[height * width * ColorRGB.ColorFormat.BytesPerPixel];
Span<ColorRGB> colorBuffer = MemoryMarshal.Cast<byte, ColorRGB>(buffer);
Rectangle rect = new(0, 0, bitmap.Width, bitmap.Height);
BitmapData bmpData = bitmap.LockBits(rect, ImageLockMode.ReadOnly, bitmap.PixelFormat);
nint ptr = bmpData.Scan0;
for (int y = 0; y < height; y++)
{
Span<ColorBGR> source = new((void*)ptr, bmpData.Stride);
Span<ColorRGB> target = colorBuffer.Slice(y * width, width);
for (int x = 0; x < width; x++)
{
ColorBGR srcColor = source[x];
target[x] = new ColorRGB(srcColor.R, srcColor.G, srcColor.B);
}
ptr += bmpData.Stride;
}
bitmap.UnlockBits(bmpData);
return new Image<ColorRGB>(buffer, 0, 0, width, height, width);
}
}