// 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.Runtime.InteropServices;
using Microsoft.Win32;
using ESRI.ArcGIS.ADF.BaseClasses;
using ESRI.ArcGIS.ADF.CATIDs;
using ESRI.ArcGIS.Controls;
using ESRI.ArcGIS.Carto;
using ESRI.ArcGIS.Display;
using ESRI.ArcGIS.Geometry;
using ESRI.ArcGIS.Geodatabase;
using ESRI.ArcGIS.DataSourcesFile;

namespace RSSWeatherLayer
{
  /// <summary>
  /// Summary description for AddWeatherItemTool.
  /// </summary>
  [ClassInterface(ClassInterfaceType.None)]
  [Guid("D74A98E8-2CAA-4068-AA3D-C4DFA25BD5BE")]
  [ProgId("RSSWeatherLayer.AddWeatherItemTool")]
  [ComVisible(true)]
  public sealed class AddWeatherItemTool: BaseTool
  {
    #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);
      MxCommands.Register(regKey);
      ControlsCommands.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);
      MxCommands.Unregister(regKey);
      ControlsCommands.Unregister(regKey);
    }

    #endregion
    #endregion

    private IHookHelper               m_hookHelper      = null;
    private RSSWeatherLayerClass    m_weatherLayer    = null;
    private IFeatureClass              m_featureClass    = null;
    private ISpatialReference         m_spatialRefWGS84 = null;

    public AddWeatherItemTool()
    {
      base.m_category = "Weather";
      base.m_caption = "Add Weather Item";
      base.m_message = "Add Weather item";
      base.m_toolTip = "Add Weather item";
      base.m_name = base.m_category + "_" + base.m_caption;

      try
      {
        base.m_bitmap = new System.Drawing.Bitmap(GetType().Assembly.GetManifestResourceStream(GetType(), "AddWeatherItemTool.bmp"));
        base.m_cursor = new System.Windows.Forms.Cursor(GetType().Assembly.GetManifestResourceStream(GetType(), "AddWeatherItemTool.cur"));
      }
      catch (Exception ex)
      {
        System.Diagnostics.Trace.WriteLine(ex.Message, "Invalid Bitmap");
      }
    }

    #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)
    {
      //Instantiate the hook helper
      if (m_hookHelper == null)
        m_hookHelper = new HookHelperClass();

      //set the hook
      m_hookHelper.Hook = hook;

      //connect to the ZipCodes featureclass
      //get the ArcGIS path from the registry
      RegistryKey key = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\ESRI\ArcObjectsSdk10.1");
      string path = Convert.ToString(key.GetValue("InstallDir"));

      path = System.IO.Path.Combine(path, @"Samples\Data\USZipCodeData\");

      IWorkspaceFactory wf = new ShapefileWorkspaceFactoryClass() as IWorkspaceFactory;
      IWorkspace ws = wf.OpenFromFile(path, 0);
      IFeatureWorkspace fw = ws as IFeatureWorkspace;
      m_featureClass = fw.OpenFeatureClass("US_ZipCodes");

      m_spatialRefWGS84 = CreateGeoCoordSys();
    }

    /// <summary>
    /// Occurs when this command is clicked
    /// </summary>
    public override void OnClick()
    {
      try
      {
        if (0 == m_hookHelper.FocusMap.LayerCount)
          return;

        //get the weather layer
        IEnumLayer layers = m_hookHelper.FocusMap.get_Layers(null, false);
        layers.Reset();
        ILayer layer = layers.Next();
        while(layer != null)
        {
          if(layer is RSSWeatherLayerClass)
          {
            m_weatherLayer = (RSSWeatherLayerClass)layer;
            break;
          }
          layer = layers.Next();
        }
      }
      catch(Exception ex)
      {
        System.Diagnostics.Trace.WriteLine(ex.Message);
      }
    }

    /// <summary>
    /// Occurs when the user click inside the globe
    /// </summary>
    /// <param name="Button"></param>
    /// <param name="Shift"></param>
    /// <param name="X"></param>
    /// <param name="Y"></param>
    public override void OnMouseDown(int Button, int Shift, int X, int Y)
    {
      //make sure that the weatherlayer and the featureclass exists
      if ((null == m_weatherLayer) || (null == m_featureClass))
        return;

      //get the current point (convert the mouse coordinate into geographics coordinate)
      IActiveView activeView = m_hookHelper.ActiveView;
      IDisplayTransformation displayTrans = activeView.ScreenDisplay.DisplayTransformation;
      IPoint point = displayTrans.ToMapPoint(X, Y);
      point.SpatialReference = m_hookHelper.FocusMap.SpatialReference;
      
      //project the point to WGS1984 CoordSys
      if (null != point.SpatialReference)
      {
        if (point.SpatialReference.FactoryCode != m_spatialRefWGS84.FactoryCode)
          point.Project(m_spatialRefWGS84);
      }

      //create the spatial filter in order to select the relevant zipCode
      ISpatialFilter spatialFilter = new SpatialFilterClass();
      spatialFilter.Geometry = point as IGeometry;
      spatialFilter.GeometryField = m_featureClass.ShapeFieldName;
      
      //The spatial filter supposed to filter all the polygons containing the point
      spatialFilter.SpatialRel = esriSpatialRelEnum.esriSpatialRelWithin;

      //query the featureclass for the containing features
      IFeatureCursor featureCursor = m_featureClass.Search(spatialFilter, true);
      IFeature feature = featureCursor.NextFeature();
      if(null == feature)
        return;
      
      //get the zip code from the containing feature
      long zipCode = Convert.ToInt64(feature.get_Value(feature.Fields.FindField("ZIP")));
      
      //add the weather item to the layer
      m_weatherLayer.AddItem(zipCode, point.Y, point.X);

      //refresh the map
      activeView.PartialRefresh(esriViewDrawPhase.esriViewGeography, m_weatherLayer, displayTrans.FittedBounds);
      activeView.ScreenDisplay.UpdateWindow();

      //release the featurecursor
      Marshal.ReleaseComObject(featureCursor);
    }
  
    /// <summary>
    /// Occurs when the user move the mouse
    /// </summary>
    /// <param name="Button"></param>
    /// <param name="Shift"></param>
    /// <param name="X"></param>
    /// <param name="Y"></param>
    public override void OnMouseMove(int Button, int Shift, int X, int Y)
    {
      // TODO:  Add AddWeatherItemTool.OnMouseMove implementation
    }
  
    /// <summary>
    /// Occurs when then user release the mouse button
    /// </summary>
    /// <param name="Button"></param>
    /// <param name="Shift"></param>
    /// <param name="X"></param>
    /// <param name="Y"></param>
    public override void OnMouseUp(int Button, int Shift, int X, int Y)
    {
      // TODO:  Add AddWeatherItemTool.OnMouseUp implementation
    }
    #endregion

    private ISpatialReference CreateGeoCoordSys()
    {
      ISpatialReferenceFactory2 spatialRefFactory = new SpatialReferenceEnvironmentClass();

      IGeographicCoordinateSystem wgs84GeoCoordSys = spatialRefFactory.CreateGeographicCoordinateSystem((int)esriSRGeoCSType.esriSRGeoCS_WGS1984);

      return (ISpatialReference)wgs84GeoCoordSys;
    }
  }
}