Author: bugman
Date: Wed Jul 6 11:35:29 2011
New Revision: 13440
URL: http://svn.gna.org/viewcvs/relax?rev=13440&view=rev
Log:
Created all the select user function pages.
Added:
branches/gui_testing/gui/user_functions/select.py
Added: branches/gui_testing/gui/user_functions/select.py
URL:
http://svn.gna.org/viewcvs/relax/branches/gui_testing/gui/user_functions/select.py?rev=13440&view=auto
==============================================================================
--- branches/gui_testing/gui/user_functions/select.py (added)
+++ branches/gui_testing/gui/user_functions/select.py Wed Jul 6 11:35:29 2011
@@ -1,0 +1,248 @@
+###############################################################################
+#
#
+# Copyright (C) 2011 Edward d'Auvergne
#
+#
#
+# This file is part of the program relax.
#
+#
#
+# relax is free software; you can redistribute it and/or modify
#
+# it under the terms of the GNU General Public License as published by
#
+# the Free Software Foundation; either version 2 of the License, or
#
+# (at your option) any later version.
#
+#
#
+# relax is distributed in the hope that it will be useful,
#
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
#
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#
+# GNU General Public License for more details.
#
+#
#
+# You should have received a copy of the GNU General Public License
#
+# along with relax; if not, write to the Free Software
#
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
+#
#
+###############################################################################
+
+# Module docstring.
+"""The select user function GUI elements."""
+
+# relax module imports.
+from prompt.select import boolean_doc
+
+# GUI module imports.
+from base import UF_base, UF_page
+from gui.misc import gui_to_bool, gui_to_int, gui_to_str, str_to_gui
+from gui.wizard import Wiz_window
+
+
+# The container class.
+class Select(UF_base):
+ """The container class for holding all GUI elements."""
+
+ def all(self, event):
+ """The select.all user function.
+
+ @param event: The wx event.
+ @type event: wx event
+ """
+
+ # Execute the wizard.
+ wizard = Wiz_window(size_x=600, size_y=300, title='Select all spins')
+ page = All_page(wizard, self.gui, self.interpreter)
+ wizard.add_page(page, apply_button=False)
+ wizard.run()
+
+
+ def read(self, event):
+ """The select.read user function.
+
+ @param event: The wx event.
+ @type event: wx event
+ """
+
+ # Execute the wizard.
+ wizard = Wiz_window(size_x=900, size_y=800, title='Select spins from
file')
+ page = Read_page(wizard, self.gui, self.interpreter)
+ wizard.add_page(page)
+ wizard.run()
+
+
+ def reverse(self, event):
+ """The select.reverse user function.
+
+ @param event: The wx event.
+ @type event: wx event
+ """
+
+ # Execute the wizard.
+ wizard = Wiz_window(size_x=700, size_y=400, title='Reverse spin
selection')
+ page = Reverse_page(wizard, self.gui, self.interpreter)
+ wizard.add_page(page, apply_button=False)
+ wizard.run()
+
+
+ def spin(self, event):
+ """The select.spin user function.
+
+ @param event: The wx event.
+ @type event: wx event
+ """
+
+ # Execute the wizard.
+ wizard = Wiz_window(size_x=700, size_y=500, title='Select spins')
+ page = Spin_page(wizard, self.gui, self.interpreter)
+ wizard.add_page(page)
+ wizard.run()
+
+
+
+
+class All_page(UF_page):
+ """The select.all() user function page."""
+
+ # Some class variables.
+ uf_path = ['select', 'all']
+
+ def add_contents(self, sizer):
+ """Add the sequence specific GUI elements.
+
+ @param sizer: A sizer object.
+ @type sizer: wx.Sizer instance
+ """
+
+
+ def on_execute(self):
+ """Execute the user function."""
+
+ # Select all.
+ self.interpreter.select.all()
+
+
+
+class Read_page(UF_page):
+ """The select.read() user function page."""
+
+ # Some class variables.
+ uf_path = ['select', 'read']
+ desc_height = 200
+
+ def add_contents(self, sizer):
+ """Add the sequence specific GUI elements.
+
+ @param sizer: A sizer object.
+ @type sizer: wx.Sizer instance
+ """
+
+ # Add a file selection.
+ self.file = self.file_selection(sizer, "The selection file:",
title="Selection file selection")
+
+ # The parameter file settings.
+ self.free_file_format(sizer)
+
+ # The spin ID restriction.
+ self.spin_id = self.spin_id_element(sizer, desc="Restrict data
loading to certain spins:")
+
+ # The boolean operator specifying how spins should be selected.
+ self.boolean = self.combo_box(sizer, "Boolean operator:",
choices=['OR', 'NOR', 'AND', 'NAND', 'XOR', 'XNOR'], tooltip=boolean_doc[2])
+ self.boolean.SetValue('OR')
+
+ # The change_all flag.
+ self.change_all = self.boolean_selector(sizer, "Change all:",
tooltip="A flag specifying if all other spins should be changed.")
+
+
+ def on_execute(self):
+ """Execute the user function."""
+
+ # The file name.
+ file = gui_to_str(self.file.GetValue())
+
+ # No file.
+ if not file:
+ return
+
+ # Get the column numbers.
+ spin_id_col = gui_to_int(self.spin_id_col.GetValue())
+ mol_name_col = gui_to_int(self.mol_name_col.GetValue())
+ res_num_col = gui_to_int(self.res_num_col.GetValue())
+ res_name_col = gui_to_int(self.res_name_col.GetValue())
+ spin_num_col = gui_to_int(self.spin_num_col.GetValue())
+ spin_name_col = gui_to_int(self.spin_name_col.GetValue())
+
+ # The column separator.
+ sep = str(self.sep.GetValue())
+ if sep == 'white space':
+ sep = None
+
+ # The spin ID.
+ spin_id = gui_to_str(self.spin_id.GetValue())
+
+ # The boolean operator and change_all flag.
+ boolean = gui_to_str(self.boolean.GetValue())
+ change_all = gui_to_bool(self.change_all.GetValue())
+
+ # Selection.
+ self.interpreter.select.read(file, spin_id_col=spin_id_col,
mol_name_col=mol_name_col, res_num_col=res_num_col,
res_name_col=res_name_col, spin_num_col=spin_num_col,
spin_name_col=spin_name_col, sep=sep, spin_id=spin_id, boolean=boolean,
change_all=change_all)
+
+
+
+class Reverse_page(UF_page):
+ """The select.reverse() user function page."""
+
+ # Some class variables.
+ uf_path = ['select', 'reverse']
+
+ def add_contents(self, sizer):
+ """Add the sequence specific GUI elements.
+
+ @param sizer: A sizer object.
+ @type sizer: wx.Sizer instance
+ """
+
+ # The spin ID restriction.
+ self.spin_id = self.spin_id_element(sizer)
+
+
+ def on_execute(self):
+ """Execute the user function."""
+
+ # The spin ID.
+ spin_id = gui_to_str(self.spin_id.GetValue())
+
+ # Select all.
+ self.interpreter.select.reverse(spin_id=spin_id)
+
+
+
+class Spin_page(UF_page):
+ """The select.spin() user function page."""
+
+ # Some class variables.
+ uf_path = ['select', 'spin']
+
+ def add_contents(self, sizer):
+ """Add the sequence specific GUI elements.
+
+ @param sizer: A sizer object.
+ @type sizer: wx.Sizer instance
+ """
+
+ # The spin ID restriction.
+ self.spin_id = self.spin_id_element(sizer)
+
+ # The boolean operator specifying how spins should be selected.
+ self.boolean = self.combo_box(sizer, "Boolean operator:",
choices=['OR', 'NOR', 'AND', 'NAND', 'XOR', 'XNOR'], tooltip=boolean_doc[2])
+ self.boolean.SetValue('OR')
+
+ # The change_all flag.
+ self.change_all = self.boolean_selector(sizer, "Change all:",
tooltip="A flag specifying if all other spins should be changed.")
+
+
+ def on_execute(self):
+ """Execute the user function."""
+
+ # The spin ID.
+ spin_id = gui_to_str(self.spin_id.GetValue())
+
+ # The boolean operator and change_all flag.
+ boolean = gui_to_str(self.boolean.GetValue())
+ change_all = gui_to_bool(self.change_all.GetValue())
+
+ # Select all.
+ self.interpreter.select.spin(spin_id=spin_id, boolean=boolean,
change_all=change_all)
r13440 - /branches/gui_testing/gui/user_functions/select.py