speckle-unity/Assets/SpeckleExamples.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using Speckle.Core;
using Speckle.Core.Models;
using Speckle.Core.Api;
using Speckle.Core.Credentials;
using System;
using System.Collections.Concurrent;
using System.Threading.Tasks;
using System.Linq;
using Speckle.Core.Transports;
using Objects.Converter.Unity;
using Objects.Geometry;
using UnityEditor;
using System.Text.RegularExpressions;
using System.IO;
using UnityEngine.UI;

namespace Speckle.ConnectorUnity
{
  public class SpeckleExamples : MonoBehaviour
  {
    public Button ReceiveBtn;
    public InputField ReceiveText;
    public Button SendBtn;
    public InputField SendText;

    private GameObject receivedGo;

    void Start()
    {
      //hardcoded cuz I'm lazy, replace with what you need
      ReceiveText.text = "d1a4748ba2";//4ad65b572e
      SendText.text = "cd83745025";

      if (ReceiveBtn == null || ReceiveText == null || SendBtn == null || SendText == null)
      {
        Debug.Log("Please set Send/Receive buttons and input fields");
        return;
      }


      Button btn = ReceiveBtn.GetComponent<Button>();
      btn.onClick.AddListener(CreateReceiver);

      Button btn2 = SendBtn.GetComponent<Button>();
      btn2.onClick.AddListener(SendData);
    }

    // Shows how to create a new Receiver from code and then pull data manually
    private void CreateReceiver()
    {
      ReceiveBtn.enabled = false;
      ReceiveText.enabled = false;

      var receiver = ScriptableObject.CreateInstance<Receiver>();
      receiver.Init(ReceiveText.text, true, onDataReceivedAction: ReceiverOnDataReceivedAction,
        onProgressAction: ReceiverProgressAction);

      //receive manually once
      receiver.Receive();
    }

    private void SendData()
    {
      if (!SelectionManager.selectedObjects.Any())
        return;

      var objs = new List<GameObject>();
      foreach (var index in SelectionManager.selectedObjects)
      {
        objs.Add(SelectionManager.selectables[index].gameObject);
      }

      Sender.Send(SendText.text, objs, onProgressAction: SenderProgressAction,
        onDataSentAction: SenderOnDataSentAction);
    }

    private void SenderProgressAction(ConcurrentDictionary<string, int> dict)
    {
      //Run on a dispatcher as GOs can only be retrieved on the main thread
      Dispatcher.Instance().Enqueue(() =>
      {
        var val = dict.Values.Average();
        SendBtn.GetComponentInChildren<Text>().text = $"Sending object #{val}";
      });
    }

    private void SenderOnDataSentAction(string commitId)
    {
      Dispatcher.Instance().Enqueue(() =>
      {
        Debug.Log($"Sent {commitId}");
        SendBtn.GetComponentInChildren<Text>().text = "Send";
      });
    }

    private void ReceiverOnDataReceivedAction(GameObject go)
    {
      Debug.Log($"Received {go.name}");

      ReceiveBtn.GetComponentInChildren<Text>().text = "Receive";

      if (receivedGo != null)
        Destroy(receivedGo);

      AddClasses(go);
      receivedGo = go;
    }

    private void ReceiverProgressAction(ConcurrentDictionary<string, int> dict)
    {
      //Run on a dispatcher as GOs can only be retrieved on the main thread
      Dispatcher.Instance().Enqueue(() =>
      {
        var val = dict.Values.Average();
        ReceiveBtn.GetComponentInChildren<Text>().text = $"Receiving object #{val}";
      });
    }

    /// <summary>
    /// Adds material and selectable script to all children of a GameObject
    /// </summary>
    /// <param name="go"></param>
    private void AddClasses(GameObject go)
    {
      for (var i = 0; i < go.transform.childCount; i++)
      {
        var child = go.transform.GetChild(i);
        child.gameObject.AddComponent<Selectable>();
      }
    }
  }
}