Initial commit - messy state, probably working.

1 files changed, 202 insertions, 0 deletions

diff --git a/sta.py b/sta.py
new file mode 100755
index 0000000..d8c883a
--- /dev/null
+++ b/sta.py
@@ -0,0 +1,202 @@
+#Note to self: I stopped while working on implementing an iterator so I could
+# do a filter() operation to get only events of a certain job so I could compare
+# to Bjoern's pdf
+# Note: Just got StDump iterator working. Remember list() in iterator.
+# Note: Just got a generator working :) Awesome, need to pretty up.
+# Note: Likely next steps:
+    # - Define convenience comparisons for ordering by earliest time for sort,
+    # - and lambdas for filtering by job or PID or CPU or whatever
+    # - and change read_trace to possibly take multiple traces.
+    # - and /maybe/ allow for saving /loading previously saved traces.
+
+
+#!/usr/bin/python3
+
+# Imports.
+import struct
+import sys
+
+
+####################################
+# Types for binary data conversion #
+####################################
+
+st_record_size = 192
+
+
+class StHeader:
+    format =  '<bbhi'
+    formatStr = struct.Struct(format)
+    keys = ['type','cpu','pid','job']
+    message = 'The header.'
+
+
+class StNameData:
+    format =  '16s'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['name']
+    message = 'The name of the executable of this process.'
+
+# Untested because this never appears in my sample files
+# ACTUALLY: None in 0, none in 1, a bunch in 2, and none in 3
+class StParamData:  
+    format =  'III8?'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['wcet','period','phase','partition']
+    message = 'Regular parameters.'
+
+
+class StReleaseData:
+    format =  'LL'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['release_time','deadline']
+    message = 'A job was/is going to be released.'
+
+ 
+class AssignedData:
+    format =  'Lh'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when','target']
+    message = 'A job was assigned to a CPU.'
+
+class StSwitchToData:
+    format =  'LI'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when','exec_time']
+    message = 'A process was switched to on a given CPU.'
+
+
+class StSwitchAwayData:
+    format =  'LI'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when','exec_time']
+
+class StCompletionData:
+    format =  'L7x?c'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when','forced?','flags']
+    message = 'A job completed.'
+
+
+class StBlockData:
+    format =  'L'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when']
+    message = 'A task blocks.'
+
+
+class StResumeData:
+    format =  'L'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when']
+    message = 'A task resumes.'
+
+
+class StSysReleaseData:
+    format =  'LL'
+    formatStr = struct.Struct(StHeader.format + format)
+    keys = StHeader.keys + ['when','release']
+    message = 'StSysReleaseData (no details).'
+
+
+
+##################
+# Business logic #
+##################
+
+
+def get_type(type_num):
+    """Return the class needed to read a record, given its numerical type"""
+    type = None
+    if type_num == 1:
+        return StNameData
+    elif type_num == 2:
+        return StParamData
+    elif type_num == 3:
+        return StReleaseData
+    elif type_num == 4:
+        return StAssignedData
+    elif type_num == 5:
+        return StSwitchToData
+    elif type_num == 6:
+        return StSwitchAwayData
+    elif type_num == 7:
+        return StCompletionData
+    elif type_num == 8:
+        return StBlockData
+    elif type_num == 9:
+        return StResumeData
+    elif type_num == 10:
+        return StSysReleaseData
+    return type
+
+
+def print_record(record):
+    """Prints a record, given the record as a dict."""
+    print(record)
+
+
+def print_record_verbose(record):
+    """Prints a record verbosely, given the record as a dict."""
+    print(50*'=')
+    print(get_type(record['type']).message)
+    for k,v in record.items():
+        print(k +":", v)
+
+
+def trace_iter(*files):
+    """Returns an iterator for iterating over trace(s) and producing records."""
+    for file in files:
+        f = open(file,'rb')
+        while True:
+            try:
+                data = f.read(st_record_size)
+                typenum = struct.unpack_from('<b',data)[0]
+                type = get_type(typenum)
+                values = struct.unpack_from(StHeader.format + type.format,data)
+                record = dict(zip(type.keys,values))
+                yield record
+            except struct.error:
+                f.close()
+                break
+
+
+def trace(*files):
+    obj = "hi"
+    iter = trace_iter(*files)
+    def filter():
+        print("Hi!")
+    obj.filter = filter
+    obj.iter = iter
+    return obj
+
+
+########
+# Main #
+########
+
+
+def main(args):
+    """Main function, called if the program is executed."""
+    if args[1] == '-v':
+        printfunc = print_record_verbose 
+        files = sys.argv[2:]
+    else:
+        printfunc = print_record
+        files = args[1:]
+    for file in files: 
+        data = open(file,'rb').read()
+        offset = 0
+        try:
+            while True:
+                record = get_record(data,offset)
+                printfunc(record)
+                offset += st_record_size
+                #input("Press Enter")
+        except struct.error:  # We've reached the EOF.
+            continue
+
+
+# Call main() if the program is executed directly.
+if __name__ == "__main__":
+    main(sys.argv)