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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
from os.path import basename, splitext
from unittest import TestCase, TestLoader
import pysimulavr
class SimulavrAdapter(object):
DEFAULT_CLOCK_SETTING = 250 # 250ns or 4MHz
def loadDevice(self, t, e):
pysimulavr.cvar.sysConHandler.SetUseExit(False)
self.__sc = pysimulavr.SystemClock.Instance()
self.__sc.ResetClock()
dev = pysimulavr.AvrFactory.instance().makeDevice(t)
dev.Load(e)
dev.SetClockFreq(self.DEFAULT_CLOCK_SETTING)
self.__sc.Add(dev)
return dev
def doRun(self, n):
ct = self.__sc.GetCurrentTime
while ct() < n:
res = self.__sc.Step()
if res is not 0: return res
return 0
def doStep(self, stepcount = 1):
while stepcount > 0:
res = self.__sc.Step()
if res is not 0: return res
stepcount -= 1
return 0
def getCurrentTime(self):
return self.__sc.GetCurrentTime()
def getAllRegisteredTraceValues(self):
os = pysimulavr.ostringstream()
pysimulavr.DumpManager.Instance().save(os)
return filter(None, [i.strip() for i in os.str().split("\n")])
def dmanSingleDeviceApplication(self):
pysimulavr.DumpManager.Instance().SetSingleDeviceApp()
def dmanStart(self):
pysimulavr.DumpManager.Instance().start()
def dmanStop(self):
pysimulavr.DumpManager.Instance().stopApplication()
def setVCDDump(self, vcdname, signals, rstrobe = False, wstrobe = False):
dman = pysimulavr.DumpManager.Instance()
sigs = ["+ " + i for i in signals]
dman.addDumpVCD(vcdname, "\n".join(sigs), "ns", rstrobe, wstrobe)
def getWordByName(self, dev, label):
addr = dev.data.GetAddressAtSymbol(label)
v = dev.getRWMem(addr)
addr += 1
v = (dev.getRWMem(addr) << 8) + v
return v
def getLongByName(self, dev, label):
addr = dev.data.GetAddressAtSymbol(label) + 3
v = 0
for i in range(4):
v = (v << 8) + dev.getRWMem(addr)
addr -= 1
return v
def getByteByName(self, dev, label):
addr = dev.data.GetAddressAtSymbol(label)
return dev.getRWMem(addr)
def setByteByName(self, dev, label, value):
addr = dev.data.GetAddressAtSymbol(label)
dev.setRWMem(addr, value)
class SimTestLoader(TestLoader):
def __init__(self, elffile):
TestLoader.__init__(self)
self.elffile = elffile
def loadTestsFromTestCase(self, testCaseClass):
testCaseNames = self.getTestCaseNames(testCaseClass)
if not testCaseNames and hasattr(testCaseClass, "runTest"):
testCaseNames = ["runTest"]
return self.suiteClass([testCaseClass(m, self.elffile) for m in testCaseNames])
class PyTestLoader(TestLoader):
def __init__(self, modname):
TestLoader.__init__(self)
self.modname = modname
def loadTestsFromTestCase(self, testCaseClass):
testCaseNames = self.getTestCaseNames(testCaseClass)
if not testCaseNames and hasattr(testCaseClass, "runTest"):
testCaseNames = ["runTest"]
return self.suiteClass([testCaseClass(m, self.modname) for m in testCaseNames])
class SimTestCase(TestCase):
def __init__(self, methodName, elfName):
TestCase.__init__(self, methodName)
self.elfName = elfName
self.processorName = splitext(basename(elfName))[0].split("_")[-1]
def shortDescription(self):
md = TestCase.shortDescription(self)
if md is None: md = self._testMethodName
d = self.elfName.split(".")[0]
return d + "::" + md
def getDevice(self):
self.sim = SimulavrAdapter()
self.dev = self.sim.loadDevice(self.processorName, self.elfName)
def setUp(self):
self.getDevice()
self.sim.dmanStart()
def tearDown(self):
self.sim.dmanStop()
del self.dev
del self.sim
def assertDevice(self):
self.assertTrue(hasattr(self, "sim") and self.sim is not None and hasattr(self, "dev") and self.dev is not None, "Simulavr device not created")
def assertStartTime(self):
self.assertEqual(self.sim.getCurrentTime(), 0)
def assertInitDone(self, limit = 20000, mainSymbol = "main"):
bpaddr = self.dev.Flash.GetAddressAtSymbol(mainSymbol)
self.dev.BP.AddBreakpoint(bpaddr)
self.sim.doRun(limit)
self.assertTrue(self.sim.getCurrentTime() < limit, "break point '%s' not arrived" % mainSymbol)
self.dev.BP.RemoveBreakpoint(bpaddr)
def assertWordValue(self, label, value):
self.assertEqual(self.sim.getWordByName(self.dev, label), value)
class PyTestCase(TestCase):
def __init__(self, methodName, modulName):
TestCase.__init__(self, methodName)
self.modulName = modulName
def shortDescription(self):
md = TestCase.shortDescription(self)
if md is None: md = self._testMethodName
return self.modulName + "::" + md
# EOF
|
