src/PythonAPI/model/sketcher/line.py

   1 from GeomDataAPI import geomDataAPI_Point2D
   2 from model.roots import Interface
   3 from model.errors import WrongNumberOfArguments
   4
   5 class Line(Interface):
   6     """Interface for editing of a sketch line feature."""
   7     def __init__(self, feature, *args):
   8         Interface.__init__(self, feature)
   9         assert(self._feature.getKind() == "SketchLine")
  10
  11         # Initialize attributes
  12         self._start_point = geomDataAPI_Point2D(
  13             self._feature.data().attribute("StartPoint")
  14             )
  15         self._end_point = geomDataAPI_Point2D(
  16             self._feature.data().attribute("EndPoint")
  17             )
  18         # Set attribute values and execute
  19         if len(args) == 4:
  20             self.__createByCoordinates(*args)
  21         elif len(args) == 2:
  22             self.__createByPoints(*args)
  23         elif len(args) == 1:
  24             self.__createByName(*args)
  25         else:
  26             raise WrongNumberOfArguments(
  27                 "Arc takes 1, 2 or 4 arguments (%s given)" % len(args)
  28                 )
  29
  30     def __createByCoordinates(self, x1, y1, x2, y2):
  31         self._start_point.setValue(x1, y1)
  32         self._end_point.setValue(x2, y2)
  33         self._feature.execute()
  34
  35     def __createByPoints(self, p1, p2):
  36         self._start_point.setValue(p1.x(), p1.y())
  37         self._end_point.setValue(p2.x(), p2.y())
  38         self._feature.execute()
  39
  40     def __createByName(self, name):
  41         self._feature.data().selection("External").selectSubShape("EDGE", name)
  42         self._feature.execute()
  43
  44     def startPointData(self):
  45         return self._start_point
  46
  47     def endPointData(self):
  48         return self._end_point
  49
  50     def result(self):
  51         return self._feature.firstResult()