Graphics/gpudriven prefab mode fix uum 62427

Fixes: Tier0 not working correctly in prefab mode and LODGroup selection visibility not working correctly. 

https://jira.unity3d.com/browse/UUM-62427

Author: alekseiunity

Packages/com.unity.render-pipelines.core/Runtime/GPUDriven/GPUResidentBatcher.SpeedTree.cs

@@ -12,13 +12,20 @@ namespace UnityEngine.Rendering
 {
     internal partial class GPUResidentBatcher : IDisposable
     {
+        private ParallelBitArray m_ProcessedThisFrameTreeBits;
+
         private void ProcessTrees()
         {
             int treeInstancesCount = m_BatchersContext.GetAliveInstancesOfType(InstanceType.SpeedTree);
 
             if (treeInstancesCount == 0)
                 return;
 
+            ParallelBitArray compactedVisibilityMasks = m_InstanceCullingBatcher.GetCompactedVisibilityMasks(syncCullingJobs: false);
+
+            if (!compactedVisibilityMasks.IsCreated)
+                return;
+
             Profiler.BeginSample("GPUResidentInstanceBatcher.ProcessTrees");
 
             int maxInstancesCount = m_BatchersContext.aliveInstances.Length;
@@ -32,9 +39,6 @@ private void ProcessTrees()
             var visibleTreeRendererIDs = new NativeList<int>(Allocator.TempJob);
             var visibleTreeInstances = new NativeList<InstanceHandle>(Allocator.TempJob);
 
-            ParallelBitArray compactedVisibilityMasks = m_InstanceCullingBatcher.GetCompactedVisibilityMasks(syncCullingJobs: false);
-            Assert.IsTrue(compactedVisibilityMasks.IsCreated);
-
             m_BatchersContext.GetVisibleTreeInstances(compactedVisibilityMasks, m_ProcessedThisFrameTreeBits, visibleTreeRendererIDs, visibleTreeInstances,
                 becomeVisibleOnly, out var becomeVisibleTreeInstancesCount);
 

Packages/com.unity.render-pipelines.core/Runtime/GPUDriven/GPUResidentBatcher.cs

@@ -21,8 +21,6 @@ internal partial class GPUResidentBatcher : IDisposable
 
         private InstanceCullingBatcher m_InstanceCullingBatcher = null;
 
-        private ParallelBitArray m_ProcessedThisFrameTreeBits;
-
         public GPUResidentBatcher(
             RenderersBatchersContext batcherContext,
             InstanceCullingBatcherDesc instanceCullerBatcherDesc,
@@ -112,6 +110,16 @@ public void UpdateRenderers(NativeArray<int> renderersID)
             m_GPUDrivenProcessor.enablePartialRendering = false;
         }
 
+#if UNITY_EDITOR
+        public void UpdateSelectedRenderers(NativeArray<int> renderersID)
+        {
+            var instances = new NativeArray<InstanceHandle>(renderersID.Length, Allocator.TempJob, NativeArrayOptions.UninitializedMemory);
+            m_BatchersContext.ScheduleQueryRendererGroupInstancesJob(renderersID, instances).Complete();
+            m_BatchersContext.UpdateSelectedInstances(instances);
+            instances.Dispose();
+        }
+#endif
+
         public void PostCullBeginCameraRendering(RenderRequestBatcherContext context)
         {
             RenderSettings.ambientProbe = context.ambientProbe;
@@ -126,9 +134,6 @@ private void UpdateRendererData(in GPUDrivenRendererGroupData rendererData, ILis
 
             Profiler.BeginSample("GPUResidentInstanceBatcher.UpdateRendererData");
             {
-                var usedMaterialIDs = new NativeList<int>(Allocator.TempJob);
-                usedMaterialIDs.AddRange(rendererData.materialID);
-
                 // --------------------------------------------------------------------------------------------------------------------------------------
                 // Allocate and Update CPU instance data
                 // --------------------------------------------------------------------------------------------------------------------------------------
@@ -166,15 +171,14 @@ private void UpdateRendererData(in GPUDrivenRendererGroupData rendererData, ILis
                 {
                     m_InstanceCullingBatcher.BuildBatch(
                         instances,
-                        usedMaterialIDs.AsArray(),
+                        rendererData.materialID,
                         rendererData.meshID,
                         rendererData);
 
                 }
                 Profiler.EndSample();
 
                 instances.Dispose();
-                usedMaterialIDs.Dispose();
             }
             Profiler.EndSample();
         }

Packages/com.unity.render-pipelines.core/Runtime/GPUDriven/GPUResidentDrawer.cs

@@ -383,6 +383,8 @@ private static void Recreate(GPUResidentDrawerSettings settings)
         private NativeList<int> m_FrameCameraIDs;
         private bool m_FrameUpdateNeeded = false;
 
+        private bool m_SelectionChanged;
+
         static GPUResidentDrawer()
         {
 			Lightmapping.bakeCompleted += Reinitialize;
@@ -437,7 +439,6 @@ private GPUResidentDrawer(GPUResidentDrawerSettings settings, int maxInstanceCou
 
             m_Dispatcher = new ObjectDispatcher();
             m_Dispatcher.EnableTypeTracking<LODGroup>(TypeTrackingFlags.SceneObjects);
-            m_Dispatcher.EnableTypeTracking<LightmapSettings>();
             m_Dispatcher.EnableTypeTracking<Mesh>();
             m_Dispatcher.EnableTypeTracking<Material>();
             m_Dispatcher.EnableTransformTracking<LODGroup>(TransformTrackingType.GlobalTRS);
@@ -454,6 +455,9 @@ private GPUResidentDrawer(GPUResidentDrawerSettings settings, int maxInstanceCou
             RenderPipelineManager.endContextRendering += OnEndContextRendering;
             RenderPipelineManager.beginCameraRendering += OnBeginCameraRendering;
             RenderPipelineManager.endCameraRendering += OnEndCameraRendering;
+#if UNITY_EDITOR
+            Selection.selectionChanged += OnSelectionChanged;
+#endif
 
             // GPU Resident Drawer only supports legacy lightmap binding.
             // Accordingly, we set the keyword globally across all shaders.
@@ -478,6 +482,9 @@ private void Dispose()
             RenderPipelineManager.endContextRendering -= OnEndContextRendering;
             RenderPipelineManager.beginCameraRendering -= OnBeginCameraRendering;
             RenderPipelineManager.endCameraRendering -= OnEndCameraRendering;
+#if UNITY_EDITOR
+            Selection.selectionChanged -= OnSelectionChanged;
+#endif
 
             RemoveFromPlayerLoop();
 
@@ -513,6 +520,7 @@ private void OnBeginContextRendering(ScriptableRenderContext context, List<Camer
         {
             if (s_Instance is null)
                 return;
+
 #if UNITY_EDITOR
             EditorFrameUpdate(cameras);
 #endif
@@ -545,8 +553,35 @@ private void EditorFrameUpdate(List<Camera> cameras)
                 else
                     m_FrameUpdateNeeded = true;
             }
+
+            ProcessSelection();
+        }
+
+        private void OnSelectionChanged()
+        {
+            m_SelectionChanged = true;
+        }
+
+        private void ProcessSelection()
+        {
+            if(!m_SelectionChanged)
+                return;
+
+            m_SelectionChanged = false;
+
+            Object[] renderers = Selection.GetFiltered(typeof(MeshRenderer), SelectionMode.Deep);
+
+            var rendererIDs = new NativeArray<int>(renderers.Length, Allocator.TempJob, NativeArrayOptions.UninitializedMemory);
+
+            for (int i = 0; i < renderers.Length; ++i)
+                rendererIDs[i] = renderers[i] ? renderers[i].GetInstanceID() : 0;
+
+            m_Batcher.UpdateSelectedRenderers(rendererIDs);
+
+            rendererIDs.Dispose();
         }
 #endif
+
         private void OnEndContextRendering(ScriptableRenderContext context, List<Camera> cameras)
         {
             if (s_Instance is null)