Newer
Older
####################################################################################################
#
# Patro - A Python library to make patterns for fashion design
# Copyright (C) 2019 Fabrice Salvaire
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
####################################################################################################
For resources on path see :ref:`this section <path-geometry-ressources-page>`.
####################################################################################################
__all__ = [
'LinearSegment',
'QuadraticBezierSegment',
'CubicBezierSegment',
'Path2D',
]
####################################################################################################
from Patro.Common.Math.Functions import sign
from .Primitive import Primitive1P, Primitive2DMixin
from .Bezier import QuadraticBezier2D, CubicBezier2D
from .Segment import Segment2D
from .Vector import Vector2D
####################################################################################################
_module_logger = logging.getLogger(__name__)
####################################################################################################
class PathPart:
##############################################
def __init__(self, path, index):
self._index = index
##############################################
def _init_absolute(self, absolute):
self._absolute = bool(absolute)
##############################################
Fabrice Salvaire
committed
def clone(self, path):
raise NotImplementedError
##############################################
return '{0}(@{1._index})'.format(self.__class__.__name__, self)
##############################################
@property
def path(self):
return self._path
@property
def index(self):
return self._index
@index.setter
def index(self, value):
self._index = int(value)
##############################################
@property
def prev_part(self):
return self._path[self._index -1]
@property
def next_part(self):
return self._path[self._index +1]
##############################################
@property
def start_point(self):
Fabrice Salvaire
committed
# Fixme: cache ???
prev_part = self.prev_part
if prev_part is not None:
return prev_part.stop_point
else:
return self._path.p0
##############################################
@property
def stop_point(self):
raise NotImplementedError
##############################################
def to_absolute_point(self, point):
# Fixme: cache ???
if self._absolute:
return point
else:
return point + self.start_point
##############################################
@property
def geometry(self):
raise NotImplementedError
##############################################
@property
def bounding_box(self):
return self.geometry.bounding_box
####################################################################################################
class OnePointMixin:
##############################################
@property
def point(self):
return self._point
@point.setter
def point(self, value):
self._point = Vector2D(value) # self._path.__vector_cls__
##############################################
@property
def stop_point(self):
return self.to_absolute_point(self._point)
##############################################
def apply_transformation(self, transformation):
# Fixme: right for relative ???
self._point = transformation * self._point
####################################################################################################
class TwoPointMixin:
##############################################
@property
def point1(self):
return self.to_absolute_point(self._point1)
@point1.setter
def point1(self, value):
self._point1 = Vector2D(value) # self._path.__vector_cls__
##############################################
@property
def point2(self):
return self.to_absolute_point(self._point2)
@point2.setter
def point2(self, value):
self._point2 = Vector2D(value)
##############################################
def apply_transformation(self, transformation):
# Fixme: right for relative ???
self._point1 = transformation * self._point1
self._point2 = transformation * self._point2
####################################################################################################
class ThreePointMixin(TwoPointMixin):
##############################################
@property
def point3(self):
return self.to_absolute_point(self._point3)
@point3.setter
def point3(self, value):
self._point3 = Vector2D(value) # self._path.__vector_cls__
##############################################
def apply_transformation(self, transformation):
# Fixme: right for relative ???
TwoPointMixin.apply_transformation(self, transformation)
self._point3 = transformation * self._point3
####################################################################################################
class LinearSegment(PathPart):
"""Class to implement a linear segment.
# Fixme:
#
# If two successive vertices share the same circle, then it should be merged to one.
#
_logger = _module_logger.getChild('LinearSegment')
##############################################
def __init__(self, path, index, radius, closing=False):
super().__init__(path, index)
self._bissector = None
self._direction = None
self._closing = bool(closing)
self.radius = radius
if self._radius is not None:
if not isinstance(self.prev_part, LinearSegment):
raise ValueError('Previous path segment must be linear')
Fabrice Salvaire
committed
self._reset_cache()
##############################################
def _reset_cache(self):
self._bulge_angle = None
self._bulge_center = None
self._start_bulge_point = None
self._stop_bulge_point = None
##############################################
def close(self, radius):
"""Set the bulge radius at the closure"""
self.radius = radius
self._reset_cache()
##############################################
@property
def prev_part(self):
return self._path.stop_segment # or [-1] don't work
else:
# Fixme: super
return self._path[self._index -1]
@property
def next_part(self):
if self._closing:
return self._path.start_segment # or [0]
else:
return self._path[self._index +1]
##############################################
if self._radius is not None:
start_point = self.bulge_stop_point
else:
start_point = self.start_point
next_part = self.next_part
if isinstance(next_part, LinearSegment) and next_part.radius is not None:
stop_point = next_part.bulge_start_point
else:
stop_point = self.stop_point
return start_point, stop_point
##############################################
@property
def radius(self):
return self._radius
@radius.setter
def radius(self, value):
if value is not None:
value = abs(float(value))
if value == 0:
radius = None
self._radius = value
##############################################
@property
def direction(self):
if self._direction is None:
self._direction = (self.stop_point - self.start_point).normalise()
return self._direction
@property
def bissector(self):
if self._radius is None:
return None
else:
if self._bissector is None:
# self._bissector = (self.prev_part.direction + self.direction).normalise().normal
self._bissector = (self.direction - self.prev_part.direction).normalise()
return self._bissector
##############################################
@property
def bulge_angle_rad(self):
if self._bulge_angle is None:
# Fixme: rad vs degree
angle = self.direction.angle_with(self.prev_part.direction)
if angle >= 0:
angle = 180 - angle
else:
angle = -(180 + angle)
self._bulge_angle = math.radians(angle)
return self._bulge_angle
@property
def bulge_angle(self):
return math.degrees(self.bulge_angle_rad)
@property
def half_bulge_angle(self):
return self.bulge_angle_rad / 2
##############################################
@property
def bulge_center(self):
if self._bulge_center is None:
offset = self.bissector * self._radius / math.sin(abs(self.half_bulge_angle))
self._bulge_center = self.start_point + offset
# Note: -offset create external loop
return self._bulge_center
##############################################
@property
def bulge_start_point(self):
if self._start_bulge_point is None:
angle = self.half_bulge_angle
offset = self.prev_part.direction * self._radius / math.tan(angle)
self._start_bulge_point = self.start_point - sign(angle) *offset
# Note: -offset create internal loop
return self._start_bulge_point
@property
def bulge_stop_point(self):
if self._stop_bulge_point is None:
angle = self.half_bulge_angle
offset = self.direction * self._radius / math.tan(self.half_bulge_angle)
self._stop_bulge_point = self.start_point + sign(angle) * offset
return self._stop_bulge_point
##############################################
@property
def bulge_geometry(self):
# Fixme: check start and stop are within segment
arc = Circle2D(self.bulge_center, self._radius)
start_angle, stop_angle = [arc.angle_for_point(point)
for point in (self.bulge_start_point, self.bulge_stop_point)]
if self.bulge_angle < 0:
start_angle, stop_angle = stop_angle, start_angle
arc.domain = AngularDomain(start_angle, stop_angle)
##############################################
def _dump_bulge(self, arc):
self._logger.info(
'Bulge @{}\n'.format(self._index) +
str(arc)
)
####################################################################################################
class PathSegment(OnePointMixin, LinearSegment):
##############################################
def __init__(self, path, index, point, radius=None, absolute=False, closing=False):
super().__init__(path, index, radius, closing)
self._init_absolute(absolute)
##############################################
Fabrice Salvaire
committed
def clone(self, path):
# Fixme: check
if obj._start_bulge:
radius = None
else:
radius = self._radius
obj = self.__class__(path, self._index, self._point, radius, self._absolute, self._closing)
if obj._start_bulge:
self.close(self._radius)
return obj
Fabrice Salvaire
committed
##############################################
def to_absolute(self):
self._point = self.stop_point
self._absolute = True
##############################################
def apply_transformation(self, transformation):
OnePointMixin.apply_transformation(self, transformation)
Fabrice Salvaire
committed
if self._radius is not None:
self._radius = transformation * self._radius
##############################################
@property
def geometry(self):
# Fixme: cache ???
return Segment2D(*self.points)
####################################################################################################
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
class DirectionalSegmentMixin(LinearSegment):
##############################################
def apply_transformation(self, transformation):
# Since a rotation will change the direction
# DirectionalSegment must be casted to PathSegment
raise NotImplementedError
##############################################
@property
def geometry(self):
# Fixme: cache ???
return Segment2D(self.start_point, self.stop_point)
####################################################################################################
class AbsoluteHVSegment(DirectionalSegmentMixin):
##############################################
def to_path_segment(self):
# Fixme: duplicted
if self._index == 0 and self._radius is not None:
radius = None
close = True
else:
radius = self._radius
close = False
path = PathSegment(self._path, self._index, self.stop_point, radius, absolute=True)
if close:
path.close(self._radius)
return path
####################################################################################################
class AbsoluteHorizontalSegment(AbsoluteHVSegment):
##############################################
def __init__(self, path, index, x, radius=None):
super().__init__(path, index, radius)
self.x = x
self._init_absolute(False) # Fixme: mix
##############################################
def __repr__(self):
return '{0}(@{1._index}, {1._x})'.format(self.__class__.__name__, self)
##############################################
def clone(self, path):
return self.__class__(path, self._index, self._x, self._radius)
##############################################
@property
def x(self):
return self._x
@x.setter
def x(self, value):
self._x = float(value)
##############################################
@property
def stop_point(self):
return Vector2D(self._x, self.start_point.y)
####################################################################################################
class AbsoluteVerticalSegment(AbsoluteHVSegment):
##############################################
def __init__(self, path, index, y, radius=None):
super().__init__(path, index, radius)
self.y = y
self._init_absolute(False) # Fixme: mix
##############################################
def __repr__(self):
return '{0}(@{1._index}, {1._y})'.format(self.__class__.__name__, self)
##############################################
def clone(self, path):
return self.__class__(path, self._index, self._y, self._radius)
##############################################
@property
def y(self):
return self._y
@y.setter
def y(self, value):
self._y = float(value)
##############################################
@property
def stop_point(self):
return Vector2D(self.start_point.x, self._y)
####################################################################################################
class DirectionalSegment(DirectionalSegmentMixin):
__angle__ = None
##############################################
def __init__(self, path, index, length, radius=None):
super().__init__(path, index, radius)
self.length = length
##############################################
def __repr__(self):
return '{0}(@{1._index}, {1.offset})'.format(self.__class__.__name__, self)
##############################################
Fabrice Salvaire
committed
def clone(self, path):
return self.__class__(path, self._index, self._length, self._radius)
Fabrice Salvaire
committed
##############################################
@property
def length(self):
return self._length
@length.setter
def length(self, value):
self._length = float(value)
##############################################
Fabrice Salvaire
committed
@property
def offset(self):
# Fixme: cache ???
return Vector2D.from_polar(self._length, self.__angle__)
@property
def stop_point(self):
# Fixme: cache ???
Fabrice Salvaire
committed
return self.start_point + self.offset
##############################################
if self._index == 0 and self._radius is not None:
radius = None
close = True
else:
radius = self._radius
close = False
Fabrice Salvaire
committed
path = PathSegment(self._path, self._index, self.offset, radius, absolute=False)
if close:
path.close(self._radius)
return path
Fabrice Salvaire
committed
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
####################################################################################################
class HorizontalSegment(DirectionalSegment):
__angle__ = 0
class VerticalSegment(DirectionalSegment):
__angle__ = 90
class NorthSegment(DirectionalSegment):
__angle__ = 90
class SouthSegment(DirectionalSegment):
__angle__ = -90
class EastSegment(DirectionalSegment):
__angle__ = 0
class WestSegment(DirectionalSegment):
__angle__ = 180
class NorthEastSegment(DirectionalSegment):
__angle__ = 45
class NorthWestSegment(DirectionalSegment):
__angle__ = 180 - 45
class SouthEastSegment(DirectionalSegment):
__angle__ = -45
class SouthWestSegment(DirectionalSegment):
__angle__ = -180 + 45
####################################################################################################
class QuadraticBezierSegment(PathPart, TwoPointMixin):
# Fixme: abs / inc
##############################################
def __init__(self, path, index, point1, point2, absolute=False):
PathPart.__init__(self, path, index)
self._init_absolute(absolute)
self.point1 = point1
self.point2 = point2
##############################################
Fabrice Salvaire
committed
def clone(self, path):
return self.__class__(path, self._index, self._point1, self._point2, self._absolute)
Fabrice Salvaire
committed
##############################################
def to_absolute(self):
self._point1 = self.point1
self._point2 = self.point2
self._absolute = True
##############################################
@property
def stop_point(self):
@property
def points(self):
return (self.start_point, self.point1, self.point2)
##############################################
@property
def geometry(self):
# Fixme: cache ???
return QuadraticBezier2D(*self.points)
####################################################################################################
class CubicBezierSegment(PathPart, ThreePointMixin):
##############################################
def __init__(self, path, index, point1, point2, point3, absolute=False):
PathPart.__init__(self, path, index)
self._init_absolute(absolute)
self.point1 = point1
self.point2 = point2
self.point3 = point3
##############################################
Fabrice Salvaire
committed
def clone(self, path):
return self.__class__(path, self._index, self._point1, self._point2, self._point3, absolute)
##############################################
def to_absolute(self):
self._point1 = self.point1
self._point2 = self.point2
self._point3 = self.point3
self._absolute = True
##############################################
@property
def stop_point(self):
@property
def points(self):
return (self.start_point, self.point1, self.point2, self.point3)
##############################################
@property
def geometry(self):
# Fixme: cache ???
return CubicBezier2D(*self.points)
####################################################################################################
class StringedQuadtraticBezierSegment(PathPart, TwoPointMixin):
##############################################
def __init__(self, path, index, point1, absolute=False):
PathPart.__init__(self, path, index)
self._init_absolute(absolute)
self.point1 = point1
##############################################
def clone(self, path):
return self.__class__(path, self._index, self._point1, absolute)
##############################################
@property
def geometry(self):
# Fixme: cache ???
# Fixme: !!!
return Segment2D(self.start_point, self._point2)
# return CubicBezier2D(self.start_point, self._point1, self._point2, self._point3)
####################################################################################################
class StringedCubicBezierSegment(PathPart, TwoPointMixin):
##############################################
def __init__(self, path, index, point1, point2, absolute=False):
PathPart.__init__(self, path, index)
self._init_absolute(absolute)
# self.point1 = point1
##############################################
def clone(self, path):
return self.__class__(path, self._index, self._point1, self._point2, absolute)
##############################################
@property
def geometry(self):
# Fixme: cache ???
# Fixme: !!!
return Segment2D(self.start_point, self._point2)
# return CubicBezier2D(self.start_point, self._point1, self._point2, self._point3)
####################################################################################################
class ArcSegment(OnePointMixin, PathPart):
##############################################
def __init__(self, path, index, point, radius_x, radius_y, angle, large_arc, sweep, absolute=False):
PathPart.__init__(self, path, index)
self._init_absolute(absolute)
self.point = point
self._large_arc = bool(large_arc)
self._sweep = bool(sweep)
self._radius_x = radius_x
self._radius_y = radius_y
self._angle = angle
##############################################
def clone(self, path):
return self.__class__(
path,
self._index,
self._point,
self._radius_x, self._radius_y,
self._angle,
self._large_arc, self._sweep,
self._absolute,
)
##############################################
def __repr__(self):
template = '{0}(@{1._index} {1._point} rx={1._radius_x} ry={1._radius_y} a={1._angle} la={1._large_arc} s={1._sweep})'
return template.format(self.__class__.__name__, self)
##############################################
def to_absolute(self):
self._point = self.stop_point
self._absolute = True
##############################################
@property
def points(self):
return self.start_point, self.stop_point
##############################################
@property
def geometry(self):
return Ellipse2D.svg_arc(
self.start_point, self.stop_point,
self._radius_x, self._radius_y,
self._angle,
self._large_arc, self._sweep,
)
####################################################################################################
class Path2D(Primitive2DMixin, Primitive1P):
"""Class to implements 2D Path."""
_logger = _module_logger.getChild('Path2D')
##############################################
def __init__(self, start_point):
Primitive1P.__init__(self, start_point)
self._parts = [] # Fixme: segment ???
self._is_closed = False
##############################################
Fabrice Salvaire
committed
def clone(self):
obj = self.__class__(self._p0)
# parts must be added sequentially to the path for bulge check
parts = obj._parts
for part in self._parts:
parts.append(part.clone(obj))
return obj
##############################################
def __len__(self):
return len(self._parts)
def __iter__(self):
return iter(self._parts)
def __getitem__(self, index):
# try:
# return self._parts[slice_]
# except IndexError:
# return None
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
index = int(index)
number_of_parts = len(self._parts)
if 0 <= index < number_of_parts:
return self._parts[index]
# elif self._is_closed:
# if index == -1:
# return self.start_segment
# elif index == number_of_parts:
# return self.stop_segment
return None
##############################################
@property
def start_segment(self):
# Fixme: start_part ???
return self._parts[0]
@property
def stop_segment(self):
return self._parts[-1]
##############################################
@property
def is_closed(self):
return self._is_closed
##############################################
def _add_part(self, part_cls, *args, **kwargs):
if not self._is_closed:
obj = part_cls(self, len(self._parts), *args, **kwargs)
self._parts.append(obj)
return obj
##############################################
def apply_transformation(self, transformation):
Fabrice Salvaire
committed
# self._logger.info(str(self) + '\n' + str(transformation.type) + '\n' + str(transformation) )
Fabrice Salvaire
committed
for part in self._parts:
# print(part)
Fabrice Salvaire
committed
if isinstance(part, PathSegment):
part._reset_cache()
if isinstance(part, DirectionalSegmentMixin):
Fabrice Salvaire
committed
# Since a rotation will change the direction
# DirectionalSegment must be casted to PathSegment
part = part.to_path_segment()
self._parts[part.index] = part
if part._absolute is False:
# print('incremental', part, part.points)
part.to_absolute()
# print('->', part.points)
# print()
self._p0 = transformation * self._p0
# print('p0', self._p0)
for part in self._parts:
# print(part)
part.apply_transformation(transformation)
Fabrice Salvaire
committed
##############################################
@property
def bounding_box(self):
# Fixme: cache
bounding_box = None
for item in self._parts:
interval = item.geometry.bounding_box
if bounding_box is None:
bounding_box = interval
else:
bounding_box |= interval
return bounding_box
##############################################
def move_to(self, point):
self.p0 = point
##############################################
def horizontal_to(self, length, radius=None, absolute=False):
return self._add_part(PathSegment, self.__vector_cls__(length, 0), radius,