using System;
using System.Collections.Generic;
using System.Linq;
using System.Windows;
using System.Collections;

namespace Microsoft.Research.DynamicDataDisplay.DataSources
{
	/// <summary>Data source that is a composer from several other data sources.</summary>
	public class CompositeDataSource : IPointDataSource {
		/// <summary>Initializes a new instance of the <see cref="CompositeDataSource"/> class</summary>
		public CompositeDataSource() { }

		/// <summary>
		/// Initializes a new instance of the <see cref="CompositeDataSource"/> class.
		/// </summary>
		/// <param name="dataSources">Data sources.</param>
		public CompositeDataSource(params IPointDataSource[] dataSources) {
			if (dataSources == null)
				throw new ArgumentNullException("dataSources");

			foreach (var dataSource in dataSources) {
				AddDataPart(dataSource);
			}
		}

		private readonly List<IPointDataSource> dataParts = new List<IPointDataSource>();

		public IEnumerable<IPointDataSource> DataParts {
			get { return dataParts; }
		}

		/// <summary>
		/// Adds data part.
		/// </summary>
		/// <param name="dataPart">The data part.</param>
		public void AddDataPart(IPointDataSource dataPart) {
			if (dataPart == null)
				throw new ArgumentNullException("dataPart");

			dataParts.Add(dataPart);
			dataPart.DataChanged += OnPartDataChanged;
		}

		private void OnPartDataChanged(object sender, EventArgs e) {
			RaiseDataChanged();
		}

		#region IPointSource Members

		public event EventHandler DataChanged;
		protected void RaiseDataChanged() {
			if (DataChanged != null) {
				DataChanged(this, EventArgs.Empty);
			}
		}

		public IPointEnumerator GetEnumerator(DependencyObject context) {
			return new CompositeEnumerator(this, context);
		}

		#endregion

		private sealed class CompositeEnumerator : IPointEnumerator {
			private readonly IEnumerable<IPointEnumerator> enumerators;

			public CompositeEnumerator(CompositeDataSource dataSource, DependencyObject context) {
				enumerators = dataSource.dataParts.Select(part => part.GetEnumerator(context)).ToList();
			}

			#region IChartPointEnumerator Members

			public bool MoveNext() {
				bool res = true;
				foreach (var enumerator in enumerators) {
					res &= enumerator.MoveNext();
				}
				return res;
			}

			public void ApplyMappings(DependencyObject glyph) {
				foreach (var enumerator in enumerators) {
					enumerator.ApplyMappings(glyph);
				}
			}

			public void GetCurrent(ref Point p) {
				foreach (var enumerator in enumerators) {
					enumerator.GetCurrent(ref p);
				}
			}

			public void Dispose() {
				foreach (var enumerator in enumerators) {
					enumerator.Dispose();
				}
			}

			#endregion
		}
	}
}