Add an option to plot bounding box diagonal on the figures

Author: orbingol

geomdl/Abstract.py

@@ -385,6 +385,7 @@ def render(self, **kwargs):
         Keyword Arguments:
             * ``cpcolor``: sets the color of the control points polygon
             * ``evalcolor``: sets the color of the curve
+            * ``bboxcolor``: sets the color of the bounding box
             * ``filename``: saves the plot with the input name
             * ``plot``: a flag to control displaying the plot window. Default is True.
 
@@ -399,6 +400,7 @@ def render(self, **kwargs):
 
         cpcolor = kwargs.get('cpcolor', 'blue')
         evalcolor = kwargs.get('evalcolor', 'black')
+        bboxcolor = kwargs.get('bboxcolor', 'darkorange')
         filename = kwargs.get('filename', None)
         plot_visible = kwargs.get('plot', True)
 
@@ -413,6 +415,7 @@ def render(self, **kwargs):
         self._vis_component.clear()
         self._vis_component.add(ptsarr=self.ctrlpts, name="Control Points", color=cpcolor, plot_type='ctrlpts')
         self._vis_component.add(ptsarr=self.evalpts, name=self.name, color=evalcolor, plot_type='evalpts')
+        self._vis_component.add(ptsarr=self.bbox, name="Bounding Box", color=bboxcolor, plot_type='bbox')
         self._vis_component.render(fig_save_as=filename, display_plot=plot_visible)
 
     def reset(self, **kwargs):
@@ -1180,6 +1183,7 @@ def render(self, **kwargs):
 
         cpcolor = kwargs.get('cpcolor', 'blue')
         evalcolor = kwargs.get('evalcolor', 'green')
+        bboxcolor = kwargs.get('bboxcolor', 'darkorange')
         trimcolor = kwargs.get('trimcolor', 'black')
         filename = kwargs.get('filename', None)
         plot_visible = kwargs.get('plot', True)
@@ -1236,6 +1240,9 @@ def render(self, **kwargs):
             self._vis_component.add(ptsarr=self.evaluate_list(trim.evalpts),
                                     name="Trim Curve " + str(idx + 1), color=trimcolor, plot_type='trimcurve')
 
+        # Bounding box
+        self._vis_component.add(ptsarr=self.bbox, name="Bounding Box", color=bboxcolor, plot_type='bbox')
+
         # Plot the surface
         self._vis_component.render(fig_save_as=filename, display_plot=plot_visible, colormap=surf_cmap)
 

geomdl/visualization/VisMPL.py

@@ -26,6 +26,7 @@ class VisConfig(Abstract.VisConfigAbstract):
 
     * ``ctrlpts`` (bool): Control points polygon/grid visibility. *Default: True*
     * ``evalpts`` (bool): Curve/surface points visibility. *Default: True*
+    * ``bbox`` (bool): Bounding box visibility. *Default: False*
     * ``legend`` (bool): Figure legend visibility. *Default: True*
     * ``axes`` (bool): Axes and figure grid visibility. *Default: True*
     * ``trims`` (bool): Trim curves visibility. *Default: True*
@@ -63,6 +64,7 @@ def __init__(self, **kwargs):
         self.dtype = np.float
         self.display_ctrlpts = kwargs.get('ctrlpts', True)
         self.display_evalpts = kwargs.get('evalpts', True)
+        self.display_bbox = kwargs.get('bbox', False)
         self.display_legend = kwargs.get('legend', True)
         self.display_axes = kwargs.get('axes', True)
         self.display_trims = kwargs.get('trims', True)
@@ -134,6 +136,12 @@ def render(self, **kwargs):
                 legend_proxy.append(curveplt)
                 legend_names.append(plot['name'])
 
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                bboxplt, = plt.plot(pts[:, 0], pts[:, 1], color=plot['color'], linestyle='--')
+                legend_proxy.append(bboxplt)
+                legend_names.append(plot['name'])
+
         # Add legend
         if self._config.display_legend:
             plt.legend(legend_proxy, legend_names)
@@ -189,15 +197,22 @@ def render(self, **kwargs):
             # Plot control points
             if plot['type'] == 'ctrlpts' and self._config.display_ctrlpts:
                 ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='-.', marker='o')
-                plot1_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
-                legend_proxy.append(plot1_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot evaluated points
             if plot['type'] == 'evalpts' and self._config.display_evalpts:
                 ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='-')
-                plot2_proxy = mpl.lines.Line2D([0], [0], linestyle='-', color=plot['color'])
-                legend_proxy.append(plot2_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-', color=plot['color'])
+                legend_proxy.append(plot_proxy)
+                legend_names.append(plot['name'])
+
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='--')
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='--', color=plot['color'])
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
         # Add legend to 3D plot, @ref: https://stackoverflow.com/a/20505720
@@ -255,8 +270,8 @@ def render(self, **kwargs):
                 pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                 cp_z = pts[:, 2] + self._ctrlpts_offset
                 ax.plot(pts[:, 0], pts[:, 1], cp_z, color=plot['color'], linestyle='-.', marker='o')
-                plot1_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
-                legend_proxy.append(plot1_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot evaluated points
@@ -265,8 +280,16 @@ def render(self, **kwargs):
                 for tri in tris:
                     pts = np.array(tri.vertices_raw, dtype=self._config.dtype)
                     ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'])
-                plot2_proxy = mpl.lines.Line2D([0], [0], linestyle='-', color=plot['color'])
-                legend_proxy.append(plot2_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-', color=plot['color'])
+                legend_proxy.append(plot_proxy)
+                legend_names.append(plot['name'])
+
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
+                ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='--')
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='--', color=plot['color'])
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot trim curves
@@ -275,8 +298,8 @@ def render(self, **kwargs):
                     pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                     ax.scatter(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], marker='o',
                                s=self._config.trim_size, depthshade=False)
-                    plot3_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
-                    legend_proxy.append(plot3_proxy)
+                    plot_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
+                    legend_proxy.append(plot_proxy)
                     legend_names.append(plot['name'])
 
         # Add legend to 3D plot, @ref: https://stackoverflow.com/a/20505720
@@ -334,16 +357,24 @@ def render(self, **kwargs):
                 pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                 cp_z = pts[:, 2] + self._ctrlpts_offset
                 ax.scatter(pts[:, 0], pts[:, 1], cp_z, color=plot['color'], s=25, depthshade=True)
-                plot1_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
-                legend_proxy.append(plot1_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot evaluated points
             if plot['type'] == 'evalpts' and self._config.display_evalpts:
                 pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                 ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'])
-                plot2_proxy = mpl.lines.Line2D([0], [0], linestyle='-', color=plot['color'])
-                legend_proxy.append(plot2_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-', color=plot['color'])
+                legend_proxy.append(plot_proxy)
+                legend_names.append(plot['name'])
+
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
+                ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='--')
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='--', color=plot['color'])
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot trim curves
@@ -352,8 +383,8 @@ def render(self, **kwargs):
                     pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                     ax.scatter(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], marker='o',
                                s=self._config.trim_size, depthshade=False)
-                    plot3_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
-                    legend_proxy.append(plot3_proxy)
+                    plot_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
+                    legend_proxy.append(plot_proxy)
                     legend_names.append(plot['name'])
 
         # Add legend to 3D plot, @ref: https://stackoverflow.com/a/20505720
@@ -426,8 +457,8 @@ def render(self, **kwargs):
                 pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                 cp_z = pts[:, 2] + self._ctrlpts_offset
                 ax.plot(pts[:, 0], pts[:, 1], cp_z, color=plot['color'], linestyle='-.', marker='o')
-                plot1_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
-                legend_proxy.append(plot1_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot evaluated points
@@ -456,8 +487,16 @@ def render(self, **kwargs):
 
                 # Use custom Triangulation object and the choice of color/colormap to plot the surface
                 ax.plot_trisurf(triangulation, pts[:, 2], **trisurf_params)
-                plot2_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='^')
-                legend_proxy.append(plot2_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='^')
+                legend_proxy.append(plot_proxy)
+                legend_names.append(plot['name'])
+
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
+                ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='--')
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='--', color=plot['color'])
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot trim curves
@@ -466,8 +505,8 @@ def render(self, **kwargs):
                     pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                     ax.scatter(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], marker='o',
                                s=self._config.trim_size, depthshade=False)
-                    plot3_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
-                    legend_proxy.append(plot3_proxy)
+                    plot_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
+                    legend_proxy.append(plot_proxy)
                     legend_names.append(plot['name'])
 
         # Add legend to 3D plot, @ref: https://stackoverflow.com/a/20505720
@@ -525,16 +564,24 @@ def render(self, **kwargs):
                 pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                 cp_z = pts[:, 2] + self._ctrlpts_offset
                 ax.plot(pts[:, 0], pts[:, 1], cp_z, color=plot['color'], linestyle='-.', marker='o')
-                plot1_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
-                legend_proxy.append(plot1_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='-.', color=plot['color'], marker='o')
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot evaluated points
             if plot['type'] == 'evalpts' and self._config.display_evalpts:
                 pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                 ax.scatter(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], s=50, depthshade=True)
-                plot2_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
-                legend_proxy.append(plot2_proxy)
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
+                legend_proxy.append(plot_proxy)
+                legend_names.append(plot['name'])
+
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
+                ax.plot(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], linestyle='--')
+                plot_proxy = mpl.lines.Line2D([0], [0], linestyle='--', color=plot['color'])
+                legend_proxy.append(plot_proxy)
                 legend_names.append(plot['name'])
 
             # Plot trim curves
@@ -543,8 +590,8 @@ def render(self, **kwargs):
                     pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
                     ax.scatter(pts[:, 0], pts[:, 1], pts[:, 2], color=plot['color'], marker='o',
                                s=self._config.trim_size, depthshade=False)
-                    plot3_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
-                    legend_proxy.append(plot3_proxy)
+                    plot_proxy = mpl.lines.Line2D([0], [0], linestyle='none', color=plot['color'], marker='o')
+                    legend_proxy.append(plot_proxy)
                     legend_names.append(plot['name'])
 
         # Add legend to 3D plot, @ref: https://stackoverflow.com/a/20505720

geomdl/visualization/VisPlotly.py

@@ -24,6 +24,7 @@ class VisConfig(Abstract.VisConfigAbstract):
 
     * ``ctrlpts`` (bool): Control points polygon/grid visibility. *Default: True*
     * ``evalpts`` (bool): Curve/surface points visibility. *Default: True*
+    * ``bbox`` (bool): Bounding box visibility. *Default: False*
     * ``legend`` (bool): Figure legend visibility. *Default: True*
     * ``axes`` (bool): Axes and figure grid visibility. *Default: True*
     * ``trims`` (bool): Trim curves visibility. *Default: True*
@@ -77,6 +78,7 @@ def __init__(self, **kwargs):
         # Get keyword arguments
         self.display_ctrlpts = kwargs.get('ctrlpts', True)
         self.display_evalpts = kwargs.get('evalpts', True)
+        self.display_bbox = kwargs.get('bbox', False)
         self.display_trims = kwargs.get('trims', True)
         self.display_legend = kwargs.get('legend', True)
         self.display_axes = kwargs.get('axes', True)
@@ -112,7 +114,7 @@ def render(self, **kwargs):
                     line=dict(
                         color=plot['color'],
                         width=self._config.line_width,
-                        dash='dashdot'
+                        dash='dash'
                     )
                 )
                 plot_data.append(figure)
@@ -131,6 +133,20 @@ def render(self, **kwargs):
                 )
                 plot_data.append(figure)
 
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                figure = graph_objs.Scatter(
+                    x=pts[:, 0],
+                    y=pts[:, 1],
+                    name=plot['name'],
+                    line=dict(
+                        color=plot['color'],
+                        width=self._config.line_width,
+                        dash='dashdot',
+                    )
+                )
+                plot_data.append(figure)
+
         plot_layout = dict(
             width=self._config.figure_size[0],
             height=self._config.figure_size[1],
@@ -203,7 +219,7 @@ def render(self, **kwargs):
                     line=dict(
                         color=plot['color'],
                         width=self._config.line_width,
-                        dash='dashdot'
+                        dash='dash'
                     ),
                     marker=dict(
                         color=plot['color'],
@@ -227,6 +243,22 @@ def render(self, **kwargs):
                 )
                 plot_data.append(figure)
 
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                figure = graph_objs.Scatter3d(
+                    x=pts[:, 0],
+                    y=pts[:, 1],
+                    z=pts[:, 2],
+                    name=plot['name'],
+                    mode='lines',
+                    line=dict(
+                        color=plot['color'],
+                        width=self._config.line_width,
+                        dash='dashdot',
+                    ),
+                )
+                plot_data.append(figure)
+
         plot_layout = dict(
             width=self._config.figure_size[0],
             height=self._config.figure_size[1],
@@ -347,6 +379,23 @@ def render(self, **kwargs):
                 )
                 plot_data.append(figure)
 
+            # Plot bounding box
+            if plot['type'] == 'bbox' and self._config.display_bbox:
+                pts = np.array(plot['ptsarr'], dtype=self._config.dtype)
+                figure = graph_objs.Scatter3d(
+                    x=pts[:, 0],
+                    y=pts[:, 1],
+                    z=pts[:, 2],
+                    name=plot['name'],
+                    mode='lines',
+                    line=dict(
+                        color=plot['color'],
+                        width=self._config.line_width,
+                        dash='dashdot',
+                    ),
+                )
+                plot_data.append(figure)
+
             # Plot trim curves
             if self._config.display_trims:
                 if plot['type'] == 'trimcurve':