diff --git a/Source/Editor/CustomEditors/LayoutElementsContainer.cs b/Source/Editor/CustomEditors/LayoutElementsContainer.cs
index 68ac9f47a..322a08d64 100644
--- a/Source/Editor/CustomEditors/LayoutElementsContainer.cs
+++ b/Source/Editor/CustomEditors/LayoutElementsContainer.cs
@@ -287,10 +287,7 @@ namespace FlaxEditor.CustomEditors
/// The created element.
public ImageElement Image(SpriteHandle sprite)
{
- var element = new ImageElement();
- element.Image.Brush = new SpriteBrush(sprite);
- OnAddElement(element);
- return element;
+ return Image(new SpriteBrush(sprite));
}
///
@@ -300,10 +297,7 @@ namespace FlaxEditor.CustomEditors
/// The created element.
public ImageElement Image(Texture texture)
{
- var element = new ImageElement();
- element.Image.Brush = new TextureBrush(texture);
- OnAddElement(element);
- return element;
+ return Image(new TextureBrush(texture));
}
///
@@ -312,9 +306,19 @@ namespace FlaxEditor.CustomEditors
/// The GPU texture.
/// The created element.
public ImageElement Image(GPUTexture texture)
+ {
+ return Image(new GPUTextureBrush(texture));
+ }
+
+ ///
+ /// Adds brush image to the layout.
+ ///
+ /// The brush.
+ /// The created element.
+ public ImageElement Image(IBrush brush)
{
var element = new ImageElement();
- element.Image.Brush = new GPUTextureBrush(texture);
+ element.Image.Brush = brush;
OnAddElement(element);
return element;
}