tuna_gui: Add isolate/include CPU socket menu entry

Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

1 files changed, 20 insertions, 0 deletions

diff --git a/tuna/tuna_gui.py b/tuna/tuna_gui.py
index a1dddd6..dbc8b3f 100644
--- a/tuna/tuna_gui.py
+++ b/tuna/tuna_gui.py
@@ -214,6 +214,18 @@ class cpu_socket_frame(gtk.Frame):
 
 		self.creator.restore_cpu()
 
+	def isolate_cpu_socket(self, a):
+
+		# Isolate all CPUs in this socket
+		cpus = [ int(cpu.name[3:]) for cpu in self.cpus ]
+		self.creator.isolate_cpus(cpus)
+
+	def include_cpu_socket(self, a):
+
+		# Include all CPUs in this socket
+		cpus = [ int(cpu.name[3:]) for cpu in self.cpus ]
+		self.creator.include_cpus(cpus)
+
 	def on_cpu_socket_frame_button_press_event(self, treeview, event):
 		if event.type != gtk.gdk.BUTTON_PRESS or event.button != 3:
 			return
@@ -224,22 +236,30 @@ class cpu_socket_frame(gtk.Frame):
 		menu = gtk.Menu()
 
 		include = gtk.MenuItem("I_nclude CPU")
+		include_socket = gtk.MenuItem("I_nclude CPU Socket")
 		isolate = gtk.MenuItem("_Isolate CPU")
+		isolate_socket = gtk.MenuItem("_Isolate CPU Socket")
 		restore = gtk.MenuItem("_Restore CPU")
 
 		menu.add(include)
+		menu.add(include_socket)
 		menu.add(isolate)
+		menu.add(isolate_socket)
 		menu.add(restore)
 
 		include.connect_object('activate', self.include_cpu, event)
+		include_socket.connect_object('activate', self.include_cpu_socket, event)
 		isolate.connect_object('activate', self.isolate_cpu, event)
+		isolate_socket.connect_object('activate', self.isolate_cpu_socket, event)
 		if not (self.creator.previous_pid_affinities or \
 			self.creator.previous_irq_affinities):
 			restore.set_sensitive(False)
 		restore.connect_object('activate', self.restore_cpu, event)
 
 		include.show()
+		include_socket.show()
 		isolate.show()
+		isolate_socket.show()
 		restore.show()
 
 		menu.popup(None, None, None, event.button, event.time)