UP-Viagg-io/Viagg-io/Assets/AVProVideo/Runtime/Scripts/Components/ApplyToMesh.cs

using System;
using UnityEngine;
using UnityEngine.Serialization;

//-----------------------------------------------------------------------------
// Copyright 2015-2022 RenderHeads Ltd.  All rights reserved.
//-----------------------------------------------------------------------------

namespace RenderHeads.Media.AVProVideo
{
	/// <summary>
	/// Sets up a mesh to display the video from a MediaPlayer
	/// </summary>
	[AddComponentMenu("AVPro Video/Apply To Mesh", 300)]
	[HelpURL("https://www.renderheads.com/products/avpro-video/")]
	public sealed class ApplyToMesh : ApplyToBase
	{
		// TODO: add specific material / material index to target in the mesh if there are multiple materials

		[Space(8f)]
		[Header("Display")]

		[Tooltip("Default texture to display when the video texture is preparing")]
		[SerializeField] Texture2D _defaultTexture = null;

		public Texture2D DefaultTexture
		{
			get
			{
				return _defaultTexture;
			}
			set
			{
				ChangeDefaultTexture(value);
			}
		}

		[Space(8f)]
		[FormerlySerializedAs("_mesh")]
		[Header("Renderer Target")]
		[SerializeField] Renderer _renderer = null;

		public Renderer MeshRenderer
		{
			get
			{
				return _renderer;
			}
			set
			{
				ChangeRenderer(value);
			}
		}

		[SerializeField]
		int _materialIndex = -1;

		public int MaterialIndex
		{
			get
			{
				return _materialIndex;
			}
			set
			{
				_materialIndex = value;
			}
		}

		private void ChangeDefaultTexture(Texture2D texture)
		{
			if (_defaultTexture != texture)
			{
				_defaultTexture = texture;
				ForceUpdate();
			}
		}
		
		private void ChangeRenderer(Renderer renderer)
		{
			if (_renderer != renderer)
			{
				if (_renderer)
				{
					// TODO: Remove from renderer
				}
				_renderer = renderer;
				if (_renderer)
				{
					ForceUpdate();
				}
			}
		}

		[SerializeField]
		string _texturePropertyName = Helper.UnityBaseTextureName;

		public string TexturePropertyName
		{
			get
			{
				return _texturePropertyName;
			}
			set
			{
				if (_texturePropertyName != value)
				{
					_texturePropertyName = value;
					_propTexture = new LazyShaderProperty(_texturePropertyName);
					_propTexture_R = new LazyShaderProperty(_texturePropertyName + "_R");
					_isDirty = true;
				}
			}
		}

		[SerializeField]
		Vector2 _offset = Vector2.zero;

		public Vector2 Offset
		{
			get
			{
				return _offset;
			}
			set
			{
				if (_offset != value)
				{
					_offset = value;
					_isDirty = true;
				}
			}
		}

		[SerializeField]
		Vector2 _scale = Vector2.one;

		public Vector2 Scale
		{
			get
			{
				return _scale;
			}
			set
			{
				if (_scale != value)
				{
					_scale = value;
					_isDirty = true;
				}
			}
		}

		private Texture _lastTextureApplied;
		private LazyShaderProperty _propTexture;
		private LazyShaderProperty _propTexture_R;	// Default property for the right-eye texture

		// We do a LateUpdate() to allow for any changes in the texture that may have happened in Update()
		private void LateUpdate()
		{
			Apply();
		}

		public override void Apply()
		{
			bool applied = false;

			// Try to apply texture from media
			if (_media != null && _media.TextureProducer != null)
			{
				Texture resamplerTex = _media.FrameResampler == null || _media.FrameResampler.OutputTexture == null ? null : _media.FrameResampler.OutputTexture[0];
				Texture texture = _media.UseResampler ? resamplerTex : _media.TextureProducer.GetTexture(0);
				if (texture != null)
				{
					// Check for changing texture
					if (texture != _lastTextureApplied)
					{
						_isDirty = true;
					}

					if (_isDirty)
					{
						bool requiresVerticalFlip = _media.TextureProducer.RequiresVerticalFlip();
						StereoPacking stereoPacking = _media.TextureProducer.GetTextureStereoPacking();

						int planeCount = 1;
						if (!_media.UseResampler)
						{
							// We're not using the resampler so the number of planes will be the texture count
							planeCount = _media.TextureProducer.GetTextureCount();
							if (stereoPacking == StereoPacking.TwoTextures)
							{
								// Unless we're using two texture stereo in which case it'll be half the texture count
								planeCount /= 2;
							}
						}

						for (int plane = 0; plane < planeCount; plane++)
						{
							Texture resamplerTexPlane = _media.FrameResampler == null || _media.FrameResampler.OutputTexture == null ? null : _media.FrameResampler.OutputTexture[plane];
							texture = _media.UseResampler ? resamplerTexPlane : _media.TextureProducer.GetTexture(plane);
							if (texture != null)
							{
								ApplyMapping(texture, _media.TextureProducer.RequiresVerticalFlip(), plane, materialIndex: _materialIndex);
							}
						}

						// Handle the right eye if we're using two texture stereo packing
						if (stereoPacking == StereoPacking.TwoTextures)
						{
							for (int plane = 0; plane < planeCount; ++plane)
							{
								texture = _media.TextureProducer.GetTexture(planeCount + plane);
								if (texture != null)
								{
									ApplyMapping(texture, requiresVerticalFlip, plane, Eye.Right);
								}
							}
						}
					}
					applied = true;
				}
			}

			// If the media didn't apply a texture, then try to apply the default texture
			if (!applied)
			{
				if (_defaultTexture != _lastTextureApplied)
				{
					_isDirty = true;
				}
				if (_isDirty)
				{
					ApplyMapping(_defaultTexture, false, 0, materialIndex: _materialIndex);
				}
			}
		}

		enum Eye
		{
			Left,
			Right
		}

		private void ApplyMapping(Texture texture, bool requiresYFlip, int plane, Eye eye = Eye.Left, int materialIndex = -1)
		{
			if (_renderer != null)
			{
				_isDirty = false;
#if UNITY_EDITOR
                Material[] meshMaterials = _renderer.sharedMaterials;
#else
				Material[] meshMaterials = _renderer.materials;
#endif

                if (meshMaterials != null)
				{
					for (int i = 0; i < meshMaterials.Length; i++)
					{
						if (_materialIndex < 0 || i == _materialIndex)
						{
							Material mat = meshMaterials[i];
							if (mat != null)
							{
								if (StereoRedGreenTint)
								{
									mat.EnableKeyword("STEREO_DEBUG");
								}
								else
								{
									mat.DisableKeyword("STEREO_DEBUG");
								}
								
								if (plane == 0)
								{
									int propTextureId = _propTexture.Id;
									if (eye == Eye.Left)
									{
										VideoRender.SetupMaterialForMedia(mat, _media, _propTexture.Id, texture, texture == _defaultTexture);
										_lastTextureApplied = texture;

#if !UNITY_EDITOR && UNITY_ANDROID
											if (texture == _defaultTexture)
											{
												mat.EnableKeyword("USING_DEFAULT_TEXTURE");
											}
											else
											{
												mat.DisableKeyword("USING_DEFAULT_TEXTURE");
											}
#endif
									}
									else
									{
										propTextureId = _propTexture_R.Id;
										mat.SetTexture(propTextureId, texture);
									}
									
									if (texture != null)
									{
										if (requiresYFlip)
										{
											mat.SetTextureScale(_propTexture.Id, new Vector2(_scale.x, -_scale.y));
											mat.SetTextureOffset(_propTexture.Id, Vector2.up + _offset);
										}
										else
										{
											mat.SetTextureScale(_propTexture.Id, _scale);
											mat.SetTextureOffset(_propTexture.Id, _offset);
										}
									}
								}
								else if (plane == 1)
								{
									if (texture != null)
									{
										if (requiresYFlip)
										{
											mat.SetTextureScale(VideoRender.PropChromaTex.Id, new Vector2(_scale.x, -_scale.y));
											mat.SetTextureOffset(VideoRender.PropChromaTex.Id, Vector2.up + _offset);
										}
										else
										{
											mat.SetTextureScale(VideoRender.PropChromaTex.Id, _scale);
											mat.SetTextureOffset(VideoRender.PropChromaTex.Id, _offset);
										}
									}
								}
							}
						}
					}
				}
			}
		}

		protected override void OnEnable()
		{
			if (_renderer == null)
			{
				_renderer = this.GetComponent<MeshRenderer>();
				if (_renderer == null)
				{
					Debug.LogWarning("[AVProVideo] No MeshRenderer set or found in gameobject");
				}
			}

			_propTexture = new LazyShaderProperty(_texturePropertyName);

			ForceUpdate();
		}

		protected override void OnDisable()
		{
			ApplyMapping(_defaultTexture, false, 0, materialIndex: _materialIndex);
		}

		protected override void SaveProperties()
		{
			_propTexture = new LazyShaderProperty(_texturePropertyName);
			_propTexture_R = new LazyShaderProperty(_texturePropertyName + "_R");
		}
	}
}