OpenRA/OpenRA.Mods.Cnc/Widgets/CncMusicPlayerLogic.cs

#region Copyright & License Information
/*
 * Copyright 2007-2011 The OpenRA Developers (see AUTHORS)
 * This file is part of OpenRA, which is free software. It is made 
 * available to you under the terms of the GNU General Public License
 * as published by the Free Software Foundation. For more information,
 * see COPYING.
 */
#endregion

using System;
using System.Drawing;
using System.Linq;
using OpenRA.FileFormats;
using OpenRA.Support;
using OpenRA.Widgets;
using OpenRA.Traits;
using OpenRA.Graphics;

namespace OpenRA.Mods.Cnc.Widgets
{
	public class CncMusicPlayerLogic : IWidgetDelegate
	{
		bool installed;
		string currentSong = null;
		Widget panel;
		string[] music;
		string[] random;

		[ObjectCreator.UseCtor]
		public CncMusicPlayerLogic([ObjectCreator.Param] Widget widget,
		                           [ObjectCreator.Param] Action onExit)
		{
			panel = widget.GetWidget("MUSIC_PANEL");
			BuildMusicTable(panel);

			currentSong = GetNextSong();
			installed = Rules.Music.Where(m => m.Value.Exists).Any();
			Func<bool> noMusic = () => !installed;
			
			panel.GetWidget<CncMenuButtonWidget>("BACK_BUTTON").OnClick = onExit;
			panel.GetWidget<CncMenuButtonWidget>("INSTALL_BUTTON").IsDisabled = () => true;
			panel.GetWidget("NO_MUSIC_LABEL").IsVisible = noMusic;

			var playButton = panel.GetWidget<CncMenuButtonWidget>("BUTTON_PLAY");
			playButton.OnClick = Play;
			playButton.IsDisabled = noMusic;
			
			var pauseButton = panel.GetWidget<CncMenuButtonWidget>("BUTTON_PAUSE");
			pauseButton.OnClick = Pause;
			pauseButton.IsDisabled = noMusic;

			var stopButton = panel.GetWidget<CncMenuButtonWidget>("BUTTON_STOP");
			stopButton.OnClick = Stop;
			stopButton.IsDisabled = noMusic;
			
			var nextButton = panel.GetWidget<CncMenuButtonWidget>("BUTTON_NEXT");
			nextButton.OnClick = () => { currentSong = GetNextSong(); Play(); };
			nextButton.IsDisabled = noMusic;
			
			var prevButton = panel.GetWidget<CncMenuButtonWidget>("BUTTON_PREV");
			prevButton.OnClick = () => { currentSong = GetPrevSong(); Play(); };
			prevButton.IsDisabled = noMusic;
			
			var shuffleCheckbox = panel.GetWidget<CncCheckboxWidget>("SHUFFLE");
			shuffleCheckbox.IsChecked = () => Game.Settings.Sound.Shuffle;
			shuffleCheckbox.OnClick = () => Game.Settings.Sound.Shuffle ^= true;
			
			var repeatCheckbox = panel.GetWidget<CncCheckboxWidget>("REPEAT");
			repeatCheckbox.IsChecked = () => Game.Settings.Sound.Repeat;
			repeatCheckbox.OnClick = () => Game.Settings.Sound.Repeat ^= true;

			panel.GetWidget<LabelWidget>("TIME_LABEL").GetText = () => (currentSong == null) ? "" : 
					"{0:D2}:{1:D2} / {2:D2}:{3:D2}".F((int)Sound.MusicSeekPosition / 60, (int)Sound.MusicSeekPosition % 60,
					    							  Rules.Music[currentSong].Length / 60, Rules.Music[currentSong].Length % 60);
		}
	
		void BuildMusicTable(Widget panel)
		{
			music = Rules.Music.Where(a => a.Value.Exists)
				.Select(a => a.Key).ToArray();
			random = music.Shuffle(Game.CosmeticRandom).ToArray();
			
			var ml = panel.GetWidget<ScrollPanelWidget>("MUSIC_LIST");
			var itemTemplate = ml.GetWidget<ContainerWidget>("MUSIC_TEMPLATE");
			
			foreach (var s in music)
			{
				var song = s;
				if (currentSong == null)
					currentSong = song;

				var template = itemTemplate.Clone() as ContainerWidget;
				template.GetBackground = () => (template.RenderBounds.Contains(Viewport.LastMousePos) ? "button-hover" : (song == currentSong) ? "button-pressed" : null);
				template.OnMouseDown = mi =>
				{
					if (mi.Button != MouseButton.Left) return false;
					currentSong = song;
					Play();
					return true;
				};

				template.IsVisible = () => true;				
				template.GetWidget<LabelWidget>("TITLE").GetText = () => Rules.Music[song].Title;
				template.GetWidget<LabelWidget>("LENGTH").GetText = () => SongLengthLabel(song);
				ml.AddChild(template);
			}
		}
		
		
		void Play()
		{
			if (currentSong == null)
				return;
			
			Sound.PlayMusicThen(Rules.Music[currentSong].Filename, () =>
			{
				if (!Game.Settings.Sound.Repeat)
					currentSong = GetNextSong();
				Play();
			});
			
			panel.GetWidget("BUTTON_PLAY").Visible = false;
			panel.GetWidget("BUTTON_PAUSE").Visible = true;
		}
		
		void Pause()
		{
			Sound.PauseMusic();
			panel.GetWidget("BUTTON_PAUSE").Visible = false;
			panel.GetWidget("BUTTON_PLAY").Visible = true;
		}
		
		void Stop()
		{
			Sound.StopMusic();
			panel.GetWidget("BUTTON_PAUSE").Visible = false;
			panel.GetWidget("BUTTON_PLAY").Visible = true;		
		}
		
		string SongLengthLabel(string song)
		{
			return "{0:D1}:{1:D2}".F(Rules.Music[song].Length / 60,
			                         Rules.Music[song].Length % 60);
		}
		
		string GetNextSong()
		{
			if (!music.Any())
				return null;
			
			var songs = Game.Settings.Sound.Shuffle ? random : music;
			return songs.SkipWhile(m => m != currentSong)
				.Skip(1).FirstOrDefault() ?? songs.FirstOrDefault();
		}

		string GetPrevSong()
		{
			if (!music.Any())
				return null;
			
			var songs = Game.Settings.Sound.Shuffle ? random : music;
			return songs.Reverse().SkipWhile(m => m != currentSong)
				.Skip(1).FirstOrDefault() ?? songs.FirstOrDefault();
		}
	}
}