public static SelectMode DoElementModeToolbar(Rect rect, SelectMode mode)
{
Styles.Init();
EditorGUI.BeginChangeCheck();
var textureMode = mode.ContainsFlag(SelectMode.TextureVertex | SelectMode.TextureEdge | SelectMode.TextureFace);
int currentSelectionMode = -1;
switch (mode)
{
case SelectMode.Object:
currentSelectionMode = 0;
break;
case SelectMode.Vertex:
case SelectMode.TextureVertex:
currentSelectionMode = 1;
break;
case SelectMode.Edge:
case SelectMode.TextureEdge:
currentSelectionMode = 2;
break;
case SelectMode.Face:
case SelectMode.TextureFace:
currentSelectionMode = 3;
break;
default:
currentSelectionMode = -1;
break;
}
currentSelectionMode = GUI.Toolbar(rect, currentSelectionMode, Styles.selectModeIcons, Styles.command);
if (EditorGUI.EndChangeCheck())
{
if (currentSelectionMode == 0)
{
mode = SelectMode.Object;
}
else if (currentSelectionMode == 1)
{
mode = textureMode ? SelectMode.TextureVertex : SelectMode.Vertex;
}
else if (currentSelectionMode == 2)
{
mode = textureMode ? SelectMode.TextureEdge : SelectMode.Edge;
}
else if (currentSelectionMode == 3)
{
mode = textureMode ? SelectMode.TextureFace : SelectMode.Face;
}
}
return(mode);
}
void SelectModeChanged(SelectMode mode)
{
if (!mode.ContainsFlag(SelectMode.InputTool) && m_IsEditing)
{
SetIsEditing(false);
Repaint();
}
}
internal static bool IsPositionMode(this SelectMode mode)
{
return(mode.ContainsFlag(
SelectMode.TextureEdge
| SelectMode.TextureFace
| SelectMode.TextureVertex
));
}
void OnSelectModeChanged(SelectMode selectMode)
{
// User changed select mode manually, remove InputTool flag
if (polygon != null &&
polygon.polyEditMode != PolyShape.PolyEditMode.None &&
!selectMode.ContainsFlag(SelectMode.InputTool))
{
SetPolyEditMode(PolyShape.PolyEditMode.None);
}
}
internal static SelectMode GetTextureMode(this SelectMode mode)
{
if (mode.ContainsFlag(SelectMode.Face))
{
mode = (mode & ~SelectMode.Face) | SelectMode.TextureFace;
}
if (mode.ContainsFlag(SelectMode.Edge))
{
mode = (mode & ~SelectMode.Edge) | SelectMode.TextureEdge;
}
if (mode.ContainsFlag(SelectMode.Vertex))
{
mode = (mode & ~SelectMode.Vertex) | SelectMode.TextureVertex;
}
return(mode);
}
// Get the object & mesh selection that the mouse is currently nearest.
// A ProBuilderMesh is returned because double click actions need to know what the last selected pb_Object was.
// If deepClickOffset is specified, the object + deepClickOffset in the deep select stack will be returned (instead of next).
internal static float MouseRayHitTest(
Vector3 mousePosition,
SelectMode selectionMode,
ScenePickerPreferences pickerOptions,
SceneSelection selection,
bool allowUnselected = false)
{
if (selectionMode.ContainsFlag(SelectMode.Edge | SelectMode.TextureEdge))
{
return(EdgeRaycast(mousePosition, pickerOptions, allowUnselected, selection));
}
if (selectionMode.ContainsFlag(SelectMode.Vertex | SelectMode.TextureVertex))
{
return(VertexRaycast(mousePosition, pickerOptions, allowUnselected, selection));
}
return(FaceRaycast(mousePosition, pickerOptions, allowUnselected, selection, 0, true));
}
/// <summary>
/// Is this mode one of the mesh element modes (vertex, edge, face, texture).
/// </summary>
/// <param name="mode"></param>
/// <returns></returns>
internal static bool IsMeshElementMode(this SelectMode mode)
{
return(mode.ContainsFlag(
SelectMode.Vertex
| SelectMode.Edge
| SelectMode.Face
| SelectMode.TextureEdge
| SelectMode.TextureFace
| SelectMode.TextureVertex
));
}
internal static List <int> GetSelectedIndicesForSelectMode(ProBuilderMesh mesh, SelectMode mode, bool collectCoincident)
{
if (mode.ContainsFlag(SelectMode.Face | SelectMode.TextureFace))
{
List <int> indices = new List <int>();
if (collectCoincident)
{
mesh.GetCoincidentVertices(mesh.selectedFacesInternal, indices);
}
else
{
Face.GetDistinctIndices(mesh.selectedFacesInternal, indices);
}
return(indices);
}
else if (mode.ContainsFlag(SelectMode.Edge | SelectMode.TextureEdge))
{
List <int> indices = new List <int>();
if (collectCoincident)
{
mesh.GetCoincidentVertices(mesh.selectedEdgesInternal, indices);
}
else
{
Edge.GetIndices(mesh.selectedEdgesInternal, indices);
}
return(indices);
}
return(collectCoincident
? mesh.GetCoincidentVertices(mesh.selectedIndexesInternal)
: new List <int>(mesh.selectedIndexesInternal));
}
public static ProBuilderMesh DoMouseClick(Event evt, SelectMode selectionMode, ScenePickerPreferences pickerPreferences)
{
bool appendModifier = EditorHandleUtility.IsAppendModifier(evt.modifiers);
if (!appendModifier)
{
MeshSelection.SetSelection((GameObject)null);
}
float pickedElementDistance = Mathf.Infinity;
if (selectionMode.ContainsFlag(SelectMode.Edge | SelectMode.TextureEdge))
{
pickedElementDistance = EdgeRaycast(evt.mousePosition, pickerPreferences, true, s_Selection);
}
else if (selectionMode.ContainsFlag(SelectMode.Vertex | SelectMode.TextureVertex))
{
pickedElementDistance = VertexRaycast(evt.mousePosition, pickerPreferences, true, s_Selection);
}
else
{
pickedElementDistance = FaceRaycast(evt.mousePosition, pickerPreferences, true, s_Selection, evt.clickCount > 1 ? -1 : 0, false);
}
evt.Use();
if (pickedElementDistance > pickerPreferences.maxPointerDistance)
{
if (appendModifier && Selection.gameObjects.Contains(s_Selection.gameObject))
{
MeshSelection.RemoveFromSelection(s_Selection.gameObject);
}
else
{
MeshSelection.AddToSelection(s_Selection.gameObject);
}
return(null);
}
MeshSelection.AddToSelection(s_Selection.gameObject);
if (s_Selection.mesh != null)
{
var mesh = s_Selection.mesh;
if (s_Selection.face != null)
{
// Check for other editor mouse shortcuts first (todo proper event handling for mouse shortcuts)
MaterialEditor matEditor = MaterialEditor.instance;
if (matEditor != null && matEditor.ClickShortcutCheck(Event.current.modifiers, mesh, s_Selection.face))
{
return(null);
}
UVEditor uvEditor = UVEditor.instance;
if (uvEditor != null && uvEditor.ClickShortcutCheck(mesh, s_Selection.face))
{
return(null);
}
var faces = mesh.faces as Face[] ?? mesh.faces.ToArray();
var ind = Array.IndexOf <Face>(faces, s_Selection.face);
var sel = mesh.selectedFaceIndexes.IndexOf(ind);
UndoUtility.RecordSelection(mesh, "Select Face");
if (sel > -1)
{
mesh.RemoveFromFaceSelectionAtIndex(sel);
}
else
{
mesh.AddToFaceSelection(ind);
}
}
else if (s_Selection.edge != Edge.Empty)
{
int ind = mesh.IndexOf(mesh.selectedEdges, s_Selection.edge);
UndoUtility.RecordSelection(mesh, "Select Edge");
if (ind > -1)
{
mesh.SetSelectedEdges(mesh.selectedEdges.ToArray().RemoveAt(ind));
}
else
{
mesh.SetSelectedEdges(mesh.selectedEdges.ToArray().Add(s_Selection.edge));
}
}
else if (s_Selection.vertex > -1)
{
int ind = Array.IndexOf(mesh.selectedIndexesInternal, s_Selection.vertex);
UndoUtility.RecordSelection(mesh, "Select Vertex");
if (ind > -1)
{
mesh.SetSelectedVertices(mesh.selectedIndexesInternal.RemoveAt(ind));
}
else
{
mesh.SetSelectedVertices(mesh.selectedIndexesInternal.Add(s_Selection.vertex));
}
}
return(mesh);
}
return(null);
}
public static ProBuilderMesh DoMouseClick(Event evt, SelectMode selectionMode, ScenePickerPreferences pickerPreferences)
{
bool appendModifier = EditorHandleUtility.IsAppendModifier(evt.modifiers);
bool addToSelectionModifier = EditorHandleUtility.IsSelectionAddModifier(evt.modifiers);
bool addOrRemoveIfPresentFromSelectionModifier = EditorHandleUtility.IsSelectionAppendOrRemoveIfPresentModifier(evt.modifiers);
bool pathSelectionModifier = EditorHandleUtility.IsSelectionPathModifier(evt.modifiers);
float pickedElementDistance;
if (selectionMode.ContainsFlag(SelectMode.Edge | SelectMode.TextureEdge))
{
pickedElementDistance = EdgeRaycast(evt.mousePosition, pickerPreferences, k_AllowUnselected, s_Selection);
}
else if (selectionMode.ContainsFlag(SelectMode.Vertex | SelectMode.TextureVertex))
{
pickedElementDistance = VertexRaycast(evt.mousePosition, pickerPreferences, k_AllowUnselected, s_Selection);
}
else
{
pickedElementDistance = FaceRaycast(evt.mousePosition, pickerPreferences, k_AllowUnselected, s_Selection, evt.clickCount > 1 ? -1 : 0, false);
}
evt.Use();
if (!appendModifier)
{
if (s_Selection.mesh != null)
{
s_Selection.mesh.ClearSelection();
}
MeshSelection.SetSelection((GameObject)null);
}
if (pickedElementDistance > ScenePickerPreferences.maxPointerDistance)
{
if (appendModifier && Selection.gameObjects.Contains(s_Selection.gameObject))
{
MeshSelection.RemoveFromSelection(s_Selection.gameObject);
}
else
{
MeshSelection.AddToSelection(s_Selection.gameObject);
}
return(null);
}
GameObject candidateNewActiveObject = s_Selection.gameObject;
bool activeObjectSelectionChanged = Selection.gameObjects.Contains(s_Selection.gameObject) && s_Selection.gameObject != Selection.activeGameObject;
if (s_Selection.mesh != null)
{
var mesh = s_Selection.mesh;
foreach (var face in s_Selection.faces)
{
// Check for other editor mouse shortcuts first (todo proper event handling for mouse shortcuts)
MaterialEditor matEditor = MaterialEditor.instance;
if (matEditor != null && matEditor.ClickShortcutCheck(Event.current.modifiers, mesh, face))
{
return(null);
}
UVEditor uvEditor = UVEditor.instance;
if (uvEditor != null && uvEditor.ClickShortcutCheck(mesh, face))
{
return(null);
}
var faces = mesh.faces as Face[] ?? mesh.faces.ToArray();
var ind = Array.IndexOf <Face>(faces, face);
var sel = mesh.selectedFaceIndexes.IndexOf(ind);
UndoUtility.RecordSelection(mesh, "Select Face");
if (sel > -1)
{
if (!appendModifier || addOrRemoveIfPresentFromSelectionModifier ||
(addToSelectionModifier && face == mesh.GetActiveFace() && !activeObjectSelectionChanged))
{
mesh.RemoveFromFaceSelectionAtIndex(sel);
if (addOrRemoveIfPresentFromSelectionModifier && activeObjectSelectionChanged)
{
candidateNewActiveObject = Selection.activeGameObject;
}
else if (mesh.selectedFaceCount == 0)
{
for (var i = MeshSelection.topInternal.Count - 1; i >= 0; i--)
{
if (MeshSelection.topInternal[i].selectedFaceCount > 0)
{
candidateNewActiveObject = MeshSelection.topInternal[i].gameObject;
activeObjectSelectionChanged = true;
break;
}
}
}
}
else
{
mesh.selectedFaceIndicesInternal = mesh.selectedFaceIndicesInternal.Remove(ind);
mesh.SetSelectedFaces(mesh.selectedFaceIndicesInternal.Add(ind));
}
}
else if (pathSelectionModifier && mesh.GetActiveFace() != null)
{
var pathFaces = SelectPathFaces.GetPath(mesh, Array.IndexOf <Face>(faces, mesh.GetActiveFace()),
Array.IndexOf <Face>(faces, face));
foreach (var pathFace in pathFaces)
{
mesh.AddToFaceSelection(pathFace);
}
}
else
{
mesh.AddToFaceSelection(ind);
}
}
foreach (var edge in s_Selection.edges)
{
int ind = mesh.IndexOf(mesh.selectedEdges, edge);
UndoUtility.RecordSelection(mesh, "Select Edge");
if (ind > -1)
{
if (!appendModifier || addOrRemoveIfPresentFromSelectionModifier ||
(addToSelectionModifier && edge == mesh.GetActiveEdge() && !activeObjectSelectionChanged))
{
mesh.SetSelectedEdges(mesh.selectedEdges.ToArray().RemoveAt(ind));
if (addOrRemoveIfPresentFromSelectionModifier && activeObjectSelectionChanged)
{
candidateNewActiveObject = Selection.activeGameObject;
}
else if (mesh.selectedEdgeCount == 0)
{
for (var i = MeshSelection.topInternal.Count - 1; i >= 0; i--)
{
if (MeshSelection.topInternal[i].selectedEdgeCount > 0)
{
candidateNewActiveObject = MeshSelection.topInternal[i].gameObject;
activeObjectSelectionChanged = true;
break;
}
}
}
}
else
{
mesh.selectedEdgesInternal = mesh.selectedEdgesInternal.Remove(edge);
mesh.SetSelectedEdges(mesh.selectedEdgesInternal.Add(edge));
}
}
else
{
mesh.SetSelectedEdges(mesh.selectedEdges.ToArray().Add(edge));
}
}
foreach (var vertex in s_Selection.vertexes)
{
int ind = Array.IndexOf(mesh.selectedIndexesInternal, vertex);
UndoUtility.RecordSelection(mesh, "Select Vertex");
if (ind > -1)
{
var sharedIndex = mesh.sharedVertexLookup[vertex];
var sharedVertex = mesh.sharedVerticesInternal[sharedIndex];
s_IndexBuffer.Clear();
foreach (var sVertex in sharedVertex)
{
var index = Array.IndexOf(mesh.selectedIndexesInternal, sVertex);
if (index < 0)
{
continue;
}
s_IndexBuffer.Add(index);
}
s_IndexBuffer.Sort();
if (!appendModifier || addOrRemoveIfPresentFromSelectionModifier ||
(addToSelectionModifier && vertex == mesh.GetActiveVertex() && !activeObjectSelectionChanged))
{
mesh.selectedIndexesInternal = mesh.selectedIndexesInternal.SortedRemoveAt(s_IndexBuffer);
mesh.SetSelectedVertices(mesh.selectedIndexesInternal);
if (addOrRemoveIfPresentFromSelectionModifier && activeObjectSelectionChanged)
{
candidateNewActiveObject = Selection.activeGameObject;
}
else if (mesh.selectedIndexesInternal.Length == 0)
{
for (var i = MeshSelection.topInternal.Count - 1; i >= 0; i--)
{
if (MeshSelection.topInternal[i].selectedIndexesInternal.Length > 0)
{
candidateNewActiveObject = MeshSelection.topInternal[i].gameObject;
activeObjectSelectionChanged = true;
break;
}
}
}
}
else
{
mesh.selectedIndexesInternal = mesh.selectedIndexesInternal.SortedRemoveAt(s_IndexBuffer);
mesh.SetSelectedVertices(mesh.selectedIndexesInternal.Add(vertex));
}
}
else
{
mesh.SetSelectedVertices(mesh.selectedIndexesInternal.Add(vertex));
}
}
if (activeObjectSelectionChanged)
{
MeshSelection.MakeActiveObject(candidateNewActiveObject);
}
else
{
MeshSelection.AddToSelection(candidateNewActiveObject);
}
return(mesh);
}
return(null);
}
