#if UNITY_5_4_OR_NEWER || (UNITY_5 && !UNITY_5_0 && !UNITY_5_1)
	#define AVPRO_MOVIECAPTURE_GLISSUEEVENT_52
#endif
#if UNITY_5_4_OR_NEWER || UNITY_5
	#define AVPRO_MOVIECAPTURE_DEFERREDSHADING
#endif
using UnityEngine;
using System.IO;
using System;
using System.Runtime.InteropServices;

//-----------------------------------------------------------------------------
// Copyright 2012-2017 RenderHeads Ltd.  All rights reserved.
//-----------------------------------------------------------------------------

namespace RenderHeads.Media.AVProMovieCapture
{
	/// <summary>
	/// Base class wrapping common capture functionality
	/// </summary>
	public class CaptureBase : MonoBehaviour
	{
		public enum FrameRate
		{
			One = 1,
			Ten = 10,
			Fifteen = 15,
			TwentyFour = 24,
			TwentyFive = 25,
			Thirty = 30,
			Fifty = 50,
			Sixty = 60,
			SeventyFive = 75,
			Ninety = 90,
			OneTwenty = 120,
		}

		public enum Resolution
		{
			POW2_8192x8192,
			POW2_8192x4096,
			POW2_4096x4096,
			POW2_4096x2048,
			POW2_2048x4096,
			UHD_3840x2160,
			UHD_3840x2048,
			UHD_3840x1920,
			POW2_2048x2048,
			POW2_2048x1024,
			HD_1920x1080,
			HD_1280x720,
			SD_1024x768,
			SD_800x600,
			SD_800x450,
			SD_640x480,
			SD_640x360,
			SD_320x240,
			Original,
			Custom,
		}

		public enum CubemapDepth
		{
			Depth_24 = 24,
			Depth_16 = 16,
			Depth_Zero = 0,
		}

		public enum CubemapResolution
		{
			POW2_4096 = 4096,
			POW2_2048 = 2048,
			POW2_1024 = 1024,
			POW2_512 = 512,
			POW2_256 = 256,
		}

		public enum AntiAliasingLevel
		{
			UseCurrent,
			ForceNone,
			ForceSample2,
			ForceSample4,
			ForceSample8,
		}

		public enum DownScale
		{
			Original = 1,
			Half = 2,
			Quarter = 4,
			Eighth = 8,
			Sixteenth = 16,
			Custom = 100,
		}

		public enum OutputPath
		{
			RelativeToProject,
			RelativeToPeristentData,
			Absolute,
		}

		public enum OutputExtension
		{
			AVI,
			MP4,
			PNG,
			Custom = 100,
		}

		public enum OutputType
		{
			VideoFile,
			NamedPipe,
		}

		// General options

		public KeyCode _captureKey = KeyCode.None;
		public bool _captureOnStart = false;
		public bool _startPaused = false;
		public bool _listVideoCodecsOnStart = false;
		public bool _isRealTime = true;

		// Stop options

		public StopMode _stopMode = StopMode.None;
		// TODO: add option to pause instread of stop?
		public int _stopFrames = 0;
		public float _stopSeconds = 0f;

		// Video options

		public bool _useMediaFoundationH264 = false;
		public string[] _videoCodecPriority = { "Lagarith Lossless Codec",
											"x264vfw - H.264/MPEG-4 AVC codec",
											"Xvid MPEG-4 Codec" };
		public FrameRate _frameRate = FrameRate.Thirty;
		public DownScale _downScale = DownScale.Original;
		public Vector2 _maxVideoSize = Vector2.zero;
		public int _forceVideoCodecIndex = -1;
		public bool _flipVertically = false;
		public bool _supportAlpha = false;

		// Audio options

		public bool _noAudio = true;
		public string[] _audioCodecPriority = { };
		public int _forceAudioCodecIndex = -1;
		public int _forceAudioDeviceIndex = -1;
		public UnityAudioCapture _audioCapture;

		// Output options

		public bool _autoGenerateFilename = true;
		public OutputPath _outputFolderType = OutputPath.RelativeToProject;
		public string _outputFolderPath;
		public string _autoFilenamePrefix = "MovieCapture";
		public string _autoFilenameExtension = "avi";
		public string _forceFilename = "movie.avi";
		public OutputType _outputType = OutputType.VideoFile;

		// Camera specific options

		public Resolution _renderResolution = Resolution.Original;
		public Vector2 _renderSize = Vector2.one;
		public int _renderAntiAliasing = -1;

		// Motion blur options

		public bool _useMotionBlur = false;
		[Range(0, 64)]
		public int _motionBlurSamples = 16;
		public Camera[] _motionBlurCameras;
		protected MotionBlur _motionBlur;

		// Performance options

		public bool _allowVSyncDisable = true;

		[SerializeField]
		protected bool _supportTextureRecreate = false;

		//public bool _allowFrameRateChange = true;

		// Cursor options
		public bool _captureMouseCursor = false;
		public MouseCursor _mouseCursor;

		[System.NonSerialized]
		public string _codecName = "uncompressed";
		[System.NonSerialized]
		public int _codecIndex = -1;

		[System.NonSerialized]
		public string _audioCodecName = "uncompressed";
		[System.NonSerialized]
		public int _audioCodecIndex = -1;

		[System.NonSerialized]
		public string _audioDeviceName = "Unity";
		[System.NonSerialized]
		public int _audioDeviceIndex = -1;

		[System.NonSerialized]
		public int _unityAudioSampleRate = -1;
		[System.NonSerialized]
		public int _unityAudioChannelCount = -1;

		protected Texture2D _texture;
		protected int _handle = -1;
		protected int _targetWidth, _targetHeight;
		protected bool _capturing = false;
		protected bool _paused = false;
		protected string _filePath;
		protected FileInfo _fileInfo;
		protected NativePlugin.PixelFormat _pixelFormat = NativePlugin.PixelFormat.YCbCr422_YUY2;
		private int _oldVSyncCount = 0;
		//private int _oldTargetFrameRate = -1;
		private float _oldFixedDeltaTime = 0f;
		protected bool _isTopDown = true;
		protected bool _isDirectX11 = false;
		private bool _queuedStartCapture = false;
		private bool _queuedStopCapture = false;
		private float _captureStartTime = 0f;
		private float _timeSinceLastFrame = 0f;

		public string LastFilePath
		{
			get { return _filePath; }
		}

		// Other options
		public int _minimumDiskSpaceMB = -1;
		private long _freeDiskSpaceMB;

		// Stats
		private uint _numDroppedFrames;
		private uint _numDroppedEncoderFrames;
		private uint _numEncodedFrames;
		private uint _totalEncodedSeconds;

#if AVPRO_MOVIECAPTURE_GLISSUEEVENT_52
		protected System.IntPtr _renderEventFunction = System.IntPtr.Zero;
		protected System.IntPtr _freeEventFunction = System.IntPtr.Zero;
#endif

		public uint NumDroppedFrames
		{
			get { return _numDroppedFrames; }
		}

		public uint NumDroppedEncoderFrames
		{
			get { return _numDroppedEncoderFrames; }
		}

		public uint NumEncodedFrames
		{
			get { return _numEncodedFrames; }
		}

		public uint TotalEncodedSeconds
		{
			get { return _totalEncodedSeconds; }
		}

		protected virtual void Awake()
		{
			try
			{
				float pluginVersion = NativePlugin.GetPluginVersion();
				bool isTrial = NativePlugin.IsTrialVersion();
				string pluginVersionString = pluginVersion.ToString("F2");
				if (isTrial)
				{
					pluginVersionString += "t";
				}

				// Check that the plugin version number is not too old
				if (pluginVersion.ToString("F2") != NativePlugin.ExpectedPluginVersion)
				{
					Debug.LogWarning("[AVProMovieCapture] Plugin version number " + pluginVersionString + " doesn't match the expected version number " + NativePlugin.ExpectedPluginVersion + ".  It looks like the plugin didn't upgrade correctly.  To resolve this please restart Unity and try to upgrade the package again.");
				}

				if (NativePlugin.Init())
				{
					Debug.Log("[AVProMovieCapture] Init plugin version: " + pluginVersionString + " (script v" + NativePlugin.ScriptVersion +") with GPU " + SystemInfo.graphicsDeviceName + " " + SystemInfo.graphicsDeviceVersion);

#if AVPRO_MOVIECAPTURE_GLISSUEEVENT_52
					_renderEventFunction = NativePlugin.GetRenderEventFunc();
					_freeEventFunction = NativePlugin.GetFreeResourcesEventFunc();
					Debug.Assert(_renderEventFunction != System.IntPtr.Zero);
					Debug.Assert(_freeEventFunction != System.IntPtr.Zero);
#endif
				}
				else
				{
					Debug.LogError("[AVProMovieCapture] Failed to initialise plugin version: " + pluginVersionString + " (script v" + NativePlugin.ScriptVersion + ") with GPU " + SystemInfo.graphicsDeviceName + " " + SystemInfo.graphicsDeviceVersion);
				}
			}
			catch (DllNotFoundException e)
			{
				string missingDllMessage = string.Empty;
#if (UNITY_5 || UNITY_5_4_OR_NEWER)
				missingDllMessage = "Unity couldn't find the plugin DLL. Please select the native plugin files in 'Plugins/RenderHeads/AVProMovieCapture/Plugins' folder and select the correct platform in the Inspector.";
#else
				missingDllMessage = "Unity couldn't find the plugin DLL, Unity 4.x requires the 'Plugins' folder to be at the root of your project.  Please move the contents of the 'Plugins' folder (in Plugins/RenderHeads/AVProMovieCapture/Plugins) to the 'Plugins' folder in the root of your project.";
#endif
				Debug.LogError("[AVProMovieCapture] " + missingDllMessage);
#if UNITY_EDITOR
				UnityEditor.EditorUtility.DisplayDialog("Plugin files not found", missingDllMessage, "Ok");
#endif
				throw e;
			}

			_isDirectX11 = SystemInfo.graphicsDeviceVersion.StartsWith("Direct3D 11");

			SelectCodec(_listVideoCodecsOnStart);
			SelectAudioCodec(_listVideoCodecsOnStart);
			SelectAudioDevice(_listVideoCodecsOnStart);
		}

		public virtual void Start()
		{
			Application.runInBackground = true;

			if (_captureOnStart)
			{
				StartCapture();
			}
		}

		public void SelectCodec(bool listCodecs)
		{
			// Enumerate video codecs
			int numVideoCodecs = NativePlugin.GetNumAVIVideoCodecs();
			if (listCodecs)
			{
				for (int i = 0; i < numVideoCodecs; i++)
				{
					Debug.Log("VideoCodec " + i + ": " + NativePlugin.GetAVIVideoCodecName(i));
				}
			}

			_codecIndex = -1;

			if (_useMediaFoundationH264)
			{
				_codecName = "Media Foundation H.264(MP4)";
			}
			else
			{
				// The user has specified their own codec index
				if (_forceVideoCodecIndex >= 0)
				{
					if (_forceVideoCodecIndex < numVideoCodecs)
					{
						_codecName = NativePlugin.GetAVIVideoCodecName(_forceVideoCodecIndex);
						_codecIndex = _forceVideoCodecIndex;
					}

					if (_codecIndex < 0)
					{
						_codecName = "Uncompressed";
						Debug.LogWarning("[AVProMovieCapture] Video codec not found.  Video will be uncompressed.");
					}
				}
				else
				{
					// Try to find the codec based on the priority list
					if (_videoCodecPriority != null && _videoCodecPriority.Length > 0)
					{
						foreach (string codec in _videoCodecPriority)
						{
							string codecName = codec.Trim();
							// Empty string means uncompressed
							if (string.IsNullOrEmpty(codecName))
								break;

							for (int i = 0; i < numVideoCodecs; i++)
							{
								if (codecName == NativePlugin.GetAVIVideoCodecName(i))
								{
									_codecName = codecName;
									_codecIndex = i;
									break;
								}
							}

							if (_codecIndex >= 0)
								break;
						}

						if (_codecIndex < 0)
						{
							_codecName = "Uncompressed";
							Debug.LogWarning("[AVProMovieCapture] Video codec not found.  Video will be uncompressed.");
						}
					}
				}
			}
		}

		public void SelectAudioCodec(bool listCodecs)
		{
			// Enumerate audio codecs
			int numAudioCodecs = NativePlugin.GetNumAVIAudioCodecs();
			if (listCodecs)
			{
				for (int i = 0; i < numAudioCodecs; i++)
				{
					Debug.Log("AudioCodec " + i + ": " + NativePlugin.GetAVIAudioCodecName(i));
				}
			}

			_audioCodecIndex = -1;

			// The user has specified their own codec index
			if (_forceAudioCodecIndex >= 0)
			{
				if (_forceAudioCodecIndex < numAudioCodecs)
				{
					_audioCodecName = NativePlugin.GetAVIAudioCodecName(_forceAudioCodecIndex);
					_audioCodecIndex = _forceAudioCodecIndex;
				}

				if (_audioCodecIndex < 0)
				{
					_audioCodecName = "Uncompressed";
					Debug.LogWarning("[AVProMovieCapture] Audio codec not found.  Audio will be uncompressed.");
				}
			}
			else
			{
				// Try to find the codec based on the priority list
				if (_audioCodecPriority != null && _audioCodecPriority.Length > 0)
				{
					foreach (string codec in _audioCodecPriority)
					{
						string codecName = codec.Trim();
						// Empty string means uncompressed
						if (string.IsNullOrEmpty(codecName))
							break;

						for (int i = 0; i < numAudioCodecs; i++)
						{
							if (codecName == NativePlugin.GetAVIAudioCodecName(i))
							{
								_audioCodecName = codecName;
								_audioCodecIndex = i;
								break;
							}
						}

						if (_audioCodecIndex >= 0)
							break;
					}

					if (_audioCodecIndex < 0)
					{
						_audioCodecName = "Uncompressed";
						Debug.LogWarning("[AVProMovieCapture] Codec not found.  Audio will be uncompressed.");
					}
				}
			}
		}

		public void SelectAudioDevice(bool display)
		{
			// Enumerate
			int num = NativePlugin.GetNumAVIAudioInputDevices();
			if (display)
			{
				for (int i = 0; i < num; i++)
				{
					Debug.Log("AudioDevice " + i + ": " + NativePlugin.GetAVIAudioInputDeviceName(i));
				}
			}

			// The user has specified their own device index
			if (_forceAudioDeviceIndex >= 0)
			{
				if (_forceAudioDeviceIndex < num)
				{
					_audioDeviceName = NativePlugin.GetAVIAudioInputDeviceName(_forceAudioDeviceIndex);
					_audioDeviceIndex = _forceAudioDeviceIndex;
				}
			}
			else
			{
				/*_audioDeviceIndex = -1;
				// Try to find one of the loopback devices
				for (int i = 0; i < num; i++)
				{
					StringBuilder sbName = new StringBuilder(512);
					if (AVProMovieCapturePlugin.GetAVIAudioInputDeviceName(i, sbName))
					{
						string[] loopbackNames = { "Stereo Mix", "What U Hear", "What You Hear", "Waveout Mix", "Mixed Output" };
						for (int j = 0; j < loopbackNames.Length; j++)
						{
							if (sbName.ToString().Contains(loopbackNames[j]))
							{
								_audioDeviceIndex = i;
								_audioDeviceName = sbName.ToString();
							}
						}
					}
					if (_audioDeviceIndex >= 0)
						break;
				}

				if (_audioDeviceIndex < 0)
				{
					// Resort to the no recording device
					_audioDeviceName = "Unity";
					_audioDeviceIndex = -1;
				}*/

				_audioDeviceName = "Unity";
				_audioDeviceIndex = -1;
			}
		}

		public static Vector2 GetRecordingResolution(int width, int height, DownScale downscale, Vector2 maxVideoSize)
		{
			int targetWidth = width;
			int targetHeight = height;
			if (downscale != DownScale.Custom)
			{
				targetWidth /= (int)downscale;
				targetHeight /= (int)downscale;
			}
			else
			{
				if (maxVideoSize.x >= 1.0f && maxVideoSize.y >= 1.0f)
				{
					targetWidth = Mathf.FloorToInt(maxVideoSize.x);
					targetHeight = Mathf.FloorToInt(maxVideoSize.y);
				}
			}

			// Some codecs like Lagarith in YUY2 mode need size to be multiple of 4
			targetWidth = NextMultipleOf4(targetWidth);
			targetHeight = NextMultipleOf4(targetHeight);

			return new Vector2(targetWidth, targetHeight);
		}

		public void SelectRecordingResolution(int width, int height)
		{
			_targetWidth = width;
			_targetHeight = height;
			if (_downScale != DownScale.Custom)
			{
				_targetWidth /= (int)_downScale;
				_targetHeight /= (int)_downScale;
			}
			else
			{
				if (_maxVideoSize.x >= 1.0f && _maxVideoSize.y >= 1.0f)
				{
					_targetWidth = Mathf.FloorToInt(_maxVideoSize.x);
					_targetHeight = Mathf.FloorToInt(_maxVideoSize.y);
				}
			}

			// Some codecs like Lagarith in YUY2 mode need size to be multiple of 4
			_targetWidth = NextMultipleOf4(_targetWidth);
			_targetHeight = NextMultipleOf4(_targetHeight);
		}

		public virtual void OnDestroy()
		{
			StopCapture();
			NativePlugin.Deinit();
		}

		private void OnApplicationQuit()
		{
			StopCapture();
			NativePlugin.Deinit();
		}

		protected void EncodeTexture(Texture2D texture)
		{
			Color32[] bytes = texture.GetPixels32();
			GCHandle _frameHandle = GCHandle.Alloc(bytes, GCHandleType.Pinned);

			EncodePointer(_frameHandle.AddrOfPinnedObject());

			if (_frameHandle.IsAllocated)
			{
				_frameHandle.Free();
			}
		}

		protected bool IsUsingUnityAudio()
		{
			return (_isRealTime && !_noAudio && _audioDeviceIndex < 0 && _audioCapture != null);
		}

		protected bool IsRecordingUnityAudio()
		{
			return (IsUsingUnityAudio() && _audioCapture.isActiveAndEnabled);
		}

		protected bool IsUsingMotionBlur()
		{
			return (_useMotionBlur && !_isRealTime && _motionBlur != null);
		}

		public virtual void EncodePointer(System.IntPtr ptr)
		{
			if (!IsRecordingUnityAudio())
			{
				NativePlugin.EncodeFrame(_handle, ptr);
			}
			else
			{
				int audioDataLength = 0;
				System.IntPtr audioDataPtr = _audioCapture.ReadData(out audioDataLength);
				if (audioDataLength > 0)
				{
					NativePlugin.EncodeFrameWithAudio(_handle, ptr, audioDataPtr, (uint)audioDataLength);
				}
				else
				{
					NativePlugin.EncodeFrame(_handle, ptr);
				}
			}
		}

		public bool IsCapturing()
		{
			return _capturing;
		}

		public bool IsPaused()
		{
			return _paused;
		}

		public int GetRecordingWidth()
		{
			return _targetWidth;
		}

		public int GetRecordingHeight()
		{
			return _targetHeight;
		}

		protected virtual string GenerateTimestampedFilename(string filenamePrefix, string filenameExtension)
		{
			TimeSpan span = (DateTime.Now - DateTime.Now.Date);
			return string.Format("{0}-{1}-{2}-{3}-{4}s-{5}x{6}.{7}", filenamePrefix, DateTime.Now.Year, DateTime.Now.Month.ToString("D2"), DateTime.Now.Day.ToString("D2"), ((int)(span.TotalSeconds)).ToString(), _targetWidth, _targetHeight, filenameExtension);
		}

		private static string GetFolder(OutputPath outputPathType, string path)
		{
			string fileFolder = string.Empty;
			if (outputPathType == OutputPath.RelativeToProject)
			{
				string projectFolder = System.IO.Path.GetFullPath(System.IO.Path.Combine(Application.dataPath, ".."));
				fileFolder = System.IO.Path.Combine(projectFolder, path);
			}
			else if (outputPathType == OutputPath.RelativeToPeristentData)
			{
				string projectFolder = System.IO.Path.GetFullPath(Application.persistentDataPath);
				fileFolder = System.IO.Path.Combine(projectFolder, path);
			}
			else if (outputPathType == OutputPath.Absolute)
			{
				fileFolder = path;
			}
			return fileFolder;
		}

		private static string AutoGenerateFilename(OutputPath outputPathType, string path, string filename)
		{
			// Create folder
			string fileFolder = GetFolder(outputPathType, path);

			// Combine path and filename
			return System.IO.Path.Combine(fileFolder, filename);
		}

		private static string ManualGenerateFilename(OutputPath outputPathType, string path, string filename)
		{
			string result = GetFolder(outputPathType, path);

			if (System.IO.Path.IsPathRooted(filename))
			{
				result = filename;
			}
			else
			{
				result = System.IO.Path.Combine(result, filename);
			}

			return result;
		}

		/*[ContextMenu("Debug GenerateFilename")]
		public void DebugGenereateFilename()
		{
			GenerateFilename();
			Debug.Log("PATH: " + _filePath);
		}*/

		protected void GenerateFilename()
		{
			if (_outputType == OutputType.VideoFile)
			{
				if (_autoGenerateFilename || string.IsNullOrEmpty(_forceFilename))
				{
					string filename = GenerateTimestampedFilename(_autoFilenamePrefix, _autoFilenameExtension);
					_filePath = AutoGenerateFilename(_outputFolderType, _outputFolderPath, filename);
				}
				else
				{
					_filePath = ManualGenerateFilename(_outputFolderType, _outputFolderPath, _forceFilename);
				}

				if (_useMediaFoundationH264 && !_filePath.EndsWith(".mp4", StringComparison.OrdinalIgnoreCase))
				{
					Debug.LogError("[AVProMovieCapture] Media Foundation H.264 MP4 Encoder selected but file extension is not set to 'mp4'");
				}

				// Create target directory if doesn't exist
				String directory = Path.GetDirectoryName(_filePath);
				if (!string.IsNullOrEmpty(directory) && !Directory.Exists(directory))
				{
					Directory.CreateDirectory(directory);
				}
			}
			else
			{
				_filePath = _forceFilename;
			}
		}

		public virtual bool PrepareCapture()
		{
			// Delete file if it already exists
			if (_outputType == OutputType.VideoFile && File.Exists(_filePath))
			{
				File.Delete(_filePath);
			}

			_numDroppedFrames = 0;
			_numDroppedEncoderFrames = 0;
			_numEncodedFrames = 0;
			_totalEncodedSeconds = 0;

#if UNITY_EDITOR_WIN || UNITY_STANDALONE_WIN
			if (_minimumDiskSpaceMB > 0 && _outputType == OutputType.VideoFile)
			{
				ulong freespace = 0;
				if (Utils.DriveFreeBytes(System.IO.Path.GetPathRoot(_filePath), out freespace))
				{
					_freeDiskSpaceMB = (long)(freespace / (1024 * 1024));
				}

				if (!IsEnoughDiskSpace())
				{
					Debug.LogError("[AVProMovieCapture] Not enough free space to start capture.  Stopping capture.");
					return false;
				}
			}
#endif

			if (_isRealTime)
			{
				/*if (_allowFrameRateChange)
				{
					_oldTargetFrameRate = Application.targetFrameRate;
					Application.targetFrameRate = (int)_frameRate;
				}*/
			}
			else
			{
				// Disable vsync
				if (_allowVSyncDisable && !Screen.fullScreen && QualitySettings.vSyncCount > 0)
				{
					_oldVSyncCount = QualitySettings.vSyncCount;
					QualitySettings.vSyncCount = 0;
				}

				if (_useMotionBlur && _motionBlurSamples > 1)
				{
					Time.captureFramerate = _motionBlurSamples * (int)_frameRate;

					// FromTexture and FromCamera360 captures don't require a camera for rendering, so set up the motion blur component differently
					if (this is CaptureFromTexture || this is CaptureFromCamera360 || this is CaptureFromCamera360ODS)
					{
						if (_motionBlur == null)
						{
							_motionBlur = this.GetComponent<MotionBlur>();
						}
						if (_motionBlur == null)
						{
							_motionBlur = this.gameObject.AddComponent<MotionBlur>();
						}
						if (_motionBlur != null)
						{
							_motionBlur.NumSamples = _motionBlurSamples;
							_motionBlur.SetTargetSize(_targetWidth, _targetHeight);
							_motionBlur.enabled = false;
						}
					}
					// FromCamera and FromScreen use this path
					else if (_motionBlurCameras.Length > 0)
					{
						// Setup the motion blur filters where cameras are used
						foreach (Camera camera in _motionBlurCameras)
						{
							MotionBlur mb = camera.GetComponent<MotionBlur>();
							if (mb == null)
							{
								mb = camera.gameObject.AddComponent<MotionBlur>();
							}
							if (mb != null)
							{
								mb.NumSamples = _motionBlurSamples;
								mb.enabled = true;
								_motionBlur = mb;
							}
						}
					}
				}
				else
				{
					Time.captureFramerate = (int)_frameRate;
				}

				// Change physics update speed
				_oldFixedDeltaTime = Time.fixedDeltaTime;
				Time.fixedDeltaTime = 1.0f / Time.captureFramerate;
			}

			int audioDeviceIndex = _audioDeviceIndex;
			int audioCodecIndex = _audioCodecIndex;
			bool noAudio = _noAudio;
			if (!_isRealTime)
			{
				noAudio = true;
			}

			if (_mouseCursor != null)
			{
				_mouseCursor.enabled = _captureMouseCursor;
			}

			// We if try to capture audio from Unity but there isn't an UnityAudioCapture component set
			if (!noAudio && _audioCapture == null && _audioDeviceIndex < 0)
			{
				// Try to find it locally
				_audioCapture = this.GetComponent<UnityAudioCapture>();
				if (_audioCapture == null)
				{
					// Try to find it globally
					_audioCapture = GameObject.FindObjectOfType<UnityAudioCapture>();
				}

				if (_audioCapture == null)
				{
					// Find an AudioListener to attach the UnityAudioCapture component to
					AudioListener audioListener = this.GetComponent<AudioListener>();
					if (audioListener == null)
					{
						audioListener = GameObject.FindObjectOfType<AudioListener>();
					}
					if (audioListener != null)
					{
						_audioCapture = audioListener.gameObject.AddComponent<UnityAudioCapture>();
						Debug.LogWarning("[AVProMovieCapture] Capturing audio from Unity without an UnityAudioCapture assigned so we had to create one manually (very slow).  Consider adding a UnityAudioCapture component to your scene and assigned it to this MovieCapture component.");
					}
					else
					{
						noAudio = true;
						Debug.LogWarning("[AVProMovieCapture] No audio listener found in scene.  Unable to capture audio from Untiy.");
					}
				}
				else
				{
					Debug.LogWarning("[AVProMovieCapture] Capturing audio from Unity without an UnityAudioCapture assigned so we had to search for one manually (very slow)");
				}
			}

			if (noAudio || (_audioCapture == null && _audioDeviceIndex < 0))
			{
				audioCodecIndex = audioDeviceIndex = -1;
				_audioDeviceName = "none";
				noAudio = true;
			}

			_unityAudioSampleRate = -1;
			_unityAudioChannelCount = -1;
			if (IsUsingUnityAudio())
			{
				if (!_audioCapture.enabled)
				{
					_audioCapture.enabled = true;
				}
				_unityAudioSampleRate = AudioSettings.outputSampleRate;
				_unityAudioChannelCount = _audioCapture.NumChannels;
			}

			string info = string.Format("{0}x{1} @ {2}fps [{3}]", _targetWidth, _targetHeight, ((int)_frameRate).ToString(), _pixelFormat.ToString());
			if (_outputType == OutputType.VideoFile)
			{
				if (!_useMediaFoundationH264)
				{
					info += string.Format(" vcodec:'{0}'", _codecName);
				}
				else
				{
					info += " vcodec:'Media Founndation H.264'";
				}
				if (!noAudio)
				{
					if (_audioDeviceIndex >= 0)
					{
						info += string.Format(" audio device:'{0}'", _audioDeviceName);
					}
					else
					{
						info += string.Format(" audio device:'Unity' {0}hz {1} channels", _unityAudioSampleRate, _unityAudioChannelCount);
					}
					info += string.Format(" acodec:'{0}'", _audioCodecName);
				}

				info += string.Format(" to file: '{0}'", _filePath);
			}
			else if (_outputType == OutputType.NamedPipe)
			{
				info += string.Format(" to pipe: '{0}'", _filePath);
			}

			// If the user has overriden the vertical flip
			if (_flipVertically)
				_isTopDown = !_isTopDown;

			if (_outputType == OutputType.VideoFile)
			{
				// TOOD: make _frameRate floating point, or add timeLapse time system
				Debug.Log("[AVProMovieCapture] Start File Capture: " + info);
				_handle = NativePlugin.CreateRecorderAVI(_filePath, (uint)_targetWidth, (uint)_targetHeight, (int)_frameRate,
																	(int)(_pixelFormat), _isTopDown, _codecIndex, !noAudio, _unityAudioSampleRate, _unityAudioChannelCount, audioDeviceIndex, audioCodecIndex, _isRealTime, _useMediaFoundationH264, _supportAlpha);
			}
			else if (_outputType == OutputType.NamedPipe)
			{
				Debug.Log("[AVProMovieCapture] Start Pipe Capture: " + info);
				_handle = NativePlugin.CreateRecorderPipe(_filePath, (uint)_targetWidth, (uint)_targetHeight, (int)_frameRate,
																	 (int)(_pixelFormat), _isTopDown, _supportAlpha);
			}

			if (_handle < 0)
			{
				Debug.LogError("[AVProMovieCapture] Failed to create recorder");

				if (!_filePath.ToLower().EndsWith(".mp4") && _useMediaFoundationH264)
				{
					Debug.LogError("[AVProMovieCapture] When using MF H.264 codec the MP4 extension must be used");
				}

				if (_filePath.ToLower().EndsWith(".mp4") && !_useMediaFoundationH264 || _codecIndex == 0)
				{
					Debug.LogError("[AVProMovieCapture] Uncompressed video codec not supported with MP4 extension, use AVI instead for uncompressed");
				}

				StopCapture();
			}

			return (_handle >= 0);
		}

		public void QueueStartCapture()
		{
			_queuedStartCapture = true;
		}

		public bool StartCapture()
		{
			if (_capturing)
			{
				return false;
			}

			if (_handle < 0)
			{
				if (!PrepareCapture())
				{
					return false;
				}
			}

			if (_handle >= 0)
			{
				if (IsUsingUnityAudio())
				{
					_audioCapture.FlushBuffer();
				}

				if (!NativePlugin.Start(_handle))
				{
					StopCapture(true);
					return false;
				}
				ResetFPS();
				_captureStartTime = Time.realtimeSinceStartup;

				// NOTE: We set this to the elapsed time so that the first frame is captured immediately
				float secondsPerFrame = 1f / (float)((int)_frameRate);
				_timeSinceLastFrame = secondsPerFrame;

				_capturing = true;
				_paused = false;
			}

			if (_startPaused)
			{
				PauseCapture();
			}

			return _capturing;
		}

		public void PauseCapture()
		{
			if (_capturing && !_paused)
			{
				if (IsUsingUnityAudio())
				{
					_audioCapture.enabled = false;
				}
				NativePlugin.Pause(_handle);

				if (!_isRealTime)
				{
					// TODO: should be store the timeScale value and restore it instead of assuming timeScale == 1.0?
					Time.timeScale = 0f;
				}

				_paused = true;
				ResetFPS();
			}
		}

		public void ResumeCapture()
		{
			if (_capturing && _paused)
			{
				if (IsUsingUnityAudio())
				{
					_audioCapture.FlushBuffer();
					_audioCapture.enabled = true;
				}

				NativePlugin.Start(_handle);

				if (!_isRealTime)
				{
					Time.timeScale = 1f;
				}

				_paused = false;
				if (_startPaused)
				{
					_captureStartTime = Time.realtimeSinceStartup;
					_startPaused = false;
				}
			}
		}

		public void CancelCapture()
		{
			StopCapture(true);

			// Delete file
			if (_outputType == OutputType.VideoFile && File.Exists(_filePath))
			{
				File.Delete(_filePath);
			}
		}

		public virtual void UnprepareCapture()
		{
			if (_mouseCursor != null)
			{
				_mouseCursor.enabled = false;
			}
		}

#if UNITY_EDITOR
		public static string LastFileSaved
		{
			get
			{
				return UnityEditor.EditorPrefs.GetString("AVProMovieCapture-LastSavedFile", string.Empty);
			}
			set
			{
				UnityEditor.EditorPrefs.SetString("AVProMovieCapture-LastSavedFile", value);
			}
		}
#endif

		protected void RenderThreadEvent(NativePlugin.PluginEvent renderEvent)
		{
			if (renderEvent == NativePlugin.PluginEvent.CaptureFrameBuffer)
			{
#if AVPRO_MOVIECAPTURE_GLISSUEEVENT_52
				GL.IssuePluginEvent(_renderEventFunction, NativePlugin.PluginID | (int)renderEvent | _handle);
#else
				GL.IssuePluginEvent(NativePlugin.PluginID | (int)renderEvent | _handle);
#endif
			}
			else if (renderEvent == NativePlugin.PluginEvent.FreeResources)
			{
#if AVPRO_MOVIECAPTURE_GLISSUEEVENT_52
				GL.IssuePluginEvent(_freeEventFunction, NativePlugin.PluginID | (int)renderEvent);
#else
				GL.IssuePluginEvent(NativePlugin.PluginID | (int)renderEvent);
#endif
			}
		}

		public virtual void StopCapture(bool skipPendingFrames = false)
		{
			UnprepareCapture();

			if (_capturing)
			{
				Debug.Log("[AVProMovieCapture] Stopping capture " + _handle);
				_capturing = false;
			}

			RenderThreadEvent(NativePlugin.PluginEvent.FreeResources);

			if (_handle >= 0)
			{
				NativePlugin.Stop(_handle, skipPendingFrames);
				//System.Threading.Thread.Sleep(100);
				NativePlugin.FreeRecorder(_handle);
				_handle = -1;

#if UNITY_EDITOR
				if (_outputType == OutputType.VideoFile && !skipPendingFrames && !string.IsNullOrEmpty(_filePath))
				{
					LastFileSaved = _filePath;
				}
#endif
			}

			_fileInfo = null;

			if (_audioCapture)
			{
				_audioCapture.enabled = false;
			}

			if (_motionBlur)
			{
				_motionBlur.enabled = false;
			}

			// Restore Unity timing
			Time.captureFramerate = 0;
			//Application.targetFrameRate = _oldTargetFrameRate;
			//_oldTargetFrameRate = -1;

			if (_oldFixedDeltaTime > 0f)
			{
				Time.fixedDeltaTime = _oldFixedDeltaTime;
			}
			_oldFixedDeltaTime = 0f;

			if (_oldVSyncCount > 0)
			{
				QualitySettings.vSyncCount = _oldVSyncCount;
				_oldVSyncCount = 0;
			}

			_motionBlur = null;

			if (_texture != null)
			{
				Destroy(_texture);
				_texture = null;
			}
		}

		private void ToggleCapture()
		{
			if (_capturing)
			{
				//_queuedStopCapture = true;
				//_queuedStartCapture = false;
				StopCapture();
			}
			else
			{
				//_queuedStartCapture = true;
				//_queuedStopCapture = false;
				StartCapture();
			}
		}

		private bool IsEnoughDiskSpace()
		{
			bool result = true;
#if UNITY_STANDALONE_WIN || UNITY_EDITOR_WIN
			long fileSizeMB = GetCaptureFileSize() / (1024 * 1024);

			if ((_freeDiskSpaceMB - fileSizeMB) < _minimumDiskSpaceMB)
			{
				result = false;
			}
#endif
			return result;
		}

		private void Update()
		{
			if (_handle >= 0 && !_paused)
			{
				CheckFreeDiskSpace();
			}

			UpdateFrame();
		}

		private void CheckFreeDiskSpace()
		{
			if (_minimumDiskSpaceMB > 0)
			{
				if (!IsEnoughDiskSpace())
				{
					Debug.LogWarning("[AVProMovieCapture] Free disk space getting too low.  Stopping capture.");
					StopCapture(true);
				}
			}
		}

		protected bool IsProgressComplete()
		{
			bool result = false;
			if (_stopMode != StopMode.None)
			{
				switch (_stopMode)
				{
					case StopMode.FramesEncoded:
						result = (_numEncodedFrames >= _stopFrames);
						break;
					case StopMode.SecondsEncoded:
						result = (_totalEncodedSeconds >= _stopSeconds);
						break;
					case StopMode.SecondsElapsed:
						if (!_startPaused && !_paused)
						{
							result = ((Time.realtimeSinceStartup - _captureStartTime) >= _stopSeconds);
						}
						break;
				}
			}
			return result;
		}

		public float GetProgress()
		{
			float result = 0f;
			if (_stopMode != StopMode.None)
			{
				switch (_stopMode)
				{
					case StopMode.FramesEncoded:
						result = (_numEncodedFrames / (float)_stopFrames);
						break;
					case StopMode.SecondsEncoded:
						result = ((_numEncodedFrames / (float)_frameRate) / _stopSeconds);
						break;
					case StopMode.SecondsElapsed:
						if (!_startPaused && !_paused)
						{
							result = ((Time.realtimeSinceStartup - _captureStartTime) / _stopSeconds);
						}
						break;
				}
			}
			return result;
		}

		protected bool CanOutputFrame()
		{
			bool result = false;
			if (_handle >= 0)
			{
				if (_isRealTime)
				{
					if (NativePlugin.IsNewFrameDue(_handle))
					{
						float secondsPerFrame = 1f / (float)((int)_frameRate);
						result = (_timeSinceLastFrame >= secondsPerFrame);
						//result = true;
					}
				}
				else
				{
					const int WatchDogLimit = 2000;
					int watchdog = 0;
					if (_outputType != OutputType.NamedPipe)
					{
						// Wait for the encoder to have an available buffer
						// The watchdog prevents an infinite while loop
						while (_handle >= 0 && !NativePlugin.IsNewFrameDue(_handle) && watchdog < WatchDogLimit)
						{
							System.Threading.Thread.Sleep(1);
							watchdog++;
						}
					}

					// Return handle status as it may have closed elsewhere
					result = (_handle >= 0) && (watchdog < WatchDogLimit);
				}
			}
			return result;
		}

		protected void TickFrameTimer()
		{
			_timeSinceLastFrame += Time.unscaledDeltaTime;
		}

		protected void RenormTimer()
		{
			float secondsPerFrame = 1f / (float)((int)_frameRate);
			if (_timeSinceLastFrame >= secondsPerFrame)
			{
				_timeSinceLastFrame = _timeSinceLastFrame - secondsPerFrame;
			}
		}

		public virtual Texture GetPreviewTexture()
		{
			return null;
		}
		
		public virtual void UpdateFrame()
		{
			if (Input.GetKeyDown(_captureKey))
			{
				ToggleCapture();
			}

			if (_handle >= 0 && !_paused)
			{
				_numDroppedFrames = NativePlugin.GetNumDroppedFrames(_handle);
				_numDroppedEncoderFrames = NativePlugin.GetNumDroppedEncoderFrames(_handle);
				_numEncodedFrames = NativePlugin.GetNumEncodedFrames(_handle);
				_totalEncodedSeconds = NativePlugin.GetEncodedSeconds(_handle);

				if (IsProgressComplete())
				{
					_queuedStopCapture = true;
				}
			}

			if (_queuedStopCapture)
			{
				_queuedStopCapture = false;
				_queuedStartCapture = false;
				StopCapture();
			}
			if (_queuedStartCapture)
			{
				_queuedStartCapture = false;
				StartCapture();
			}
		}

		private float _fps;
		private int _frameTotal;

		public float FPS { get { return _fps; } }
		public float FramesTotal { get { return _frameTotal; } }

		private int _frameCount;
		private float _startFrameTime;

		protected void ResetFPS()
		{
			_frameCount = 0;
			_frameTotal = 0;
			_fps = 0.0f;
			_startFrameTime = 0.0f;
		}

		public void UpdateFPS()
		{
			_frameCount++;
			_frameTotal++;

			float timeNow = Time.realtimeSinceStartup;
			float timeDelta = timeNow - _startFrameTime;
			if (timeDelta >= 1.0f)
			{
				_fps = (float)_frameCount / timeDelta;
				_frameCount = 0;
				_startFrameTime = timeNow;
			}
		}

		protected int GetCameraAntiAliasingLevel(Camera camera)
		{
			int aaLevel = QualitySettings.antiAliasing;
			if (aaLevel == 0)
			{
				aaLevel = 1;
			}

			if (_renderAntiAliasing > 0)
			{
				aaLevel = _renderAntiAliasing;
			}

			if (aaLevel != 1 && aaLevel != 2 && aaLevel != 4 && aaLevel != 8)
			{
				Debug.LogError("[AVProMovieCapture] Invalid antialiasing value, must be 1, 2, 4 or 8.  Defaulting to 1. >> " + aaLevel);
				aaLevel = 1;
			}

			if (aaLevel != 1)
			{
				if (camera.actualRenderingPath == RenderingPath.DeferredLighting
#if AVPRO_MOVIECAPTURE_DEFERREDSHADING
					|| camera.actualRenderingPath == RenderingPath.DeferredShading
#endif
					)
				{
					Debug.LogWarning("[AVProMovieCapture] Not using antialiasing because MSAA is not supported by camera render path " + camera.actualRenderingPath);
					aaLevel = 1;
				}
			}
			return aaLevel;
		}

		private void ConfigureCodec()
		{
			NativePlugin.Init();
			SelectCodec(false);
			if (_codecIndex >= 0)
			{
				NativePlugin.ConfigureVideoCodec(_codecIndex);
			}
			//AVProMovieCapture.Deinit();
		}

		public long GetCaptureFileSize()
		{
			long result = 0;
#if !UNITY_WEBPLAYER
			if (_handle >= 0 && _outputType == OutputType.VideoFile)
			{
				if (_fileInfo == null && File.Exists(_filePath))
				{
					_fileInfo = new System.IO.FileInfo(_filePath);
				}
				if (_fileInfo != null)
				{
					_fileInfo.Refresh();
					result = _fileInfo.Length;
				}
			}
#endif
			return result;
		}

		public static void GetResolution(Resolution res, ref int width, ref int height)
		{
			switch (res)
			{
				case Resolution.POW2_8192x8192:
					width = 8192; height = 8192;
					break;
				case Resolution.POW2_8192x4096:
					width = 8192; height = 4096;
					break;
				case Resolution.POW2_4096x4096:
					width = 4096; height = 4096;
					break;
				case Resolution.POW2_4096x2048:
					width = 4096; height = 2048;
					break;
				case Resolution.POW2_2048x4096:
					width = 2048; height = 4096;
					break;
				case Resolution.UHD_3840x2160:
					width = 3840; height = 2160;
					break;
				case Resolution.UHD_3840x2048:
					width = 3840; height = 2048;
					break;
				case Resolution.UHD_3840x1920:
					width = 3840; height = 1920;
					break;
				case Resolution.POW2_2048x2048:
					width = 2048; height = 2048;
					break;
				case Resolution.POW2_2048x1024:
					width = 2048; height = 1024;
					break;
				case Resolution.HD_1920x1080:
					width = 1920; height = 1080;
					break;
				case Resolution.HD_1280x720:
					width = 1280; height = 720;
					break;
				case Resolution.SD_1024x768:
					width = 1024; height = 768;
					break;
				case Resolution.SD_800x600:
					width = 800; height = 600;
					break;
				case Resolution.SD_800x450:
					width = 800; height = 450;
					break;
				case Resolution.SD_640x480:
					width = 640; height = 480;
					break;
				case Resolution.SD_640x360:
					width = 640; height = 360;
					break;
				case Resolution.SD_320x240:
					width = 320; height = 240;
					break;
			}
		}

		// Returns the next multiple of 4 or the same value if it's already a multiple of 4
		protected static int NextMultipleOf4(int value)
		{
			return (value + 3) & ~0x03;
		}
	}
}