You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
828 B
Python
46 lines
828 B
Python
import numpy as np
|
|
|
|
from cmsisdsp.cg.scheduler import *
|
|
|
|
|
|
class Processing(GenericNode):
|
|
def __init__(self,name,outLength):
|
|
GenericNode.__init__(self,name)
|
|
self.addInput("i",CType(Q15),outLength)
|
|
self.addOutput("o",CType(Q15),outLength)
|
|
|
|
@property
|
|
def typeName(self):
|
|
return "Processing"
|
|
|
|
|
|
BUFSIZE=128
|
|
### Define nodes
|
|
src=VHTSource("src",BUFSIZE,0)
|
|
processing=Processing("proc",BUFSIZE)
|
|
sink=VHTSink("sink",BUFSIZE,0)
|
|
|
|
|
|
g = Graph()
|
|
|
|
g.connect(src.o, processing.i)
|
|
g.connect(processing.o, sink.i)
|
|
|
|
|
|
|
|
print("Generate graphviz and code")
|
|
|
|
|
|
|
|
conf=Configuration()
|
|
sched = g.computeSchedule(conf)
|
|
print("Schedule length = %d" % sched.scheduleLength)
|
|
print("Memory usage %d bytes" % sched.memory)
|
|
|
|
|
|
sched.pythoncode(".",config=conf)
|
|
|
|
with open("test.dot","w") as f:
|
|
sched.graphviz(f)
|
|
|