Wes xx mediapipe integration
This commit is contained in:
@@ -0,0 +1,83 @@
|
||||
// Copyright (c) 2021 homuler
|
||||
//
|
||||
// Use of this source code is governed by an MIT-style
|
||||
// license that can be found in the LICENSE file or at
|
||||
// https://opensource.org/licenses/MIT.
|
||||
|
||||
using UnityEngine;
|
||||
|
||||
namespace Mediapipe.Unity
|
||||
{
|
||||
#pragma warning disable IDE0065
|
||||
using Color = UnityEngine.Color;
|
||||
#pragma warning restore IDE0065
|
||||
|
||||
public class LineAnnotation : HierarchicalAnnotation
|
||||
{
|
||||
[SerializeField] private LineRenderer _lineRenderer;
|
||||
[SerializeField] private Color _color = Color.green;
|
||||
[SerializeField, Range(0, 1)] private float _lineWidth = 1.0f;
|
||||
|
||||
private void OnEnable()
|
||||
{
|
||||
ApplyColor(_color);
|
||||
ApplyLineWidth(_lineWidth);
|
||||
}
|
||||
|
||||
private void OnDisable()
|
||||
{
|
||||
ApplyLineWidth(0.0f);
|
||||
}
|
||||
|
||||
#if UNITY_EDITOR
|
||||
private void OnValidate()
|
||||
{
|
||||
if (!UnityEditor.PrefabUtility.IsPartOfAnyPrefab(this))
|
||||
{
|
||||
ApplyColor(_color);
|
||||
ApplyLineWidth(_lineWidth);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
public void SetColor(Color color)
|
||||
{
|
||||
_color = color;
|
||||
ApplyColor(_color);
|
||||
}
|
||||
|
||||
public void SetLineWidth(float lineWidth)
|
||||
{
|
||||
_lineWidth = lineWidth;
|
||||
ApplyLineWidth(_lineWidth);
|
||||
}
|
||||
|
||||
public void Draw(Vector3 a, Vector3 b)
|
||||
{
|
||||
_lineRenderer.SetPositions(new Vector3[] { a, b });
|
||||
}
|
||||
|
||||
public void Draw(GameObject a, GameObject b)
|
||||
{
|
||||
_lineRenderer.SetPositions(new Vector3[] { a.transform.localPosition, b.transform.localPosition });
|
||||
}
|
||||
|
||||
public void ApplyColor(Color color)
|
||||
{
|
||||
if (_lineRenderer != null)
|
||||
{
|
||||
_lineRenderer.startColor = color;
|
||||
_lineRenderer.endColor = color;
|
||||
}
|
||||
}
|
||||
|
||||
private void ApplyLineWidth(float lineWidth)
|
||||
{
|
||||
if (_lineRenderer != null)
|
||||
{
|
||||
_lineRenderer.startWidth = lineWidth;
|
||||
_lineRenderer.endWidth = lineWidth;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user