ObjectContainerHelper.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;
using System.Collections.Generic;
using System.ComponentModel;
#if !VS2008
using System.ComponentModel.DataAnnotations;
#endif
using System.Diagnostics;
using System.Linq;
using Xceed.Wpf.Toolkit.PropertyGrid.Attributes;

namespace Xceed.Wpf.Toolkit.PropertyGrid
{
  internal class ObjectContainerHelper : ObjectContainerHelperBase
  {
    private object _selectedObject;

    public ObjectContainerHelper( IPropertyContainer propertyContainer, object selectedObject )
      : base( propertyContainer )
    {
      _selectedObject = selectedObject;
    }

    private object SelectedObject
    {
      get
      {
        return _selectedObject;
      }
    }

    protected override string GetDefaultPropertyName()
    {
      object selectedObject = SelectedObject;
      return ( selectedObject != null ) ? ObjectContainerHelperBase.GetDefaultPropertyName( SelectedObject ) : ( string )null;
    }

    protected override void GenerateSubPropertiesCore( Action<IEnumerable<PropertyItem>> updatePropertyItemsCallback )
    {
      var propertyItems = new List<PropertyItem>();

      if( SelectedObject != null )
      {
        try
        {
          var descriptors = new List<PropertyDescriptor>();
          {
            descriptors = ObjectContainerHelperBase.GetPropertyDescriptors( SelectedObject, this.PropertyContainer.HideInheritedProperties );
          }

          foreach( var descriptor in descriptors )
          {
            var propertyDef = this.GetPropertyDefinition( descriptor );
            bool isBrowsable = false;

            var isPropertyBrowsable = this.PropertyContainer.IsPropertyVisible( descriptor );
            if( isPropertyBrowsable.HasValue )
            {
              isBrowsable = isPropertyBrowsable.Value;
            }
            else
            {
#if !VS2008
              var displayAttribute = PropertyGridUtilities.GetAttribute<DisplayAttribute>( descriptor );
              if( displayAttribute != null )
              {
                var autoGenerateField = displayAttribute.GetAutoGenerateField();
                isBrowsable = this.PropertyContainer.AutoGenerateProperties 
                              && ((autoGenerateField.HasValue && autoGenerateField.Value) || !autoGenerateField.HasValue);
              }
              else
#endif
              {
                isBrowsable = descriptor.IsBrowsable && this.PropertyContainer.AutoGenerateProperties;
              }

              if( propertyDef != null )
              {
                isBrowsable = propertyDef.IsBrowsable.GetValueOrDefault( isBrowsable );
              }
            }

            if( isBrowsable )
            {
              var prop = this.CreatePropertyItem( descriptor, propertyDef );
              if( prop != null )
              {
                propertyItems.Add( prop );
              }
            }
          }
        }
        catch( Exception e )
        {
          //TODO: handle this some how
          Debug.WriteLine( "Property creation failed." );
          Debug.WriteLine( e.StackTrace );
        }
      }

      updatePropertyItemsCallback.Invoke( propertyItems );
    }


    private PropertyItem CreatePropertyItem( PropertyDescriptor property, PropertyDefinition propertyDef )
    {
      DescriptorPropertyDefinition definition = new DescriptorPropertyDefinition( property, SelectedObject, this.PropertyContainer );                                                                                 
      definition.InitProperties();

      this.InitializeDescriptorDefinition( definition, propertyDef );
      PropertyItem propertyItem = new PropertyItem( definition );
      Debug.Assert( SelectedObject != null );
      propertyItem.Instance = SelectedObject;
      propertyItem.CategoryOrder = this.GetCategoryOrder( definition.CategoryValue );

      propertyItem.WillRefreshPropertyGrid = this.GetWillRefreshPropertyGrid( property );
      return propertyItem;
    }

    private int GetCategoryOrder(object categoryValue)
    {
      Debug.Assert(SelectedObject != null);

      if (categoryValue == null)
        return int.MaxValue;

      object selectedObject = SelectedObject;

      var orderAttribute = TypeDescriptor.GetAttributes(selectedObject)
        .OfType<CategoryOrderAttribute>()
        .FirstOrDefault(a => Equals(a.CategoryValue, categoryValue));

      return orderAttribute?.Order ?? int.MaxValue;
    }
  }
}