| | | 1 | | using System; |
| | | 2 | | using System.IO; |
| | | 3 | | using UnityEngine; |
| | | 4 | | using UnityEngine.Networking; |
| | | 5 | | |
| | | 6 | | public static class ConfigLoader { |
| | | 7 | | // Delegate for loading from a Protobuf text file and returning the length of the serialized |
| | | 8 | | // message as an output argument. |
| | | 9 | | private delegate Plugin.StatusCode SerializedProtobufLengthDelegate(string file, |
| | | 10 | | out int serializedLength); |
| | | 11 | | // Delegate for loading from a Protobuf text file to binary format and returning the length of the |
| | | 12 | | // serialized message as an output argument. |
| | | 13 | | private delegate Plugin.StatusCode LoadProtobufDelegate(string file, byte[] buffer, |
| | | 14 | | int bufferSize, out int serializedLength); |
| | | 15 | | |
| | 1950 | 16 | | public static string GetStreamingAssetsFilePath(string relativePath) { |
| | 1950 | 17 | | return Path.Combine(Application.streamingAssetsPath, relativePath); |
| | 1950 | 18 | | } |
| | | 19 | | |
| | 0 | 20 | | public static string LoadFromStreamingAssets(string relativePath) { |
| | 0 | 21 | | string streamingAssetsPath = GetStreamingAssetsFilePath(relativePath); |
| | | 22 | | |
| | | 23 | | #if UNITY_STANDALONE_OSX || UNITY_EDITOR_OSX || UNITY_STANDALONE_LINUX || UNITY_EDITOR_LINUX || UNITY_IOS |
| | 0 | 24 | | if (!streamingAssetsPath.StartsWith("file://")) { |
| | 0 | 25 | | streamingAssetsPath = "file://" + streamingAssetsPath; |
| | 0 | 26 | | } |
| | | 27 | | #endif |
| | | 28 | | |
| | 0 | 29 | | using (UnityWebRequest www = UnityWebRequest.Get(streamingAssetsPath)) { |
| | 0 | 30 | | www.SendWebRequest(); |
| | 0 | 31 | | while (!www.isDone) {} |
| | | 32 | | |
| | 0 | 33 | | if (www.result != UnityWebRequest.Result.Success) { |
| | 0 | 34 | | Debug.LogError($"Failed to load {streamingAssetsPath}: {www.error}."); |
| | 0 | 35 | | return null; |
| | | 36 | | } |
| | 0 | 37 | | return www.downloadHandler.text; |
| | | 38 | | } |
| | 0 | 39 | | } |
| | | 40 | | |
| | 955 | 41 | | public static Configs.AttackBehaviorConfig LoadAttackBehaviorConfig(string configFile) { |
| | 955 | 42 | | return LoadProtobufConfig<Configs.AttackBehaviorConfig>( |
| | | 43 | | Path.Combine("Configs/Attacks", configFile), |
| | | 44 | | Protobuf.Protobuf_AttackBehaviorConfig_GetSerializedLength, |
| | | 45 | | Protobuf.Protobuf_AttackBehaviorConfig_LoadToBinary); |
| | 955 | 46 | | } |
| | | 47 | | |
| | 0 | 48 | | public static Configs.RunConfig LoadRunConfig(string configFile) { |
| | 0 | 49 | | return LoadProtobufConfig<Configs.RunConfig>(Path.Combine("Configs/Runs", configFile), |
| | | 50 | | Protobuf.Protobuf_RunConfig_GetSerializedLength, |
| | | 51 | | Protobuf.Protobuf_RunConfig_LoadToBinary); |
| | 0 | 52 | | } |
| | | 53 | | |
| | 12 | 54 | | public static Configs.SimulationConfig LoadSimulationConfig(string configFile) { |
| | 12 | 55 | | var config = LoadProtobufConfig<Configs.SimulationConfig>( |
| | | 56 | | Path.Combine("Configs/Simulations", configFile), |
| | | 57 | | Protobuf.Protobuf_SimulationConfig_GetSerializedLength, |
| | | 58 | | Protobuf.Protobuf_SimulationConfig_LoadToBinary); |
| | 24 | 59 | | if (config != null) { |
| | 12 | 60 | | UIManager.Instance.LogActionMessage($"[SIM] Loaded simulation configuration: {configFile}."); |
| | 12 | 61 | | } |
| | 12 | 62 | | return config; |
| | 12 | 63 | | } |
| | | 64 | | |
| | 12 | 65 | | public static Configs.SimulatorConfig LoadSimulatorConfig(string configFile) { |
| | 12 | 66 | | return LoadProtobufConfig<Configs.SimulatorConfig>( |
| | | 67 | | configFile, Protobuf.Protobuf_SimulatorConfig_GetSerializedLength, |
| | | 68 | | Protobuf.Protobuf_SimulatorConfig_LoadToBinary); |
| | 12 | 69 | | } |
| | | 70 | | |
| | 969 | 71 | | public static Configs.StaticConfig LoadStaticConfig(string configFile) { |
| | 969 | 72 | | return LoadProtobufConfig<Configs.StaticConfig>( |
| | | 73 | | Path.Combine("Configs/Models", configFile), |
| | | 74 | | Protobuf.Protobuf_StaticConfig_GetSerializedLength, |
| | | 75 | | Protobuf.Protobuf_StaticConfig_LoadToBinary); |
| | 969 | 76 | | } |
| | | 77 | | |
| | | 78 | | private static T LoadProtobufConfig<T>(string relativePath, |
| | | 79 | | SerializedProtobufLengthDelegate serializedLengthFunction, |
| | | 80 | | LoadProtobufDelegate loadFunction) |
| | 1948 | 81 | | where T : Google.Protobuf.IMessage<T>, new() { |
| | 1948 | 82 | | string streamingAssetsPath = GetStreamingAssetsFilePath(relativePath); |
| | | 83 | | |
| | | 84 | | // Determine the length of the serialized Protobuf message. |
| | 1948 | 85 | | Plugin.StatusCode status = |
| | | 86 | | serializedLengthFunction(streamingAssetsPath, out int serializedLength); |
| | 1948 | 87 | | if (status != Plugin.StatusCode.StatusOk) { |
| | 0 | 88 | | Debug.Log( |
| | | 89 | | $"Failed to get the length of the serialized message from the Protobuf text file {relativePath} with status co |
| | 0 | 90 | | return default; |
| | | 91 | | } |
| | | 92 | | |
| | | 93 | | // Load the Protobuf message to binary format. |
| | 1948 | 94 | | byte[] buffer = new byte[serializedLength]; |
| | 1948 | 95 | | status = loadFunction(streamingAssetsPath, buffer, serializedLength, out int loadedLength); |
| | 1948 | 96 | | if (status != Plugin.StatusCode.StatusOk) { |
| | 0 | 97 | | Debug.Log($"Failed to load the Protobuf text file {relativePath} with status code {status}."); |
| | 0 | 98 | | return default; |
| | | 99 | | } |
| | 1948 | 100 | | return new Google.Protobuf.MessageParser<T>(() => new T()).ParseFrom(buffer, 0, loadedLength); |
| | 1948 | 101 | | } |
| | | 102 | | } |