minesweeper_game/Library/PackageCache/com.unity.2d.tilemap@91afae292a4e/Editor/GridBrushEditor.cs
2025-03-15 14:30:26 -04:00

907 lines
42 KiB
C#

using System;
using System.Linq;
using UnityEditor.EditorTools;
using UnityEngine;
using UnityEngine.Tilemaps;
using UnityEditor.SceneManagement;
using UnityEngine.Scripting.APIUpdating;
using Object = UnityEngine.Object;
namespace UnityEditor.Tilemaps
{
/// <summary>Editor for GridBrush.</summary>
[MovedFrom(true, "UnityEditor", "UnityEditor")]
[CustomEditor(typeof(GridBrush))]
public class GridBrushEditor : GridBrushEditorBase
{
private static class Styles
{
public static readonly GUIContent tileLabel = EditorGUIUtility.TrTextContent("Tile", "Tile set in tilemap");
public static readonly GUIContent spriteLabel = EditorGUIUtility.TrTextContent("Sprite", "Sprite set when tile is set in tilemap");
public static readonly GUIContent colorLabel = EditorGUIUtility.TrTextContent("Color", "Color set when tile is set in tilemap");
public static readonly GUIContent colliderTypeLabel = EditorGUIUtility.TrTextContent("Collider Type", "Collider shape used for tile");
public static readonly GUIContent gameObjectToInstantiateLabel = EditorGUIUtility.TrTextContent("GameObject to Instantiate", "GameObject to instantiate for tile");
public static readonly GUIContent lockColorLabel = EditorGUIUtility.TrTextContent("Lock Color", "Prevents tilemap from changing color of tile");
public static readonly GUIContent lockTransformLabel = EditorGUIUtility.TrTextContent("Lock Transform", "Prevents tilemap from changing transform of tile");
public static readonly GUIContent gridSelectionPropertiesLabel = EditorGUIUtility.TrTextContent("Grid Selection Properties");
public static readonly GUIContent modifyTilemapLabel = EditorGUIUtility.TrTextContent("Modify Tilemap");
public static readonly GUIContent modifyLabel = EditorGUIUtility.TrTextContent("Modify");
public static readonly GUIContent deleteSelectionLabel = EditorGUIUtility.TrTextContent("Delete Selection");
public static readonly GUIContent noTool =
EditorGUIUtility.TrTextContentWithIcon("None", "No Gizmo in the Scene view", "RectTool");
public static readonly GUIContent moveTool =
EditorGUIUtility.TrTextContentWithIcon("Move", "Shows a Gizmo in the Scene view for changing the offset for the Grid Selection", "MoveTool");
public static readonly GUIContent rotateTool =
EditorGUIUtility.TrTextContentWithIcon("Rotate", "Shows a Gizmo in the Scene view for changing the rotation for the Grid Selection", "RotateTool");
public static readonly GUIContent scaleTool =
EditorGUIUtility.TrTextContentWithIcon("Scale", "Shows a Gizmo in the Scene view for changing the scale for the Grid Selection", "ScaleTool");
public static readonly GUIContent transformTool =
EditorGUIUtility.TrTextContentWithIcon("Transform", "Shows a Gizmo in the Scene view for changing the transform for the Grid Selection", "TransformTool");
public static readonly GUIContent[] selectionTools = new[]
{
noTool
, moveTool
, rotateTool
, scaleTool
, transformTool
};
public static readonly Type[] selectionTypes = new[]
{
typeof(SelectTool)
, typeof(GridSelectionMoveTool)
, typeof(GridSelectionRotateTool)
, typeof(GridSelectionScaleTool)
, typeof(GridSelectionTransformTool)
};
public static readonly string tooltipText = L10n.Tr("Use this brush to paint and erase Tiles from a Tilemap.");
public static readonly string iconPath = "Packages/com.unity.2d.tilemap/Editor/Icons/Tilemap.DefaultBrush.png";
}
/// <summary>
/// Identifiers for operations modifying the Tilemap.
/// </summary>
public enum ModifyCells
{
/// <summary>
/// Inserts a row at the target position.
/// </summary>
InsertRow,
/// <summary>
/// Inserts a column at the target position.
/// </summary>
InsertColumn,
/// <summary>
/// Inserts a row before the target position.
/// </summary>
InsertRowBefore,
/// <summary>
/// Inserts a column before the target position.
/// </summary>
InsertColumnBefore,
/// <summary>
/// Delete a row at the target position.
/// </summary>
DeleteRow,
/// <summary>
/// Delete a column at the target position.
/// </summary>
DeleteColumn,
/// <summary>
/// Delete a row before the target position.
/// </summary>
DeleteRowBefore,
/// <summary>
/// Delete a column before the target position.
/// </summary>
DeleteColumnBefore,
}
private class GridBrushProperties
{
public static readonly GUIContent floodFillPreviewLabel = EditorGUIUtility.TrTextContent("Show Flood Fill Preview", "Whether a preview is shown while painting a Tilemap when Flood Fill mode is enabled");
public static readonly GUIContent floodFillPreviewFillExtentsLabel = EditorGUIUtility.TrTextContent("Flood Fill Preview Fill Extents", "Extents from the selected position when flood filling with a Tile. Set this to 0 to flood fill to the full extents.");
public static readonly GUIContent floodFillPreviewEraseExtentsLabel = EditorGUIUtility.TrTextContent("Flood Fill Preview Erase Extents", "Extents from the selected position when flood filling without a Tile. Set this to 0 to flood erase to the full extents.");
public static readonly string floodFillPreviewEditorPref = "GridBrush.EnableFloodFillPreview";
public static readonly string floodFillPreviewFillExtentsEditorPref = "GridBrush.FloodFillPreviewFillExtents";
public static readonly string floodFillPreviewEraseExtentsEditorPref = "GridBrush.FloodFillPreviewEraseExtents";
}
internal static bool showFloodFillPreview
{
get => EditorPrefs.GetBool(GridBrushProperties.floodFillPreviewEditorPref, true);
set => EditorPrefs.SetBool(GridBrushProperties.floodFillPreviewEditorPref, value);
}
internal static int floodFillPreviewFillExtents
{
get => Math.Max(0, EditorPrefs.GetInt(GridBrushProperties.floodFillPreviewFillExtentsEditorPref, 0));
set => EditorPrefs.SetInt(GridBrushProperties.floodFillPreviewFillExtentsEditorPref, Math.Max(0, value));
}
internal static int floodFillPreviewEraseExtents
{
get => Math.Max(0, EditorPrefs.GetInt(GridBrushProperties.floodFillPreviewEraseExtentsEditorPref, 0));
set => EditorPrefs.SetInt(GridBrushProperties.floodFillPreviewEraseExtentsEditorPref, Math.Max(0, value));
}
/// <summary>The GridBrush that is the target for this editor.</summary>
public GridBrush brush { get { return target as GridBrush; } }
private int m_LastPreviewRefreshHash;
// These are used to clean out previews that happened on previous update
private GridLayout m_LastGrid;
private GameObject m_LastBrushTarget;
private BoundsInt? m_LastBounds;
private GridBrushBase.Tool? m_LastTool;
// These are used to handle selection in Selection Inspector
private TileBase[] m_SelectionTiles;
private Color[] m_SelectionColors;
private Matrix4x4[] m_SelectionMatrices;
private TileFlags[] m_SelectionFlagsArray;
private Sprite[] m_SelectionSprites;
private Tile.ColliderType[] m_SelectionColliderTypes;
private GameObject[] m_SelectionGameObjectToInstantiate;
private int selectionCellCount => Math.Abs(GridSelection.position.size.x * GridSelection.position.size.y * GridSelection.position.size.z);
// These are used to handle insert/delete cells on the Tilemap
private int m_CellCount = 1;
private ModifyCells m_ModifyCells = ModifyCells.InsertRow;
private Texture2D m_Icon;
private static GridSelectionTool[] s_GridSelectionTools;
private static Tile s_EmptySpriteTile;
/// <summary>
/// Initializes the GridBrushEditor.
/// </summary>
protected virtual void OnEnable()
{
Undo.undoRedoPerformed += ClearLastPreview;
if (s_GridSelectionTools == null || s_GridSelectionTools[0] == null)
{
s_GridSelectionTools = new GridSelectionTool[]
{
CreateInstance<GridSelectionMoveTool>(),
CreateInstance<GridSelectionRotateTool>(),
CreateInstance<GridSelectionScaleTool>(),
CreateInstance<GridSelectionTransformTool>()
};
}
if (s_EmptySpriteTile == null)
{
s_EmptySpriteTile = ScriptableObject.CreateInstance<Tile>();
s_EmptySpriteTile.sprite = null;
}
}
/// <summary>
/// Deinitialises the GridBrushEditor.
/// </summary>
protected virtual void OnDisable()
{
Undo.undoRedoPerformed -= ClearLastPreview;
ClearLastPreview();
}
private void ClearLastPreview()
{
ClearPreview();
m_LastPreviewRefreshHash = 0;
}
/// <summary>Callback for painting the GUI for the GridBrush in the Scene View.</summary>
/// <param name="gridLayout">Grid that the brush is being used on.</param>
/// <param name="brushTarget">Target of the GridBrushBase::ref::Tool operation. By default the currently selected GameObject.</param>
/// <param name="position">Current selected location of the brush.</param>
/// <param name="tool">Current GridBrushBase::ref::Tool selected.</param>
/// <param name="executing">Whether brush is being used.</param>
public override void OnPaintSceneGUI(GridLayout gridLayout, GameObject brushTarget, BoundsInt position, GridBrushBase.Tool tool, bool executing)
{
BoundsInt gizmoRect = position;
bool refreshPreviews = false;
if (Event.current.type == EventType.Layout || Event.current.type == EventType.Repaint)
{
int newPreviewRefreshHash = GetHash(gridLayout, brushTarget, position, tool, brush);
refreshPreviews = newPreviewRefreshHash != m_LastPreviewRefreshHash;
if (refreshPreviews)
m_LastPreviewRefreshHash = newPreviewRefreshHash;
}
if (tool == GridBrushBase.Tool.Move)
{
if (refreshPreviews && executing)
{
ClearPreview();
PaintPreview(gridLayout, brushTarget, position.min);
}
}
else if (tool == GridBrushBase.Tool.Paint || tool == GridBrushBase.Tool.Erase)
{
if (refreshPreviews)
{
ClearPreview();
if (tool != GridBrushBase.Tool.Erase)
{
PaintPreview(gridLayout, brushTarget, position.min);
}
else
{
ErasePreview(gridLayout, brushTarget, position.min);
}
}
gizmoRect = new BoundsInt(position.min - brush.pivot, brush.size);
}
else if (tool == GridBrushBase.Tool.Box)
{
if (refreshPreviews)
{
ClearPreview();
BoxFillPreview(gridLayout, brushTarget, position);
}
}
else if (tool == GridBrushBase.Tool.FloodFill)
{
if (refreshPreviews)
{
if (CheckFloodFillPreview(gridLayout, brushTarget, position.min))
ClearPreview();
FloodFillPreview(gridLayout, brushTarget, position.min);
}
}
base.OnPaintSceneGUI(gridLayout, brushTarget, gizmoRect, tool, executing);
}
private void UpdateSelection(Tilemap tilemap)
{
var selection = GridSelection.position;
var cellCount = selectionCellCount;
if (m_SelectionTiles == null || m_SelectionTiles.Length != selectionCellCount)
{
m_SelectionTiles = new TileBase[cellCount];
m_SelectionColors = new Color[cellCount];
m_SelectionMatrices = new Matrix4x4[cellCount];
m_SelectionFlagsArray = new TileFlags[cellCount];
m_SelectionSprites = new Sprite[cellCount];
m_SelectionColliderTypes = new Tile.ColliderType[cellCount];
m_SelectionGameObjectToInstantiate = new GameObject[cellCount];
}
int index = 0;
foreach (var p in selection.allPositionsWithin)
{
m_SelectionTiles[index] = tilemap.GetTile(p);
m_SelectionColors[index] = tilemap.GetColor(p);
m_SelectionMatrices[index] = tilemap.GetTransformMatrix(p);
m_SelectionFlagsArray[index] = tilemap.GetTileFlags(p);
m_SelectionSprites[index] = tilemap.GetSprite(p);
m_SelectionColliderTypes[index] = tilemap.GetColliderType(p);
m_SelectionGameObjectToInstantiate[index] = tilemap.GetObjectToInstantiate(p);
index++;
}
}
/// <summary>Callback for drawing the Inspector GUI when there is an active GridSelection made in a Tilemap.</summary>
public override void OnSelectionInspectorGUI()
{
BoundsInt selection = GridSelection.position;
Tilemap tilemap = GridSelection.target.GetComponent<Tilemap>();
int cellCount = selectionCellCount;
if (tilemap != null && cellCount > 0)
{
base.OnSelectionInspectorGUI();
if (!EditorGUIUtility.editingTextField
&& Event.current.type == EventType.KeyDown
&& (Event.current.keyCode == KeyCode.Delete
|| Event.current.keyCode == KeyCode.Backspace))
{
DeleteSelection(tilemap, selection);
Event.current.Use();
}
GUILayout.Space(10f);
EditorGUILayout.LabelField(Styles.gridSelectionPropertiesLabel, EditorStyles.boldLabel);
UpdateSelection(tilemap);
EditorGUI.BeginChangeCheck();
EditorGUI.showMixedValue = m_SelectionTiles.Any(tile => tile != m_SelectionTiles.First());
var position = new Vector3Int(selection.xMin, selection.yMin, selection.zMin);
TileBase newTile = EditorGUILayout.ObjectField(Styles.tileLabel, tilemap.GetTile(position), typeof(TileBase), false) as TileBase;
if (EditorGUI.EndChangeCheck())
{
Undo.RecordObject(tilemap, "Edit Tilemap");
foreach (var p in selection.allPositionsWithin)
tilemap.SetTile(p, newTile);
}
using (new EditorGUI.DisabledScope(true))
{
EditorGUI.showMixedValue = m_SelectionSprites.Any(sprite => sprite != m_SelectionSprites.First());
EditorGUILayout.ObjectField(Styles.spriteLabel, m_SelectionSprites[0], typeof(Sprite), false, GUILayout.Height(EditorGUI.kSingleLineHeight));
}
bool colorFlagsAllEqual = m_SelectionFlagsArray.All(flags => (flags & TileFlags.LockColor) == (m_SelectionFlagsArray.First() & TileFlags.LockColor));
using (new EditorGUI.DisabledScope(!colorFlagsAllEqual || (m_SelectionFlagsArray[0] & TileFlags.LockColor) != 0))
{
EditorGUI.showMixedValue = m_SelectionColors.Any(color => color != m_SelectionColors.First());
EditorGUI.BeginChangeCheck();
Color newColor = EditorGUILayout.ColorField(Styles.colorLabel, m_SelectionColors[0]);
if (EditorGUI.EndChangeCheck())
{
Undo.RecordObject(tilemap, "Edit Tilemap");
foreach (var p in selection.allPositionsWithin)
tilemap.SetColor(p, newColor);
}
}
using (new EditorGUI.DisabledScope(true))
{
EditorGUI.showMixedValue = m_SelectionColliderTypes.Any(colliderType => colliderType != m_SelectionColliderTypes.First());
EditorGUILayout.EnumPopup(Styles.colliderTypeLabel, m_SelectionColliderTypes[0]);
EditorGUI.showMixedValue = m_SelectionGameObjectToInstantiate.Any(gameObject => gameObject != m_SelectionGameObjectToInstantiate.First());
EditorGUILayout.ObjectField(Styles.gameObjectToInstantiateLabel, m_SelectionGameObjectToInstantiate[0], typeof(GameObject), false);
}
bool transformFlagsAllEqual = m_SelectionFlagsArray.All(flags => (flags & TileFlags.LockTransform) == (m_SelectionFlagsArray.First() & TileFlags.LockTransform));
using (new EditorGUI.DisabledScope(!transformFlagsAllEqual || (m_SelectionFlagsArray[0] & TileFlags.LockTransform) != 0))
{
EditorGUI.showMixedValue = m_SelectionMatrices.Any(matrix => matrix != m_SelectionMatrices.First());
EditorGUI.BeginChangeCheck();
Matrix4x4 newTransformMatrix = TileEditor.TransformMatrixOnGUI(m_SelectionMatrices[0]);
if (EditorGUI.EndChangeCheck())
{
Undo.RecordObject(tilemap, "Edit Tilemap");
foreach (var p in selection.allPositionsWithin)
tilemap.SetTransformMatrix(p, newTransformMatrix);
}
}
var lockedTransform = (m_SelectionFlagsArray[0] & TileFlags.LockTransform) != 0;
using (new EditorGUI.DisabledScope(true))
{
EditorGUI.showMixedValue = !colorFlagsAllEqual;
EditorGUILayout.Toggle(Styles.lockColorLabel, (m_SelectionFlagsArray[0] & TileFlags.LockColor) != 0);
EditorGUI.showMixedValue = !transformFlagsAllEqual;
EditorGUILayout.Toggle(Styles.lockTransformLabel, lockedTransform);
}
EditorGUI.showMixedValue = false;
if (GUILayout.Button(Styles.deleteSelectionLabel))
{
DeleteSelection(tilemap, selection);
}
EditorGUILayout.Space();
EditorGUILayout.LabelField(Styles.modifyTilemapLabel, EditorStyles.boldLabel);
EditorGUILayout.Space();
var active = -1;
for (var i = 0; i < Styles.selectionTypes.Length; ++i)
{
if (ToolManager.activeToolType == Styles.selectionTypes[i])
{
active = i;
break;
}
}
using (new EditorGUI.DisabledScope(lockedTransform))
{
EditorGUI.BeginChangeCheck();
var selected = GUILayout.Toolbar(active, Styles.selectionTools);
if (EditorGUI.EndChangeCheck() && selected != -1)
{
ToolManager.SetActiveTool(Styles.selectionTypes[selected]);
}
}
EditorGUILayout.Space();
GUILayout.BeginHorizontal();
m_ModifyCells = (ModifyCells)EditorGUILayout.EnumPopup(m_ModifyCells);
m_CellCount = EditorGUILayout.IntField(m_CellCount);
if (GUILayout.Button(Styles.modifyLabel))
{
RegisterUndoForTilemap(tilemap, Enum.GetName(typeof(ModifyCells), m_ModifyCells));
switch (m_ModifyCells)
{
case ModifyCells.InsertRow:
{
tilemap.InsertCells(GridSelection.position.position, 0, m_CellCount, 0);
break;
}
case ModifyCells.InsertRowBefore:
{
tilemap.InsertCells(GridSelection.position.position, 0, -m_CellCount, 0);
break;
}
case ModifyCells.InsertColumn:
{
tilemap.InsertCells(GridSelection.position.position, m_CellCount, 0, 0);
break;
}
case ModifyCells.InsertColumnBefore:
{
tilemap.InsertCells(GridSelection.position.position, -m_CellCount, 0, 0);
break;
}
case ModifyCells.DeleteRow:
{
tilemap.DeleteCells(GridSelection.position.position, 0, m_CellCount, 0);
break;
}
case ModifyCells.DeleteRowBefore:
{
tilemap.DeleteCells(GridSelection.position.position, 0, -m_CellCount, 0);
break;
}
case ModifyCells.DeleteColumn:
{
tilemap.DeleteCells(GridSelection.position.position, m_CellCount, 0, 0);
break;
}
case ModifyCells.DeleteColumnBefore:
{
tilemap.DeleteCells(GridSelection.position.position, -m_CellCount, 0, 0);
break;
}
}
}
GUILayout.EndHorizontal();
}
}
private void DeleteSelection(Tilemap tilemap, BoundsInt selection)
{
if (tilemap == null)
return;
RegisterUndo(tilemap.gameObject, GridBrushBase.Tool.Erase);
brush.BoxErase(tilemap.layoutGrid, tilemap.gameObject, selection);
}
/// <summary> Callback when the mouse cursor leaves and editing area. </summary>
/// <remarks> Cleans up brush previews. </remarks>
public override void OnMouseLeave()
{
ClearPreview();
}
/// <summary> Callback when the GridBrush Tool is deactivated. </summary>
/// <param name="tool">GridBrush Tool that is deactivated.</param>
/// <remarks> Cleans up brush previews. </remarks>
public override void OnToolDeactivated(GridBrushBase.Tool tool)
{
ClearPreview();
}
/// <summary> Describes the usage of the GridBrush. </summary>
public override string tooltip
{
get { return Styles.tooltipText; }
}
/// <summary> Returns an icon identifying the Grid Brush. </summary>
public override Texture2D icon
{
get
{
if (m_Icon == null)
{
m_Icon = EditorGUIUtility.LoadIcon(Styles.iconPath);
}
return m_Icon;
}
}
/// <summary> Whether the GridBrush can change Z Position. </summary>
public override bool canChangeZPosition
{
get { return brush.canChangeZPosition; }
set { brush.canChangeZPosition = value; }
}
/// <summary>
/// Whether the Brush is in a state that should be saved for selection.
/// </summary>
public override bool shouldSaveBrushForSelection
{
get
{
if (brush.cells != null)
{
foreach (var cell in brush.cells)
{
if (cell != null && cell.tile != null)
return true;
}
}
return false;
}
}
/// <summary>Callback for registering an Undo action before the GridBrushBase does the current GridBrushBase::ref::Tool action.</summary>
/// <param name="brushTarget">Target of the GridBrushBase::ref::Tool operation. By default the currently selected GameObject.</param>
/// <param name="tool">Current GridBrushBase::ref::Tool selected.</param>
/// <remarks>Implement this for any special Undo behaviours when a brush is used.</remarks>
public override void RegisterUndo(GameObject brushTarget, GridBrushBase.Tool tool)
{
if (brushTarget != null)
{
var tilemap = brushTarget.GetComponent<Tilemap>();
if (tilemap != null)
{
RegisterUndoForTilemap(tilemap, tool.ToString());
}
}
}
/// <summary>Returns all valid targets that the brush can edit.</summary>
/// <remarks>Valid targets for the GridBrush are any GameObjects with a Tilemap component.</remarks>
public override GameObject[] validTargets
{
get
{
StageHandle currentStageHandle = StageUtility.GetCurrentStageHandle();
return currentStageHandle.FindComponentsOfType<Tilemap>().Where(x =>
{
GameObject gameObject;
return (gameObject = x.gameObject).scene.isLoaded
&& gameObject.activeInHierarchy
&& !gameObject.hideFlags.HasFlag(HideFlags.NotEditable);
}).Select(x => x.gameObject).ToArray();
}
}
/// <summary>Paints preview data into a cell of a grid given the coordinates of the cell.</summary>
/// <param name="gridLayout">The grid to paint data to.</param>
/// <param name="brushTarget">The target of the paint operation. This is the currently selected GameObject by default.</param>
/// <param name="position">The coordinates of the cell to paint data to.</param>
/// <remarks>The GridBrush will paint preview Sprites in its brush cells onto an associated Tilemap. This will not instantiate objects associated with the painted Tiles.</remarks>
public virtual void PaintPreview(GridLayout gridLayout, GameObject brushTarget, Vector3Int position)
{
var min = position - brush.pivot;
var max = min + brush.size;
var bounds = new BoundsInt(min, max - min);
if (brushTarget != null)
{
var map = brushTarget.GetComponent<Tilemap>();
if (map != null)
{
foreach (var location in bounds.allPositionsWithin)
{
var brushPosition = location - min;
var cell = brush.cells[brush.GetCellIndex(brushPosition)];
if (cell.tile != null)
{
SetTilemapPreviewCell(map, location, cell.tile, cell.matrix, cell.color);
}
}
}
}
m_LastGrid = gridLayout;
m_LastBounds = bounds;
m_LastBrushTarget = brushTarget;
m_LastTool = GridBrushBase.Tool.Paint;
}
/// <summary>Displays a preview of the Tile (after erasure) on the cell of a grid at the given coordinates of the cell.</summary>
/// <param name="gridLayout">The grid to paint data to.</param>
/// <param name="brushTarget">The target of the erase operation. This is the currently selected GameObject by default.</param>
/// <param name="position">The coordinates of the cell to paint data to.</param>
/// <remarks>The GridBrush will paint preview Sprites of the Tiles (after erasure) into its brush cells on an associated Tilemap. This will not instantiate objects associated with the painted Tiles.</remarks>
public virtual void ErasePreview(GridLayout gridLayout, GameObject brushTarget, Vector3Int position)
{
var min = position - brush.pivot;
var max = min + brush.size;
var bounds = new BoundsInt(min, max - min);
if (brushTarget != null)
{
var map = brushTarget.GetComponent<Tilemap>();
if (map != null)
{
foreach (var location in bounds.allPositionsWithin)
{
var brushPosition = location - min;
var cell = brush.cells[brush.GetCellIndex(brushPosition)];
SetTilemapPreviewCell(map, location, s_EmptySpriteTile, cell.matrix, cell.color);
}
}
}
m_LastGrid = gridLayout;
m_LastBounds = bounds;
m_LastBrushTarget = brushTarget;
m_LastTool = GridBrushBase.Tool.Erase;
}
/// <summary>Does a preview of what happens when a GridBrush.BoxFill is done with the same parameters.</summary>
/// <param name="gridLayout">Grid to box fill data to.</param>
/// <param name="brushTarget">Target of box fill operation. By default the currently selected GameObject.</param>
/// <param name="position">The bounds to box fill data to.</param>
public virtual void BoxFillPreview(GridLayout gridLayout, GameObject brushTarget, BoundsInt position)
{
if (brushTarget != null)
{
var map = brushTarget.GetComponent<Tilemap>();
if (map != null)
{
foreach (var location in position.allPositionsWithin)
{
var local = location - position.min;
var cell = brush.cells[brush.GetCellIndexWrapAround(local.x, local.y, local.z)];
if (cell.tile != null)
{
SetTilemapPreviewCell(map, location, cell.tile, cell.matrix, cell.color);
}
}
}
}
m_LastGrid = gridLayout;
m_LastBounds = position;
m_LastBrushTarget = brushTarget;
m_LastTool = GridBrushBase.Tool.Box;
}
private bool CheckFloodFillPreview(GridLayout gridLayout, GameObject brushTarget, Vector3Int position)
{
if (m_LastGrid == gridLayout
&& m_LastBrushTarget == brushTarget
&& m_LastBounds.HasValue && m_LastBounds.Value.Contains(position)
&& brushTarget != null && brush.cellCount > 0)
{
var map = brushTarget.GetComponent<Tilemap>();
if (map != null)
{
var cell = brush.cells[0];
var hasTile = cell.tile != null;
if ((hasTile && floodFillPreviewFillExtents == 0 || !hasTile && floodFillPreviewEraseExtents == 0)
&& (cell.tile == map.GetEditorPreviewTile(position)))
return false;
}
}
return true;
}
/// <summary>Does a preview of what happens when a GridBrush.FloodFill is done with the same parameters.</summary>
/// <param name="gridLayout">Grid to paint data to.</param>
/// <param name="brushTarget">Target of the flood fill operation. By default the currently selected GameObject.</param>
/// <param name="position">The coordinates of the cell to flood fill data to.</param>
public virtual void FloodFillPreview(GridLayout gridLayout, GameObject brushTarget, Vector3Int position)
{
// This can be quite taxing on a large Tilemap, so users can choose whether to do this or not
if (!showFloodFillPreview)
return;
var bounds = new BoundsInt(position, Vector3Int.one);
if (brushTarget != null && brush.cellCount > 0)
{
var map = brushTarget.GetComponent<Tilemap>();
if (map != null)
{
var cell = brush.cells[0];
var eraseExtents = floodFillPreviewEraseExtents;
var fillExtents = floodFillPreviewFillExtents;
var validFloodFillTile = cell.tile != null;
var floodFillTile = validFloodFillTile ? cell.tile : s_EmptySpriteTile;
if (!validFloodFillTile && floodFillPreviewEraseExtents > 0)
{
map.EditorPreviewBoxFill(position, floodFillTile, position.x - eraseExtents, position.y - eraseExtents, position.x + eraseExtents, position.y + eraseExtents);
}
else if (validFloodFillTile && floodFillPreviewFillExtents > 0)
{
map.EditorPreviewBoxFill(position, floodFillTile, position.x - fillExtents, position.y - fillExtents, position.x + fillExtents, position.y + fillExtents);
}
else
{
map.EditorPreviewFloodFill(position, floodFillTile);
}
// Set floodfill bounds as tilemap bounds
var origin = map.origin;
bounds.min = origin;
bounds.max = origin + map.size;
}
}
m_LastGrid = gridLayout;
m_LastBounds = bounds;
m_LastBrushTarget = brushTarget;
m_LastTool = GridBrushBase.Tool.FloodFill;
}
[SettingsProvider]
internal static SettingsProvider CreateSettingsProvider()
{
var settingsProvider = new SettingsProvider("Preferences/2D/Grid Brush", SettingsScope.User, SettingsProvider.GetSearchKeywordsFromGUIContentProperties<GridBrushProperties>()) {
guiHandler = _ =>
{
PreferencesGUI();
}
};
return settingsProvider;
}
private static void PreferencesGUI()
{
using (new SettingsWindow.GUIScope())
{
EditorGUI.BeginChangeCheck();
var val = EditorGUILayout.Toggle(GridBrushProperties.floodFillPreviewLabel, showFloodFillPreview);
if (EditorGUI.EndChangeCheck())
{
showFloodFillPreview = val;
}
EditorGUI.indentLevel++;
using (new EditorGUI.DisabledScope(!val))
{
EditorGUI.BeginChangeCheck();
var fill = EditorGUILayout.IntField(GridBrushProperties.floodFillPreviewFillExtentsLabel, floodFillPreviewFillExtents);
if (EditorGUI.EndChangeCheck())
{
floodFillPreviewFillExtents = fill;
}
EditorGUI.BeginChangeCheck();
var erase = EditorGUILayout.IntField(GridBrushProperties.floodFillPreviewEraseExtentsLabel, floodFillPreviewEraseExtents);
if (EditorGUI.EndChangeCheck())
{
floodFillPreviewEraseExtents = erase;
}
}
EditorGUI.indentLevel--;
}
}
/// <summary>Clears any preview drawn previously by the GridBrushEditor.</summary>
public virtual void ClearPreview()
{
if (m_LastGrid == null || m_LastBounds == null || m_LastBrushTarget == null || m_LastTool == null)
return;
Tilemap map = m_LastBrushTarget.GetComponent<Tilemap>();
if (map != null)
{
switch (m_LastTool)
{
case GridBrushBase.Tool.FloodFill:
{
map.ClearAllEditorPreviewTiles();
break;
}
case GridBrushBase.Tool.Box:
{
Vector3Int min = m_LastBounds.Value.position;
Vector3Int max = min + m_LastBounds.Value.size;
BoundsInt bounds = new BoundsInt(min, max - min);
foreach (Vector3Int location in bounds.allPositionsWithin)
{
ClearTilemapPreview(map, location);
}
break;
}
case GridBrushBase.Tool.Erase:
case GridBrushBase.Tool.Paint:
{
BoundsInt bounds = m_LastBounds.Value;
foreach (Vector3Int location in bounds.allPositionsWithin)
{
ClearTilemapPreview(map, location);
}
break;
}
}
}
m_LastBrushTarget = null;
m_LastGrid = null;
m_LastBounds = null;
m_LastTool = null;
}
/// <summary>
/// Creates a static preview of the GridBrush with its current selection.
/// </summary>
/// <param name="assetPath">The asset to operate on.</param>
/// <param name="subAssets">An array of all Assets at assetPath.</param>
/// <param name="width">Width of the created texture.</param>
/// <param name="height">Height of the created texture.</param>
/// <returns>Generated texture or null.</returns>
public override Texture2D RenderStaticPreview(string assetPath, Object[] subAssets, int width, int height)
{
if (brush == null)
return null;
var previewInstance = new GameObject("Brush Preview", typeof(Grid), typeof(Tilemap), typeof(TilemapRenderer));
var previewGrid = previewInstance.GetComponent<Grid>();
previewGrid.cellLayout = brush.lastPickedCellLayout;
previewGrid.cellSize = brush.lastPickedCellSize;
if (previewGrid.cellLayout != GridLayout.CellLayout.Hexagon)
{
previewGrid.cellGap = brush.lastPickedCellGap;
}
else
{
var tilemap = previewInstance.GetComponent<Tilemap>();
tilemap.tileAnchor = Vector3.zero;
}
previewGrid.cellSwizzle = brush.lastPickedCellSwizzle;
brush.Paint(previewGrid, previewInstance, Vector3Int.zero);
var bounds = previewGrid.GetBoundsLocal(Vector3.zero, brush.size);
var pivotLocal = previewGrid.CellToLocal(brush.pivot);
var center = bounds.center - pivotLocal;
center.z -= 10f;
var rect = new Rect(0, 0, width, height);
var previewUtility = new PreviewRenderUtility(true, true);
previewUtility.camera.orthographic = true;
previewUtility.camera.orthographicSize = 0.5f * Math.Max(brush.size.x, brush.size.y);
if (rect.height > rect.width)
previewUtility.camera.orthographicSize *= rect.height / rect.width;
previewUtility.camera.transform.position = center;
previewUtility.AddSingleGO(previewInstance);
previewUtility.BeginStaticPreview(rect);
previewUtility.camera.Render();
var tex = previewUtility.EndStaticPreview();
previewUtility.Cleanup();
DestroyImmediate(previewInstance);
return tex;
}
private void RegisterUndoForTilemap(Tilemap tilemap, string undoMessage)
{
Undo.RegisterCompleteObjectUndo(new Object[] { tilemap, tilemap.gameObject }, undoMessage);
}
private static void SetTilemapPreviewCell(Tilemap map, Vector3Int location, TileBase tile, Matrix4x4 transformMatrix, Color color)
{
if (map == null)
return;
map.SetEditorPreviewTile(location, tile);
map.SetEditorPreviewTransformMatrix(location, transformMatrix);
map.SetEditorPreviewColor(location, color);
}
private static void ClearTilemapPreview(Tilemap map, Vector3Int location)
{
if (map == null)
return;
map.SetEditorPreviewTile(location, null);
map.SetEditorPreviewTransformMatrix(location, Matrix4x4.identity);
map.SetEditorPreviewColor(location, Color.white);
}
private static int GetHash(GridLayout gridLayout, GameObject brushTarget, BoundsInt position, GridBrushBase.Tool tool, GridBrush brush)
{
int hash;
unchecked
{
hash = gridLayout != null ? gridLayout.GetHashCode() : 0;
hash = hash * 33 + (brushTarget != null ? brushTarget.GetHashCode() : 0);
hash = hash * 33 + position.GetHashCode();
hash = hash * 33 + tool.GetHashCode();
hash = hash * 33 + (brush != null ? brush.GetHashCode() : 0);
}
return hash;
}
}
}