swangnice/rotor_py_control
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added optional args for world visualization.

Browse Source
This commit is contained in:
spencerfolk
2024-10-25 09:23:05 -04:00
parent 4cf54daaff
commit 9d62b6d96d
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rotorpy/world.py
View File

@@ -158,7 +158,7 @@ class World(object):
c.transform(position=(xmin, ymin, zmin)) c.transform(position=(xmin, ymin, zmin))
return list(c.artists) return list(c.artists)
def draw(self, ax): def draw(self, ax, facecolor=None, edgecolor=None, alpha=0.6):
""" """
Draw world onto existing Axes3D axes and return artists corresponding to the Draw world onto existing Axes3D axes and return artists corresponding to the
blocks. blocks.
@@ -177,7 +177,11 @@ class World(object):
block_artists = [] block_artists = []
for b in self.world.get('blocks', []): for b in self.world.get('blocks', []):
(xmin, xmax, ymin, ymax, zmin, zmax) = b['extents'] (xmin, xmax, ymin, ymax, zmin, zmax) = b['extents']
c = Cuboid(ax, xmax-xmin, ymax-ymin, zmax-zmin, alpha=0.6, linewidth=1, edgecolors='k', facecolors=b.get('color', None)) if facecolor is None:
facecolor = b.get('color', None)
if edgecolor is None:
edgecolor = 'k'
c = Cuboid(ax, xmax-xmin, ymax-ymin, zmax-zmin, alpha=alpha, linewidth=1, edgecolors=edgecolor, facecolors=facecolor)
c.transform(position=(xmin, ymin, zmin)) c.transform(position=(xmin, ymin, zmin))
block_artists.extend(c.artists) block_artists.extend(c.artists)
return bounds_artists + block_artists return bounds_artists + block_artists