wpftoolkit/ExtendedWPFToolkitSolution/Src/Xceed.Wpf.Toolkit/PropertyGrid/Implementation/PropertyDefinitionBaseCollection.cs

/*************************************************************************************

   Extended WPF Toolkit

   Copyright (C) 2007-2013 Xceed Software Inc.

   This program is provided to you under the terms of the Microsoft Public
   License (Ms-PL) as published at http://wpftoolkit.codeplex.com/license 

   For more features, controls, and fast professional support,
   pick up the Plus Edition at http://xceed.com/wpf_toolkit

   Stay informed: follow @datagrid on Twitter or Like http://facebook.com/datagrids

  ***********************************************************************************/

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

namespace Xceed.Wpf.Toolkit.PropertyGrid
{
  public class PropertyDefinitionCollection : PropertyDefinitionBaseCollection<PropertyDefinition>
  {
  }
  public class EditorDefinitionCollection : PropertyDefinitionBaseCollection<EditorDefinitionBase>
  {
  }


  public abstract class PropertyDefinitionBaseCollection<T> : DefinitionCollectionBase<T> where T : PropertyDefinitionBase
  {
    protected PropertyDefinitionBaseCollection()
    {
    }

    public virtual T this[ object propertyId ]
    {
      get
      {
        foreach( var item in Items )
        {
          if( item.TargetProperties.Contains( propertyId ) )
            return item;

          // Using the special character "*" in a string of TargetProperties will
          // return all the items containing the string (before or after) the "*".
          // ex : Prop* will return properties named Prop1, Prop2, Prop3...
          List<string> stringTargetProperties = item.TargetProperties.OfType<string>().ToList();
          if( ( stringTargetProperties != null ) && ( stringTargetProperties.Count > 0 ) )
          {
            if( propertyId is string )
            {
              string stringPropertyID = ( string )propertyId;
              foreach( var targetPropertyString in stringTargetProperties )
              {
                if( targetPropertyString.Contains( "*" ) )
                {
                  string searchString = targetPropertyString.Replace( "*", "" );
                  if( stringPropertyID.StartsWith( searchString ) || stringPropertyID.EndsWith( searchString ) )
                    return item;
                }
              }
            }
          }
          else
          {
            //Manage Interfaces
            var type = propertyId as Type;
            if( type != null )
            {
              foreach( Type targetProperty in item.TargetProperties )
              {
                if( targetProperty.IsAssignableFrom( type ) )
                  return item;
              }
            }
          }
        }

        return null;
      }
    }

    internal T GetRecursiveBaseTypes( Type type )
    {
      // If no definition for the current type, fall back on base type editor recursively.
      T ret = null;
      while( ret == null && type != null )
      {
        ret = this[ type ];
        type = type.BaseType;
      }
      return ret;
    }
  }

  public abstract class DefinitionCollectionBase<T> : ObservableCollection<T> where T : DefinitionBase
  {
    internal DefinitionCollectionBase()
    {
    }

    protected override void InsertItem( int index, T item )
    {
      if( item == null )
        throw new InvalidOperationException( @"Cannot insert null items in the collection." );

      item.Lock();
      base.InsertItem( index, item );
    }

    protected override void SetItem( int index, T item )
    {
      if( item == null )
        throw new InvalidOperationException( @"Cannot insert null items in the collection." );

      item.Lock();
      base.SetItem( index, item );
    }
  }
}