Avalonia/Perspex/Controls/ControlTemplate.cs

// -----------------------------------------------------------------------
// <copyright file="ControlTemplate.cs" company="Steven Kirk">
// Copyright 2014 MIT Licence. See licence.md for more information.
// </copyright>
// -----------------------------------------------------------------------

namespace Perspex.Controls
{
    using System;
    using System.Linq;
    using Perspex.Controls;

    public class ControlTemplate
    {
        private Func<ITemplatedControl, Control> build;

        public ControlTemplate(Func<ITemplatedControl, Control> build)
        {
            Contract.Requires<NullReferenceException>(build != null);

            this.build = build;
        }

        public static ControlTemplate Create<TControl>(Func<TControl, Control> build)
            where TControl : ITemplatedControl
        {
            Contract.Requires<NullReferenceException>(build != null);

            return new ControlTemplate(c => build((TControl)c));
        }

        public Control Build(ITemplatedControl templatedParent)
        {
            Contract.Requires<NullReferenceException>(templatedParent != null);

            Control root = this.build(templatedParent);
            this.SetTemplatedParent(root, templatedParent);
            return root;
        }

        private void SetTemplatedParent(Control control, ITemplatedControl templatedParent)
        {
            Contract.Requires<NullReferenceException>(control != null);
            Contract.Requires<NullReferenceException>(templatedParent != null);

            control.TemplatedParent = templatedParent;

            if (!(control is ContentPresenter))
            {
                foreach (Control child in ((IVisual)control).VisualChildren.OfType<Control>())
                {
                    this.SetTemplatedParent(child, templatedParent);
                }
            }
        }
    }
}