From cd626917090e99ad552edf9a0fcc7b0d2b2b5c84 Mon Sep 17 00:00:00 2001
From: Darth Affe <darthaffe@wyrez.org>
Date: Fri, 15 Dec 2023 17:46:47 +0100
Subject: [PATCH] Renamed Image to StableDiffusionImage

---
 README.md                                                 | 2 +-
 StableDiffusion.NET/StableDiffusion.cs                    | 8 ++++----
 StableDiffusion.NET/{Image.cs => StableDiffusionImage.cs} | 6 +++---
 3 files changed, 8 insertions(+), 8 deletions(-)
 rename StableDiffusion.NET/{Image.cs => StableDiffusionImage.cs} (83%)

diff --git a/README.md b/README.md
index 10f6ba1..131d453 100644
--- a/README.md
+++ b/README.md
@@ -12,5 +12,5 @@ Run `build.bat` to build the native libs (modify params like CUDA-builds if need
 ```csharp
 using StableDiffusionModel sd = new(@"<path_to_model>", new ModelParameter());
 using StableDiffusionParameter parameter = new StableDiffusionParameter();
-using Image image = sd.TextToImage("<prompt>", parameter);
+using StableDiffusionImage image = sd.TextToImage("<prompt>", parameter);
 ```
diff --git a/StableDiffusion.NET/StableDiffusion.cs b/StableDiffusion.NET/StableDiffusion.cs
index a56ff9c..026c6c8 100644
--- a/StableDiffusion.NET/StableDiffusion.cs
+++ b/StableDiffusion.NET/StableDiffusion.cs
@@ -41,22 +41,22 @@ public sealed unsafe class StableDiffusionModel : IDisposable
         if (!success) throw new IOException("Failed to load model");
     }
 
-    public Image TextToImage(string prompt, StableDiffusionParameter parameter)
+    public StableDiffusionImage TextToImage(string prompt, StableDiffusionParameter parameter)
     {
         ObjectDisposedException.ThrowIf(_disposed, this);
 
         byte* result = Native.stable_diffusion_predict_image(_ctx, parameter.ParamPtr, prompt);
-        return new Image(result, parameter.Width, parameter.Height);
+        return new StableDiffusionImage(result, parameter.Width, parameter.Height);
     }
 
-    public Image ImageToImage(string prompt, Span<byte> image, StableDiffusionParameter parameter)
+    public StableDiffusionImage ImageToImage(string prompt, Span<byte> image, StableDiffusionParameter parameter)
     {
         ObjectDisposedException.ThrowIf(_disposed, this);
 
         fixed (byte* imagePtr = image)
         {
             byte* result = Native.stable_diffusion_image_predict_image(_ctx, parameter.ParamPtr, imagePtr, prompt);
-            return new Image(result, parameter.Width, parameter.Height);
+            return new StableDiffusionImage(result, parameter.Width, parameter.Height);
         }
     }
 
diff --git a/StableDiffusion.NET/Image.cs b/StableDiffusion.NET/StableDiffusionImage.cs
similarity index 83%
rename from StableDiffusion.NET/Image.cs
rename to StableDiffusion.NET/StableDiffusionImage.cs
index 3f4dfbe..5219e31 100644
--- a/StableDiffusion.NET/Image.cs
+++ b/StableDiffusion.NET/StableDiffusionImage.cs
@@ -2,7 +2,7 @@
 
 namespace StableDiffusion.NET;
 
-public sealed unsafe class Image : IDisposable
+public sealed unsafe class StableDiffusionImage : IDisposable
 {
     #region Constants
 
@@ -33,14 +33,14 @@ public sealed unsafe class Image : IDisposable
 
     #region Constructors
 
-    internal Image(byte* ptr, int width, int height)
+    internal StableDiffusionImage(byte* ptr, int width, int height)
     {
         this._imagePtr = ptr;
         this.Width = width;
         this.Height = height;
     }
 
-    ~Image() => Dispose();
+    ~StableDiffusionImage() => Dispose();
 
     #endregion