108 lines
3.1 KiB
C#
108 lines
3.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using UnityEngine.XR.ARSubsystems;
|
|
|
|
namespace UnityEngine.XR.ARFoundation
|
|
{
|
|
/// <summary>
|
|
/// Renders an <see cref="ARPointCloud"/> as a <c>ParticleSystem</c>.
|
|
/// </summary>
|
|
/// <remarks>
|
|
/// Related information: <a href="xref:arfoundation-point-clouds">AR Point Cloud Manager component</a>
|
|
/// </remarks>
|
|
[RequireComponent(typeof(ARPointCloud))]
|
|
[RequireComponent(typeof(ParticleSystem))]
|
|
[HelpURL(typeof(ARPointCloudParticleVisualizer))]
|
|
public sealed class ARPointCloudParticleVisualizer : MonoBehaviour
|
|
{
|
|
void OnPointCloudChanged(ARPointCloudUpdatedEventArgs eventArgs)
|
|
{
|
|
var points = s_Vertices;
|
|
points.Clear();
|
|
|
|
if (m_PointCloud.positions.HasValue)
|
|
{
|
|
foreach (var point in m_PointCloud.positions.Value)
|
|
s_Vertices.Add(point);
|
|
}
|
|
|
|
int numParticles = points.Count;
|
|
if (m_Particles == null || m_Particles.Length < numParticles)
|
|
m_Particles = new ParticleSystem.Particle[numParticles];
|
|
|
|
var color = m_ParticleSystem.main.startColor.color;
|
|
var size = m_ParticleSystem.main.startSize.constant;
|
|
|
|
for (int i = 0; i < numParticles; ++i)
|
|
{
|
|
m_Particles[i].startColor = color;
|
|
m_Particles[i].startSize = size;
|
|
m_Particles[i].position = points[i];
|
|
m_Particles[i].remainingLifetime = 1f;
|
|
}
|
|
|
|
// Remove any existing particles by setting remainingLifetime
|
|
// to a negative value.
|
|
for (int i = numParticles; i < m_NumParticles; ++i)
|
|
{
|
|
m_Particles[i].remainingLifetime = -1f;
|
|
}
|
|
|
|
m_ParticleSystem.SetParticles(m_Particles, Math.Max(numParticles, m_NumParticles));
|
|
m_NumParticles = numParticles;
|
|
}
|
|
|
|
void Awake()
|
|
{
|
|
m_PointCloud = GetComponent<ARPointCloud>();
|
|
m_ParticleSystem = GetComponent<ParticleSystem>();
|
|
}
|
|
|
|
void OnEnable()
|
|
{
|
|
m_PointCloud.updated += OnPointCloudChanged;
|
|
UpdateVisibility();
|
|
}
|
|
|
|
void OnDisable()
|
|
{
|
|
m_PointCloud.updated -= OnPointCloudChanged;
|
|
UpdateVisibility();
|
|
}
|
|
|
|
void Update()
|
|
{
|
|
UpdateVisibility();
|
|
}
|
|
|
|
void UpdateVisibility()
|
|
{
|
|
var visible =
|
|
enabled &&
|
|
(m_PointCloud.trackingState != TrackingState.None);
|
|
|
|
SetVisible(visible);
|
|
}
|
|
|
|
void SetVisible(bool visible)
|
|
{
|
|
if (m_ParticleSystem == null)
|
|
return;
|
|
|
|
var renderer = m_ParticleSystem.GetComponent<Renderer>();
|
|
if (renderer != null)
|
|
renderer.enabled = visible;
|
|
}
|
|
|
|
ARPointCloud m_PointCloud;
|
|
|
|
ParticleSystem m_ParticleSystem;
|
|
|
|
ParticleSystem.Particle[] m_Particles;
|
|
|
|
int m_NumParticles;
|
|
|
|
static List<Vector3> s_Vertices = new ();
|
|
}
|
|
}
|