Subset network evaluators
SubsetNetworkEvaluatorsUI\AddSubsetAttributesCommand.cs
// Copyright 2012 ESRI
// 
// All rights reserved under the copyright laws of the United States
// and applicable international laws, treaties, and conventions.
// 
// You may freely redistribute and use this sample code, with or
// without modification, provided you include the original copyright
// notice and use restrictions.
// 
// See the use restrictions.
// 

using System;
using System.Drawing;
using System.Collections.Generic;
using System.Runtime.InteropServices;
using System.Windows.Forms;
using ESRI.ArcGIS.ADF.BaseClasses;
using ESRI.ArcGIS.ADF.CATIDs;
using ESRI.ArcGIS.Framework;
using ESRI.ArcGIS.esriSystem;
using ESRI.ArcGIS.Geodatabase;
using ESRI.ArcGIS.Catalog;
using ESRI.ArcGIS.CatalogUI;
using ESRI.ArcGIS.NetworkAnalystUI;
using SubsetNetworkEvaluators;

namespace SubsetNetworkEvaluatorsUI
{
  /// <summary>
  /// The AddSubsetAttributesCommand is a context menu item automatically added to the ArcCatalog
  /// Network Dataset context menu.  If the network analyst extension license is checked out
  /// you can use this command to quickly add new attributes with default subset evaluator assignments
  /// based on the attributes currently present in the context network dataset.  You can always just add
  /// the attributes along with the necessary parameters and assign these evaluators manually using the
  /// network dataset property pages instead of using this command.  The command is just a shortcut to quickly
  /// set up some subset attributes with default parameters and assignments.
  /// </summary>
  [Guid("E2B0245E-F707-4779-BEB3-9BA62D5325D6")]
  [ClassInterface(ClassInterfaceType.None)]
  [ProgId("SubsetNetworkEvaluatorsUI.AddSubsetAttributesCommand")]
  public sealed class AddSubsetAttributesCommand : BaseCommand
  {
    #region COM Registration Function(s)
    [ComRegisterFunction()]
    [ComVisible(false)]
    static void RegisterFunction(Type registerType)
    {
      // Required for ArcGIS Component Category Registrar support
      ArcGISCategoryRegistration(registerType);

      //
      // TODO: Add any COM registration code here
      //
    }

    [ComUnregisterFunction()]
    [ComVisible(false)]
    static void UnregisterFunction(Type registerType)
    {
      // Required for ArcGIS Component Category Registrar support
      ArcGISCategoryUnregistration(registerType);
    }

    #region ArcGIS Component Category Registrar generated code
    /// <summary>
    /// Required method for ArcGIS Component Category registration -
    /// Do not modify the contents of this method with the code editor.
    /// </summary>
    private static void ArcGISCategoryRegistration(Type registerType)
    {
      string regKey = string.Format("HKEY_CLASSES_ROOT\\CLSID\\{{{0}}}", registerType.GUID);
      GxCommands.Register(regKey);
      GxNetworkDatasetContextMenuCommands.Register(regKey);
    }
    /// <summary>
    /// Required method for ArcGIS Component Category unregistration -
    /// Do not modify the contents of this method with the code editor.
    /// </summary>
    private static void ArcGISCategoryUnregistration(Type registerType)
    {
      string regKey = string.Format("HKEY_CLASSES_ROOT\\CLSID\\{{{0}}}", registerType.GUID);
      GxCommands.Unregister(regKey);
      GxNetworkDatasetContextMenuCommands.Unregister(regKey);
    }

    #endregion
    #endregion

    private IApplication m_application = null;
    private INetworkAnalystExtension m_nax = null;

    public AddSubsetAttributesCommand()
    {
      base.m_category = "Network Analyst Samples";   //localizable text
      base.m_caption = "Add Subset Attributes";      //localizable text
      base.m_message = "Add Subset Attributes";      //localizable text 
      base.m_toolTip = "Add Subset Attributes";      //localizable text 
      base.m_name = "NASamples_AddSubsetAttributes"; //unique id, non-localizable (e.g. "MyCategory_ArcMapCommand")
    }

    #region Overriden Class Methods

    /// <summary>
    /// Occurs when this command is created
    /// </summary>
    /// <param name="hook">Instance of the application</param>
    public override void OnCreate(object hook)
    {
      if (hook == null)
        return;

      m_application = hook as IApplication;
      m_nax = SubsetHelperUI.GetNAXConfiguration(m_application) as INetworkAnalystExtension;
    }

    /// <summary>
    /// Occurs when this command is clicked
    /// </summary>
    public override void OnClick()
    {
      IGxApplication gxApp = m_application as IGxApplication;
      IGxDataset gxDataset = null;
      esriDatasetType dsType = esriDatasetType.esriDTAny;

      if (gxApp != null)
      {
        gxDataset = gxApp.SelectedObject as IGxDataset;
        dsType = gxDataset.Type;
      }

      if (dsType != esriDatasetType.esriDTNetworkDataset)
        return;

      IDataset ds = gxDataset.Dataset;
      if (ds == null)
        return;

      INetworkDataset nds = ds as INetworkDataset;
      if (nds == null)
        return;

      if (!nds.Buildable)
        return;

      INetworkBuild netBuild = nds as INetworkBuild;
      if (netBuild == null)
        return;

      IDatasetComponent dsComponent = nds as IDatasetComponent;
      IDENetworkDataset deNet = null;
      if (dsComponent != null)
        deNet = dsComponent.DataElement as IDENetworkDataset;

      if (deNet == null)
        return;

      FilterSubsetEvaluator.RemoveFilterSubsetAttribute(deNet);
      ScaleSubsetEvaluator.RemoveScaleSubsetAttributes(deNet);

      FilterSubsetEvaluator.AddFilterSubsetAttribute(deNet);
      ScaleSubsetEvaluator.AddScaleSubsetAttributes(deNet);

      netBuild.UpdateSchema(deNet);
    }

    public override bool Enabled
    {
      get
      {
        IGxApplication gxApp = m_application as IGxApplication;
        IGxDataset gxDataset = null;
        esriDatasetType dsType = esriDatasetType.esriDTAny;

        bool naxEnabled = false;
        IExtensionConfig naxConfig = m_nax as IExtensionConfig;
        if (naxConfig != null)
          naxEnabled = naxConfig.State == esriExtensionState.esriESEnabled;

        if (naxEnabled)
        {
          if (gxApp != null)
          {
            gxDataset = gxApp.SelectedObject as IGxDataset;
            dsType = gxDataset.Type;
          }
        }

        return (dsType == esriDatasetType.esriDTNetworkDataset);
      }
    }

    #endregion
  }
}