82 lines
2.4 KiB
C#
82 lines
2.4 KiB
C#
using System;
|
|
using UnityEngine;
|
|
using UI = UnityEngine.UI;
|
|
using Klak.TestTools;
|
|
using System.Linq;
|
|
|
|
namespace UltraFace
|
|
{
|
|
public sealed class Visualizer : MonoBehaviour
|
|
{
|
|
#region Editable attributes
|
|
|
|
[SerializeField] ImageSource _source = null;
|
|
[SerializeField, Range(0, 1)] float _threshold = 0.5f;
|
|
[SerializeField] ResourceSet _resources = null;
|
|
[SerializeField] Shader _visualizer = null;
|
|
[SerializeField] Texture2D _texture = null;
|
|
[SerializeField] UI.RawImage _previewUI = null;
|
|
|
|
public Action<bool> OnDetectionStatusChanged;
|
|
private bool _isDetected;
|
|
|
|
#endregion
|
|
|
|
#region Private objects
|
|
|
|
FaceDetector _detector;
|
|
Material _material;
|
|
ComputeBuffer _drawArgs;
|
|
|
|
#endregion
|
|
|
|
#region MonoBehaviour implementation
|
|
|
|
void Start()
|
|
{
|
|
_detector = new FaceDetector(_resources);
|
|
_material = new Material(_visualizer);
|
|
_drawArgs = new ComputeBuffer(4, sizeof(uint), ComputeBufferType.IndirectArguments);
|
|
_drawArgs.SetData(new [] {6, 0, 0, 0});
|
|
}
|
|
|
|
void OnDestroy()
|
|
{
|
|
_detector?.Dispose();
|
|
Destroy(_material);
|
|
_drawArgs?.Dispose();
|
|
}
|
|
|
|
void Update()
|
|
{
|
|
_detector.ProcessImage(_source.Texture, _threshold);
|
|
_previewUI.texture = _source.Texture;
|
|
|
|
if(!_isDetected && _detector.Detections.ToList().Count > 0)
|
|
{
|
|
_isDetected = true;
|
|
float height = _detector.Detections.Max(x => x.y2 - x.y1);
|
|
Debug.LogWarning(height);
|
|
OnDetectionStatusChanged?.Invoke(_isDetected);
|
|
}
|
|
else if(_isDetected && _detector.Detections.ToList().Count == 0)
|
|
{
|
|
_isDetected = false;
|
|
OnDetectionStatusChanged?.Invoke(_isDetected);
|
|
Debug.LogWarning("йобла потеряны");
|
|
}
|
|
}
|
|
|
|
void OnRenderObject()
|
|
{
|
|
_detector.SetIndirectDrawCount(_drawArgs);
|
|
_material.SetFloat("_Threshold", _threshold);
|
|
_material.SetTexture("_Texture", _texture);
|
|
_material.SetBuffer("_Detections", _detector.DetectionBuffer);
|
|
_material.SetPass(_texture == null ? 0 : 1);
|
|
Graphics.DrawProceduralIndirectNow(MeshTopology.Triangles, _drawArgs, 0);
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
} |