[ArcGIS]포인트에 래스터 픽셀값 계산하기 VBA Script

포인트 레이어의 필드에 래스터의 픽셀값을 계산하는 VBA 스크립트 예입니다.

가령, DEM(Digital Elevation Model)과 사용자가 수집한 포인트 레이어가 있을 경우, 포인트 위치의 표고값을  필드값으로 계산하고자 할 때 사용할 수 있습니다.

소스를 수정/확장하면 더 많은 결과를 만들어 낼 수 있을 것입니다.

▣ 전제사항
 - ArcMap에 포인트 레이어 및 래스터 레이어가 추가되어 있을 것
 - 포인트 레이어는 Shapefile, Personal GDB(mdb), File GDB일 것
 - 래스터 레이어는 단일밴드일 것
 - 스크립트 실행 조건은 편집 상태가 아닐것(Update Cursor를 사용하기 때문임)

▣ 메인 코드

펼쳐두기..

[code vb]
Option Explicit
Private m_ipMap As IMap

Public Sub CalculateRasterValue()
    Dim ipDoc As IMxDocument
    Set ipDoc = ThisDocument
    Set m_ipMap = ipDoc.FocusMap

    Dim ipFeatureLayer As IFeatureLayer, ipFeatureClass As IFeatureClass
    Dim fieldName As String, idxField As Long, ipField As IField
    Dim ipRasterLayer As IRasterLayer
   
    Set ipFeatureLayer = GetFeatureLayer(m_ipMap, "POINT")
    Set ipRasterLayer = GetRasterLayer(m_ipMap, "DEM")
   
    If (ipFeatureLayer Is Nothing Or ipRasterLayer Is Nothing) Then
        MsgBox "포인트 또는 래스터 레이어가 없습니다. 확인 바랍니다."
        Exit Sub
    End If
   
    Set ipFeatureClass = ipFeatureLayer.FeatureClass
           
    '1. 필드가 없으면 Double 필드를 추가함
    fieldName = "GRID_VALUE"
    AddField ipFeatureClass, fieldName, esriFieldTypeDouble
   
    idxField = ipFeatureClass.FindField(fieldName)
    Set ipField = ipFeatureClass.Fields.Field(idxField)
   
    '2. prepare identify
    Dim ipIdentify As IIdentify, ipRasterIDObj As IRasterIdentifyObj
    Dim ipArray As IArray, rasterValue As Variant
   
    Set ipIdentify = ipRasterLayer
   
    '3. 계산
    Dim totalCnt As Long, step As Long
    Dim ipQf As IQueryFilter, ipCursor As IFeatureCursor
    Dim ipFeature As IFeature, ipCurPoint As IPoint
       
    totalCnt = ipFeatureClass.FeatureCount(Nothing)
   
    Set ipQf = New QueryFilter
    ipQf.SubFields = ipFeatureClass.ShapeFieldName & ", " & fieldName
   
    Set ipCursor = ipFeatureClass.Update(ipQf, False)
    Set ipFeature = ipCursor.NextFeature
    Do Until ipFeature Is Nothing
        DoEvents
        Application.StatusBar.Message(0) = step & " / " & totalCnt & " calculated..."
        step = step + 1
       
        rasterValue = 0#
        Set ipCurPoint = ipFeature.ShapeCopy
       
        '래스터 레이어의 값 가져오기
        Set ipArray = ipIdentify.Identify(ipCurPoint)
        If (Not ipArray Is Nothing) Then
            If (ipArray.Count > 0) Then
                Set ipRasterIDObj = ipArray.Element(0)
                If (IsNumeric(ipRasterIDObj.Name)) Then
                    rasterValue = ipRasterIDObj.Name
                End If
            End If
        End If
       
        '필드 유형에 맞게 값 확인
        Select Case ipField.Type
            Case esriFieldTypeSmallInteger
                rasterValue = CInt(rasterValue)
            Case esriFieldTypeInteger
                rasterValue = CLng(rasterValue)
            Case esriFieldTypeDouble
                rasterValue = CDbl(rasterValue)
            Case esriFieldTypeString
                rasterValue = CStr(rasterValue)
        End Select
       
        '저장
        ipFeature.Value(idxField) = rasterValue
        ipCursor.UpdateFeature ipFeature
       
        Set ipFeature = ipCursor.NextFeature
    Loop
   
    MsgBox "계산을 완료하였습니다 !"
End Sub
[/code]

▣ 전체 소스코드

RasterValueCalculator.bas

▣ 계산결과 예시