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; }