SelectByLineTool

SelectByLineTool.cs

using System;
using System.Collections.Generic;
using System.Text;
using ESRI.ArcGIS.ArcMapUI;
using ESRI.ArcGIS.Carto;
using ESRI.ArcGIS.Geometry;
using ESRI.ArcGIS.esriSystem;

namespace SelectionSample
{
  public class SelectByLineTool : ESRI.ArcGIS.Desktop.AddIns.Tool
  {
    private bool m_isMouseDown = false;
    private ESRI.ArcGIS.Display.INewLineFeedback m_lineFeedback;
    private IActiveView m_focusMap;

    public SelectByLineTool()
    {

    }

    protected override void OnMouseDown(MouseEventArgs arg)
    {
      IMxDocument mxDoc = ArcMap.Document;
      m_focusMap = mxDoc.FocusMap as IActiveView;
      IPoint point = m_focusMap.ScreenDisplay.DisplayTransformation.ToMapPoint(arg.X, arg.Y) as IPoint;

      if (m_lineFeedback == null)
      {
        m_lineFeedback = new ESRI.ArcGIS.Display.NewLineFeedback();
        m_lineFeedback.Display = m_focusMap.ScreenDisplay;
        m_lineFeedback.Start(point);
      }
      else
      {
        m_lineFeedback.AddPoint(point);
      }

      m_isMouseDown = true;
    }

    protected override void OnDoubleClick()
    {
      m_focusMap.PartialRefresh(esriViewDrawPhase.esriViewGeoSelection, null, null);

      IPolyline polyline;

      if (m_lineFeedback != null)
      {
        polyline = m_lineFeedback.Stop();
        if (polyline != null)
          ArcMap.Document.FocusMap.SelectByShape(polyline, null, false);
      }


      m_focusMap.PartialRefresh(esriViewDrawPhase.esriViewGeoSelection, null, null);

      m_lineFeedback = null;
      m_isMouseDown = false;
    }

    protected override void OnMouseMove(MouseEventArgs arg)
    {
      if (!m_isMouseDown) return;

      IPoint point = m_focusMap.ScreenDisplay.DisplayTransformation.ToMapPoint(arg.X, arg.Y) as IPoint;
      m_lineFeedback.MoveTo(point);
    }

    protected override void OnUpdate()
    {
      if (!SelectionExtension.IsExtensionEnabled())
      {
        this.Enabled = false;
        return;
      }

      this.Enabled = SelectionExtension.HasSelectableLayer();
    }
  }
}

SelectByLineTool.vb

Imports Microsoft.VisualBasic
Imports System
Imports System.Collections.Generic
Imports System.Text
Imports ESRI.ArcGIS.ArcMapUI
Imports ESRI.ArcGIS.Carto
Imports ESRI.ArcGIS.Geometry
Imports ESRI.ArcGIS.esriSystem

Imports My

Namespace SelectionSample
  Public Class SelectByLineTool
	  Inherits ESRI.ArcGIS.Desktop.AddIns.Tool
	Private m_isMouseDown As Boolean = False
	Private m_lineFeedback As ESRI.ArcGIS.Display.INewLineFeedback
	Private m_focusMap As IActiveView

	Public Sub New()

	End Sub

	Protected Overrides Sub OnMouseDown(ByVal arg As MouseEventArgs)
      Dim mxDoc As IMxDocument = ArcMap.Document
	  m_focusMap = TryCast(mxDoc.FocusMap, IActiveView)
	  Dim point As IPoint = TryCast(m_focusMap.ScreenDisplay.DisplayTransformation.ToMapPoint(arg.X, arg.Y), IPoint)

	  If m_lineFeedback Is Nothing Then
		m_lineFeedback = New ESRI.ArcGIS.Display.NewLineFeedback()
		m_lineFeedback.Display = m_focusMap.ScreenDisplay
		m_lineFeedback.Start(point)
	  Else
		m_lineFeedback.AddPoint(point)
	  End If

	  m_isMouseDown = True
	End Sub

	Protected Overrides Sub OnDoubleClick()
	  m_focusMap.PartialRefresh(esriViewDrawPhase.esriViewGeoSelection, Nothing, Nothing)

	  Dim polyline As IPolyline

	  If m_lineFeedback IsNot Nothing Then
		polyline = m_lineFeedback.Stop()
		If polyline IsNot Nothing Then
		  ArcMap.Document.FocusMap.SelectByShape(polyline, Nothing, False)
		End If
	  End If


	  m_focusMap.PartialRefresh(esriViewDrawPhase.esriViewGeoSelection, Nothing, Nothing)

	  m_lineFeedback = Nothing
	  m_isMouseDown = False
	End Sub

	Protected Overrides Sub OnMouseMove(ByVal arg As MouseEventArgs)
	  If (Not m_isMouseDown) Then
		  Return
	  End If

	  Dim point As IPoint = TryCast(m_focusMap.ScreenDisplay.DisplayTransformation.ToMapPoint(arg.X, arg.Y), IPoint)
	  m_lineFeedback.MoveTo(point)
	End Sub

	Protected Overrides Sub OnUpdate()
	  If (Not SelectionExtension.IsExtensionEnabled()) Then
		Me.Enabled = False
		Return
	  End If

	  Me.Enabled = SelectionExtension.HasSelectableLayer()
	End Sub
  End Class
End Namespace