149 lines
3.8 KiB
C#
149 lines
3.8 KiB
C#
// 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 Mediapipe.Unity.CoordinateSystem;
|
|
using UnityEngine;
|
|
|
|
using mplt = Mediapipe.LocationData.Types;
|
|
|
|
namespace Mediapipe.Unity
|
|
{
|
|
#pragma warning disable IDE0065
|
|
using Color = UnityEngine.Color;
|
|
#pragma warning restore IDE0065
|
|
|
|
public class RectangleAnnotation : HierarchicalAnnotation
|
|
{
|
|
[SerializeField] private LineRenderer _lineRenderer;
|
|
[SerializeField] private Color _color = Color.red;
|
|
[SerializeField, Range(0, 1)] private float _lineWidth = 1.0f;
|
|
|
|
private static readonly Vector3[] _EmptyPositions = new Vector3[] { Vector3.zero, Vector3.zero, Vector3.zero, Vector3.zero };
|
|
|
|
private void OnEnable()
|
|
{
|
|
ApplyColor(_color);
|
|
ApplyLineWidth(_lineWidth);
|
|
}
|
|
|
|
private void OnDisable()
|
|
{
|
|
ApplyLineWidth(0.0f);
|
|
_lineRenderer.SetPositions(_EmptyPositions);
|
|
}
|
|
|
|
#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[] positions)
|
|
{
|
|
_lineRenderer.SetPositions(positions ?? _EmptyPositions);
|
|
}
|
|
|
|
public void Draw(Rect target, Vector2Int imageSize)
|
|
{
|
|
if (ActivateFor(target))
|
|
{
|
|
Draw(GetScreenRect().GetRectVertices(target, imageSize, rotationAngle, isMirrored));
|
|
}
|
|
}
|
|
|
|
public void Draw(NormalizedRect target)
|
|
{
|
|
if (ActivateFor(target))
|
|
{
|
|
Draw(GetScreenRect().GetRectVertices(target, rotationAngle, isMirrored));
|
|
}
|
|
}
|
|
|
|
public void Draw(LocationData target, Vector2Int imageSize)
|
|
{
|
|
if (ActivateFor(target))
|
|
{
|
|
switch (target.Format)
|
|
{
|
|
case mplt.Format.BoundingBox:
|
|
{
|
|
Draw(GetScreenRect().GetRectVertices(target.BoundingBox, imageSize, rotationAngle, isMirrored));
|
|
break;
|
|
}
|
|
case mplt.Format.RelativeBoundingBox:
|
|
{
|
|
Draw(GetScreenRect().GetRectVertices(target.RelativeBoundingBox, rotationAngle, isMirrored));
|
|
break;
|
|
}
|
|
case mplt.Format.Global:
|
|
case mplt.Format.Mask:
|
|
default:
|
|
{
|
|
throw new System.ArgumentException($"The format of the LocationData must be BoundingBox or RelativeBoundingBox, but {target.Format}");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void Draw(LocationData target)
|
|
{
|
|
if (ActivateFor(target))
|
|
{
|
|
switch (target.Format)
|
|
{
|
|
case mplt.Format.RelativeBoundingBox:
|
|
{
|
|
Draw(GetScreenRect().GetRectVertices(target.RelativeBoundingBox, rotationAngle, isMirrored));
|
|
break;
|
|
}
|
|
case mplt.Format.BoundingBox:
|
|
case mplt.Format.Global:
|
|
case mplt.Format.Mask:
|
|
default:
|
|
{
|
|
throw new System.ArgumentException($"The format of the LocationData must be RelativeBoundingBox, but {target.Format}");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private 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;
|
|
}
|
|
}
|
|
}
|
|
}
|