def render_locally(self,
addr="localhost",
server_port=8080,
open_webbrowser=False,
render_engine="threejs",
resolution=(1800, 900)):
from OCC.Display.WebGl.simple_server import start_server
if render_engine == "xdom":
from OCC.Display.WebGl import x3dom_renderer
my_renderer = x3dom_renderer.X3DomRenderer()
# TODO: Find similarities in build processing as done for THREE.js (tesselate geom etc..).
# my_renderer.DisplayShape(shape.profile_curve_outer.wire)
# my_renderer.DisplayShape(shape.sweep_curve.wire)
# my_renderer.DisplayShape(shape.geom)
my_renderer.render()
else: # Assume THREEJS
from ipywidgets.embed import embed_minimal_html
from ada.base.renderer import MyRenderer
_path = pathlib.Path("temp/index.html").resolve().absolute()
renderer = MyRenderer(resolution)
renderer.DisplayObj(self)
renderer.build_display()
os.makedirs(_path.parent, exist_ok=True)
embed_minimal_html(_path,
views=renderer._renderer,
title="Pythreejs Viewer")
start_server(addr, server_port, str(_path.parent), open_webbrowser)
def get_render_snippet(self, view_size=None):
"""
Return the html snippet containing threejs renderer
"""
from ipywidgets.embed import embed_snippet
from ada.base.renderer import MyRenderer
renderer = MyRenderer()
renderer.DisplayObj(self)
renderer.build_display()
return embed_snippet(renderer._renderer)
def _repr_html_(self):
from IPython.display import display
from ipywidgets import HBox, VBox
from ada.base.renderer import MyRenderer
renderer = MyRenderer()
renderer.DisplayObj(self)
renderer.build_display()
self._renderer = renderer
display(
HBox([
VBox([HBox(renderer._controls), renderer._renderer]),
renderer.html
]))
# renderer._reset()
# self._renderer.Display()
return ""
def _repr_html_(self):
from ada.base.renderer import MyRenderer
self._renderer = MyRenderer()
self._renderer.DisplayObj(self)
self._renderer.Display()
class BackendGeom(Backend):
"""
The shared backend of all physical components (Beam, Plate) or aggregate of components (Part, Assembly).
"""
_renderer = None
def __init__(self,
name,
guid=None,
metadata=None,
units="m",
parent=None,
colour=None):
super().__init__(name, guid, metadata, units, parent)
self._penetrations = []
self.colour = colour
def add_penetration(self, pen):
"""
:param pen:
"""
from ada import Penetration, PrimBox, PrimCyl, PrimExtrude, PrimRevolve
if type(pen) not in [
Penetration, PrimExtrude, PrimRevolve, PrimCyl, PrimBox
]:
raise ValueError(f'Unsupported penetration type "{type(pen)}"')
pen.parent = self
if type(pen) in (PrimExtrude, PrimRevolve, PrimCyl, PrimBox):
self._penetrations.append(Penetration(pen, parent=self))
else:
self._penetrations.append(pen)
return pen
def to_stp(self, destination_file, geom_repr="solid", schema="AP242"):
"""
Write current assembly to STEP file
OpenCascade reference:
https://www.opencascade.com/doc/occt-7.4.0/overview/html/occt_user_guides__step.html#occt_step_3
:param destination_file:
:param geom_repr:
:param schema: STEP Schemas.
"""
from OCC.Core.IFSelect import IFSelect_RetError
from OCC.Core.Interface import Interface_Static_SetCVal
from OCC.Core.STEPConstruct import stepconstruct_FindEntity
from OCC.Core.STEPControl import STEPControl_AsIs, STEPControl_Writer
from OCC.Core.TCollection import TCollection_HAsciiString
from ada.core.utils import Counter
if geom_repr not in ["shell", "solid"]:
raise ValueError(
'Geometry representation can only accept either "solid" or "shell" as input'
)
destination_file = pathlib.Path(destination_file).with_suffix(".stp")
assembly_mode = 1
shp_names = Counter(1, "shp")
writer = STEPControl_Writer()
fp = writer.WS().TransferWriter().FinderProcess()
Interface_Static_SetCVal("write.step.schema", schema)
# Interface_Static_SetCVal('write.precision.val', '1e-5')
Interface_Static_SetCVal("write.precision.mode", "1")
Interface_Static_SetCVal("write.step.assembly", str(assembly_mode))
from ada import Assembly, Beam, Part, Pipe, Plate, Shape, Wall
def add_geom(geo, o):
name = o.name if o.name is not None else next(shp_names)
Interface_Static_SetCVal("write.step.product.name", name)
stat = writer.Transfer(geo, STEPControl_AsIs)
if int(stat) > int(IFSelect_RetError):
raise Exception("Some Error occurred")
item = stepconstruct_FindEntity(fp, geo)
if not item:
logging.debug("STEP item not found for FindEntity")
else:
item.SetName(TCollection_HAsciiString(name))
if type(self) is Shape:
assert isinstance(self, Shape)
add_geom(self.geom, self)
elif type(self) in (Beam, Plate, Wall):
assert isinstance(self, (Beam, Plate, Wall))
if geom_repr == "shell":
add_geom(self.shell, self)
else:
add_geom(self.solid, self)
elif type(self) is Pipe:
assert isinstance(self, Pipe)
for geom in self.geometries:
add_geom(geom, self)
elif type(self) in (Part, Assembly):
assert isinstance(self, Part)
for p in self.get_all_subparts() + [self]:
for obj in list(p.plates) + list(p.beams) + list(
p.shapes) + list(p.pipes) + list(p.walls):
if type(obj) in (Plate, Beam, Wall):
try:
if geom_repr == "shell":
add_geom(obj.shell, self)
else:
add_geom(obj.solid, self)
except BaseException as e:
logging.info(f'passing pl "{obj.name}" due to {e}')
continue
elif type(obj) in (Pipe, ):
assert isinstance(obj, Pipe)
for geom in obj.geometries:
add_geom(geom, self)
elif type(obj) is Shape:
add_geom(obj.geom, self)
else:
raise ValueError("Unkown Geometry type")
os.makedirs(destination_file.parent, exist_ok=True)
status = writer.Write(str(destination_file))
if int(status) > int(IFSelect_RetError):
raise Exception("Error during write operation")
print(f'step file created at "{destination_file}"')
def render_locally(self,
addr="localhost",
server_port=8080,
open_webbrowser=False,
render_engine="threejs"):
from OCC.Display.WebGl.simple_server import start_server
if render_engine == "xdom":
from OCC.Display.WebGl import x3dom_renderer
my_renderer = x3dom_renderer.X3DomRenderer()
# TODO: Find similarities in build processing as done for THREE.js (tesselate geom etc..).
# my_renderer.DisplayShape(shape.profile_curve_outer.wire)
# my_renderer.DisplayShape(shape.sweep_curve.wire)
# my_renderer.DisplayShape(shape.geom)
my_renderer.render()
else: # Assume THREEJS
from ipywidgets.embed import embed_minimal_html
from ada.base.renderer import MyRenderer
_path = pathlib.Path("temp/index.html").resolve().absolute()
renderer = MyRenderer((1800, 900))
renderer.DisplayObj(self)
renderer.build_display()
os.makedirs(_path.parent, exist_ok=True)
embed_minimal_html(_path,
views=renderer.renderer,
title="Pythreejs Viewer")
start_server(addr, server_port, str(_path.parent), open_webbrowser)
def get_render_snippet(self, view_size=None):
"""
Return the html snippet containing threejs renderer
"""
from ipywidgets.embed import embed_snippet
from ada.base.renderer import MyRenderer
renderer = MyRenderer()
renderer.DisplayObj(self)
renderer.build_display()
return embed_snippet(renderer.renderer)
@property
def colour(self):
return self._colour
@colour.setter
def colour(self, value):
if type(value) is str:
if value.lower() not in _cmap.keys():
raise ValueError("Currently unsupported")
self._colour = _cmap[value.lower()]
else:
self._colour = value
@property
def colour_webgl(self):
from OCC.Display.WebGl.jupyter_renderer import format_color
if self.colour is None:
return None
if self.colour[0] == -1 and self.colour[1] == -1 and self.colour[
2] == -1:
return None
if self.colour[0] <= 1.0:
colour = [int(x * 255) for x in self.colour]
else:
colour = [int(x) for x in self.colour]
colour_formatted = format_color(*colour)
return colour_formatted
@property
def penetrations(self):
return self._penetrations
def _repr_html_(self):
from ada.base.renderer import MyRenderer
self._renderer = MyRenderer()
self._renderer.DisplayObj(self)
self._renderer.Display()