Move MatplotlibRenderer.create_window parameter into cfg

pull/357/head
nyanpasu64 2018-08-14 05:31:12 -07:00
rodzic 06159555a1
commit 4a25b41d22
3 zmienionych plików z 5 dodań i 7 usunięć

Wyświetl plik

@ -33,7 +33,6 @@ class Config:
render: RendererConfig
outputs: List[outputs.IOutputConfig]
create_window: bool = False
@property
def render_width_s(self) -> float:
@ -97,12 +96,11 @@ class Ovgen:
# Calculate number of frames (TODO master file?)
render_width_s = self.cfg.render_width_s
fps = self.cfg.fps
create_window = self.cfg.create_window
nframes = fps * self.waves[0].get_s()
nframes = int(nframes) + 1
renderer = MatplotlibRenderer(self.cfg.render, self.nchan, create_window)
renderer = MatplotlibRenderer(self.cfg.render, self.nchan)
if RENDER_PROFILING:
begin = time.perf_counter()

Wyświetl plik

@ -24,6 +24,7 @@ class RendererConfig:
nrows: Optional[int] = None
ncols: Optional[int] = None
create_window: bool = False
def __post_init__(self):
if not self.nrows:
@ -61,10 +62,9 @@ class MatplotlibRenderer:
DPI = 96
def __init__(self, cfg: RendererConfig, nplots: int, create_window: bool):
def __init__(self, cfg: RendererConfig, nplots: int):
self.cfg = cfg
self.nplots = nplots
self.create_window = create_window
# Setup layout
# "ncols=1" is good for vertical layouts.
@ -121,7 +121,7 @@ class MatplotlibRenderer:
self.cfg.width / self.DPI,
self.cfg.height / self.DPI
)
if self.create_window:
if self.cfg.create_window:
plt.show(block=False)
def _calc_layout(self) -> Tuple[int, int]:

Wyświetl plik

@ -36,7 +36,7 @@ def test_renderer():
cfg = RendererConfig(WIDTH, HEIGHT, ncols=2)
nplots = 16
r = MatplotlibRenderer(cfg, nplots, create_window=False)
r = MatplotlibRenderer(cfg, nplots)
# 2 columns, 8 rows
assert r.ncols == 2