Initial Commit

Assets/Oculus/VR/Editor/BuildingBlocks/CustomHierarchyIcon.cs

@@ -0,0 +1,69 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * Licensed under the Oculus SDK License Agreement (the "License");
+ * you may not use the Oculus SDK except in compliance with the License,
+ * which is provided at the time of installation or download, or which
+ * otherwise accompanies this software in either electronic or hard copy form.
+ *
+ * You may obtain a copy of the License at
+ *
+ * https://developer.oculus.com/licenses/oculussdk/
+ *
+ * Unless required by applicable law or agreed to in writing, the Oculus SDK
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using UnityEngine;
+using UnityEditor;
+
+namespace Meta.XR.BuildingBlocks.Editor
+{
+    [InitializeOnLoad]
+    public class CustomHierarchyIcon
+    {
+        private const string HierarchyIconPath =
+            "Assets/Oculus/VR/Editor/BuildingBlocks/Icons/ovr_bb_icon.png";
+
+        private static Texture2D _hierarchyIcon;
+        private static Texture2D HierarchyIcon
+        {
+            get
+            {
+                if (_hierarchyIcon == null)
+                {
+                    _hierarchyIcon = AssetDatabase.LoadAssetAtPath<Texture2D>(HierarchyIconPath);
+                }
+
+                return _hierarchyIcon;
+            }
+        }
+
+        static CustomHierarchyIcon()
+        {
+            EditorApplication.hierarchyWindowItemOnGUI -= HierarchyItemOnGUI;
+            EditorApplication.hierarchyWindowItemOnGUI += HierarchyItemOnGUI;
+        }
+
+        private static void HierarchyItemOnGUI(int instanceID, Rect selectionRect)
+        {
+            var gameObject = EditorUtility.InstanceIDToObject(instanceID) as GameObject;
+
+            if (gameObject == null)
+                return;
+
+            if (gameObject.GetComponent<BuildingBlock>() == null)
+                return;
+
+            if (HierarchyIcon == null)
+                return;
+
+            Rect iconRect = new Rect(selectionRect.x + selectionRect.width - 20, selectionRect.y, 16, 16);
+            GUI.DrawTexture(iconRect, HierarchyIcon);
+        }
+    }
+}