def test_angle_to_param(): random_tests_count = 100 random.seed(0) angle = 1.23 assert math.isclose(angle_to_param(1.0, angle), angle) angle = 1.23 + math.pi / 2 assert math.isclose(angle_to_param(1.0, angle), angle) angle = 1.23 + math.pi assert math.isclose(angle_to_param(1.0, angle), angle) angle = 1.23 + 3 * math.pi / 2 assert math.isclose(angle_to_param(1.0, angle), angle) angle = math.pi / 2 + 1e-15 assert math.isclose(angle_to_param(1.0, angle), angle) for _ in range(random_tests_count): ratio = random.uniform(1e-6, 1) angle = random.uniform(0, math.tau) param = angle_to_param(ratio, angle) ellipse = Ellipse.new( dxfattribs={ # avoid (0, 0, 0) as major axis 'major_axis': (non_zero_random(), non_zero_random(), 0), 'ratio': ratio, 'start_param': 0, 'end_param': param, 'extrusion': (0, 0, random.choice((1, -1))), }) calculated_angle = ellipse.dxf.extrusion.angle_about( ellipse.dxf.major_axis, ellipse.end_point) calculated_angle_without_direction = ellipse.dxf.major_axis.angle_between( ellipse.end_point) assert math.isclose(calculated_angle, angle, abs_tol=1e-9) assert (math.isclose(calculated_angle, calculated_angle_without_direction) or math.isclose(math.tau - calculated_angle, calculated_angle_without_direction))
def test_angle_to_param(): random_tests_count = 100 random.seed(0) angle = 1.23 assert math.isclose(angle_to_param(1.0, angle), angle) angle = 1.23 + math.pi / 2 assert math.isclose(angle_to_param(1.0, angle), angle) angle = 1.23 + math.pi assert math.isclose(angle_to_param(1.0, angle), angle) angle = 1.23 + 3 * math.pi / 2 assert math.isclose(angle_to_param(1.0, angle), angle) angle = math.pi / 2 + 1e-15 assert math.isclose(angle_to_param(1.0, angle), angle) for _ in range(random_tests_count): ratio = random.uniform(1e-6, 1) angle = random.uniform(0, math.tau) param = angle_to_param(ratio, angle) ellipse = ConstructionEllipse( # avoid (0, 0, 0) as major axis major_axis=(non_zero_random(), non_zero_random(), 0), ratio=ratio, start_param=0, end_param=param, extrusion=(0, 0, random.choice((1, -1))), ) calculated_angle = ellipse.extrusion.angle_about( ellipse.major_axis, ellipse.end_point) calculated_angle_without_direction = ellipse.major_axis.angle_between( ellipse.end_point) assert math.isclose(calculated_angle, angle, abs_tol=1e-9) assert math.isclose( calculated_angle, calculated_angle_without_direction) or math.isclose( math.tau - calculated_angle, calculated_angle_without_direction)
def virtual_block_reference_entities( block_ref: 'Insert', uniform_scaling_factor: float = None, skipped_entity_callback: Optional[Callable[['DXFGraphic', str], None]] = None ) -> Iterable['DXFGraphic']: """ Yields 'virtual' parts of block reference `block_ref`. This method is meant to examine the the block reference entities without the need to explode the block reference. The `skipped_entity_callback()` will be called for all entities which are not processed, signature: :code:`skipped_entity_callback(entity: DXFEntity, reason: str)`, `entity` is the original (untransformed) DXF entity of the block definition, the `reason` string is an explanation why the entity was skipped. This entities are located at the 'exploded' positions, but are not stored in the entity database, have no handle and are not assigned to any layout. Args: block_ref: Block reference entity (INSERT) uniform_scaling_factor: override uniform scaling factor for text entities (TEXT, ATTRIB, MTEXT) and HATCH pattern, default is ``max(abs(xscale), abs(yscale), abs(zscale))`` skipped_entity_callback: called whenever the transformation of an entity is not supported and so was skipped. .. warning:: **Non uniform scaling** returns incorrect results for text entities (TEXT, MTEXT, ATTRIB) and some other entities like ELLIPSE, SHAPE, HATCH with arc or ellipse path segments and POLYLINE/LWPOLYLINE with arc segments. (internal API) """ assert block_ref.dxftype() == 'INSERT' Ellipse = cast('Ellipse', factory.cls('ELLIPSE')) if skipped_entity_callback is None: def skipped_entity_callback(entity, reason): logger.debug( f'(Virtual Block Reference Entities) Ignoring {str(entity)}: "{reason}"' ) def disassemble(layout) -> Generator['DXFGraphic', None, None]: for entity in layout: dxftype = entity.dxftype() if dxftype == 'ATTDEF': # do not explode ATTDEF entities. Already available in Insert.attribs continue if has_non_uniform_scaling: if dxftype in {'ARC', 'CIRCLE'}: # convert ARC to ELLIPSE yield Ellipse.from_arc(entity) continue if dxftype in {'LWPOLYLINE', 'POLYLINE'} and entity.has_arc: # disassemble (LW)POLYLINE into LINE and ARC segments for segment in entity.virtual_entities(): # convert ARC to ELLIPSE if segment.dxftype() == 'ARC': yield Ellipse.from_arc(segment) else: yield segment continue # Copy entity with all DXF attributes try: copy = entity.copy() except DXFTypeError: skipped_entity_callback(entity, 'non copyable') continue # non copyable entities will be ignored if copy.dxftype() == 'HATCH': if copy.dxf.associative: # remove associations copy.dxf.associative = 0 for path in copy.paths: path.source_boundary_objects = [] if has_non_uniform_scaling and copy.paths.has_critical_elements( ): # None uniform scaling produces incorrect results for the arc and ellipse transformations. # This causes an DXF structure error for AutoCAD. # todo: requires testing skipped_entity_callback(entity, 'unsupported non-uniform scaling') continue # For the case that arc and ellipse transformation works correct someday: # copy.paths.arc_edges_to_ellipse_edges() yield copy brcs = block_ref.brcs() block_layout = block_ref.block() if block_layout is None: raise DXFStructureError( f'Required block definition for "{block_ref.dxf.name}" does not exist.' ) has_scaling = block_ref.has_scaling if has_scaling: # Non uniform scaling will produce incorrect results for some entities! # Mirroring about an axis is handled like non uniform scaling! (-1, 1, 1) # (-1, -1, -1) is uniform scaling! has_non_uniform_scaling = not block_ref.has_uniform_scaling xscale = block_ref.dxf.xscale yscale = block_ref.dxf.yscale zscale = block_ref.dxf.zscale if block_ref.has_uniform_scaling and xscale < 0: # handle reflection about all three axis -x, -y, -z explicit as non uniform scaling has_non_uniform_scaling = True if uniform_scaling_factor is not None: uniform_scaling_factor = float(uniform_scaling_factor) else: uniform_scaling_factor = block_ref.text_scaling else: xscale, yscale, zscale = (1, 1, 1) uniform_scaling_factor = 1 has_non_uniform_scaling = False for entity in disassemble(block_layout): dxftype = entity.dxftype() original_ellipse: Optional[Ellipse] = None if has_non_uniform_scaling and dxftype == 'ELLIPSE': original_ellipse = entity.copy() # Basic transformation from BRCS to WCS try: entity.transform_to_wcs(brcs) except NotImplementedError: # entities without 'transform_to_wcs' support will be ignored skipped_entity_callback(entity, 'non transformable') continue if has_scaling: # Apply DXF attribute scaling: # Simple entities without properties to scale if dxftype in { 'LINE', 'POINT', 'LWPOLYLINE', 'POLYLINE', 'MESH', 'SPLINE', 'SOLID', '3DFACE', 'TRACE', 'IMAGE', 'WIPEOUT', 'XLINE', 'RAY', 'LIGHT', 'HELIX' }: pass # nothing else to do elif dxftype in {'CIRCLE', 'ARC'}: # Non uniform scaling: ARC and CIRCLE converted to ELLIPSE # TODO: since non uniform scale => ellipse, scaling by (-s, -s, -s) is the only possible reflection here # TODO: handle reflections about z entity.dxf.radius = entity.dxf.radius * uniform_scaling_factor elif dxftype == 'ELLIPSE': # TODO: handle reflections about z if has_non_uniform_scaling: open_ellipse = not math.isclose( normalize_angle(original_ellipse.dxf.start_param), normalize_angle(original_ellipse.dxf.end_param), ) minor_axis = brcs.direction_to_wcs( original_ellipse.minor_axis) ellipse = cast('Ellipse', entity) # Transform axis major_axis = ellipse.dxf.major_axis if not math.isclose( major_axis.dot(minor_axis), 0, abs_tol=1e-9): try: major_axis, _, ratio = rytz_axis_construction( major_axis, minor_axis) except ArithmeticError: # axis construction error - skip entity skipped_entity_callback( entity, 'axis construction error - please send a bug report.' ) continue else: ratio = minor_axis.magnitude / major_axis.magnitude ellipse.dxf.major_axis = major_axis # AutoCAD does not accept a ratio < 1e-6 -> invalid DXF file ellipse.dxf.ratio = max(ratio, 1e-6) if open_ellipse: original_start_param = original_ellipse.dxf.start_param original_end_param = original_ellipse.dxf.end_param start_point, end_point = brcs.points_to_wcs( original_ellipse.vertices( (original_start_param, original_end_param))) # adjusting start- and end parameter center = ellipse.dxf.center # transformed center point extrusion = ellipse.dxf.extrusion # transformed extrusion vector, default is (0, 0, 1) start_angle = extrusion.angle_about( major_axis, start_point - center) end_angle = extrusion.angle_about( major_axis, end_point - center) start_param = angle_to_param(ratio, start_angle) end_param = angle_to_param(ratio, end_angle) # if drawing the wrong side of the ellipse if (start_param > end_param) != (original_start_param > original_end_param): start_param, end_param = end_param, start_param ellipse.dxf.start_param = start_param ellipse.dxf.end_param = end_param if ellipse.dxf.ratio > 1: ellipse.swap_axis() elif dxftype == 'MTEXT': # TODO: handle reflections. Note that the entity does store enough information to represent being # reflected. This can be seen by reflecting then exploding in Autocad. # The text will no longer be reflected. # Scale MTEXT height/width just by uniform_scaling. entity.dxf.char_height *= uniform_scaling_factor entity.dxf.width *= uniform_scaling_factor elif dxftype in {'TEXT', 'ATTRIB'}: # TODO: handle reflections. Note that the entity does store enough information to represent being # reflected. This can be seen by reflecting then exploding in Autocad. # The text will no longer be reflected. # Scale TEXT height just by uniform_scaling. entity.dxf.height *= uniform_scaling_factor elif dxftype == 'INSERT': # Set scaling of child INSERT to scaling of parent INSERT entity.dxf.xscale *= xscale entity.dxf.yscale *= yscale entity.dxf.zscale *= zscale # Scale attached ATTRIB entities: for attrib in entity.attribs: attrib.dxf.height *= uniform_scaling_factor elif dxftype == 'SHAPE': # Scale SHAPE size just by uniform_scaling. entity.dxf.size *= uniform_scaling_factor elif dxftype == 'HATCH': # Non uniform scaling produces incorrect results for boundary paths containing ARC or ELLIPSE segments. # Scale HATCH pattern: hatch = cast('Hatch', entity) if uniform_scaling_factor != 1 and hatch.has_pattern_fill and hatch.pattern is not None: hatch.dxf.pattern_scale *= uniform_scaling_factor # hatch.pattern is already scaled by the stored pattern_scale value hatch.set_pattern_definition(hatch.pattern.as_list(), uniform_scaling_factor) else: # unsupported entity will be ignored skipped_entity_callback(entity, 'unsupported entity') continue yield entity
def end_param(self) -> float: return angle_to_param(self.ratio, math.radians(self.end_angle))
def start_param(self) -> float: return angle_to_param(self.ratio, math.radians(self.start_angle))