CrowdControl/Assets/AlmostEngine/UltimateScreenshotCreator/Assets/Scripts/ScreenshotCamera.cs

using UnityEngine;
using System.Collections;
using System.Collections.Generic;

namespace AlmostEngine.Screenshot
{
        [System.Serializable]
        /// <summary>
        /// Screenshot cameras are used to control the cameras used for rendering the screenshot.
        /// </summary>
        public class ScreenshotCamera
        {
                public enum CustomSettings
                {
                    KEEP_CAMERA_SETTINGS,
                    CUSTOM}
                ;

				// Camera
                public bool m_Active = true;
                public Camera m_Camera = null;

                // Clear
                public CustomSettings m_ClearSettings;
                public CameraClearFlags m_ClearFlags = CameraClearFlags.Skybox;
                public Color m_BackgroundColor = Color.white;

        		// Culling
                public CustomSettings m_CullingSettings;
                public int m_CullingMask = ~0;

				// FOV
                public CustomSettings m_FOVSettings;
                public float m_FOV = 70;

        
                public class Settings
                {
                        public bool m_Enabled;
                        public bool m_GameObjectEnabled;

                        public CameraClearFlags m_ClearFlags;
                        public Color m_BackgroundColor;
                        public int m_CullingMask;
                        public float m_FOV;

                        public Settings (bool enabled, bool go, CameraClearFlags clear, Color color, int culling, float fov)
                        {
                                m_Enabled = enabled;
                                m_GameObjectEnabled = go;
                
                                m_ClearFlags = clear;
                                m_BackgroundColor = color;
                                m_CullingMask = culling;
                                m_FOV = fov;
                        }
                };

                public Stack<Settings> m_SettingStack = new Stack<Settings> ();


                public ScreenshotCamera ()
                {
                }

                public ScreenshotCamera (Camera cam)
                {
                        m_Camera = cam;
                }

                /// <summary>
                /// Apply the custom settings to the camera.
                /// </summary>
                public void ApplySettings ()
                {
                        if (m_Camera == null)
                                return;

                        // Save current settings
                        m_SettingStack.Push (new Settings (m_Camera.enabled, m_Camera.gameObject.activeSelf, m_Camera.clearFlags, m_Camera.backgroundColor, m_Camera.cullingMask, m_Camera.fieldOfView));

                        // Apply settings
                        m_Camera.enabled = true;
                        m_Camera.gameObject.SetActive (true);

                        if (m_ClearSettings == CustomSettings.CUSTOM) {
                                m_Camera.clearFlags = m_ClearFlags;
                                m_Camera.backgroundColor = m_BackgroundColor;
                        }
                        if (m_CullingSettings == CustomSettings.CUSTOM) {
                                m_Camera.cullingMask = m_CullingMask;
                        }
                        if (m_FOVSettings == CustomSettings.CUSTOM) {
                                m_Camera.fieldOfView = m_FOV;
                        }
                }

                public void Disable ()
                {
                        if (m_Camera == null)
                                return;
            
                        // Save current settings
                        m_SettingStack.Push (new Settings (m_Camera.enabled, m_Camera.gameObject.activeSelf, m_Camera.clearFlags, m_Camera.backgroundColor, m_Camera.cullingMask, m_Camera.fieldOfView));
            
                        // Apply settings
                        m_Camera.enabled = false;

                }

                /// <summary>
                /// Restaure the original camera settings.
                /// </summary>
                public void RestoreSettings ()
                {
                        if (m_Camera == null)
                                return;

                        if (m_SettingStack.Count <= 0)
                                return;

            
                        Settings s = m_SettingStack.Pop ();
                        m_Camera.enabled = s.m_Enabled;
                        m_Camera.gameObject.SetActive (s.m_GameObjectEnabled);
                        m_Camera.clearFlags = s.m_ClearFlags;
                        m_Camera.backgroundColor = s.m_BackgroundColor;
                        m_Camera.cullingMask = s.m_CullingMask;
                        m_Camera.fieldOfView = s.m_FOV;
                }


        }
}