Dim oDoc As Document = ThisDoc.Document
Dim oDef As Object = oDoc.ComponentDefinition
Dim oDcc As ComponentOccurrence = Nothing
On Error Resume Next

'부품과 어셈블리 모두 적용할수 있는 코드
Dim oBody As SurfaceBody
If TypeOf oDoc Is PartDocument Then
oDef = DirectCast(oDoc, PartDocument).ComponentDefinition
oBody = oDef.SurfaceBodies(1)
ElseIf TypeOf oDoc Is AssemblyDocument Then
oDef = DirectCast(oDoc, AssemblyDocument).ComponentDefinition
Dim oOcc As ComponentOccurrence = oDoc.ComponentDefinition.Occurrences.Item(1)
Dim oPartDef As PartComponentDefinition = oOcc.Definition
oBody = oPartDef.SurfaceBodies.Item(1)
Else
'MessageBox.Show("이 코드는 부품 또는 어셈블리 문서", "ilogic")
End If