// 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.
// 

//*************************************************************************************
//       ArcGIS Network Analyst extension - Load network layer from active analysis to the table of contents
//
//   This simple code shows how to load the network layer and its associated sources
//    to a map
//
//************************************************************************************

using ESRI.ArcGIS.Carto;
using ESRI.ArcGIS.Geodatabase;
using ESRI.ArcGIS.NetworkAnalyst;
using ESRI.ArcGIS.NetworkAnalystUI;

namespace LoadNetworkLayerFromActiveAnalysisToTOC
{
  public class LoadNetworkLayerFromActiveAnalysisToTOC : ESRI.ArcGIS.Desktop.AddIns.Button
  {
    public LoadNetworkLayerFromActiveAnalysisToTOC()
    {
    }

    /// <summary>
    /// OnClick is the main function for the add-in.  When the button is clicked in ArcMap,
    ///  this code will execute
    /// </summary>
    protected override void OnClick()
    {
      // Verify that the network extension has been loaded properly.
      var naExt = ArcMap.Application.FindExtensionByName("Network Analyst") as INetworkAnalystExtension;
      if (naExt == null || naExt.NAWindow == null)
        return;

      // There must be an active analysis layer
      INALayer naLayer = naExt.NAWindow.ActiveAnalysis;
      if (naLayer == null || naLayer.Context == null)
        return;

      // Use the network layer factory to generate the layers
      ILayerFactory layerFactory = new EngineNetworkLayerFactoryClass();
      var dataset = naLayer.Context.NetworkDataset as IDataset;
      if (layerFactory.get_CanCreate(dataset.FullName))
      {
        // Calling create will open a popup asking the user if the network sources
        //  should be added as well.  If the user clicks NO, then the returned
        //  IEnumLayer will only return a network layer for the network dataset.
        //  If the user clicks YES, then IEnumLayer will also return layers for 
        //  the network sources
        IEnumLayer enumLayer = layerFactory.Create(dataset.FullName);
        ILayer layer = enumLayer.Next();
        while (layer != null)
        {
          ArcMap.Document.FocusMap.AddLayer(layer);
          layer = enumLayer.Next();
        }
      }
    }

    protected override void OnUpdate()
    {
      Enabled = (ArcMap.Application != null);
    }
  }

}