custom.py 417 B

123456789101112131415161718192021
  1. from cmsisdsp.sdf.nodes.simu import *
  2. import numpy as np
  3. import cmsisdsp as dsp
  4. class Processing(GenericNode):
  5. def __init__(self,inputSize,outputSize,fifoin,fifoout):
  6. GenericNode.__init__(self,inputSize,outputSize,fifoin,fifoout)
  7. def run(self):
  8. i=self.getReadBuffer()
  9. o=self.getWriteBuffer()
  10. b=dsp.arm_scale_q15(i,0x6000,1)
  11. o[:]=b[:]
  12. return(0)