ZHGD_Web/Assets/UniversalMediaPlayer/Scripts/Sources/Players/MediaPlayerHelper.cs

using System.IO;
using System.Text.RegularExpressions;
using UnityEngine;
using UnityEngine.Networking;
using UnityEngine.UI;

namespace UMP
{
    public class MediaPlayerHelper
    {
        private const string UMP_FOLDER_NAME = "/UniversalMediaPlayer";
        private const string LOCAL_FILE_ROOT = "file:///";

        private delegate void ManageLogCallback(string msg, ManageLogLevel level);
        private ManageLogCallback _manageLogCallback;

        private static Regex _androidStorageRoot = new Regex("(^\\/.*)Android");

        private enum ManageLogLevel
        {
            DEBUG = 0,
            WARNING = 1,
            ERROR = 2
        };

        private void DebugLogHandler(string msg, ManageLogLevel level)
        {
            Debug.LogError(msg);
        }

        /// <summary>
        /// Apply texture to Unity game objects that has 'RawImage' or 'MeshRenderer' component
        /// </summary>
        /// <param name="texture">Texture to render video output</param>
        /// <param name="renderingObjects">Game objects where will be rendering video output</param>
        /// <returns></returns>
        public static void ApplyTextureToRenderingObjects(Texture2D texture, GameObject[] renderingObjects)
        {
            if (renderingObjects == null)
                return;

            foreach (var gameObject in renderingObjects)
            {
                if (gameObject == null)
                    continue;

                var rawImage = gameObject.GetComponent<RawImage>();

                if (rawImage != null)
                {
                    rawImage.texture = texture;
                    continue;
                }

                var meshRenderer = gameObject.GetComponent<MeshRenderer>();

                if (meshRenderer != null && meshRenderer.material != null)
                    meshRenderer.material.mainTexture = texture;
                else
                    Debug.LogError(gameObject.name + ": don't have 'RawImage' or 'MeshRenderer' component - ignored");
            }
        }

        /// <summary>
        /// Generate correct texture for current runtime platform
        /// </summary>
        /// <param name="width">Width in pixels</param>
        /// <param name="height">Height in pixels</param>
        /// <returns></returns>
        public static Texture2D GenVideoTexture(int width, int height)
        {
            if (UMPSettings.RuntimePlatform == UMPSettings.Platforms.Win ||
                UMPSettings.RuntimePlatform == UMPSettings.Platforms.Mac ||
                UMPSettings.RuntimePlatform == UMPSettings.Platforms.Linux)
                return new Texture2D(width, height, TextureFormat.BGRA32, false);

            return new Texture2D(width, height, TextureFormat.RGBA32, false);
        }

        /// <summary>
        /// Generate correct texture for native plugin
        /// </summary>
        /// <param name="width">Width in pixels</param>
        /// <param name="height">Height in pixels</param>
        /// <returns></returns>
        internal static Texture2D GenPluginTexture(int width, int height)
        {
            if (UMPSettings.RuntimePlatform == UMPSettings.Platforms.Win ||
                UMPSettings.RuntimePlatform == UMPSettings.Platforms.Mac ||
                UMPSettings.RuntimePlatform == UMPSettings.Platforms.Linux)
                return new Texture2D(width, height, TextureFormat.BGRA32, false);

            return new Texture2D(width, height, TextureFormat.RGBA32, false);
        }

        /// <summary>
        /// Getting average color from frame buffer array
        /// </summary>
        /// <returns></returns>
        public static Color GetAverageColor(byte[] frameBuffer)
        {
            if (frameBuffer == null)
                return Color.black;

            long redBucket = 0;
            long greenBucket = 0;
            long blueBucket = 0;
            long alphaBucket = 0;
            int pixelCount = frameBuffer.Length / 4;

            if (pixelCount <= 0 || pixelCount % 4 != 0)
                return Color.black;

            for (int x = 0; x < frameBuffer.Length; x+=4)
            {
                redBucket += frameBuffer[x];
                greenBucket += frameBuffer[x + 1];
                blueBucket += frameBuffer[x + 2];
                alphaBucket += frameBuffer[x + 3];
            }

            return new Color(redBucket / pixelCount, 
                greenBucket / pixelCount, 
                blueBucket / pixelCount, 
                alphaBucket / pixelCount);
        }

        /// <summary>
        /// Getting colors from frame buffer array
        /// </summary>
        /// <returns></returns>
        public static Color32[] GetFrameColors(byte[] frameBuffer)
        {
            var colorsArray = new Color32[frameBuffer.Length / 4];
            for (var i = 0; i < frameBuffer.Length; i += 4)
            {
                var color = new Color32(frameBuffer[i + 2], frameBuffer[i + 1], frameBuffer[i + 0], frameBuffer[i + 3]);
                colorsArray[i / 4] = color;
            }
            return colorsArray;
        }

        /// <summary>
        /// Getting root storage menory for current platform
        /// Windows, Mac OS, Linux will return path to project folder
        /// Android, iOS will return root of internal/external memory root
        /// </summary>
        /// <returns></returns>
        public static string GetDeviceRootPath()
        {
            Match match = _androidStorageRoot.Match(Application.persistentDataPath);

            if (match.Length > 1)
                return match.Groups[1].Value;

            return Application.persistentDataPath;
        }

        /// <summary>
        /// Check if file exists in 'StreamingAssets' folder
        /// </summary>
        /// <param name="filePath">File name or path to file in 'StreamingAssets' folder</param>
        /// <returns></returns>
        public static bool IsAssetsFile(string filePath)
        {
            if (filePath.StartsWith(LOCAL_FILE_ROOT))
                filePath = filePath.Substring(LOCAL_FILE_ROOT.Length);

            if (!filePath.Contains(Application.streamingAssetsPath))
                filePath = Path.Combine(Application.streamingAssetsPath, filePath);

            if (UMPSettings.RuntimePlatform != UMPSettings.Platforms.Android)
            {
                return File.Exists(filePath);
            }
            else
            {
#if UNITY_2017_2_OR_NEWER
                var www = UnityWebRequest.Get(filePath);
                www.SendWebRequest();
                while (!www.isDone && www.downloadProgress <= 0) { }
#else
                var www = new WWW(filePath);
                while (!www.isDone && www.progress <= 0) { }
#endif
                bool result = string.IsNullOrEmpty(www.error);
                www.Dispose();

                return result;
            }
        }

        /// <summary>
        /// Get correct madia data source path for file path
        /// </summary>
        /// <param name="relativePath">Relative path to the data file</param>
        /// <returns></returns>
        public static string GetDataSourcePath(string relativePath)
        {
            if (relativePath.StartsWith(LOCAL_FILE_ROOT))
            {
                relativePath = relativePath.Substring(LOCAL_FILE_ROOT.Length);

                if (IsAssetsFile(relativePath))
                    relativePath = Path.Combine(Application.streamingAssetsPath, relativePath);
            }

            if (UMPSettings.RuntimePlatform == UMPSettings.Platforms.Android)
            {
                var pathDR = GetDeviceRootPath() + relativePath;

                if (File.Exists(pathDR))
                    relativePath = pathDR;
            }

            if (File.Exists(relativePath))
                relativePath = LOCAL_FILE_ROOT + relativePath;

            return relativePath;
        }
    }
}