ScreenShotMaker.cs

Go to the documentation of this file.

using UnityEngine;
using System.IO;
using System.Collections;

namespace HKI.Core.Misc
{
    public class ScreenShotMaker : MonoBehaviour
    {
        // Public variables
        [SerializeField] KeyCode CaptureKey = KeyCode.F8;
        [SerializeField] string ScreenShotsPath = "Screenshots";
        [SerializeField] string Name = "Screen";

        [SerializeField] GameObject[] HudFreeScreenShotHideObjects;
        [SerializeField] KeyCode CaptureKeyForHudFree = KeyCode.F9;


        // Private Variables
        int count = 0;
        bool[] oldStatus;

        // Update function
        void Update()
        {
            if (Input.GetKeyDown(CaptureKey))
                StartCoroutine(CaptureScreenshoot(false));

            if (Input.GetKeyDown(CaptureKeyForHudFree))
                StartCoroutine(CaptureScreenshoot(true));

        }

        IEnumerator CaptureScreenshoot(bool withoutHUD)
        {
            count++;

            if (HudFreeScreenShotHideObjects != null && withoutHUD)
            {
                oldStatus = new bool[HudFreeScreenShotHideObjects.Length];

                for (int i = 0; i < oldStatus.Length; i++)
                {
                    oldStatus[i] = HudFreeScreenShotHideObjects[i].activeSelf;
                    HudFreeScreenShotHideObjects[i].SetActive(false);
                }

                yield return null;
            }

            yield return null;
            if (!Directory.Exists(ScreenShotsPath))
                Directory.CreateDirectory(ScreenShotsPath);

            ScreenCapture.CaptureScreenshot(ScreenShotsPath + "/" + Name + "_" + count.ToString() + ".png");

            yield return null;
            if (HudFreeScreenShotHideObjects != null && withoutHUD)
            {
                for (int i = 0; i < oldStatus.Length; i++)
                    HudFreeScreenShotHideObjects[i].SetActive(oldStatus[i]);
            }

        }
    }
}