61 lines
2.6 KiB
C#
61 lines
2.6 KiB
C#
using FishNet.Object;
|
|
using GameKit.Utilities;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using UnityEngine.SceneManagement;
|
|
|
|
namespace FishNet.Utility.Extension
|
|
{
|
|
|
|
public static class Scenes
|
|
{
|
|
/// <summary>
|
|
/// Gets all NetworkObjects in a scene.
|
|
/// </summary>
|
|
/// <param name="s">Scene to get objects in.</param>
|
|
/// <param name="firstOnly">True to only return the first NetworkObject within an object chain. False will return nested NetworkObjects.</param>
|
|
/// <param name="cache">ListCache of found NetworkObjects.</param>
|
|
/// <returns></returns>
|
|
public static void GetSceneNetworkObjects(Scene s, bool firstOnly, ref List<NetworkObject> result)
|
|
{
|
|
List<NetworkObject> nobCacheA = CollectionCaches<NetworkObject>.RetrieveList();
|
|
List<NetworkObject> nobCacheB = CollectionCaches<NetworkObject>.RetrieveList();
|
|
List<GameObject> gameObjectCache = CollectionCaches<GameObject>.RetrieveList();
|
|
//Iterate all root objects for the scene.
|
|
s.GetRootGameObjects(gameObjectCache);
|
|
foreach (GameObject go in gameObjectCache)
|
|
{
|
|
//Get NetworkObjects within children of each root.
|
|
go.GetComponentsInChildren<NetworkObject>(true, nobCacheA);
|
|
//If network objects are found.
|
|
if (nobCacheA.Count > 0)
|
|
{
|
|
//Add only the first networkobject
|
|
if (firstOnly)
|
|
{
|
|
/* The easiest way to see if a nob is nested is to
|
|
* get nobs in parent and if the count is greater than 1, then
|
|
* it is nested. The technique used here isn't exactly fast but
|
|
* it will only occur during scene loads, so I'm trading off speed
|
|
* for effort and readability. */
|
|
foreach (NetworkObject nob in nobCacheA)
|
|
{
|
|
nob.GetComponentsInParent<NetworkObject>(true, nobCacheB);
|
|
//No extra nobs, only this one.
|
|
if (nobCacheB.Count == 1)
|
|
result.Add(nob);
|
|
}
|
|
}
|
|
//Not first only, add them all.
|
|
else
|
|
{
|
|
result.AddRange(nobCacheA);
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
} |