You're breathtaking!
This commit is contained in:
Wojtek Figat
524
Source/Editor/Windows/Assets/MaterialInstanceWindow.cs
Normal file
524
Source/Editor/Windows/Assets/MaterialInstanceWindow.cs
Normal file
@@ -0,0 +1,524 @@
|
||||
// Copyright (c) 2012-2020 Wojciech Figat. All rights reserved.
|
||||
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Reflection;
|
||||
using System.Xml;
|
||||
using FlaxEditor.Content;
|
||||
using FlaxEditor.CustomEditors;
|
||||
using FlaxEditor.CustomEditors.Editors;
|
||||
using FlaxEditor.CustomEditors.GUI;
|
||||
using FlaxEditor.GUI;
|
||||
using FlaxEditor.Surface;
|
||||
using FlaxEditor.Viewport.Previews;
|
||||
using FlaxEngine;
|
||||
using FlaxEngine.GUI;
|
||||
|
||||
namespace FlaxEditor.Windows.Assets
|
||||
{
|
||||
/// <summary>
|
||||
/// Material window allows to view and edit <see cref="MaterialInstance"/> asset.
|
||||
/// Note: it uses actual asset to modify so changes are visible live in the game/editor preview.
|
||||
/// </summary>
|
||||
/// <seealso cref="MaterialInstance" />
|
||||
/// <seealso cref="FlaxEditor.Windows.Assets.AssetEditorWindow" />
|
||||
public sealed class MaterialInstanceWindow : AssetEditorWindowBase<MaterialInstance>
|
||||
{
|
||||
private sealed class EditParamOverrideAction : IUndoAction
|
||||
{
|
||||
public MaterialInstanceWindow Window;
|
||||
public string Name;
|
||||
public bool Before;
|
||||
|
||||
/// <inheritdoc />
|
||||
public string ActionString => "Edit Override";
|
||||
|
||||
private void Set(bool value)
|
||||
{
|
||||
Window.Asset.GetParameter(Name).IsOverride = value;
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public void Do()
|
||||
{
|
||||
Set(!Before);
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public void Undo()
|
||||
{
|
||||
Set(Before);
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public void Dispose()
|
||||
{
|
||||
Window = null;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// The material properties proxy object.
|
||||
/// </summary>
|
||||
[CustomEditor(typeof(ParametersEditor))]
|
||||
private sealed class PropertiesProxy
|
||||
{
|
||||
private MaterialBase _restoreBase;
|
||||
private Dictionary<string, object> _restoreParams;
|
||||
|
||||
[EditorDisplay("General"), Tooltip("The base material used to override it's properties")]
|
||||
public MaterialBase BaseMaterial
|
||||
{
|
||||
get => Window?.Asset?.BaseMaterial;
|
||||
set
|
||||
{
|
||||
var asset = Window?.Asset;
|
||||
if (asset)
|
||||
{
|
||||
if (value == asset)
|
||||
{
|
||||
MessageBox.Show("Cannot use material itself as instance base.", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
|
||||
return;
|
||||
}
|
||||
asset.BaseMaterial = value;
|
||||
Window._editor.BuildLayoutOnUpdate();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// The window reference. Used to handle some special logic.
|
||||
/// </summary>
|
||||
[NoSerialize, HideInEditor]
|
||||
public MaterialInstanceWindow Window;
|
||||
|
||||
/// <summary>
|
||||
/// The material parameter values collection. Used to record undo changes.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Contains only items with raw values excluding Flax Objects.
|
||||
/// </remarks>
|
||||
[HideInEditor]
|
||||
public object[] Values
|
||||
{
|
||||
get => Window?.Asset?.Parameters.Select(x => x.Value).ToArray();
|
||||
set
|
||||
{
|
||||
var parameters = Window?.Asset?.Parameters;
|
||||
if (value != null && parameters != null)
|
||||
{
|
||||
if (value.Length != parameters.Length)
|
||||
return;
|
||||
|
||||
for (int i = 0; i < value.Length; i++)
|
||||
{
|
||||
var p = parameters[i].Value;
|
||||
if (p is FlaxEngine.Object || p == null)
|
||||
continue;
|
||||
|
||||
parameters[i].Value = value[i];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// The material parameter values collection. Used to record undo changes.
|
||||
/// </summary>
|
||||
/// <remarks>
|
||||
/// Contains only items with references to Flax Objects identified by ID.
|
||||
/// </remarks>
|
||||
[HideInEditor]
|
||||
public FlaxEngine.Object[] ValuesRef
|
||||
{
|
||||
get => Window?.Asset?.Parameters.Select(x => x.Value as FlaxEngine.Object).ToArray();
|
||||
set
|
||||
{
|
||||
var parameters = Window?.Asset?.Parameters;
|
||||
if (value != null && parameters != null)
|
||||
{
|
||||
if (value.Length != parameters.Length)
|
||||
return;
|
||||
|
||||
for (int i = 0; i < value.Length; i++)
|
||||
{
|
||||
var p = parameters[i].Value;
|
||||
if (!(p is FlaxEngine.Object || p == null))
|
||||
continue;
|
||||
|
||||
parameters[i].Value = value[i];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Gathers parameters from the specified material.
|
||||
/// </summary>
|
||||
/// <param name="materialWin">The material window.</param>
|
||||
public void OnLoad(MaterialInstanceWindow materialWin)
|
||||
{
|
||||
// Link
|
||||
Window = materialWin;
|
||||
|
||||
// Prepare restore data
|
||||
PeekState();
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Records the current state to restore it on DiscardChanges.
|
||||
/// </summary>
|
||||
public void PeekState()
|
||||
{
|
||||
if (Window == null)
|
||||
return;
|
||||
|
||||
var material = Window.Asset;
|
||||
_restoreBase = material.BaseMaterial;
|
||||
var parameters = material.Parameters;
|
||||
_restoreParams = new Dictionary<string, object>();
|
||||
for (int i = 0; i < parameters.Length; i++)
|
||||
_restoreParams[parameters[i].Name] = parameters[i].Value;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// On discard changes
|
||||
/// </summary>
|
||||
public void DiscardChanges()
|
||||
{
|
||||
if (Window == null)
|
||||
return;
|
||||
|
||||
var material = Window.Asset;
|
||||
material.BaseMaterial = _restoreBase;
|
||||
var parameters = material.Parameters;
|
||||
for (int i = 0; i < parameters.Length; i++)
|
||||
{
|
||||
var p = parameters[i];
|
||||
if (p.IsPublic && _restoreParams.TryGetValue(p.Name, out var value))
|
||||
{
|
||||
p.Value = value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Clears temporary data.
|
||||
/// </summary>
|
||||
public void OnClean()
|
||||
{
|
||||
// Unlink
|
||||
Window = null;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Custom editor for editing material parameters collection.
|
||||
/// </summary>
|
||||
/// <seealso cref="FlaxEditor.CustomEditors.CustomEditor" />
|
||||
public class ParametersEditor : GenericEditor
|
||||
{
|
||||
/// <inheritdoc />
|
||||
public override void Initialize(LayoutElementsContainer layout)
|
||||
{
|
||||
// Prepare
|
||||
var proxy = (PropertiesProxy)Values[0];
|
||||
var materialInstance = proxy.Window?.Asset;
|
||||
if (materialInstance == null)
|
||||
{
|
||||
layout.Label("No parameters");
|
||||
return;
|
||||
}
|
||||
if (!materialInstance.IsLoaded || materialInstance.BaseMaterial && !materialInstance.BaseMaterial.IsLoaded)
|
||||
{
|
||||
layout.Label("Loading...");
|
||||
return;
|
||||
}
|
||||
var parameters = materialInstance.Parameters;
|
||||
|
||||
base.Initialize(layout);
|
||||
|
||||
if (parameters.Length == 0)
|
||||
return;
|
||||
var parametersGroup = layout.Group("Parameters");
|
||||
var baseMaterial = materialInstance.BaseMaterial;
|
||||
var material = baseMaterial;
|
||||
if (material)
|
||||
{
|
||||
while (material is MaterialInstance instance)
|
||||
material = instance.BaseMaterial;
|
||||
}
|
||||
|
||||
var data = SurfaceUtils.InitGraphParameters(parameters, (Material)material);
|
||||
SurfaceUtils.DisplayGraphParameters(parametersGroup, data,
|
||||
(instance, parameter, tag) =>
|
||||
{
|
||||
// Get material parameter
|
||||
var p = (MaterialParameter)tag;
|
||||
var proxyEx = (PropertiesProxy)instance;
|
||||
var array = proxyEx.Window.Asset.Parameters;
|
||||
if (array == null || !array.Contains(p))
|
||||
throw new TargetException("Material parameters collection has been changed.");
|
||||
return p.Value;
|
||||
},
|
||||
(instance, value, parameter, tag) =>
|
||||
{
|
||||
// Set material parameter and surface parameter
|
||||
var p = (MaterialParameter)tag;
|
||||
var proxyEx = (PropertiesProxy)instance;
|
||||
p.Value = value;
|
||||
proxyEx.Window._paramValueChange = true;
|
||||
},
|
||||
Values,
|
||||
null,
|
||||
(LayoutElementsContainer itemLayout, ValueContainer valueContainer, ref SurfaceUtils.GraphParameterData e) =>
|
||||
{
|
||||
var p = (MaterialParameter)e.Tag;
|
||||
|
||||
// Try to get default value (from the base material)
|
||||
var pBase = baseMaterial?.GetParameter(p.Name);
|
||||
if (pBase != null && pBase.ParameterType == p.ParameterType)
|
||||
{
|
||||
valueContainer.SetDefaultValue(pBase.Value);
|
||||
}
|
||||
|
||||
// Add label with checkbox for parameter value override
|
||||
var label = new CheckablePropertyNameLabel(e.DisplayName);
|
||||
label.CheckBox.Checked = p.IsOverride;
|
||||
label.CheckBox.Tag = new KeyValuePair<PropertiesProxy, MaterialParameter>(proxy.Window._properties, p);
|
||||
label.CheckChanged += nameLabel =>
|
||||
{
|
||||
var pair = (KeyValuePair<PropertiesProxy, MaterialParameter>)nameLabel.CheckBox.Tag;
|
||||
var proxyEx = pair.Key;
|
||||
var pEx = pair.Value;
|
||||
pEx.IsOverride = nameLabel.CheckBox.Checked;
|
||||
proxyEx.Window._undo.AddAction(new EditParamOverrideAction
|
||||
{
|
||||
Window = proxyEx.Window,
|
||||
Name = pEx.Name,
|
||||
Before = !nameLabel.CheckBox.Checked,
|
||||
});
|
||||
};
|
||||
itemLayout.Property(label, valueContainer, null, e.Tooltip?.Text);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
private readonly SplitPanel _split;
|
||||
private readonly MaterialPreview _preview;
|
||||
private readonly ToolStripButton _saveButton;
|
||||
private readonly ToolStripButton _undoButton;
|
||||
private readonly ToolStripButton _redoButton;
|
||||
|
||||
private readonly CustomEditorPresenter _editor;
|
||||
private readonly Undo _undo;
|
||||
private readonly PropertiesProxy _properties;
|
||||
private bool _isWaitingForLoad;
|
||||
internal bool _paramValueChange;
|
||||
|
||||
/// <inheritdoc />
|
||||
public MaterialInstanceWindow(Editor editor, AssetItem item)
|
||||
: base(editor, item)
|
||||
{
|
||||
// Undo
|
||||
_undo = new Undo();
|
||||
_undo.UndoDone += OnUndoRedo;
|
||||
_undo.RedoDone += OnUndoRedo;
|
||||
_undo.ActionDone += OnAction;
|
||||
|
||||
// Toolstrip
|
||||
_saveButton = (ToolStripButton)_toolstrip.AddButton(Editor.Icons.Save32, Save).LinkTooltip("Save");
|
||||
_toolstrip.AddSeparator();
|
||||
_undoButton = (ToolStripButton)_toolstrip.AddButton(Editor.Icons.Undo32, _undo.PerformUndo).LinkTooltip("Undo (Ctrl+Z)");
|
||||
_redoButton = (ToolStripButton)_toolstrip.AddButton(Editor.Icons.Redo32, _undo.PerformRedo).LinkTooltip("Redo (Ctrl+Y)");
|
||||
_toolstrip.AddSeparator();
|
||||
_toolstrip.AddButton(Editor.Icons.Rotate32, OnRevertAllParameters).LinkTooltip("Revert all the parameters to the default values");
|
||||
_toolstrip.AddSeparator();
|
||||
_toolstrip.AddButton(editor.Icons.Docs32, () => Platform.OpenUrl(Utilities.Constants.DocsUrl + "manual/graphics/materials/instanced-materials/index.html")).LinkTooltip("See documentation to learn more");
|
||||
|
||||
// Split Panel
|
||||
_split = new SplitPanel(Orientation.Horizontal, ScrollBars.None, ScrollBars.Vertical)
|
||||
{
|
||||
AnchorPreset = AnchorPresets.StretchAll,
|
||||
Offsets = new Margin(0, 0, _toolstrip.Bottom, 0),
|
||||
SplitterValue = 0.5f,
|
||||
Parent = this
|
||||
};
|
||||
|
||||
// Material preview
|
||||
_preview = new MaterialPreview(true)
|
||||
{
|
||||
Parent = _split.Panel1
|
||||
};
|
||||
|
||||
// Material properties editor
|
||||
_editor = new CustomEditorPresenter(_undo);
|
||||
_editor.Panel.Parent = _split.Panel2;
|
||||
_properties = new PropertiesProxy();
|
||||
_editor.Select(_properties);
|
||||
_editor.Modified += OnMaterialPropertyEdited;
|
||||
|
||||
// Setup input actions
|
||||
InputActions.Add(options => options.Undo, _undo.PerformUndo);
|
||||
InputActions.Add(options => options.Redo, _undo.PerformRedo);
|
||||
}
|
||||
|
||||
private void OnRevertAllParameters()
|
||||
{
|
||||
var baseMaterial = Asset.BaseMaterial;
|
||||
if (!baseMaterial)
|
||||
return;
|
||||
var parameters = Asset.Parameters;
|
||||
var actions = new List<IUndoAction>();
|
||||
for (var i = 0; i < parameters.Length; i++)
|
||||
{
|
||||
var p = parameters[i];
|
||||
if (p.IsOverride)
|
||||
{
|
||||
p.IsOverride = false;
|
||||
actions.Add(new EditParamOverrideAction
|
||||
{
|
||||
Window = this,
|
||||
Name = p.Name,
|
||||
Before = true,
|
||||
});
|
||||
}
|
||||
}
|
||||
using (new UndoMultiBlock(_undo, _editor.Selection, "Revert all the parameters to the default values", new MultiUndoAction(actions)))
|
||||
{
|
||||
for (var i = 0; i < parameters.Length; i++)
|
||||
{
|
||||
var p = parameters[i];
|
||||
var pBase = baseMaterial.GetParameter(p.Name);
|
||||
if (pBase != null && pBase.ParameterType == p.ParameterType)
|
||||
p.Value = pBase.Value;
|
||||
}
|
||||
}
|
||||
_editor.BuildLayoutOnUpdate();
|
||||
}
|
||||
|
||||
private void OnAction(IUndoAction action)
|
||||
{
|
||||
_paramValueChange = false;
|
||||
MarkAsEdited();
|
||||
UpdateToolstrip();
|
||||
}
|
||||
|
||||
private void OnUndoRedo(IUndoAction action)
|
||||
{
|
||||
_paramValueChange = false;
|
||||
MarkAsEdited();
|
||||
UpdateToolstrip();
|
||||
_editor.BuildLayoutOnUpdate();
|
||||
}
|
||||
|
||||
private void OnMaterialPropertyEdited()
|
||||
{
|
||||
_paramValueChange = false;
|
||||
//MarkAsEdited();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override void Save()
|
||||
{
|
||||
if (!IsEdited)
|
||||
return;
|
||||
|
||||
if (Asset.Save())
|
||||
{
|
||||
Editor.LogError("Cannot save asset.");
|
||||
return;
|
||||
}
|
||||
|
||||
_properties.PeekState();
|
||||
ClearEditedFlag();
|
||||
_item.RefreshThumbnail();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void UpdateToolstrip()
|
||||
{
|
||||
_saveButton.Enabled = IsEdited;
|
||||
_undoButton.Enabled = _undo.CanUndo;
|
||||
_redoButton.Enabled = _undo.CanRedo;
|
||||
|
||||
base.UpdateToolstrip();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void UnlinkItem()
|
||||
{
|
||||
_properties.OnClean();
|
||||
_preview.Material = null;
|
||||
_isWaitingForLoad = false;
|
||||
|
||||
base.UnlinkItem();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void OnAssetLinked()
|
||||
{
|
||||
_preview.Material = _asset;
|
||||
_isWaitingForLoad = true;
|
||||
|
||||
base.OnAssetLinked();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
protected override void OnClose()
|
||||
{
|
||||
// Discard unsaved changes
|
||||
_properties.DiscardChanges();
|
||||
|
||||
// Cleanup
|
||||
_undo.Clear();
|
||||
|
||||
base.OnClose();
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override void Update(float deltaTime)
|
||||
{
|
||||
// Check if need to load
|
||||
if (_isWaitingForLoad && _asset.IsLoaded && (_asset.BaseMaterial == null || _asset.BaseMaterial.IsLoaded))
|
||||
{
|
||||
// Clear flag
|
||||
_isWaitingForLoad = false;
|
||||
|
||||
// Init material properties and parameters proxy
|
||||
_properties.OnLoad(this);
|
||||
|
||||
// Setup
|
||||
ClearEditedFlag();
|
||||
_undo.Clear();
|
||||
_editor.BuildLayout();
|
||||
}
|
||||
|
||||
base.Update(deltaTime);
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override bool UseLayoutData => true;
|
||||
|
||||
/// <inheritdoc />
|
||||
public override void OnLayoutSerialize(XmlWriter writer)
|
||||
{
|
||||
writer.WriteAttributeString("Split", _split.SplitterValue.ToString());
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override void OnLayoutDeserialize(XmlElement node)
|
||||
{
|
||||
float value1;
|
||||
|
||||
if (float.TryParse(node.GetAttribute("Split"), out value1))
|
||||
_split.SplitterValue = value1;
|
||||
}
|
||||
|
||||
/// <inheritdoc />
|
||||
public override void OnLayoutDeserialize()
|
||||
{
|
||||
_split.SplitterValue = 0.5f;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user