// Does not work on the Xbox360, no Save method in the Texture class!
#if !XBOX360

#region Using Statements
using System;
using System.Collections.Generic;
using Microsoft.Xna.Framework;
using XnaShooter.Helpers;
using Microsoft.Xna.Framework.Input;
using System.IO;
using Microsoft.Xna.Framework.Graphics;
#endregion

namespace XnaShooter.Game
{
	/// <summary>
	/// This is a game component that implements IUpdateable.
	/// </summary>
	public partial class ScreenshotCapturer : GameComponent
	{
		#region Variables
		/// <summary>
		/// Internal screenshot number (will increase by one each screenshot)
		/// </summary>
		private int screenshotNum = 0;
		/// <summary>
		/// Link to BaseGame class instance. Also holds windows title,
		/// which is used instead of Application.ProgramName.
		/// </summary>
		BaseGame game;
		#endregion

		#region Constructor
		public ScreenshotCapturer(BaseGame setGame)
			: base(setGame)
		{
			game = setGame;
			screenshotNum = GetCurrentScreenshotNum();
		} // ScreenshotCapturer(game)
		#endregion

		#region Make screenshot
		#region Screenshot name builder
		/// <summary>
		/// Screenshot name builder
		/// </summary>
		/// <param name="num">Num</param>
		/// <returns>String</returns>
		private string ScreenshotNameBuilder(int num)
		{
			return Directories.ScreenshotsDirectory + "\\" +
				game.Window.Title + " Screenshot " +
				num.ToString("0000") + ".jpg";
		} // ScreenshotNameBuilder(num)
		#endregion

		#region Get current screenshot num
		/// <summary>
		/// Get current screenshot num
		/// </summary>
		/// <returns>Int</returns>
		private int GetCurrentScreenshotNum()
		{
			// We must search for last screenshot we can found in list using own
			// fast filesearch
			int i = 0, j = 0, k = 0, l = -1;
			// First check if at least 1 screenshot exist
			if (File.Exists(ScreenshotNameBuilder(0)) == true)
			{
				// First scan for screenshot num/1000
				for (i = 1; i < 10; i++)
				{
					if (File.Exists(ScreenshotNameBuilder(i * 1000)) == false)
						break;
				} // for (i)

				// This i*1000 does not exist, continue scan next level
				// screenshotnr/100
				i--;
				for (j = 1; j < 10; j++)
				{
					if (File.Exists(ScreenshotNameBuilder(i * 1000 + j * 100)) == false)
						break;
				} // for (j)

				// This i*1000+j*100 does not exist, continue scan next level
				// screenshotnr/10
				j--;
				for (k = 1; k < 10; k++)
				{
					if (File.Exists(ScreenshotNameBuilder(
							i * 1000 + j * 100 + k * 10)) == false)
						break;
				} // for (k)

				// This i*1000+j*100+k*10 does not exist, continue scan next level
				// screenshotnr/1
				k--;
				for (l = 1; l < 10; l++)
				{
					if (File.Exists(ScreenshotNameBuilder(
							i * 1000 + j * 100 + k * 10 + l)) == false)
						break;
				} // for (l)

				// This i*1000+j*100+k*10+l does not exist, we have now last
				// screenshot nr!!!
				l--;
			} // if (File.Exists)

			return i * 1000 + j * 100 + k * 10 + l;
		} // GetCurrentScreenshotNum()
		#endregion

		/// <summary>
		/// Make screenshot
		/// </summary>
		private void MakeScreenshot()
		{
			try
			{
				//NOTE: This doesn't always work on all cards, especially if
				// desktop mode switches in fullscreen mode!

				screenshotNum++;
				// Make sure screenshots directory exists
				if (Directory.Exists(Directories.ScreenshotsDirectory) == false)
					Directory.CreateDirectory(Directories.ScreenshotsDirectory);

				using (Texture2D dstTexture = new Texture2D(
					BaseGame.Device,
					BaseGame.Width,
					BaseGame.Height,
					1,
					ResourceUsage.ResolveTarget,
					SurfaceFormat.Color,
					ResourceManagementMode.Manual))
				{
					// Get data with help of the resolve method
					BaseGame.Device.ResolveBackBuffer(dstTexture);

					dstTexture.Save(
						ScreenshotNameBuilder(screenshotNum),
						ImageFileFormat.Jpg);
				} // using
			} // try
			catch (Exception ex)
			{
				Log.Write("Failed to save Screenshot: " + ex.ToString());
			} // catch (ex)
		} // MakeScreenshot()
		#endregion

		#region Update
		/// <summary>
		/// Allows the game component to update itself.
		/// </summary>
		/// <param name="gameTime">Provides a snapshot of timing values.</param>
		public override void Update(GameTime gameTime)
		{
			if (Input.KeyboardKeyJustPressed(Keys.PrintScreen))
				MakeScreenshot();

			base.Update(gameTime);

			/*TODO?
			// Make a screenshot?
			if (remToMakeScreenshot)
			{
				remToMakeScreenshot = false;
				MakeScreenshot();
			} // if (remToMakeScreenshot)
			 */
		} // Update(gameTime)
		#endregion
	} // class ScreenshotCapturer
} // namespace XnaShooter.Game
#endif