SubsetNetworkEvaluatorsUI\RemoveSubsetAttributesCommand.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 RemoveSubsetAttributesCommand 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 reverse the affects of the AddSubsetAttributesCommand. It only /// removes those subset attributes that were or could have been added by the add command. You can always /// just remove the subset attributes manually using the attributes page of the network dataset property sheet. /// Removing the attribute will also remove the parameters and evaluator assignments. /// </summary> [Guid("EDD273A2-2AC2-4f2f-B096-7088B58F0667")] [ClassInterface(ClassInterfaceType.None)] [ProgId("SubsetNetworkEvaluatorsUI.RemoveSubsetAttributesCommand")] public sealed class RemoveSubsetAttributesCommand : 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); // // TODO: Add any COM unregistration code here // } #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 RemoveSubsetAttributesCommand() { base.m_category = "Network Analyst Samples"; //localizable text base.m_caption = "Remove Subset Attributes"; //localizable text base.m_message = "Remove Subset Attributes"; //localizable text base.m_toolTip = "Remove Subset Attributes"; //localizable text base.m_name = "NASamples_RemoveSubsetAttributes"; //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); 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 } }