Merge pull request #361 from Unity-Technologies/pyrception-integration

Added the Dataset Visualizer installer and corresponding UI

Author: mkamalza

com.unity.perception/Editor/GroundTruth/PerceptionCameraEditor.cs

@@ -1,9 +1,13 @@
-using System;
+using System;
 using System.Collections.Generic;
 using UnityEditorInternal;
 using UnityEngine;
 using UnityEngine.Perception.GroundTruth;
 
+#if UNITY_EDITOR_WIN || UNITY_EDITOR_OSX
+using UnityEditor.Perception.Visualizer;
+#endif
+
 namespace UnityEditor.Perception.GroundTruth
 {
     [CustomEditor(typeof(PerceptionCamera))]
@@ -189,6 +193,27 @@ public override void OnInspectorGUI()
             GUILayout.EndHorizontal();
             GUILayout.EndVertical();
 
+#if UNITY_EDITOR_WIN || UNITY_EDITOR_OSX
+            GUILayout.Space(10);
+            EditorGUILayout.LabelField("Visualizer Tool");
+
+            GUILayout.BeginHorizontal("TextArea");
+            if (GUILayout.Button("Open Visualizer"))
+            {
+                var project = Application.dataPath;
+                _=VisualizerInstaller.RunVisualizer(project);
+            }
+
+            if (GUILayout.Button("Check For Updates"))
+            {
+                var project = Application.dataPath;
+                _=VisualizerInstaller.CheckForUpdates();
+            }
+
+            GUILayout.EndHorizontal();
+            GUILayout.Space(10);
+#endif
+
 
             if (EditorSettings.asyncShaderCompilation)
             {

com.unity.perception/Editor/Pyrception.meta

@@ -0,0 +1,73 @@
+#if UNITY_EDITOR_WIN || UNITY_EDITOR_OSX
+using System;
+using System.Net.Http;
+using System.Threading.Tasks;
+using Newtonsoft.Json;
+using UnityEngine;
+
+namespace UnityEditor.Perception.Visualizer
+{
+    static class PipAPI
+    {
+        static readonly HttpClient k_HttpClient;
+        const string k_PypiServer = "https://pypi.org";
+        const string k_PackageName = "unity-cv-datasetvisualizer";
+
+        static PipAPI()
+        {
+            k_HttpClient = new HttpClient();
+        }
+
+        internal static async Task<string> GetLatestVersionNumber()
+        {
+            var requestUri = new Uri($"{k_PypiServer}/pypi/{k_PackageName}/json?Accept=application/json");
+			try
+            {
+            	var httpResponse = await k_HttpClient.GetAsync(requestUri);
+                if (httpResponse.IsSuccessStatusCode)
+                {
+                    var responseString = httpResponse.Content.ReadAsStringAsync().Result;
+                    dynamic responseJson = JsonConvert.DeserializeObject(responseString);
+                    return responseJson.info.version;
+                }
+
+                HandleApiErrors(httpResponse);
+            }
+            catch (HttpRequestException e)
+            {
+                Debug.LogException(e);
+            }
+			return null;
+        }
+
+        static void HandleApiErrors(HttpResponseMessage responseMessage)
+        {
+            Debug.LogError("A request to PyPI.org did not successfully complete: " + responseMessage.ReasonPhrase);
+        }
+
+        internal static int CompareVersions(string version1, string version2)
+        {
+            var split1 = version1.Split('.');
+            var split2 = version2.Split('.');
+
+            int i;
+            for(i = 0; i < Math.Min(split1.Length, split2.Length); i++)
+            {
+                var compare = Int32.Parse(split1[i]) - Int32.Parse(split2[i]);
+                if (compare != 0)
+                {
+                    return compare;
+                }
+            }
+
+            if (i < split1.Length)
+                return 1;
+
+            if (i < split2.Length)
+                return -1;
+
+            return 0;
+        }
+    }
+}
+#endif

…/Editor/Pyrception/pyrception-utils.meta ….unity.perception/Editor/Visualizer.metacom.unity.perception/Editor/Pyrception/pyrception-utils.meta renamed to com.unity.perception/Editor/Visualizer.meta com.unity.perception/Editor/Pyrception/pyrception-utils.meta renamed to com.unity.perception/Editor/Visualizer.meta

@@ -0,0 +1,58 @@
+#if UNITY_EDITOR_WIN || UNITY_EDITOR_OSX
+using System;
+using UnityEditor.PackageManager.Requests;
+using UnityEditor.PackageManager;
+using UnityEngine;
+using System.Threading;
+
+namespace UnityEditor.Perception.Visualizer
+{
+    [InitializeOnLoad]
+    static class PythonForUnityInstaller
+    {
+        static PythonForUnityInstaller()
+        {
+            Add();
+        }
+
+        static void Add()
+        {
+            if (!CheckIfPackageInstalled())
+            {
+                var request = Client.Add("[email protected]");
+
+                while (!request.IsCompleted)
+                {
+                    Thread.Sleep(100);
+                }
+                if (request.Status == StatusCode.Success)
+                    Debug.Log("Installed: " + request.Result.packageId);
+                else if (request.Status >= StatusCode.Failure)
+                    Debug.Log(request.Error.message);
+            }
+        }
+
+        static bool CheckIfPackageInstalled()
+        {
+            var request = Client.List();
+            while (!request.IsCompleted)
+            {
+                Thread.Sleep(100);
+            }
+            if (request.Status == StatusCode.Success)
+            {
+                foreach (var package in request.Result)
+                {
+                    if (package.packageId.Contains("com.unity.scripting.python"))
+                    {
+                        return true;
+                    }
+                }
+            }
+            else if (request.Status >= StatusCode.Failure)
+                Debug.LogError(request.Error.message);
+            return false;
+        }
+    }
+}
+#endif