Add zero on increment support

vlm_plotz_backup
Tyler Goodlet 2021-09-30 11:19:50 -04:00
parent 3ed0739bbe
commit fc12e72906
1 changed files with 9 additions and 0 deletions

View File

@ -172,6 +172,7 @@ async def cascade(
symbol: str,
func_name: str,
zero_on_step: bool = False,
loglevel: Optional[str] = None,
@ -233,6 +234,11 @@ async def cascade(
)
cs, index = await n.start(fsp_target)
if zero_on_step:
last = dst.array[-1:]
zeroed = np.zeros(last.shape, dtype=last.dtype)
await ctx.started(index)
profiler(f'{func_name}: fsp up')
@ -264,6 +270,9 @@ async def cascade(
# TODO: some signals, like vlm should be reset to
# zero every step.
last = array[-1:].copy()
if zero_on_step:
last = zeroed
dst.push(last)
last_len = new_len