Snaparazzi/Assets/MPUIKit/Runtime/Scripts/MPImageBasic.cs

421 lines
16 KiB
C#
Raw Permalink Normal View History

2024-02-29 20:33:27 +00:00
using System;
using UnityEngine;
using UnityEngine.UI;
namespace MPUIKIT {
[AddComponentMenu("UI/MPUI/MPImageBasic")]
public class MPImageBasic : Image {
#region SerializedFields
[SerializeField] private DrawShape m_DrawShape = DrawShape.None;
[SerializeField] private Type m_ImageType = Type.Simple; // Mapping in Vertex Stream
[SerializeField] private float m_StrokeWidth; // MapTo -> UV2.x
[SerializeField] private float m_FalloffDistance = 0.5f; // MapTo -> UV2.y
[SerializeField] private float m_OutlineWidth; // MapTo -> Normal.x
[SerializeField] private Color m_OutlineColor = Color.black; // MapTo -> Tangent.x, Tangent.y, Tangent.z, Tangent.w
[SerializeField] private float m_ShapeRotation; // MapTo -> UV3.x Compressed
[SerializeField] private bool m_ConstrainRotation = true; // MapTo -> UV3.x Compressed
[SerializeField] private bool m_FlipHorizontal; // MapTo -> UV3.x Compressed
[SerializeField] private bool m_FlipVertical; // MapTo -> UV3.x Compressed
[SerializeField] private CornerStyleType m_CornerStyle; // MapTo -> UV3.y
[SerializeField] private Vector4 m_RectangleCornerRadius; // MapTo -> Normal.y, Normal.z compressed
[SerializeField] private Vector3 m_TriangleCornerRadius; // MapTo -> Normal.y, Normal.z compressed
#pragma warning disable
// ReSharper disable once NotAccessedField.Local
[SerializeField] private bool m_TriangleUniformCornerRadius = true;
// ReSharper disable once NotAccessedField.Local
[SerializeField] private bool m_RectangleUniformCornerRadius = true;
#pragma warning restore
[SerializeField] private float m_CircleRadius; // MapTo -> Normal.y
[SerializeField] private bool m_CircleFitToRect = true; // MapTo -> Normal.z
[SerializeField] private int m_NStarPolygonSideCount = 3; // MapTo -> Normal.y compressed
[SerializeField] private float m_NStarPolygonInset = 2f; // MapTo -> Normal.y compressed
[SerializeField] private float m_NStarPolygonCornerRadius; // MapTo -> Normal.z
#endregion
#region Public Accessors
public DrawShape Shape { get => m_DrawShape;
set {
m_DrawShape = value;
m_Material = null;
base.SetMaterialDirty();
base.SetVerticesDirty();
}
}
public float StrokeWidth {
get => m_StrokeWidth;
set {
Vector2 size = GetPixelAdjustedRect().size;
m_StrokeWidth = Mathf.Clamp(value, 0, Mathf.Min(size.x, size.y) * 0.5f);
base.SetVerticesDirty();
}
}
public float FallOffDistance {
get => m_FalloffDistance;
set {
m_FalloffDistance = Mathf.Max(0, value);
base.SetVerticesDirty();
}
}
public float OutlineWidth {
get => m_OutlineWidth;
set {
m_OutlineWidth = Mathf.Max(0,value);
base.SetVerticesDirty();
}
}
public Color OutlineColor {
get => m_OutlineColor;
set {
m_OutlineColor = value;
base.SetVerticesDirty();
}
}
public float ShapeRotation {
get => m_ShapeRotation;
set {
m_ShapeRotation = value % 360;
ConstrainRotationValue();
base.SetVerticesDirty();
}
}
public bool ConstrainRotation {
get => m_ConstrainRotation;
set {
m_ConstrainRotation = value;
ConstrainRotationValue();
base.SetVerticesDirty();
}
}
public bool FlipHorizontal {
get => m_FlipHorizontal;
set {
m_FlipHorizontal = value;
base.SetVerticesDirty();
}
}
public bool FlipVertical {
get => m_FlipVertical;
set {
m_FlipVertical = value;
base.SetVerticesDirty();
}
}
/// <summary>
/// Type of the image. Only two types are supported. Simple and Filled.
/// Default and fallback value is Simple.
/// </summary>
public new Type type {
get => m_ImageType;
set {
if (m_ImageType != value) {
switch (value) {
case Type.Simple:
case Type.Filled:
if (sprite) m_ImageType = value;
break;
case Type.Tiled:
case Type.Sliced:
break;
default:
throw new ArgumentOutOfRangeException(value.ToString(), value, null);
}
}
if(base.type != m_ImageType) base.type = m_ImageType;
base.SetAllDirty();
}
}
public CornerStyleType CornerStyle {
get => m_CornerStyle;
set {
m_CornerStyle = value;
base.SetVerticesDirty();
}
}
public Vector3 TriangleCornerRadius {
get => m_TriangleCornerRadius;
set {
Vector2 size = GetPixelAdjustedRect().size;
float zMax = size.x * 0.5f;
m_TriangleCornerRadius.z = Mathf.Clamp(value.z, 0, zMax);
float hMax = Mathf.Min(size.x, size.y) * 0.3f;
m_TriangleCornerRadius.x = Mathf.Clamp(value.x, 0, hMax);
m_TriangleCornerRadius.y = Mathf.Clamp(value.y, 0, hMax);
base.SetVerticesDirty();
}
}
public Vector4 RectangleCornerRadius {
get => m_RectangleCornerRadius;
set
{
m_RectangleCornerRadius = value;
base.SetVerticesDirty();
}
}
public float CircleRadius {
get => m_CircleRadius;
set {
m_CircleRadius = Mathf.Clamp(value, 0, GetMinSize());
base.SetVerticesDirty();
}
}
public bool CircleFitToRect {
get => m_CircleFitToRect;
set {
m_CircleFitToRect = value;
base.SetVerticesDirty();
}
}
public float NStarPolygonCornerRadius {
get => m_NStarPolygonCornerRadius;
set {
float halfHeight = GetPixelAdjustedRect().height * 0.5f;
m_NStarPolygonCornerRadius = Mathf.Clamp(value, m_NStarPolygonSideCount == 2? 0.1f : 0f, halfHeight);
base.SetVerticesDirty();
}
}
public float NStarPolygonInset {
get => m_NStarPolygonInset;
set {
m_NStarPolygonInset = Mathf.Clamp(value, 2f, m_NStarPolygonSideCount);
base.SetVerticesDirty();
}
}
public int NStarPolygonSideCount {
get => m_NStarPolygonSideCount;
set {
m_NStarPolygonSideCount = Mathf.Clamp(value, 2, 10);
base.SetVerticesDirty();
}
}
#endregion
public override Material material {
get
{
switch (m_DrawShape)
{
case DrawShape.None:
return Canvas.GetDefaultCanvasMaterial();
case DrawShape.Circle:
case DrawShape.Triangle:
case DrawShape.Rectangle:
return MPMaterials.GetMaterial((int)m_DrawShape - 1, m_StrokeWidth > 0f, m_OutlineWidth > 0f);
case DrawShape.Pentagon:
case DrawShape.Hexagon:
case DrawShape.NStarPolygon:
return MPMaterials.GetMaterial(3, m_StrokeWidth > 0f, m_OutlineWidth > 0f);
default:
throw new ArgumentOutOfRangeException();
}
}
set => Debug.LogWarning("Setting Material of MPImageBasic has no effect.");
}
public override float preferredWidth => sprite == MPImageUtility.EmptySprite ? 0 : base.preferredWidth;
public override float preferredHeight => sprite == MPImageUtility.EmptySprite ? 0 : base.preferredHeight;
protected override void OnEnable() {
base.OnEnable();
MPImageUtility.FixAdditionalShaderChannelsInCanvas(canvas);
if (sprite == null) sprite = MPImageUtility.EmptySprite;
}
#if UNITY_EDITOR
protected override void OnValidate() {
base.OnValidate();
Shape = m_DrawShape;
if (sprite == null) sprite = MPImageUtility.EmptySprite;
type = m_ImageType;
StrokeWidth = m_StrokeWidth;
FallOffDistance = m_FalloffDistance;
OutlineWidth = m_OutlineWidth;
OutlineColor = m_OutlineColor;
ShapeRotation = m_ShapeRotation;
ConstrainRotation = m_ConstrainRotation;
FlipHorizontal = m_FlipHorizontal;
FlipVertical = m_FlipVertical;
CornerStyle = m_CornerStyle;
}
#endif
private float GetMinSizeHalf() {
return GetMinSize() * 0.5f;
}
private float GetMinSize() {
Vector2 size = GetPixelAdjustedRect().size;
return Mathf.Min(size.x, size.y);
}
private void ConstrainRotationValue() {
if (!m_ConstrainRotation) return;
float finalRotation = m_ShapeRotation - (m_ShapeRotation % 90);
if (Mathf.Abs(finalRotation) >= 360) finalRotation = 0;
m_ShapeRotation = finalRotation;
}
protected override void OnTransformParentChanged() {
base.OnTransformParentChanged();
MPImageUtility.FixAdditionalShaderChannelsInCanvas(canvas);
base.SetVerticesDirty();
}
private MPVertexStream CreateVertexStream() {
MPVertexStream stream = new MPVertexStream();
RectTransform rectT = rectTransform;
stream.RectTransform = rectT;
Rect r = GetPixelAdjustedRect();
stream.Uv1 = new Vector2(r.width + m_FalloffDistance, r.height + m_FalloffDistance);
float packedRotData =
PackRotationData(m_ShapeRotation, m_ConstrainRotation, m_FlipHorizontal, m_FlipVertical);
stream.Uv3 = new Vector2(packedRotData, (float)m_CornerStyle);
stream.Tangent = QualitySettings.activeColorSpace == ColorSpace.Linear? m_OutlineColor.linear : m_OutlineColor;
Vector3 normal = new Vector3();
normal.x = m_OutlineWidth;
normal.y = m_StrokeWidth;
normal.z = m_FalloffDistance;
Vector4 data;
Vector2 shapeProps;
switch (m_DrawShape) {
case DrawShape.Circle:
shapeProps = new Vector2(m_CircleRadius, m_CircleFitToRect ? 1 : 0);
break;
case DrawShape.Triangle:
data = m_TriangleCornerRadius;
data = data / Mathf.Min(r.width, r.height);
shapeProps = MPImageUtility.Encode_0_1_16(data);
break;
case DrawShape.Rectangle:
data = FixRadius(m_RectangleCornerRadius);
data = data / Mathf.Min(r.width, r.height);
shapeProps = MPImageUtility.Encode_0_1_16(data);
break;
case DrawShape.NStarPolygon:
data = new Vector4(m_NStarPolygonSideCount, m_NStarPolygonCornerRadius, m_NStarPolygonInset);
data = data / Mathf.Min(r.width, r.height);
shapeProps = MPImageUtility.Encode_0_1_16(data);
break;
default:
shapeProps = Vector2.zero;
break;
}
stream.Uv2 = shapeProps;
stream.Normal = normal;
return stream;
}
private float PackRotationData(float rotation, bool constrainRotation, bool flipH, bool flipV) {
int c = constrainRotation ? 1 : 0;
c += flipH ? 10 : 0;
c += flipV ? 100 : 0;
float cr = rotation % 360f;
float sign = cr >= 0 ? 1 : -1;
cr = Mathf.Abs(cr) / 360f;
cr = (cr + c) * sign;
return cr;
}
void UnPackRotation(float f) {
float r = 0, x = 0, y = 0, z = 0;
float sign = f >= 0.0f ? 1 : -1;
f = Mathf.Abs(f);
r = fract(f) * 360f * sign;
f = Mathf.Floor(f);
float p = f / 100f;
z = Mathf.Floor(p);
p = fract(p) * 10f;
y = Mathf.Floor(p);
p = fract(p) * 10f;
x = Mathf.Round(p);
// Debug.Log($"Rotation: {r}, X: {x}, Y: {y}, Z: {z}");
float fract(float val) {
val = Mathf.Abs(val);
float ret = val - Mathf.Floor(val);
return ret;
}
}
protected override void OnRectTransformDimensionsChange()
{
base.OnRectTransformDimensionsChange();
base.SetVerticesDirty();
}
private Vector4 FixRadius(Vector4 radius)
{
Rect rect = rectTransform.rect;
radius = Vector4.Max(radius, Vector4.zero);
radius = Vector4.Min(radius, Vector4.one * Mathf.Min(rect.width, rect.height));
float scaleFactor =
Mathf.Min (
Mathf.Min (
Mathf.Min (
Mathf.Min (
rect.width / (radius.x + radius.y),
rect.width / (radius.z + radius.w)),
rect.height / (radius.x + radius.w)),
rect.height / (radius.z + radius.y)),
1f);
return radius * scaleFactor;
}
protected override void OnPopulateMesh(VertexHelper toFill) {
base.OnPopulateMesh(toFill);
MPVertexStream stream = CreateVertexStream();
UIVertex uiVert = new UIVertex();
for (int i = 0; i < toFill.currentVertCount; i++) {
toFill.PopulateUIVertex(ref uiVert, i);
//uiVert.position += ((Vector3)uiVert.uv0 - new Vector3(0.5f, 0.5f)) * m_FalloffDistance;
uiVert.uv1 = stream.Uv1;
uiVert.uv2 = stream.Uv2;
uiVert.uv3 = stream.Uv3;
uiVert.normal = stream.Normal;
uiVert.tangent = stream.Tangent;
toFill.SetUIVertex(uiVert, i);
}
}
#if UNITY_EDITOR
protected override void Reset() {
base.Reset();
if (sprite == null) sprite = MPImageUtility.EmptySprite;
}
#else
void Reset() {
if (sprite == null) sprite = MPImageUtility.EmptySprite;
}
#endif
}
}