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

namespace Panorama180Render {
    [RequireComponent(typeof(Camera))]
    public class MultipleCapture : MonoBehaviour
    {
        // Camera start position.
        [SerializeField] Vector3 startPosition = new Vector3(0, 0, 0);
        
        // Camera end position.
        [SerializeField] Vector3 endPosition = new Vector3(1, 0, 0);

        // capture count.
        [SerializeField, TooltipAttribute("Capture count"), Range(1, 100)] int captureCount = 4;

        // Save filename.
        [SerializeField, TooltipAttribute("Save file name")] string fileName = "capture";

        // Camera near distance.
        [SerializeField] float nearDistance = 0.1f;
        
        // Camera far distance.
        [SerializeField] float farDistance = 100.0f;

        // Synchronize gaze direction (ver.1.1.0).
        [SerializeField] bool synchronizeGazeDirection = false;

        // Camera position.
        private Vector3 m_baseCameraPos;

        void Start()
        {
            m_baseCameraPos = this.transform.position;

            // Coroutine that outputs still images.
            StartCoroutine(m_OutputStillImages());
        }

        void Update()
        {
        }

        /**
         * Output still images.
         */
        IEnumerator m_OutputStillImages () {
            Panorama180Render panorama180Render = this.GetComponent<Panorama180Render>();

            Vector3 cameraPos = this.transform.position;

            Vector3 pD = (endPosition - startPosition) / (float)Mathf.Max(1, (captureCount - 1));

            // Camera near,far distance.
            panorama180Render.SetCameraNearFarDistance(nearDistance, farDistance);

            // Set the depth linear type to linear.
            panorama180Render.SetDepthLinearType(Panorama180Render.OutputDepthLinearType.Linear);

            // Depth type is one quarter resolution.
            panorama180Render.SetDepthTextureType(Panorama180Render.OutputDepthTextureType.DepthOneQuarter);

            // Synchronize gaze direction (ver.1.1.0 -).
            panorama180Render.SetSynchronizeGazeDirection(synchronizeGazeDirection);

            for (int loop = 0; loop < captureCount; ++loop) {
                // Change camera position.
                Vector3 newPos = startPosition + (pD * loop);
                yield return StartCoroutine(panorama180Render.SetCameraPosition(newPos));

                // Set output file name. 
                panorama180Render.SetOutputImageFileName(fileName + "_" + (loop).ToString());

                // Save image.
                panorama180Render.OutputStillImage();           // RGB image.
                panorama180Render.OutputStillImage(true);       // Depth image(exr).
            }

            this.transform.position = m_baseCameraPos;
        }
    }
}