Skip to content

fullcontrol: visualize: add TubeMesh #18

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
AndyGlx merged 1 commit into from
Jun 8, 2023
Merged

fullcontrol: visualize: add TubeMesh #18

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions fullcontrol/visualize/controls.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,9 @@ class PlotControls(BaseModel):
'control to adjust the style of the plot'
color_type: Optional[str] = 'z_gradient'
line_width: Optional[float] = 2
style: Optional[str] = None # 'tube'/'line'
tube_type: Optional[str] = None # 'flow'/'cylinders'
tube_sides: Optional[int] = None
zoom: Optional[float] = 1
hide_annotations: Optional[bool] = False
hide_travel: Optional[bool] = False
Expand Down
119 changes: 75 additions & 44 deletions fullcontrol/visualize/plotly.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,75 +1,106 @@

import numpy as np
import plotly.graph_objects as go
from fullcontrol.visualize.plot_data import PlotData
from fullcontrol.visualize.controls import PlotControls
from fullcontrol.visualize.tube_mesh import CylindersMesh, FlowTubeMesh


def plot(data: PlotData, controls: PlotControls):
'plot data for x y z lines with RGB colors and annotations. style of plot governed by controls'
fig = go.Figure()

def column(matrix: list, i: int):
'return list of ith elements from a 2D array'
return [row[i] for row in matrix]
if controls.style is None:
if controls.tube_type is None:
print('Plot shows printed line width - use `fc.PlotControls(style="line")` for a simple path, or'
' `fc.PlotControls(style="tube")` to disable this message.')
controls.style = 'tube'

if controls.tube_type is not None:
Mesh = {'flow': FlowTubeMesh, 'cylinders': CylindersMesh}[controls.tube_type]
else:
Mesh = FlowTubeMesh

# generate line plots
path_count = len(data.paths)
for i in range(path_count):
line_point_count = len(data.paths[i].xvals)
colors_now = []
for j in range(line_point_count):
colors_now.append(
f'rgb({data.paths[i].colors[j][0]*255:.2f}, {data.paths[i].colors[j][1]*255:.2f}, {data.paths[i].colors[j][2]*255:.2f})')
any_mesh_plots = False
max_width = 0
for path in data.paths:
colors_now = [f'rgb({color[0]*255:.2f}, {color[1]*255:.2f}, {color[2]*255:.2f})' for color in path.colors]
linewidth_now = controls.line_width * \
2 if data.paths[i].extruder.on == True else controls.line_width*0.5
if not controls.hide_travel or data.paths[i].extruder.on == True:
fig.add_trace(go.Scatter3d(mode='lines', x=data.paths[i].xvals, y=data.paths[i].yvals,
z=data.paths[i].zvals, showlegend=False, line=dict(width=linewidth_now, color=colors_now)))
2 if path.extruder.on == True else controls.line_width*0.5
if path.extruder.on and controls.style == 'tube':
path_points = np.array([path.xvals, path.yvals, path.zvals]).T
good_points = np.ones(len(path_points), dtype=bool)
dups = np.all(np.diff(path_points, axis=0)==0, axis=1)
if np.any(dups):
# remove successive duplicate points so TubeMesh can be generated
good_points[1:] = ~dups
colors_now = np.array(colors_now, dtype=object)[good_points]
path_points = path_points[good_points]
num_path_points = len(path_points)
neat = bool(controls.neat_for_publishing)
# if automatic, dynamically reduce the number of sides when plotting large numbers of path points
sides = controls.tube_sides or (6 if num_path_points < 10 else 4 if num_path_points < 1_000_000 else 2)
capped = neat or num_path_points < 100
widths = path.widths
if not widths:
local_max = widths = linewidth_now/10
else:
widths = np.array(widths)[good_points]
if Mesh == CylindersMesh:
widths = widths[1:]
local_max = max(widths)
heights = path.heights or None
if heights:
heights = np.array(heights)[good_points]
if Mesh == CylindersMesh:
heights = heights[1:]
fig.add_trace(Mesh(path_points, widths=widths, heights=heights, sides=sides, capped=capped, inplace_path=True)
.to_Mesh3d(colors=colors_now))
any_mesh_plots = True
max_width = max(max_width, local_max)
elif not controls.hide_travel or path.extruder.on:
fig.add_trace(go.Scatter3d(mode='lines', x=path.xvals, y=path.yvals, z=path.zvals,
showlegend=False, line=dict(width=linewidth_now, color=colors_now)))

# find a bounding box, to create a plot with equally proportioned X Y Z scales (so a cuboid looks like a cuboid, not a cube)
bounding_box_size = max(data.bounding_box.maxx-data.bounding_box.minx, data.bounding_box.maxy -
data.bounding_box.miny, data.bounding_box.maxz-min(0, data.bounding_box.minz))
bounding_box_size += 0.002
bounding_box_size += max_width

# generate annotations
annotations_count = len(data.annotations)
annotations_pts = []
annotations = []
if controls.hide_annotations == False and not controls.neat_for_publishing:
for i in range(annotations_count):
annotations_pts.append(
[data.annotations[i]['x'], data.annotations[i]['y'], data.annotations[i]['z']])
for annotation in data.annotations:
x, y, z = (annotation[axis] for axis in 'xyz')
annotations_pts.append([x, y, z])
annotations.append(dict(
showarrow=False,
x=annotations_pts[i][0],
y=annotations_pts[i][1],
z=annotations_pts[i][2],
text=data.annotations[i]['label'],
x=x, y=y, z=z,
text=annotation['label'],
yshift=10))
fig.add_trace(go.Scatter3d(mode='markers', x=column(annotations_pts, 0), y=column(
annotations_pts, 1), z=column(annotations_pts, 2), showlegend=False, marker=dict(size=2, color='red')))
xs, ys, zs = zip(*annotations_pts) if annotations_pts else [[]]*3
fig.add_trace(go.Scatter3d(mode='markers', x=xs, y=ys, z=zs, showlegend=False, marker=dict(size=2, color='red')))

# make sure the bounding box is big enough for the annotations
# the 0.001 is to make sure the annotations don't lie on the boundary
for i in range(annotations_count):
if annotations_pts[i][0] < data.bounding_box.midx-bounding_box_size/2+0.001:
bounding_box_size = 2 * \
(data.bounding_box.midx-annotations_pts[i][0])+0.002
if annotations_pts[i][0] > data.bounding_box.midx+bounding_box_size/2-0.001:
bounding_box_size = 2 * \
(annotations_pts[i][0]-data.bounding_box.midx)+0.002
if annotations_pts[i][1] < data.bounding_box.midy-bounding_box_size/2+0.001:
bounding_box_size = 2 * \
(data.bounding_box.midy-annotations_pts[i][1])+0.002
if annotations_pts[i][1] > data.bounding_box.midy+bounding_box_size/2-0.001:
bounding_box_size = 2 * \
(annotations_pts[i][1]-data.bounding_box.midy)+0.002
if annotations_pts[i][2] < data.bounding_box.midz-bounding_box_size/2+0.001:
bounding_box_size = 2 * \
(data.bounding_box.midz-annotations_pts[i][2])+0.002
if annotations_pts[i][2] > data.bounding_box.midz+bounding_box_size/2-0.001:
bounding_box_size = 2 * \
(annotations_pts[i][2]-data.bounding_box.midz)+0.002
midx, midy, midz = (getattr(data.bounding_box, f'mid{axis}') for axis in 'xyz')
offset = 0.001
offset_both_sides = 2 * offset
for (x, y, z) in annotations_pts:
if x < midx - bounding_box_size / 2 + offset:
bounding_box_size = 2 * (midx - x) + offset_both_sides
if x > midx + bounding_box_size / 2 - offset:
bounding_box_size = 2 * (x - midx) + offset_both_sides
if y < midy - bounding_box_size / 2 + offset:
bounding_box_size = 2 * (midy - y) + offset_both_sides
if y > midy + bounding_box_size / 2 - offset:
bounding_box_size = 2 * (y - midy) + offset_both_sides
if z < midz - bounding_box_size / 2 + offset:
bounding_box_size = 2 * (midz - z) + offset_both_sides
if z > midz + bounding_box_size / 2 - offset:
bounding_box_size = 2 * (z - midz) + offset_both_sides

camera = dict(eye=dict(x=-0.5/controls.zoom, y=-1/controls.zoom, z=-0.5+0.5/controls.zoom),
center=dict(x=0, y=0, z=-0.5))
Expand Down
Loading