1 files changed, 93 insertions, 0 deletions

diff --git a/scripts/pybootchartgui/pybootchartgui/samples.py b/scripts/pybootchartgui/pybootchartgui/samples.py
new file mode 100644
index 0000000000..c94b30d032
--- /dev/null
+++ b/scripts/pybootchartgui/pybootchartgui/samples.py
@@ -0,0 +1,93 @@
+class DiskStatSample:
+        def __init__(self, time):
+                self.time = time
+                self.diskdata = [0, 0, 0]
+        def add_diskdata(self, new_diskdata):
+                self.diskdata = [ a + b for a, b in zip(self.diskdata, new_diskdata) ]
+
+class CPUSample:
+        def __init__(self, time, user, sys, io):
+                self.time = time
+                self.user = user
+                self.sys = sys
+                self.io = io
+
+        def __str__(self):
+                return str(self.time) + "\t" + str(self.user) + "\t" + str(self.sys) + "\t" + str(self.io);
+                
+class ProcessSample:
+        def __init__(self, time, state, cpu_sample):
+                self.time = time
+                self.state = state
+                self.cpu_sample = cpu_sample
+                
+        def __str__(self):
+                return str(self.time) + "\t" + str(self.state) + "\t" + str(self.cpu_sample);
+
+class ProcessStats:
+    def __init__(self, process_list, sample_period, start_time, end_time):
+                self.process_list = process_list
+                self.sample_period = sample_period
+                self.start_time = start_time
+                self.end_time = end_time
+
+class Process:  
+        def __init__(self, pid, cmd, ppid, start_time):
+                self.pid = pid
+                self.cmd = cmd.strip('(').strip(')')
+                self.ppid = ppid
+                self.start_time = start_time
+                self.samples = []
+                self.parent = None
+                self.child_list = []
+                
+                self.duration = 0
+                self.active = None
+                
+                self.last_user_cpu_time = None
+                self.last_sys_cpu_time = None
+        
+        def __str__(self):
+                return " ".join([str(self.pid), self.cmd, str(self.ppid), '[ ' + str(len(self.samples)) + ' samples ]' ])
+        
+        def calc_stats(self, samplePeriod):
+                if self.samples:
+                        firstSample = self.samples[0]
+                        lastSample = self.samples[-1]
+                        self.start_time = min(firstSample.time, self.start_time)
+                        self.duration = lastSample.time - self.start_time + samplePeriod
+                        
+                activeCount = sum( [1 for sample in self.samples if sample.cpu_sample and sample.cpu_sample.sys + sample.cpu_sample.user + sample.cpu_sample.io > 0.0] )
+                activeCount = activeCount + sum( [1 for sample in self.samples if sample.state == 'D'] )
+                self.active = (activeCount>2)
+                
+        def calc_load(self, userCpu, sysCpu, interval):
+                
+                userCpuLoad = float(userCpu - self.last_user_cpu_time) / interval
+                sysCpuLoad = float(sysCpu - self.last_sys_cpu_time) / interval
+                cpuLoad = userCpuLoad + sysCpuLoad
+                # normalize
+                if cpuLoad > 1.0:
+                        userCpuLoad = userCpuLoad / cpuLoad;
+                        sysCpuLoad = sysCpuLoad / cpuLoad;
+                return (userCpuLoad, sysCpuLoad)
+        
+        def set_parent(self, processMap):
+                if self.ppid != None:
+                        self.parent = processMap.get(self.ppid)
+                        if self.parent == None and self.pid > 1:
+                                print "warning: no parent for pid '%i' with ppid '%i'" % (self.pid,self.ppid)
+        def get_end_time(self):
+                return self.start_time + self.duration
+
+class DiskSample:
+        def __init__(self, time, read, write, util):
+                self.time = time
+                self.read = read
+                self.write = write
+                self.util = util
+                self.tput = read + write
+
+        def __str__(self):
+                return "\t".join([str(self.time), str(self.read), str(self.write), str(self.util)])
+