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)])
+