##Copyright 2008-2013 Jelle Feringa (jelleferinga@gmail.com)

##

##This file is part of pythonOCC.

##

##pythonOCC is free software: you can redistribute it and/or modify

##it under the terms of the GNU Lesser General Public License as published by

##the Free Software Foundation, either version 3 of the License, or

##(at your option) any later version.

##

##pythonOCC is distributed in the hope that it will be useful,

##but WITHOUT ANY WARRANTY; without even the implied warranty of

##MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

##GNU Lesser General Public License for more details.

##

##You should have received a copy of the GNU Lesser General Public License

##along with pythonOCC. If not, see <http://www.gnu.org/licenses/>

from OCC.gp import gp_Pnt, gp_Vec, gp_Dir, gp_XYZ, gp_Pnt2d

from OCC.TopoDS import TopoDS_Vertex

from OCC.ShapeBuild import ShapeBuild_ReShape

from base import KbeObject

from Construct import make_vertex

from Common import vertex2pnt

class Vertex(TopoDS_Vertex, KbeObject):

"""

wraps gp_Pnt

"""

_n = 0

def __init__(self, x, y, z):

super(Vertex, self).__init__()

"""Constructor for KbeVertex"""

KbeObject.__init__(self, name='Vertex #{0}'.format(self._n))

self._n += 1 # should be a property of KbeObject

self._pnt = gp_Pnt(x, y, z)

self._vertex = make_vertex(self._pnt)

TopoDS_Vertex.__init__(self, self._vertex)

def _update(self):

"""

"""

# TODO: perhaps should take an argument until which topological level

# topological entities bound to the vertex should be updated too...

reshape = ShapeBuild_ReShape()

reshape.Replace(self._vertex, make_vertex(self._pnt))

@staticmethod

def from_vertex(cls, vertex):

return Vertex.from_pnt(vertex2pnt(vertex))

@staticmethod

def from_pnt(cls, pnt):

x, y, z = pnt.Coord()

return cls(x, y, z)

@classmethod

def from_vec(cls):

raise NotImplementedError

@property

def x(self):

return self._pnt.X()

@x.setter

def x(self, val):

self._pnt.SetX(val)

self._update()

@property

def y(self):

return self._pnt.Y()

@y.setter

def y(self, val):

self._pnt.SetY(val)

self._update()

@property

def z(self):

return self._pnt.Z()

@z.setter

def z(self, val):

self._pnt.SetZ(val)

self._update()

@property

def xyz(self):

return self._pnt.Coord()

@xyz.setter

def xyz(self, *val):

self._pnt.SetXYZ(*val)

self._update()

def __repr__(self):

return self.name

@property

def as_vec(self):

'''returns a gp_Vec version of self'''

return gp_Vec(*self._pnt.Coord())

@property

def as_dir(self):

'''returns a gp_Dir version of self'''

return gp_Dir(*self._pnt.Coord())

@property

def as_xyz(self):

'''returns a gp_XYZ version of self'''

return gp_XYZ(*self._pnt.Coord())

@property

def as_pnt(self):

return self._pnt

@property

def as_2d(self):

'''returns a gp_Pnt2d version of self'''

return gp_Pnt2d(*self._pnt.Coord()[:2])