Jedd Morgan
80 lines
2.6 KiB
C#
80 lines
2.6 KiB
C#
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using Speckle.ConnectorUnity.Components;
|
|
using Speckle.Core.Api;
|
|
using Speckle.Core.Models;
|
|
using Speckle.Core.Transports;
|
|
using UnityEngine;
|
|
|
|
namespace VRSample.Speckle_Helpers
|
|
{
|
|
|
|
/// <summary>
|
|
/// A custom speckle sender that adds sending as a coroutine
|
|
/// This functionality might be added to SpeckleSender sometime.
|
|
/// </summary>
|
|
[RequireComponent(typeof(RecursiveConverter))]
|
|
public class VRSender : MonoBehaviour
|
|
{
|
|
private RecursiveConverter converter;
|
|
private CancellationTokenSource tokenSource;
|
|
void Awake()
|
|
{
|
|
converter = GetComponent<RecursiveConverter>();
|
|
}
|
|
|
|
public IEnumerator ConvertAndSend(GameObject environment, Client client, Stream stream, Branch branch)
|
|
{
|
|
//Convert one (top level) object per coroutine update
|
|
List<Base> convertedRootObjects = new List<Base>();
|
|
foreach (Transform rootObject in environment.transform)
|
|
{
|
|
converter.RecurseTreeToSpeckle(rootObject.gameObject,
|
|
converter.ConverterInstance.CanConvertToSpeckle,
|
|
convertedRootObjects);
|
|
yield return null;
|
|
}
|
|
|
|
Base b = new Base()
|
|
{
|
|
["@objects"] = convertedRootObjects,
|
|
};
|
|
|
|
Task.Run(async () =>
|
|
{
|
|
string commitId = await SendDataAsync(b, client, stream, branch);
|
|
});
|
|
}
|
|
|
|
|
|
public async Task<string> SendDataAsync(Base data, Client client, Stream stream, Branch branch)
|
|
{
|
|
tokenSource?.Cancel();
|
|
tokenSource?.Dispose();
|
|
tokenSource = new CancellationTokenSource();
|
|
|
|
ServerTransport transport = new ServerTransport(client.Account, stream.id);
|
|
transport.CancellationToken = tokenSource.Token;
|
|
|
|
return await SpeckleSender.SendDataAsync(tokenSource.Token,
|
|
remoteTransport: transport,
|
|
data: data,
|
|
client: client,
|
|
branchName: branch.name,
|
|
createCommit: true,
|
|
onProgressAction: dict => _ = 0,
|
|
onErrorAction: (m, e) => Debug.LogError($"{m}\n{e}", this)
|
|
);
|
|
}
|
|
|
|
|
|
public void OnDestroy()
|
|
{
|
|
tokenSource?.Cancel();
|
|
tokenSource?.Dispose();
|
|
}
|
|
}
|
|
}
|