WPF Control Factory

using System.Collections.Generic;
using System.Windows;
using System.Windows.Controls;

namespace Pfz.WpfControls.Extensions.ElementExtensions
{
	/// <summary>
	/// Adds methods to find enumerate all controls and sub-controls.
	/// </summary>
	public static class PfzElementExtensions
	{
		/// <summary>
		/// Gets the actual element and all it's sub-elements recursivelly.
		/// </summary>
		public static IEnumerable<UIElement> EnumerateElementsRecursive(this UIElement element)
		{
			if (element == null)
				yield break;
			
			yield return element;
			
			var contentControl = element as ContentControl;
			if (contentControl != null)
			{
				var content = contentControl.Content as UIElement;
				foreach(var subControl in content.EnumerateElementsRecursive())
					yield return subControl;
					
				yield break;
			}
			
			Decorator decorator = element as Decorator;
			if (decorator != null)
			{
				var content = decorator.Child;
				foreach(var subControl in content.EnumerateElementsRecursive())
					yield return subControl;
				
				yield break;
			}
			
			Panel panel = element as Panel;
			if (panel != null)
			{
				foreach(var child in panel.Children)
				{
					var content = child as UIElement;
					foreach(var item in content.EnumerateElementsRecursive())
						yield return item;
				}
				
				yield break;
			}
		}
		
		/// <summary>
		/// Gets all elements of the given type, recursivelly, from the given base element.
		/// </summary>
		public static IEnumerable<T> EnumerateElementsRecursive<T>(this UIElement baseElement)
		where
			T: UIElement
		{
			foreach(var element in baseElement.EnumerateElementsRecursive())
			{
				var typedElement = element as T;
				if (typedElement != null)
					yield return typedElement;
			}
		}
		
		/// <summary>
		/// Gets the first parent found of the given type.
		/// </summary>
		public static T GetParentOfType<T>(this FrameworkElement element)
		where
			T: class
		{
			while(element != null)
			{
				var parent = element.Parent;
				if (parent == null)
					return null;
				
				T result = parent as T;
				if (result != null)
					return result;
				
				element = parent as FrameworkElement;
			}
			
			return null;
		}
	}
}