h3n3/jmwright-CQ-Editor
1
0
Fork 0
mirror of https://github.com/jdegenstein/jmwright-CQ-Editor.git synced 2025-12-20 22:47:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Bring up to date with mainline CQ-editor ec81ad9

Browse Source 
partial PR for cq_utils.py only
This commit is contained in:
jdegenstein
2022-10-19 13:15:49 -05:00
committed by GitHub
parent b6532d0cae
commit 4ee59fdeb1
1 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
cq_editor/cq_utils.py
View File

@@ -7,7 +7,7 @@ from types import SimpleNamespace
from OCP.XCAFPrs import XCAFPrs_AISObject from OCP.XCAFPrs import XCAFPrs_AISObject
from OCP.TopoDS import TopoDS_Shape from OCP.TopoDS import TopoDS_Shape
from OCP.AIS import AIS_InteractiveObject, AIS_Shape, AIS_ColoredShape from OCP.AIS import AIS_InteractiveObject, AIS_Shape
from OCP.Quantity import \ from OCP.Quantity import \
Quantity_TOC_RGB as TOC_RGB, Quantity_Color Quantity_TOC_RGB as TOC_RGB, Quantity_Color
@@ -66,7 +66,7 @@ def make_AIS(obj : Union[cq.Workplane, List[cq.Workplane], cq.Shape, List[cq.Sha
ais = obj ais = obj
else: else:
shape = to_compound(obj) shape = to_compound(obj)
ais = AIS_ColoredShape(shape.wrapped) ais = AIS_Shape(shape.wrapped)
if 'alpha' in options: if 'alpha' in options:
ais.SetTransparency(options['alpha']) ais.SetTransparency(options['alpha'])
@@ -109,13 +109,23 @@ def to_occ_color(color) -> Quantity_Color:
color.blueF(), color.blueF(),
TOC_RGB) TOC_RGB)
def get_occ_color(ais : AIS_ColoredShape) -> QColor: def get_occ_color(obj : Union[AIS_InteractiveObject, Quantity_Color]) -> QColor:
if isinstance(obj, AIS_InteractiveObject):
color = Quantity_Color() color = Quantity_Color()
ais.Color(color) obj.Color(color)
else:
color = obj
return QColor.fromRgbF(color.Red(), color.Green(), color.Blue()) return QColor.fromRgbF(color.Red(), color.Green(), color.Blue())
def set_color(ais : AIS_Shape, color : Quantity_Color) -> AIS_Shape:
drawer = ais.Attributes()
drawer.ShadingAspect().SetColor(color)
return ais
def reload_cq(): def reload_cq():
# NB: order of reloads is important # NB: order of reloads is important