2019-05-28 12:00:15 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
2020-07-31 06:33:54 +00:00
|
|
|
# cython: language_level=3
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
"""Wrapper source code of Solvespace.
|
|
|
|
|
|
|
|
author: Yuan Chang
|
|
|
|
copyright: Copyright (C) 2016-2019
|
|
|
|
license: GPLv3+
|
|
|
|
email: pyslvs@gmail.com
|
|
|
|
"""
|
|
|
|
|
|
|
|
from cpython.object cimport Py_EQ, Py_NE
|
|
|
|
from collections import Counter
|
|
|
|
|
|
|
|
|
2021-12-14 01:37:51 +00:00
|
|
|
def _create_sys(dof_v, g, param_list, entity_list, cons_list):
|
|
|
|
cdef SolverSystem s = SolverSystem.__new__(SolverSystem)
|
|
|
|
s.dof_v = dof_v
|
|
|
|
s.g = g
|
|
|
|
s.param_list = param_list
|
|
|
|
s.entity_list = entity_list
|
|
|
|
s.cons_list = cons_list
|
|
|
|
return s
|
|
|
|
|
|
|
|
|
2019-05-28 12:00:15 +00:00
|
|
|
cpdef tuple quaternion_u(double qw, double qx, double qy, double qz):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Input quaternion, return unit vector of U axis.
|
|
|
|
|
2020-07-31 06:33:54 +00:00
|
|
|
Where `qw`, `qx`, `qy`, `qz` are corresponded to the W, X, Y, Z value of
|
2020-02-22 07:02:20 +00:00
|
|
|
quaternion.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef double x, y, z
|
2019-11-20 14:00:27 +00:00
|
|
|
Slvs_QuaternionU(qw, qx, qy, qz, &x, &y, &z)
|
2019-05-28 12:00:15 +00:00
|
|
|
return x, y, z
|
|
|
|
|
|
|
|
|
|
|
|
cpdef tuple quaternion_v(double qw, double qx, double qy, double qz):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Input quaternion, return unit vector of V axis.
|
|
|
|
|
|
|
|
Signature is same as [quaternion_u](#quaternion_u).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef double x, y, z
|
|
|
|
Slvs_QuaternionV(qw, qx, qy, qz, &x, &y, &z)
|
|
|
|
return x, y, z
|
|
|
|
|
|
|
|
|
|
|
|
cpdef tuple quaternion_n(double qw, double qx, double qy, double qz):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Input quaternion, return unit vector of normal.
|
|
|
|
|
|
|
|
Signature is same as [quaternion_u](#quaternion_u).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef double x, y, z
|
|
|
|
Slvs_QuaternionN(qw, qx, qy, qz, &x, &y, &z)
|
|
|
|
return x, y, z
|
|
|
|
|
|
|
|
|
|
|
|
cpdef tuple make_quaternion(double ux, double uy, double uz, double vx, double vy, double vz):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Input two unit vector, return quaternion.
|
|
|
|
|
|
|
|
Where `ux`, `uy`, `uz` are corresponded to the value of U vector;
|
|
|
|
`vx`, `vy`, `vz` are corresponded to the value of V vector.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef double qw, qx, qy, qz
|
|
|
|
Slvs_MakeQuaternion(ux, uy, uz, vx, vy, vz, &qw, &qx, &qy, &qz)
|
|
|
|
return qw, qx, qy, qz
|
|
|
|
|
|
|
|
|
|
|
|
cdef class Params:
|
|
|
|
|
|
|
|
"""Python object to handle multiple parameter handles."""
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
cdef Params create(Slvs_hParam *p, size_t count):
|
|
|
|
"""Constructor."""
|
|
|
|
cdef Params params = Params.__new__(Params)
|
|
|
|
cdef size_t i
|
|
|
|
for i in range(count):
|
|
|
|
params.param_list.push_back(p[i])
|
|
|
|
return params
|
|
|
|
|
2019-10-15 08:13:16 +00:00
|
|
|
def __richcmp__(self, Params other, int op) -> bint:
|
|
|
|
"""Compare the parameters."""
|
|
|
|
if op == Py_EQ:
|
|
|
|
return self.param_list == other.param_list
|
|
|
|
elif op == Py_NE:
|
|
|
|
return self.param_list != other.param_list
|
|
|
|
else:
|
|
|
|
raise TypeError(
|
|
|
|
f"'{op}' not support between instances of "
|
|
|
|
f"{type(self)} and {type(other)}"
|
|
|
|
)
|
|
|
|
|
2019-05-28 12:00:15 +00:00
|
|
|
def __repr__(self) -> str:
|
2019-10-15 08:13:16 +00:00
|
|
|
m = f"{type(self).__name__}(["
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef size_t i
|
|
|
|
cdef size_t s = self.param_list.size()
|
|
|
|
for i in range(s):
|
|
|
|
m += str(<int>self.param_list[i])
|
|
|
|
if i != s - 1:
|
|
|
|
m += ", "
|
|
|
|
m += "])"
|
|
|
|
return m
|
|
|
|
|
|
|
|
# A virtual work plane that present 3D entity or constraint.
|
|
|
|
cdef Entity _E_FREE_IN_3D = Entity.__new__(Entity)
|
|
|
|
_E_FREE_IN_3D.t = SLVS_E_WORKPLANE
|
|
|
|
_E_FREE_IN_3D.h = SLVS_FREE_IN_3D
|
|
|
|
_E_FREE_IN_3D.g = 0
|
|
|
|
_E_FREE_IN_3D.params = Params.create(NULL, 0)
|
|
|
|
|
|
|
|
# A "None" entity used to fill in constraint option.
|
|
|
|
cdef Entity _E_NONE = Entity.__new__(Entity)
|
|
|
|
_E_NONE.t = 0
|
|
|
|
_E_NONE.h = 0
|
|
|
|
_E_NONE.g = 0
|
|
|
|
_E_NONE.params = Params.create(NULL, 0)
|
|
|
|
|
|
|
|
# Entity names
|
2019-10-15 08:13:16 +00:00
|
|
|
_NAME_OF_ENTITIES = {
|
2019-05-28 12:00:15 +00:00
|
|
|
SLVS_E_POINT_IN_3D: "point 3d",
|
|
|
|
SLVS_E_POINT_IN_2D: "point 2d",
|
|
|
|
SLVS_E_NORMAL_IN_2D: "normal 2d",
|
|
|
|
SLVS_E_NORMAL_IN_3D: "normal 3d",
|
|
|
|
SLVS_E_DISTANCE: "distance",
|
|
|
|
SLVS_E_WORKPLANE: "work plane",
|
|
|
|
SLVS_E_LINE_SEGMENT: "line segment",
|
|
|
|
SLVS_E_CUBIC: "cubic",
|
|
|
|
SLVS_E_CIRCLE: "circle",
|
|
|
|
SLVS_E_ARC_OF_CIRCLE: "arc",
|
|
|
|
}
|
|
|
|
|
|
|
|
# Constraint names
|
2019-10-15 08:13:16 +00:00
|
|
|
_NAME_OF_CONSTRAINTS = {
|
|
|
|
SLVS_C_POINTS_COINCIDENT: "points coincident",
|
|
|
|
SLVS_C_PT_PT_DISTANCE: "point point distance",
|
|
|
|
SLVS_C_PT_PLANE_DISTANCE: "point plane distance",
|
|
|
|
SLVS_C_PT_LINE_DISTANCE: "point line distance",
|
|
|
|
SLVS_C_PT_FACE_DISTANCE: "point face distance",
|
|
|
|
SLVS_C_PT_IN_PLANE: "point in plane",
|
|
|
|
SLVS_C_PT_ON_LINE: "point on line",
|
|
|
|
SLVS_C_PT_ON_FACE: "point on face",
|
|
|
|
SLVS_C_EQUAL_LENGTH_LINES: "equal length lines",
|
|
|
|
SLVS_C_LENGTH_RATIO: "length ratio",
|
|
|
|
SLVS_C_EQ_LEN_PT_LINE_D: "equal length point line distance",
|
|
|
|
SLVS_C_EQ_PT_LN_DISTANCES: "equal point line distance",
|
|
|
|
SLVS_C_EQUAL_ANGLE: "equal angle",
|
|
|
|
SLVS_C_EQUAL_LINE_ARC_LEN: "equal line arc length",
|
|
|
|
SLVS_C_SYMMETRIC: "symmetric",
|
|
|
|
SLVS_C_SYMMETRIC_HORIZ: "symmetric horizontal",
|
|
|
|
SLVS_C_SYMMETRIC_VERT: "symmetric vertical",
|
|
|
|
SLVS_C_SYMMETRIC_LINE: "symmetric line",
|
|
|
|
SLVS_C_AT_MIDPOINT: "at midpoint",
|
|
|
|
SLVS_C_HORIZONTAL: "horizontal",
|
|
|
|
SLVS_C_VERTICAL: "vertical",
|
|
|
|
SLVS_C_DIAMETER: "diameter",
|
|
|
|
SLVS_C_PT_ON_CIRCLE: "point on circle",
|
|
|
|
SLVS_C_SAME_ORIENTATION: "same orientation",
|
|
|
|
SLVS_C_ANGLE: "angle",
|
|
|
|
SLVS_C_PARALLEL: "parallel",
|
|
|
|
SLVS_C_PERPENDICULAR: "perpendicular",
|
|
|
|
SLVS_C_ARC_LINE_TANGENT: "arc line tangent",
|
|
|
|
SLVS_C_CUBIC_LINE_TANGENT: "cubic line tangent",
|
|
|
|
SLVS_C_EQUAL_RADIUS: "equal radius",
|
|
|
|
SLVS_C_PROJ_PT_DISTANCE: "project point distance",
|
|
|
|
SLVS_C_WHERE_DRAGGED: "where dragged",
|
|
|
|
SLVS_C_CURVE_CURVE_TANGENT: "curve curve tangent",
|
|
|
|
SLVS_C_LENGTH_DIFFERENCE: "length difference",
|
2019-05-28 12:00:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
cdef class Entity:
|
|
|
|
|
2021-12-12 06:15:06 +00:00
|
|
|
"""The handles of entities.
|
|
|
|
|
|
|
|
This handle **should** be dropped after system removed.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
FREE_IN_3D = _E_FREE_IN_3D
|
|
|
|
NONE = _E_NONE
|
|
|
|
|
|
|
|
@staticmethod
|
2021-12-12 06:15:06 +00:00
|
|
|
cdef Entity create(Slvs_Entity *e):
|
2019-05-28 12:00:15 +00:00
|
|
|
"""Constructor."""
|
|
|
|
cdef Entity entity = Entity.__new__(Entity)
|
|
|
|
with nogil:
|
|
|
|
entity.t = e.type
|
|
|
|
entity.h = e.h
|
|
|
|
entity.wp = e.wrkpl
|
|
|
|
entity.g = e.group
|
2021-12-12 06:15:06 +00:00
|
|
|
cdef size_t p_size
|
|
|
|
if e.type == SLVS_E_DISTANCE:
|
|
|
|
p_size = 1
|
|
|
|
elif e.type == SLVS_E_POINT_IN_2D:
|
|
|
|
p_size = 2
|
|
|
|
elif e.type == SLVS_E_POINT_IN_3D:
|
|
|
|
p_size = 3
|
|
|
|
elif e.type == SLVS_E_NORMAL_IN_3D:
|
|
|
|
p_size = 4
|
|
|
|
else:
|
|
|
|
p_size = 0
|
2019-05-28 12:00:15 +00:00
|
|
|
entity.params = Params.create(e.param, p_size)
|
|
|
|
return entity
|
|
|
|
|
2019-10-10 07:44:59 +00:00
|
|
|
def __richcmp__(self, Entity other, int op) -> bint:
|
2019-05-28 12:00:15 +00:00
|
|
|
"""Compare the entities."""
|
|
|
|
if op == Py_EQ:
|
|
|
|
return (
|
|
|
|
self.t == other.t and
|
|
|
|
self.h == other.h and
|
|
|
|
self.wp == other.wp and
|
|
|
|
self.g == other.g and
|
|
|
|
self.params == other.params
|
|
|
|
)
|
|
|
|
elif op == Py_NE:
|
2019-10-15 08:13:16 +00:00
|
|
|
return not (self == other)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(
|
|
|
|
f"'{op}' not support between instances of "
|
|
|
|
f"{type(self)} and {type(other)}"
|
|
|
|
)
|
|
|
|
|
|
|
|
cpdef bint is_3d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 3D entity."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.wp == SLVS_FREE_IN_3D
|
|
|
|
|
|
|
|
cpdef bint is_none(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a empty entity."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.h == 0
|
|
|
|
|
|
|
|
cpdef bint is_point_2d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 2D point."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_POINT_IN_2D
|
|
|
|
|
|
|
|
cpdef bint is_point_3d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 3D point."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_POINT_IN_3D
|
|
|
|
|
|
|
|
cpdef bint is_point(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a point."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.is_point_2d() or self.is_point_3d()
|
|
|
|
|
|
|
|
cpdef bint is_normal_2d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 2D normal."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_NORMAL_IN_2D
|
|
|
|
|
|
|
|
cpdef bint is_normal_3d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 3D normal."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_NORMAL_IN_3D
|
|
|
|
|
|
|
|
cpdef bint is_normal(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a normal."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.is_normal_2d() or self.is_normal_3d()
|
|
|
|
|
|
|
|
cpdef bint is_distance(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a distance."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_DISTANCE
|
|
|
|
|
|
|
|
cpdef bint is_work_plane(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a work plane."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_WORKPLANE
|
|
|
|
|
|
|
|
cpdef bint is_line_2d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 2D line."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.is_line() and not self.is_3d()
|
|
|
|
|
|
|
|
cpdef bint is_line_3d(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a 3D line."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.is_line() and self.is_3d()
|
|
|
|
|
|
|
|
cpdef bint is_line(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a line."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_LINE_SEGMENT
|
|
|
|
|
|
|
|
cpdef bint is_cubic(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a cubic."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_CUBIC
|
|
|
|
|
|
|
|
cpdef bint is_circle(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a circle."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_CIRCLE
|
|
|
|
|
|
|
|
cpdef bint is_arc(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return True if this is a arc."""
|
2019-05-28 12:00:15 +00:00
|
|
|
return self.t == SLVS_E_ARC_OF_CIRCLE
|
|
|
|
|
|
|
|
def __repr__(self) -> str:
|
|
|
|
cdef int h = <int>self.h
|
|
|
|
cdef int g = <int>self.g
|
|
|
|
cdef str t = _NAME_OF_ENTITIES[<int>self.t]
|
|
|
|
return (
|
2019-09-10 09:34:12 +00:00
|
|
|
f"{type(self).__name__}"
|
2019-05-28 12:00:15 +00:00
|
|
|
f"(handle={h}, group={g}, type=<{t}>, is_3d={self.is_3d()}, params={self.params})"
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
cdef class SolverSystem:
|
|
|
|
|
2020-02-22 07:02:20 +00:00
|
|
|
"""A solver system of Python-Solvespace.
|
|
|
|
|
2021-12-12 06:15:06 +00:00
|
|
|
The operation of entities and constraints are using the methods of this class.
|
2020-02-22 07:02:20 +00:00
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
|
2021-12-14 01:37:51 +00:00
|
|
|
def __cinit__(self):
|
|
|
|
self.g = 0
|
2019-05-28 12:00:15 +00:00
|
|
|
|
2021-12-11 10:49:15 +00:00
|
|
|
def __reduce__(self):
|
2021-12-14 01:37:51 +00:00
|
|
|
return (_create_sys, (self.dof_v, self.g, self.param_list, self.entity_list, self.cons_list))
|
2021-12-11 10:49:15 +00:00
|
|
|
|
2021-12-12 06:15:06 +00:00
|
|
|
def entity(self, int i) -> Entity:
|
|
|
|
"""Generate entity handle, it can only be used with this system.
|
|
|
|
|
|
|
|
Negative index is allowed.
|
|
|
|
"""
|
|
|
|
if i < 0:
|
|
|
|
return Entity.create(&self.entity_list[self.entity_list.size() + i])
|
|
|
|
elif i >= self.entity_list.size():
|
|
|
|
raise IndexError(f"index {i} is out of bound {self.entity_list.size()}")
|
|
|
|
else:
|
|
|
|
return Entity.create(&self.entity_list[i])
|
|
|
|
|
2021-12-11 12:53:13 +00:00
|
|
|
cpdef SolverSystem copy(self):
|
2021-12-11 10:49:15 +00:00
|
|
|
"""Copy the solver."""
|
2021-12-14 01:37:51 +00:00
|
|
|
return _create_sys(self.dof_v, self.g, self.param_list, self.entity_list, self.cons_list)
|
2021-12-11 10:49:15 +00:00
|
|
|
|
2019-05-28 12:00:15 +00:00
|
|
|
cpdef void clear(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Clear the system."""
|
2021-12-12 02:51:01 +00:00
|
|
|
self.dof_v = 0
|
2019-05-28 12:00:15 +00:00
|
|
|
self.g = 0
|
|
|
|
self.param_list.clear()
|
|
|
|
self.entity_list.clear()
|
|
|
|
self.cons_list.clear()
|
|
|
|
self.failed_list.clear()
|
|
|
|
|
|
|
|
cpdef void set_group(self, size_t g):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Set the current group (`g`)."""
|
2021-12-11 12:53:13 +00:00
|
|
|
self.g = g
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef int group(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return the current group."""
|
2021-12-11 12:53:13 +00:00
|
|
|
return self.g
|
2019-05-28 12:00:15 +00:00
|
|
|
|
2019-07-11 12:40:33 +00:00
|
|
|
cpdef void set_params(self, Params p, object params):
|
2021-12-11 12:53:13 +00:00
|
|
|
"""Set the parameters from a [Params] handle (`p`) belong to this system.
|
2020-02-22 07:02:20 +00:00
|
|
|
The values is come from `params`, length must be equal to the handle.
|
|
|
|
"""
|
2021-12-11 12:53:13 +00:00
|
|
|
params = list(params)
|
|
|
|
if p.param_list.size() != len(params):
|
|
|
|
raise ValueError(f"number of parameters {len(params)} are not match {p.param_list.size()}")
|
2019-07-11 12:40:33 +00:00
|
|
|
|
2021-12-11 12:53:13 +00:00
|
|
|
cdef int i = 0
|
2019-07-11 12:40:33 +00:00
|
|
|
cdef Slvs_hParam h
|
|
|
|
for h in p.param_list:
|
2021-12-11 12:26:36 +00:00
|
|
|
self.param_list[h - 1].val = params[i]
|
2019-07-11 12:40:33 +00:00
|
|
|
i += 1
|
|
|
|
|
2021-12-11 12:53:13 +00:00
|
|
|
cpdef list params(self, Params p):
|
2020-07-31 06:33:54 +00:00
|
|
|
"""Get the parameters from a [Params] handle (`p`) belong to this
|
2020-02-22 07:02:20 +00:00
|
|
|
system.
|
|
|
|
The length of tuple is decided by handle.
|
|
|
|
"""
|
2019-10-15 08:13:16 +00:00
|
|
|
param_list = []
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef Slvs_hParam h
|
|
|
|
for h in p.param_list:
|
2021-12-11 12:26:36 +00:00
|
|
|
param_list.append(self.param_list[h - 1].val)
|
2021-12-11 12:53:13 +00:00
|
|
|
return param_list
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef int dof(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return the degrees of freedom of current group.
|
2021-12-12 02:51:01 +00:00
|
|
|
Only can be called after solved.
|
2020-02-22 07:02:20 +00:00
|
|
|
"""
|
2021-12-12 02:51:01 +00:00
|
|
|
return self.dof_v
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef object constraints(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return the number of each constraint type.
|
|
|
|
The name of constraints is represented by string.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cons_list = []
|
|
|
|
cdef Slvs_Constraint con
|
|
|
|
for con in self.cons_list:
|
|
|
|
cons_list.append(_NAME_OF_CONSTRAINTS[con.type])
|
|
|
|
return Counter(cons_list)
|
|
|
|
|
2020-02-02 13:44:32 +00:00
|
|
|
cpdef list failures(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Return a list of failed constraint numbers."""
|
2021-12-11 12:53:13 +00:00
|
|
|
return self.failed_list
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef int solve(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Start the solving, return the result flag."""
|
2021-12-12 02:51:01 +00:00
|
|
|
cdef Slvs_System sys
|
2019-05-28 12:00:15 +00:00
|
|
|
# Parameters
|
2021-12-12 02:51:01 +00:00
|
|
|
sys.param = self.param_list.data()
|
|
|
|
sys.params = self.param_list.size()
|
2019-05-28 12:00:15 +00:00
|
|
|
# Entities
|
2021-12-12 02:51:01 +00:00
|
|
|
sys.entity = self.entity_list.data()
|
|
|
|
sys.entities = self.entity_list.size()
|
2019-05-28 12:00:15 +00:00
|
|
|
# Constraints
|
2021-12-12 02:51:01 +00:00
|
|
|
sys.constraint = self.cons_list.data()
|
|
|
|
sys.constraints = self.cons_list.size()
|
2021-12-11 12:53:13 +00:00
|
|
|
# Faileds
|
|
|
|
self.failed_list.reserve(self.cons_list.size())
|
2021-12-12 02:51:01 +00:00
|
|
|
sys.failed = self.failed_list.data()
|
|
|
|
sys.faileds = self.cons_list.size()
|
2019-05-28 12:00:15 +00:00
|
|
|
# Solve
|
2021-12-12 02:51:01 +00:00
|
|
|
Slvs_Solve(&sys, self.g)
|
|
|
|
self.dof_v = sys.dof
|
|
|
|
return sys.result
|
2019-05-28 12:00:15 +00:00
|
|
|
|
2021-12-11 10:49:15 +00:00
|
|
|
cpdef size_t param_len(self):
|
|
|
|
"""The length of parameter list."""
|
|
|
|
return self.param_list.size()
|
|
|
|
|
|
|
|
cpdef size_t entity_len(self):
|
|
|
|
"""The length of parameter list."""
|
|
|
|
return self.entity_list.size()
|
|
|
|
|
|
|
|
cpdef size_t cons_len(self):
|
|
|
|
"""The length of parameter list."""
|
|
|
|
return self.cons_list.size()
|
|
|
|
|
2019-05-28 12:00:15 +00:00
|
|
|
cpdef Entity create_2d_base(self):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Create a 2D system on current group,
|
|
|
|
return the handle of work plane.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef double qw, qx, qy, qz
|
|
|
|
qw, qx, qy, qz = make_quaternion(1, 0, 0, 0, 1, 0)
|
|
|
|
cdef Entity nm = self.add_normal_3d(qw, qx, qy, qz)
|
|
|
|
return self.add_work_plane(self.add_point_3d(0, 0, 0), nm)
|
|
|
|
|
|
|
|
cdef inline Slvs_hParam new_param(self, double val) nogil:
|
|
|
|
"""Add a parameter."""
|
2021-12-11 10:49:15 +00:00
|
|
|
cdef Slvs_hParam h = <Slvs_hParam>self.param_list.size() + 1
|
2021-12-11 12:26:36 +00:00
|
|
|
self.param_list.push_back(Slvs_MakeParam(h, self.g, val))
|
2019-05-28 12:00:15 +00:00
|
|
|
return h
|
|
|
|
|
|
|
|
cdef inline Slvs_hEntity eh(self) nogil:
|
|
|
|
"""Return new entity handle."""
|
2021-12-11 10:49:15 +00:00
|
|
|
return <Slvs_hEntity>self.entity_list.size() + 1
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_point_2d(self, double u, double v, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a 2D point to specific work plane (`wp`) then return the handle.
|
|
|
|
|
|
|
|
Where `u`, `v` are corresponded to the value of U, V axis on the work
|
|
|
|
plane.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
|
|
|
|
cdef Slvs_hParam u_p = self.new_param(u)
|
|
|
|
cdef Slvs_hParam v_p = self.new_param(v)
|
2021-12-11 10:49:15 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakePoint2d(self.eh(), self.g, wp.h, u_p, v_p))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_point_3d(self, double x, double y, double z):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a 3D point then return the handle.
|
|
|
|
|
|
|
|
Where `x`, `y`, `z` are corresponded to the value of X, Y, Z axis.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef Slvs_hParam x_p = self.new_param(x)
|
|
|
|
cdef Slvs_hParam y_p = self.new_param(y)
|
|
|
|
cdef Slvs_hParam z_p = self.new_param(z)
|
2021-12-11 10:49:15 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakePoint3d(self.eh(), self.g, x_p, y_p, z_p))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_normal_2d(self, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a 2D normal orthogonal to specific work plane (`wp`)
|
|
|
|
then return the handle.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
2021-12-11 10:49:15 +00:00
|
|
|
|
|
|
|
self.entity_list.push_back(Slvs_MakeNormal2d(self.eh(), self.g, wp.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_normal_3d(self, double qw, double qx, double qy, double qz):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a 3D normal from quaternion then return the handle.
|
|
|
|
|
|
|
|
Where `qw`, `qx`, `qy`, `qz` are corresponded to
|
|
|
|
the W, X, Y, Z value of quaternion.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
cdef Slvs_hParam w_p = self.new_param(qw)
|
|
|
|
cdef Slvs_hParam x_p = self.new_param(qx)
|
|
|
|
cdef Slvs_hParam y_p = self.new_param(qy)
|
|
|
|
cdef Slvs_hParam z_p = self.new_param(qz)
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeNormal3d(
|
|
|
|
self.eh(), self.g, w_p, x_p, y_p, z_p))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_distance(self, double d, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a distance to specific work plane (`wp`) then return the handle.
|
|
|
|
|
|
|
|
Where `d` is distance value.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
|
|
|
|
cdef Slvs_hParam d_p = self.new_param(d)
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeDistance(
|
|
|
|
self.eh(), self.g, wp.h, d_p))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_line_2d(self, Entity p1, Entity p2, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a 2D line to specific work plane (`wp`) then return the handle.
|
|
|
|
|
|
|
|
Where `p1` is the start point; `p2` is the end point.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
if p1 is None or not p1.is_point_2d():
|
|
|
|
raise TypeError(f"{p1} is not a 2d point")
|
|
|
|
if p2 is None or not p2.is_point_2d():
|
|
|
|
raise TypeError(f"{p2} is not a 2d point")
|
|
|
|
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeLineSegment(
|
|
|
|
self.eh(), self.g, wp.h, p1.h, p2.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_line_3d(self, Entity p1, Entity p2):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a 3D line then return the handle.
|
|
|
|
|
|
|
|
Where `p1` is the start point; `p2` is the end point.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if p1 is None or not p1.is_point_3d():
|
|
|
|
raise TypeError(f"{p1} is not a 3d point")
|
|
|
|
if p2 is None or not p2.is_point_3d():
|
|
|
|
raise TypeError(f"{p2} is not a 3d point")
|
|
|
|
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeLineSegment(
|
|
|
|
self.eh(), self.g, SLVS_FREE_IN_3D, p1.h, p2.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_cubic(self, Entity p1, Entity p2, Entity p3, Entity p4, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a cubic curve to specific work plane (`wp`) then return the
|
|
|
|
handle.
|
|
|
|
|
|
|
|
Where `p1` to `p4` is the control points.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
if p1 is None or not p1.is_point_2d():
|
|
|
|
raise TypeError(f"{p1} is not a 2d point")
|
|
|
|
if p2 is None or not p2.is_point_2d():
|
|
|
|
raise TypeError(f"{p2} is not a 2d point")
|
|
|
|
if p3 is None or not p3.is_point_2d():
|
|
|
|
raise TypeError(f"{p3} is not a 2d point")
|
|
|
|
if p4 is None or not p4.is_point_2d():
|
|
|
|
raise TypeError(f"{p4} is not a 2d point")
|
|
|
|
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeCubic(
|
|
|
|
self.eh(), self.g, wp.h, p1.h, p2.h, p3.h, p4.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_arc(self, Entity nm, Entity ct, Entity start, Entity end, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add an arc to specific work plane (`wp`) then return the handle.
|
|
|
|
|
|
|
|
Where `nm` is the orthogonal normal; `ct` is the center point;
|
|
|
|
`start` is the start point; `end` is the end point.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
if nm is None or not nm.is_normal_3d():
|
|
|
|
raise TypeError(f"{nm} is not a 3d normal")
|
|
|
|
if ct is None or not ct.is_point_2d():
|
|
|
|
raise TypeError(f"{ct} is not a 2d point")
|
|
|
|
if start is None or not start.is_point_2d():
|
|
|
|
raise TypeError(f"{start} is not a 2d point")
|
|
|
|
if end is None or not end.is_point_2d():
|
|
|
|
raise TypeError(f"{end} is not a 2d point")
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeArcOfCircle(
|
|
|
|
self.eh(), self.g, wp.h, nm.h, ct.h, start.h, end.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_circle(self, Entity nm, Entity ct, Entity radius, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add an circle to specific work plane (`wp`) then return the handle.
|
|
|
|
|
|
|
|
Where `nm` is the orthogonal normal;
|
|
|
|
`ct` is the center point;
|
|
|
|
`radius` is the distance value represent radius.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
if nm is None or not nm.is_normal_3d():
|
|
|
|
raise TypeError(f"{nm} is not a 3d normal")
|
|
|
|
if ct is None or not ct.is_point_2d():
|
|
|
|
raise TypeError(f"{ct} is not a 2d point")
|
|
|
|
if radius is None or not radius.is_distance():
|
|
|
|
raise TypeError(f"{radius} is not a distance")
|
|
|
|
|
2019-10-15 08:13:16 +00:00
|
|
|
self.entity_list.push_back(Slvs_MakeCircle(self.eh(), self.g, wp.h,
|
|
|
|
ct.h, nm.h, radius.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef Entity add_work_plane(self, Entity origin, Entity nm):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a work plane then return the handle.
|
|
|
|
|
|
|
|
Where `origin` is the origin point of the plane;
|
|
|
|
`nm` is the orthogonal normal.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if origin is None or origin.t != SLVS_E_POINT_IN_3D:
|
|
|
|
raise TypeError(f"{origin} is not a 3d point")
|
|
|
|
if nm is None or nm.t != SLVS_E_NORMAL_IN_3D:
|
|
|
|
raise TypeError(f"{nm} is not a 3d normal")
|
|
|
|
|
|
|
|
self.entity_list.push_back(Slvs_MakeWorkplane(self.eh(), self.g, origin.h, nm.h))
|
2021-12-12 06:15:06 +00:00
|
|
|
return Entity.create(&self.entity_list.back())
|
2019-05-28 12:00:15 +00:00
|
|
|
|
|
|
|
cpdef void add_constraint(
|
|
|
|
self,
|
2019-10-15 08:13:16 +00:00
|
|
|
int c_type,
|
2019-05-28 12:00:15 +00:00
|
|
|
Entity wp,
|
|
|
|
double v,
|
|
|
|
Entity p1,
|
|
|
|
Entity p2,
|
|
|
|
Entity e1,
|
|
|
|
Entity e2,
|
|
|
|
Entity e3 = _E_NONE,
|
|
|
|
Entity e4 = _E_NONE,
|
|
|
|
int other = 0,
|
|
|
|
int other2 = 0
|
|
|
|
):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Add a constraint by type code `c_type`.
|
|
|
|
This is an origin function mapping to different constraint methods.
|
|
|
|
|
|
|
|
Where `wp` represents work plane; `v` represents constraint value;
|
2020-07-31 06:33:54 +00:00
|
|
|
`p1` and `p2` represent point entities; `e1` to `e4` represent other
|
2020-02-22 07:02:20 +00:00
|
|
|
types of entity;
|
|
|
|
`other` and `other2` are control options of the constraint.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is None or not wp.is_work_plane():
|
|
|
|
raise TypeError(f"{wp} is not a work plane")
|
|
|
|
|
|
|
|
cdef Entity e
|
|
|
|
for e in (p1, p2):
|
|
|
|
if e is None or not (e.is_none() or e.is_point()):
|
|
|
|
raise TypeError(f"{e} is not a point")
|
|
|
|
for e in (e1, e2, e3, e4):
|
|
|
|
if e is None:
|
|
|
|
raise TypeError(f"{e} is not a entity")
|
|
|
|
|
|
|
|
cdef Slvs_Constraint c
|
2021-12-11 10:49:15 +00:00
|
|
|
c.h = <Slvs_hConstraint>self.cons_list.size() + 1
|
2019-05-28 12:00:15 +00:00
|
|
|
c.group = self.g
|
|
|
|
c.type = c_type
|
|
|
|
c.wrkpl = wp.h
|
|
|
|
c.valA = v
|
|
|
|
c.ptA = p1.h
|
|
|
|
c.ptB = p2.h
|
|
|
|
c.entityA = e1.h
|
|
|
|
c.entityB = e2.h
|
|
|
|
c.entityC = e3.h
|
|
|
|
c.entityD = e4.h
|
|
|
|
c.other = other
|
|
|
|
c.other2 = other2
|
|
|
|
self.cons_list.push_back(c)
|
|
|
|
|
|
|
|
#####
|
|
|
|
# Constraint methods.
|
|
|
|
#####
|
|
|
|
|
|
|
|
cpdef void coincident(self, Entity e1, Entity e2, Entity wp = _E_FREE_IN_3D):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Coincident two entities.
|
|
|
|
|
|
|
|
| Entity 1 (`e1`) | Entity 2 (`e2`) | Work plane (`wp`) |
|
|
|
|
|:---------------:|:---------------:|:-----------------:|
|
|
|
|
| [is_point] | [is_point] | Optional |
|
|
|
|
| [is_point] | [is_work_plane] | [Entity.FREE_IN_3D] |
|
|
|
|
| [is_point] | [is_line] | Optional |
|
|
|
|
| [is_point] | [is_circle] | Optional |
|
|
|
|
"""
|
2019-10-15 08:13:16 +00:00
|
|
|
cdef int t
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_point() and e2.is_point():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_POINTS_COINCIDENT, wp, 0., e1, e2,
|
|
|
|
_E_NONE, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_point() and e2.is_work_plane() and wp is _E_FREE_IN_3D:
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_PT_IN_PLANE, e2, 0., e1, _E_NONE, e2,
|
|
|
|
_E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_point() and (e2.is_line() or e2.is_circle()):
|
|
|
|
if e2.is_line():
|
2019-10-15 08:13:16 +00:00
|
|
|
t = SLVS_C_PT_ON_LINE
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
2019-10-15 08:13:16 +00:00
|
|
|
t = SLVS_C_PT_ON_CIRCLE
|
2019-05-28 12:00:15 +00:00
|
|
|
self.add_constraint(t, wp, 0., e1, _E_NONE, e2, _E_NONE)
|
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void distance(
|
|
|
|
self,
|
|
|
|
Entity e1,
|
|
|
|
Entity e2,
|
|
|
|
double value,
|
|
|
|
Entity wp = _E_FREE_IN_3D
|
|
|
|
):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Distance constraint between two entities.
|
|
|
|
|
|
|
|
If `value` is equal to zero, then turn into
|
|
|
|
[coincident](#solversystemcoincident)
|
|
|
|
|
|
|
|
| Entity 1 (`e1`) | Entity 2 (`e2`) | Work plane (`wp`) |
|
|
|
|
|:---------------:|:---------------:|:-----------------:|
|
|
|
|
| [is_point] | [is_point] | Optional |
|
|
|
|
| [is_point] | [is_work_plane] | [Entity.FREE_IN_3D] |
|
|
|
|
| [is_point] | [is_line] | Optional |
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if value == 0.:
|
|
|
|
self.coincident(e1, e2, wp)
|
|
|
|
return
|
|
|
|
if e1.is_point() and e2.is_point():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_PT_PT_DISTANCE, wp, value, e1, e2,
|
|
|
|
_E_NONE, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_point() and e2.is_work_plane() and wp is _E_FREE_IN_3D:
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_PT_PLANE_DISTANCE, e2, value, e1,
|
|
|
|
_E_NONE, e2, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_point() and e2.is_line():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_PT_LINE_DISTANCE, wp, value, e1,
|
|
|
|
_E_NONE, e2, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void equal(self, Entity e1, Entity e2, Entity wp = _E_FREE_IN_3D):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Equal constraint between two entities.
|
|
|
|
|
|
|
|
| Entity 1 (`e1`) | Entity 2 (`e2`) | Work plane (`wp`) |
|
|
|
|
|:---------------:|:---------------:|:-----------------:|
|
|
|
|
| [is_line] | [is_line] | Optional |
|
|
|
|
| [is_line] | [is_arc] | Optional |
|
|
|
|
| [is_line] | [is_circle] | Optional |
|
|
|
|
| [is_arc] | [is_arc] | Optional |
|
|
|
|
| [is_arc] | [is_circle] | Optional |
|
|
|
|
| [is_circle] | [is_circle] | Optional |
|
|
|
|
| [is_circle] | [is_arc] | Optional |
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_line() and e2.is_line():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_EQUAL_LENGTH_LINES, wp, 0., _E_NONE,
|
|
|
|
_E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_line() and (e2.is_arc() or e2.is_circle()):
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_EQUAL_LINE_ARC_LEN, wp, 0., _E_NONE,
|
|
|
|
_E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif (e1.is_arc() or e1.is_circle()) and (e2.is_arc() or e2.is_circle()):
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_EQUAL_RADIUS, wp, 0., _E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void equal_included_angle(
|
|
|
|
self,
|
|
|
|
Entity e1,
|
|
|
|
Entity e2,
|
|
|
|
Entity e3,
|
|
|
|
Entity e4,
|
|
|
|
Entity wp
|
|
|
|
):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Constraint that 2D line 1 (`e1`) and line 2 (`e2`),
|
2020-07-31 06:33:54 +00:00
|
|
|
line 3 (`e3`) and line 4 (`e4`) must have same included angle on work
|
2020-02-22 07:02:20 +00:00
|
|
|
plane `wp`.
|
2019-05-28 12:00:15 +00:00
|
|
|
"""
|
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_line_2d() and e2.is_line_2d() and e3.is_line_2d() and e4.is_line_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_EQUAL_ANGLE, wp, 0., _E_NONE, _E_NONE,
|
|
|
|
e1, e2, e3, e4)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {e3}, {e4}, {wp}")
|
|
|
|
|
|
|
|
cpdef void equal_point_to_line(
|
|
|
|
self,
|
|
|
|
Entity e1,
|
|
|
|
Entity e2,
|
|
|
|
Entity e3,
|
|
|
|
Entity e4,
|
|
|
|
Entity wp
|
|
|
|
):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Constraint that point 1 (`e1`) and line 1 (`e2`),
|
2020-07-31 06:33:54 +00:00
|
|
|
point 2 (`e3`) and line 2 (`e4`) must have same distance on work
|
2020-02-22 07:02:20 +00:00
|
|
|
plane `wp`.
|
2019-05-28 12:00:15 +00:00
|
|
|
"""
|
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_point_2d() and e2.is_line_2d() and e3.is_point_2d() and e4.is_line_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_EQ_PT_LN_DISTANCES, wp, 0., e1, e3, e2, e4)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {e3}, {e4}, {wp}")
|
|
|
|
|
|
|
|
cpdef void ratio(self, Entity e1, Entity e2, double value, Entity wp):
|
2020-07-31 06:33:54 +00:00
|
|
|
"""The ratio (`value`) constraint between two 2D lines (`e1` and
|
2020-02-22 07:02:20 +00:00
|
|
|
`e2`).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_line_2d() and e2.is_line_2d():
|
2021-02-11 15:51:18 +00:00
|
|
|
self.add_constraint(SLVS_C_LENGTH_RATIO, wp, value, _E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void symmetric(
|
|
|
|
self,
|
|
|
|
Entity e1,
|
|
|
|
Entity e2,
|
|
|
|
Entity e3 = _E_NONE,
|
|
|
|
Entity wp = _E_FREE_IN_3D
|
|
|
|
):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Symmetric constraint between two points.
|
|
|
|
|
|
|
|
| Entity 1 (`e1`) | Entity 2 (`e2`) | Entity 3 (`e3`) | Work plane (`wp`) |
|
|
|
|
|:---------------:|:---------------:|:---------------:|:-----------------:|
|
|
|
|
| [is_point_3d] | [is_point_3d] | [is_work_plane] | [Entity.FREE_IN_3D] |
|
|
|
|
| [is_point_2d] | [is_point_2d] | [is_work_plane] | [Entity.FREE_IN_3D] |
|
|
|
|
| [is_point_2d] | [is_point_2d] | [is_line_2d] | Is not [Entity.FREE_IN_3D] |
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_point_3d() and e2.is_point_3d() and e3.is_work_plane() and wp is _E_FREE_IN_3D:
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_SYMMETRIC, wp, 0., e1, e2, e3, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_point_2d() and e2.is_point_2d() and e3.is_work_plane() and wp is _E_FREE_IN_3D:
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_SYMMETRIC, e3, 0., e1, e2, e3, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_point_2d() and e2.is_point_2d() and e3.is_line_2d():
|
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_SYMMETRIC_LINE, wp, 0., e1, e2, e3, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {e3}, {wp}")
|
|
|
|
|
|
|
|
cpdef void symmetric_h(self, Entity e1, Entity e2, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Symmetric constraint between two 2D points (`e1` and `e2`)
|
|
|
|
with horizontal line on the work plane (`wp` can not be
|
|
|
|
[Entity.FREE_IN_3D]).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_point_2d() and e2.is_point_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_SYMMETRIC_HORIZ, wp, 0., e1, e2, _E_NONE, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void symmetric_v(self, Entity e1, Entity e2, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Symmetric constraint between two 2D points (`e1` and `e2`)
|
|
|
|
with vertical line on the work plane (`wp` can not be
|
|
|
|
[Entity.FREE_IN_3D]).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_point_2d() and e2.is_point_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_SYMMETRIC_VERT, wp, 0., e1, e2, _E_NONE, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void midpoint(
|
|
|
|
self,
|
|
|
|
Entity e1,
|
|
|
|
Entity e2,
|
|
|
|
Entity wp = _E_FREE_IN_3D
|
|
|
|
):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Midpoint constraint between a point (`e1`) and
|
|
|
|
a line (`e2`) on work plane (`wp`).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_point() and e2.is_line():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_AT_MIDPOINT, wp, 0., e1, _E_NONE, e2, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void horizontal(self, Entity e1, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Vertical constraint of a 2d point (`e1`) on
|
|
|
|
work plane (`wp` can not be [Entity.FREE_IN_3D]).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_line_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_HORIZONTAL, wp, 0., _E_NONE, _E_NONE, e1, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {wp}")
|
|
|
|
|
|
|
|
cpdef void vertical(self, Entity e1, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Vertical constraint of a 2d point (`e1`) on
|
|
|
|
work plane (`wp` can not be [Entity.FREE_IN_3D]).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_line_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_VERTICAL, wp, 0., _E_NONE, _E_NONE, e1, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {wp}")
|
|
|
|
|
|
|
|
cpdef void diameter(self, Entity e1, double value, Entity wp):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Diameter (`value`) constraint of a circular entities.
|
|
|
|
|
|
|
|
| Entity 1 (`e1`) | Work plane (`wp`) |
|
|
|
|
|:---------------:|:-----------------:|
|
|
|
|
| [is_arc] | Optional |
|
|
|
|
| [is_circle] | Optional |
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_arc() or e1.is_circle():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_DIAMETER, wp, value, _E_NONE, _E_NONE,
|
|
|
|
e1, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {wp}")
|
|
|
|
|
|
|
|
cpdef void same_orientation(self, Entity e1, Entity e2):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Equal orientation constraint between two 3d normals (`e1` and
|
|
|
|
`e2`).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_normal_3d() and e2.is_normal_3d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_SAME_ORIENTATION, _E_FREE_IN_3D, 0.,
|
|
|
|
_E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}")
|
|
|
|
|
|
|
|
cpdef void angle(self, Entity e1, Entity e2, double value, Entity wp, bint inverse = False):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Degrees angle (`value`) constraint between two 2d lines (`e1` and
|
|
|
|
`e2`) on the work plane (`wp` can not be [Entity.FREE_IN_3D]).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_line_2d() and e2.is_line_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_ANGLE, wp, value, _E_NONE, _E_NONE,
|
2019-05-28 12:00:15 +00:00
|
|
|
e1, e2, _E_NONE, _E_NONE, inverse)
|
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void perpendicular(self, Entity e1, Entity e2, Entity wp, bint inverse = False):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Perpendicular constraint between two 2d lines (`e1` and `e2`)
|
|
|
|
on the work plane (`wp` can not be [Entity.FREE_IN_3D]) with
|
|
|
|
`inverse` option.
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
|
|
|
if e1.is_line_2d() and e2.is_line_2d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_PERPENDICULAR, wp, 0., _E_NONE, _E_NONE,
|
2019-05-28 12:00:15 +00:00
|
|
|
e1, e2, _E_NONE, _E_NONE, inverse)
|
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void parallel(self, Entity e1, Entity e2, Entity wp = _E_FREE_IN_3D):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Parallel constraint between two lines (`e1` and `e2`) on
|
|
|
|
the work plane (`wp`).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_line() and e2.is_line():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_PARALLEL, wp, 0., _E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void tangent(self, Entity e1, Entity e2, Entity wp = _E_FREE_IN_3D):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Parallel constraint between two entities (`e1` and `e2`) on the
|
|
|
|
work plane (`wp`).
|
|
|
|
|
|
|
|
| Entity 1 (`e1`) | Entity 2 (`e2`) | Work plane (`wp`) |
|
|
|
|
|:---------------:|:---------------:|:-----------------:|
|
|
|
|
| [is_arc] | [is_line_2d] | Is not [Entity.FREE_IN_3D] |
|
|
|
|
| [is_cubic] | [is_line_3d] | [Entity.FREE_IN_3D] |
|
|
|
|
| [is_arc] | [is_cubic] | Is not [Entity.FREE_IN_3D] |
|
|
|
|
| [is_arc] | [is_arc] | Is not [Entity.FREE_IN_3D] |
|
|
|
|
| [is_cubic] | [is_cubic] | Optional |
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_arc() and e2.is_line_2d():
|
|
|
|
if wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_ARC_LINE_TANGENT, wp, 0., _E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif e1.is_cubic() and e2.is_line_3d() and wp is _E_FREE_IN_3D:
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_CUBIC_LINE_TANGENT, wp, 0., _E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
elif (e1.is_arc() or e1.is_cubic()) and (e2.is_arc() or e2.is_cubic()):
|
|
|
|
if (e1.is_arc() or e2.is_arc()) and wp is _E_FREE_IN_3D:
|
|
|
|
raise ValueError("this is a 2d constraint")
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_CURVE_CURVE_TANGENT, wp, 0., _E_NONE, _E_NONE, e1, e2)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}, {wp}")
|
|
|
|
|
|
|
|
cpdef void distance_proj(self, Entity e1, Entity e2, double value):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Projected distance (`value`) constraint between
|
|
|
|
two 3d points (`e1` and `e2`).
|
|
|
|
"""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_point_3d() and e2.is_point_3d():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_CURVE_CURVE_TANGENT, _E_FREE_IN_3D,
|
|
|
|
value, e1, e2, _E_NONE, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {e2}")
|
|
|
|
|
|
|
|
cpdef void dragged(self, Entity e1, Entity wp = _E_FREE_IN_3D):
|
2020-02-22 07:02:20 +00:00
|
|
|
"""Dragged constraint of a point (`e1`) on the work plane (`wp`)."""
|
2019-05-28 12:00:15 +00:00
|
|
|
if e1.is_point():
|
2019-10-15 08:13:16 +00:00
|
|
|
self.add_constraint(SLVS_C_WHERE_DRAGGED, wp, 0., e1, _E_NONE, _E_NONE, _E_NONE)
|
2019-05-28 12:00:15 +00:00
|
|
|
else:
|
|
|
|
raise TypeError(f"unsupported entities: {e1}, {wp}")
|