File size: 821 Bytes
663494c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import matplotlib.pyplot as plt
from pyquaternion import Quaternion
class BaseRender:
"""
BaseRender class
"""
def __init__(
self,
figsize=(10, 10)):
self.figsize = figsize
self.fig, self.axes = None, None
def reset_canvas(self, dx=1, dy=1, tight_layout=False):
plt.close()
plt.gca().set_axis_off()
plt.axis('off')
self.fig, self.axes = plt.subplots(dx, dy, figsize=self.figsize)
if tight_layout:
plt.tight_layout()
def close_canvas(self):
plt.close()
def save_fig(self, filename):
plt.subplots_adjust(top=1, bottom=0, right=1, left=0,
hspace=0, wspace=0)
plt.margins(0, 0)
print(f'saving to {filename}')
plt.savefig(filename)
|