def add_middle_frame_widgets(self): # Entryfield for selecting the residues range. self.residue_range_enf = PyMod_entryfield_qt( label_text="Residue(s)", value=str(self.selected_residue.db_index)) self.middle_formlayout.add_widget_to_align(self.residue_range_enf) # Entryfield for the feature name. self.feature_name_enf = PyMod_entryfield_qt(label_text="Feature Name", value="new feature") self.middle_formlayout.add_widget_to_align(self.feature_name_enf) # Widgets for choosing a color for the feature. self.selected_rgb = (1.0, 0.0, 0.0) self.selected_hex = '#ff0000' self.feature_color_enf = PyMod_entryfield_button_qt( label_text="Feature Color", readonly=True, button_text="Pick", button_command=self.pick_color_dialog) self.middle_formlayout.add_widget_to_align(self.feature_color_enf) self.feature_color_enf.entry.setStyleSheet("background-color: %s" % self.selected_hex) # Select in PyMOL. if self.pymod_element.has_structure(): self.select_in_pymol_rds = PyMod_radioselect_qt( label_text="Select in PyMOL", buttons=("Yes", "No")) self.select_in_pymol_rds.setvalue("No") self.middle_formlayout.add_widget_to_align( self.select_in_pymol_rds) self.middle_formlayout.set_input_widgets_width(150)
def build_algorithm_standard_options_widgets(self): db_list = [text for (text, val) in self.protocol.ncbi_databases] self.ncbiblast_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=db_list) self.ncbiblast_database_rds.setvalue('PDB') self.middle_formlayout.add_widget_to_align(self.ncbiblast_database_rds)
class Clustalo_base_window_qt: """ Base class for ClustalOmega protocols. """ def build_algorithm_options_widgets(self): # Use full distance matrix. self.use_full_dm_rds = PyMod_radioselect_qt(label_text="Use Full Distance Matrix", buttons=('Yes', 'No')) self.use_full_dm_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.use_full_dm_rds) # Number of (combined guide-tree/HMM) iterations. self.clustalo_iterations_enf = PyMod_entryfield_qt(label_text="Combined Iterations", value='0', validate={'validator': 'integer', 'min': 0, 'max': 5}) self.middle_formlayout.add_widget_to_align(self.clustalo_iterations_enf) self.middle_formlayout.set_input_widgets_width("auto") def get_iterations_value(self): return self.clustalo_iterations_enf.getvalue(validate=True) def get_use_full_dm_value(self): return self.use_full_dm_rds.getvalue() == "Yes"
def add_middle_frame_widgets(self): # Scoring matrix combobox. self.matrix_cbx = PyMod_combobox_qt(label_text="Scoring Matrix Selection", items=self.protocol.campo_matrices) self.matrix_cbx.combobox.setCurrentIndex(2) self.middle_formlayout.add_widget_to_align(self.matrix_cbx) # Gap open entryfield. self.campo_gap_penalty_enf = PyMod_entryfield_qt(label_text="Gap Score", value="-1", validate={'validator': 'integer', 'min': -1000, 'max': 0}) self.middle_formlayout.add_widget_to_align(self.campo_gap_penalty_enf) # Gap extension entryfield. self.campo_gap_to_gap_score_enf = PyMod_entryfield_qt(label_text="Gap to Gap Score", value="0", validate={'validator': 'integer', 'min': -1000, 'max': 0}) self.middle_formlayout.add_widget_to_align(self.campo_gap_to_gap_score_enf) # Toss gaps. self.campo_exclude_gaps_rds = PyMod_radioselect_qt(label_text="Toss gaps", buttons=('Yes', 'No')) self.campo_exclude_gaps_rds.setvalue("Yes") self.middle_formlayout.add_widget_to_align(self.campo_exclude_gaps_rds) self.middle_formlayout.set_input_widgets_width("auto", padding=10)
class Hmmscan_options_window_qt(PyMod_protocol_window_qt): def build_protocol_middle_frame(self): # Add the buttons to choose the database in which to search for domain profiles. if self.protocol.father_protocol.domain_search_mode == 'remote': self.hmmer_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=("PFAM", "Gene3D")) for button in self.hmmer_database_rds.get_buttons(): button.clicked.connect(self.database_opt_cmd) elif self.protocol.father_protocol.domain_search_mode == 'local': # Build the list of database names. self.protocol.hmmscan_db_dict = { } # This dictionary associates a database code (displayed in the GUI) to its filename. db_list = [] for db_filename in self.protocol.hmmscan_db_list: db_name = "".join(db_filename.split(".")[0:-2]) db_list.append(db_name) self.protocol.hmmscan_db_dict[db_name] = db_filename # Packs the PHMMER database selection widget. self.hmmer_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=db_list) self.middle_formlayout.add_widget_to_align(self.hmmer_database_rds) # E-value selection. self.e_value_threshold_enf = PyMod_entryfield_qt( label_text="E-value Threshold", value="1.0", validate={ 'validator': 'real', 'min': 0.0, 'max': 1000.0 }) self.middle_formlayout.add_widget_to_align(self.e_value_threshold_enf) # Note about the Gene3D and Evalues. if self.protocol.father_protocol.domain_search_mode == 'remote': info_note = ('Note: The Gene3D online database will\n' 'ignore custom cut-off parameters since\n' 'they use a post processing step that\n' 'involves preset thresholds.') self.notelabel = QtWidgets.QLabel(info_note) self.middle_formlayout.addRow(self.notelabel) self.middle_formlayout.set_input_widgets_width(140) def database_opt_cmd(self): if self.hmmer_database_rds.getvalue() == 'Gene3D': self.e_value_threshold_enf.entry.setStyleSheet( inactive_entry_style) self.e_value_threshold_enf.entry.setEnabled(False) else: self.e_value_threshold_enf.entry.setStyleSheet(active_entry_style) self.e_value_threshold_enf.entry.setEnabled(True)
def build_algorithm_options_widgets(self): if self.protocol.structures_are_selected: # Use structure information to guide sequence alignment. self.salign_seq_struct_rds = PyMod_radioselect_qt( label_text="Use structural information", buttons=('Yes', 'No')) self.salign_seq_struct_rds.setvalue("No") self.middle_formlayout.add_widget_to_align( self.salign_seq_struct_rds) self.middle_formlayout.set_input_widgets_width("auto")
class Structural_alignment_base_window_qt: def build_rmsd_option(self): # Decide whether to compute the RMSD matrix if the structural alignment. self.compute_rmsd_rds = PyMod_radioselect_qt(label_text="Compute RMSD Matrix", buttons=('Yes', 'No')) self.compute_rmsd_rds.setvalue("Yes") self.middle_formlayout.add_widget_to_align(self.compute_rmsd_rds) def get_compute_rmsd_option_value(self): return pmdt.yesno_dict[self.compute_rmsd_rds.getvalue()]
class Clustalw_base_window_qt: """ Base class for ClustalW protocols. """ def build_algorithm_options_widgets(self): # Scoring matrix radioselect. self.clustal_matrices = ["Blosum", "Pam", "Gonnet", "Id"] self.clustal_matrices_dict = { "Blosum": "blosum", "Pam": "pam", "Gonnet": "gonnet", "Id": "id" } self.matrix_rds = PyMod_radioselect_qt( label_text="Scoring Matrix Selection", buttons=self.clustal_matrices) self.matrix_rds.setvalue("Blosum") self.middle_formlayout.add_widget_to_align(self.matrix_rds) # Gap open entryfield. self.gapopen_enf = PyMod_entryfield_qt( label_text="Gap Opening Penalty", value="10.0", validate={ 'validator': 'real', 'min': 0, 'max': 1000 }) self.middle_formlayout.add_widget_to_align(self.gapopen_enf) # Gap extension entryfield. self.gapextension_enf = PyMod_entryfield_qt( label_text="Gap Extension Penalty", value="0.2", validate={ 'validator': 'real', 'min': 0, 'max': 1000 }) self.middle_formlayout.add_widget_to_align(self.gapextension_enf) self.middle_formlayout.set_input_widgets_width("auto") def get_matrix_value(self): return self.clustal_matrices_dict[self.matrix_rds.getvalue()] def get_gapopen_value(self): return self.gapopen_enf.getvalue(validate=True) def get_gapextension_value(self): return self.gapextension_enf.getvalue(validate=True)
class SCR_FIND_window_qt(PyMod_protocol_window_qt): def add_middle_frame_widgets(self): # SC-score scalebar. self.sc_scale = PyMod_scalebar_qt( label_text="SC-score Limit", slider_value=3.0, slider_from=0.5, slider_to=5.0, slider_resoution=0.25, slider_tickinterval=1.0, slider_use_float=True, slider_binding=self.protocol.scr_window_submit, slider_width=375) self.middle_formlayout.add_widget_to_align(self.sc_scale) # Sliding Window size scalebar. self.sw_scale = PyMod_scalebar_qt( label_text="Sliding Window Min. Size", slider_value=3, slider_from=2, slider_to=50, slider_resoution=1, slider_tickinterval=5, slider_binding=self.protocol.scr_window_submit, slider_width=375) self.middle_formlayout.add_widget_to_align(self.sw_scale) # Gap penalty entry field. self.scr_find_gap_penalty_enf = PyMod_entryfield_qt( label_text="Gap Penalty", value='100', validate={ 'validator': 'integer', 'min': 0, 'max': 1000 }, enter_command=self.protocol.scr_window_submit) self.middle_formlayout.add_widget_to_align( self.scr_find_gap_penalty_enf) # Hide non-SCR residues or show them white. self.scr_find_hide_non_scrs = PyMod_radioselect_qt( label_text="Hide non SCRs", buttons=('Yes', 'No')) self.scr_find_hide_non_scrs.setvalue("No") self.scr_find_hide_non_scrs.buttons_dict["No"].clicked.connect( self.protocol.scr_window_submit) self.scr_find_hide_non_scrs.buttons_dict["Yes"].clicked.connect( self.protocol.scr_window_submit) self.middle_formlayout.add_widget_to_align(self.scr_find_hide_non_scrs) self.middle_formlayout.set_input_widgets_width(110)
def build_protocol_middle_frame(self): """ Allow to choose between plotting all amino acids types or only a subset of them. """ # Radioselect. aa_select_choices = ("Use all amino acids", "Select amino acids types") aa_select_choices_values = ["all", "single"] self.aa_select_choices_dict = dict([ (k, v) for (k, v) in zip(aa_select_choices, aa_select_choices_values) ]) self.aa_select_rds = PyMod_radioselect_qt( label_text="Select Amino Acids", buttons=aa_select_choices) self.aa_select_rds.setvalue(aa_select_choices[0]) self.aa_select_rds.buttons_dict[aa_select_choices[0]].clicked.connect( self.hide_select_single_aa_frame) self.aa_select_rds.buttons_dict[aa_select_choices[1]].clicked.connect( self.show_select_single_aa_frame) self.middle_formlayout.add_widget_to_align(self.aa_select_rds) # Checkboxes for selecting single amino acids. self.aa_select_grid_layout = QtWidgets.QGridLayout() self.aa_select_rds.input.addLayout(self.aa_select_grid_layout) self.aa_checkbutton = {} self.aa_freq_dict = {} self.aa_checkbutton_list = [] for i, aa in enumerate(prot_standard_one_letter): # Get the number of aa in the sequence. aa_freq = str(self.protocol.target_sequence.my_sequence).count(aa) self.aa_freq_dict[aa] = aa_freq # Build a checkbox for the aa. checkbox = QtWidgets.QCheckBox( pmsm.one2three(aa) + " (" + str(aa_freq) + ")") checkbox.setEnabled(False) self.aa_select_grid_layout.addWidget(checkbox, int(i % 10), int(i / 10)) self.aa_checkbutton[aa] = checkbox self.aa_checkbutton_list.append(checkbox) self.aa_select_grid_layout.setAlignment(QtCore.Qt.AlignLeft) self.middle_formlayout.set_input_widgets_width("auto", padding=40)
class CAMPO_options_window_qt(PyMod_protocol_window_qt): """ Window for CAMPO options. """ def add_middle_frame_widgets(self): # Scoring matrix combobox. self.matrix_cbx = PyMod_combobox_qt(label_text="Scoring Matrix Selection", items=self.protocol.campo_matrices) self.matrix_cbx.combobox.setCurrentIndex(2) self.middle_formlayout.add_widget_to_align(self.matrix_cbx) # Gap open entryfield. self.campo_gap_penalty_enf = PyMod_entryfield_qt(label_text="Gap Score", value="-1", validate={'validator': 'integer', 'min': -1000, 'max': 0}) self.middle_formlayout.add_widget_to_align(self.campo_gap_penalty_enf) # Gap extension entryfield. self.campo_gap_to_gap_score_enf = PyMod_entryfield_qt(label_text="Gap to Gap Score", value="0", validate={'validator': 'integer', 'min': -1000, 'max': 0}) self.middle_formlayout.add_widget_to_align(self.campo_gap_to_gap_score_enf) # Toss gaps. self.campo_exclude_gaps_rds = PyMod_radioselect_qt(label_text="Toss gaps", buttons=('Yes', 'No')) self.campo_exclude_gaps_rds.setvalue("Yes") self.middle_formlayout.add_widget_to_align(self.campo_exclude_gaps_rds) self.middle_formlayout.set_input_widgets_width("auto", padding=10) def validate_input(self): params_from_gui = {} try: params_from_gui["mutational_matrix"] = self.protocol.campo_matrices_dict[self.matrix_cbx.get()] params_from_gui["gap_score"] = self.campo_gap_penalty_enf.getvalue(validate=True) params_from_gui["gap_gap_score"] = self.campo_gap_to_gap_score_enf.getvalue(validate=True) params_from_gui["toss_gaps"] = pymod_vars.yesno_dict[self.campo_exclude_gaps_rds.getvalue()] except (ValueError, KeyError) as e: return None, str(e) return params_from_gui, None
def build_algorithm_options_widgets(self): # Use full distance matrix. self.use_full_dm_rds = PyMod_radioselect_qt(label_text="Use Full Distance Matrix", buttons=('Yes', 'No')) self.use_full_dm_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.use_full_dm_rds) # Number of (combined guide-tree/HMM) iterations. self.clustalo_iterations_enf = PyMod_entryfield_qt(label_text="Combined Iterations", value='0', validate={'validator': 'integer', 'min': 0, 'max': 5}) self.middle_formlayout.add_widget_to_align(self.clustalo_iterations_enf) self.middle_formlayout.set_input_widgets_width("auto")
class SALIGN_seq_base_window_qt: def build_algorithm_options_widgets(self): if self.protocol.structures_are_selected: # Use structure information to guide sequence alignment. self.salign_seq_struct_rds = PyMod_radioselect_qt( label_text="Use structural information", buttons=('Yes', 'No')) self.salign_seq_struct_rds.setvalue("No") self.middle_formlayout.add_widget_to_align( self.salign_seq_struct_rds) self.middle_formlayout.set_input_widgets_width("auto") def get_use_str_information_var(self): if self.protocol.structures_are_selected: return pmdt.yesno_dict[self.salign_seq_struct_rds.getvalue()] else: return False
def build_algorithm_options_widgets(self): # MUSCLE mode radioselect (for more information see: https://www.drive5.com/muscle/manual/). self.muscle_modes = ["Highest Accuracy", "Large Datasets", "Fastest"] self.muscle_modes_short = [ "highest_accuracy", "large_datasets", "fastest" ] self.muscle_modes_dict = dict( (k, v) for (k, v) in zip(self.muscle_modes, self.muscle_modes_short)) self.muscle_mode_rds = PyMod_radioselect_qt(label_text="MUSCLE Mode", buttons=self.muscle_modes) self.muscle_mode_rds.setvalue(self.muscle_modes[0]) self.middle_formlayout.add_widget_to_align(self.muscle_mode_rds) self.middle_formlayout.set_input_widgets_width("auto")
class Phmmer_options_window_qt(BLAST_base_options_window_qt): """ Window for PHMMER searches. """ def build_algorithm_standard_options_widgets(self): self.hmmer_database_labels_dict = {} # A list containing information about the databases present in PyMod hmmer database folder. db_list = [] for db in self.protocol.databases_directories_list: db_label = "".join(db.split(".")[0:-1]) db_list.append(db_label) self.hmmer_database_labels_dict[db_label] = db # Makes the user choose the folder where the hmmer database files are stored locally. self.hmmer_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=db_list) self.middle_formlayout.add_widget_to_align(self.hmmer_database_rds) self.build_additional_hmmer_widgets() def build_additional_hmmer_widgets(self): pass def build_algorithm_advanced_options_widgets(self): pass def get_phmmer_database(self): return self.hmmer_database_labels_dict.get( self.hmmer_database_rds.getvalue(), None) def get_additional_hmmer_parameters(self): return {}
def build_algorithm_standard_options_widgets(self): self.hmmer_database_labels_dict = {} # A list containing information about the databases present in PyMod hmmer database folder. db_list = [] for db in self.protocol.databases_directories_list: db_label = "".join(db.split(".")[0:-1]) db_list.append(db_label) self.hmmer_database_labels_dict[db_label] = db # Makes the user choose the folder where the hmmer database files are stored locally. self.hmmer_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=db_list) self.middle_formlayout.add_widget_to_align(self.hmmer_database_rds) self.build_additional_hmmer_widgets()
def add_middle_frame_widgets(self): # Type of entropy measure. self.entropy_scorer_function_rds = PyMod_radioselect_qt( label_text="Entropy Measure", buttons=self.protocol.entropy_func_labels) self.entropy_scorer_function_rds.setvalue( self.protocol.entropy_func_labels[0]) self.middle_formlayout.add_widget_to_align( self.entropy_scorer_function_rds) # Toss gaps. self.entropy_scorer_exclude_gaps_rds = PyMod_radioselect_qt( label_text="Toss gaps", buttons=('Yes', 'No')) self.entropy_scorer_exclude_gaps_rds.setvalue("Yes") self.middle_formlayout.add_widget_to_align( self.entropy_scorer_exclude_gaps_rds) self.middle_formlayout.set_input_widgets_width("auto", padding=10)
class BLAST_options_window_qt(BLAST_base_options_window_qt): """ Window for BLAST searches. """ # input_widget_width = 16 # geometry_string = "450x650" def build_algorithm_standard_options_widgets(self): db_list = [text for (text, val) in self.protocol.ncbi_databases] self.ncbiblast_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=db_list) self.ncbiblast_database_rds.setvalue('PDB') self.middle_formlayout.add_widget_to_align(self.ncbiblast_database_rds) def build_algorithm_advanced_options_widgets(self): pass
class Tree_building_window_qt(PyMod_protocol_window_qt): def build_protocol_middle_frame(self): # Add some options. self.algorithm_rds = PyMod_radioselect_qt( label_text="Clustering Algorithm", buttons=list(sorted(tree_building_alg_dict.keys()))) self.algorithm_rds.setvalue("Neighbor Joining") self.middle_formlayout.add_widget_to_align(self.algorithm_rds) if self.protocol.tree_building_software == "clustalw": # Kimura distance correction. self.distance_correction_rds = PyMod_radioselect_qt( label_text="Use Distance Correction", buttons=('Yes', 'No')) self.distance_correction_rds.setvalue("No") self.middle_formlayout.add_widget_to_align( self.distance_correction_rds) # Toss gaps. self.exclude_gaps_rds = PyMod_radioselect_qt( label_text="Exclude Gaps", buttons=('Yes', 'No')) self.exclude_gaps_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.exclude_gaps_rds) self.middle_formlayout.set_input_widgets_width("auto", padding=10)
def add_middle_frame_widgets(self): #------------------------------------- # Form layout for input and results. - #------------------------------------- entries_width = 340 # Entryfield for inserting a subsequence. self.search_string_enf = PyMod_entryfield_qt( label_text="Search For", value="", enter_command=self.submit_command) self.search_string_enf.set_input_widget_width(entries_width) self.middle_formlayout.add_widget_to_align(self.search_string_enf, align=False) # Entryfield for showing the results. self.results_enf = PyMod_entryfield_qt(label_text="", value=self.default_message, readonly=True) self.set_results_entry_bg(self.inactive_results_color) self.results_enf.set_input_widget_width(entries_width) self.middle_formlayout.add_widget_to_align(self.results_enf, align=False) #--------------------------- # Form layout for options. - #--------------------------- # Use regular expressions. self.use_regex_rds = PyMod_radioselect_qt(label_text="Use Regex", buttons=("Yes", "No")) self.use_regex_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.use_regex_rds) # Highlight color selection. color_buttons = ("yellow", "red", "green", "cyan") # "violet" self.highlight_color_rds = PyMod_radioselect_qt( label_text='Highlight Color', buttons=color_buttons) self.highlight_color_rds.setvalue('yellow') self.middle_formlayout.add_widget_to_align(self.highlight_color_rds) if self.pymod_element.has_structure(): self.select_in_pymol_rds = PyMod_radioselect_qt( label_text='Select in PyMOL', buttons=("Yes", "No")) self.select_in_pymol_rds.setvalue('No') self.middle_formlayout.add_widget_to_align( self.select_in_pymol_rds) self.middle_formlayout.set_input_widgets_width(width="auto")
def build_algorithm_standard_options_widgets(self): # Makes the user chose the folder where the BLAST database files are stored locally. # A list containing information about the databases present in PyMod BLAST database # folder. db_list = [ k["prefix"] for k in self.protocol.databases_directories_list ] self.psiblast_database_rds = PyMod_radioselect_qt( label_text="Database Selection", buttons=db_list) self.middle_formlayout.add_widget_to_align(self.psiblast_database_rds) # Number of PSI-BLAST iterations. if self.protocol.blast_version == "psi-blast": self.psiblast_iterations_enf = PyMod_entryfield_qt( label_text="PSI-BLAST Iterations", value='3', validate={ 'validator': 'integer', 'min': 1, 'max': 10 }) self.middle_formlayout.add_widget_to_align( self.psiblast_iterations_enf, validate=True)
class Add_feature_window_qt(PyMod_tool_window_qt): def __init__(self, parent, pymod_element, selected_residue, *args, **configs): self.pymod_element = pymod_element self.selected_residue = selected_residue PyMod_tool_window_qt.__init__(self, parent, *args, **configs) def add_middle_frame_widgets(self): # Entryfield for selecting the residues range. self.residue_range_enf = PyMod_entryfield_qt( label_text="Residue(s)", value=str(self.selected_residue.db_index)) self.middle_formlayout.add_widget_to_align(self.residue_range_enf) # Entryfield for the feature name. self.feature_name_enf = PyMod_entryfield_qt(label_text="Feature Name", value="new feature") self.middle_formlayout.add_widget_to_align(self.feature_name_enf) # Widgets for choosing a color for the feature. self.selected_rgb = (1.0, 0.0, 0.0) self.selected_hex = '#ff0000' self.feature_color_enf = PyMod_entryfield_button_qt( label_text="Feature Color", readonly=True, button_text="Pick", button_command=self.pick_color_dialog) self.middle_formlayout.add_widget_to_align(self.feature_color_enf) self.feature_color_enf.entry.setStyleSheet("background-color: %s" % self.selected_hex) # Select in PyMOL. if self.pymod_element.has_structure(): self.select_in_pymol_rds = PyMod_radioselect_qt( label_text="Select in PyMOL", buttons=("Yes", "No")) self.select_in_pymol_rds.setvalue("No") self.middle_formlayout.add_widget_to_align( self.select_in_pymol_rds) self.middle_formlayout.set_input_widgets_width(150) def get_residue_range(self): return self.residue_range_enf.getvalue() def get_feature_name(self): return self.feature_name_enf.getvalue() def pick_color_dialog(self): selected_color = open_color_dialog(color_format="all") if selected_color is not None: self.selected_rgb, self.selected_hex = selected_color # self.feature_color_enf.setvalue(self.selected_hex) self.feature_color_enf.entry.setStyleSheet("background-color: %s" % self.selected_hex) def get_selected_colors(self): return self.selected_rgb, self.selected_hex def get_select_in_pymol(self): return _get_select_in_pymol(self)
class Search_string_window_qt(PyMod_tool_window_qt): inactive_results_color = inactive_bg_color found_results_color = success_bg_color not_found_results_color = failure_bg_color default_message = "Type a pattern and press Enter..." def __init__(self, parent, pymod_elements, *args, **configs): self.pymod_element = pymod_elements[0] PyMod_tool_window_qt.__init__(self, parent, *args, **configs) def add_middle_frame_widgets(self): #------------------------------------- # Form layout for input and results. - #------------------------------------- entries_width = 340 # Entryfield for inserting a subsequence. self.search_string_enf = PyMod_entryfield_qt( label_text="Search For", value="", enter_command=self.submit_command) self.search_string_enf.set_input_widget_width(entries_width) self.middle_formlayout.add_widget_to_align(self.search_string_enf, align=False) # Entryfield for showing the results. self.results_enf = PyMod_entryfield_qt(label_text="", value=self.default_message, readonly=True) self.set_results_entry_bg(self.inactive_results_color) self.results_enf.set_input_widget_width(entries_width) self.middle_formlayout.add_widget_to_align(self.results_enf, align=False) #--------------------------- # Form layout for options. - #--------------------------- # Use regular expressions. self.use_regex_rds = PyMod_radioselect_qt(label_text="Use Regex", buttons=("Yes", "No")) self.use_regex_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.use_regex_rds) # Highlight color selection. color_buttons = ("yellow", "red", "green", "cyan") # "violet" self.highlight_color_rds = PyMod_radioselect_qt( label_text='Highlight Color', buttons=color_buttons) self.highlight_color_rds.setvalue('yellow') self.middle_formlayout.add_widget_to_align(self.highlight_color_rds) if self.pymod_element.has_structure(): self.select_in_pymol_rds = PyMod_radioselect_qt( label_text='Select in PyMOL', buttons=("Yes", "No")) self.select_in_pymol_rds.setvalue('No') self.middle_formlayout.add_widget_to_align( self.select_in_pymol_rds) self.middle_formlayout.set_input_widgets_width(width="auto") def get_search_string(self): return self.search_string_enf.getvalue() def get_regex_use(self): use_regex_val = self.use_regex_rds.getvalue() if use_regex_val == "Yes": return True elif use_regex_val == "No": return False else: raise KeyError(use_regex_val) def get_highlight_color(self): return self.highlight_color_rds.getvalue() def show_results(self, message, state): self.results_enf.setvalue(message) if state == "found": self.set_results_entry_bg(self.found_results_color) elif state == "not_found": self.set_results_entry_bg(self.not_found_results_color) elif state == "empty": self.set_results_entry_bg(self.inactive_results_color) else: raise KeyError(state) def set_results_entry_bg(self, color): self.results_enf.input.setStyleSheet( "background-color: %s; color: black" % color) def get_select_in_pymol(self): return _get_select_in_pymol(self)
def build_protocol_middle_frame(self): # Feature type selection. if len(self.protocol.target_sequences) == 1: # Contact and distance. sel_option_idx = 0 all_options_idx = (0, 1) elif len(self.protocol.target_sequences ) == 2: # Distance difference, mean and std. sel_option_idx = 2 all_options_idx = (2, 3, 4) elif len(self.protocol.target_sequences) > 2: # Distance mean and std. sel_option_idx = 3 all_options_idx = (3, 4) features_buttons = [ self.feature_types_list[i] for i in all_options_idx ] self.feature_type_rds = PyMod_radioselect_qt(label_text="Feature Type", buttons=features_buttons) for button, option_idx in zip(self.feature_type_rds.get_buttons(), all_options_idx): button.clicked.connect( lambda e=None, o=option_idx: self.feature_type_state(o)) self.feature_type_rds.setvalue(self.feature_types_list[sel_option_idx]) self.middle_formlayout.add_widget_to_align(self.feature_type_rds) # Distance threshold for contacts. if len(self.protocol.target_sequences) == 1: threshold = self.default_contact_threshold elif len(self.protocol.target_sequences) == 2: threshold = self.default_distance_diff_threshold else: threshold = self.default_distance_threshold self.dist_threshold_enf = PyMod_entryfield_qt( label_text="Contact Threshold (%s)" % ("\u212B"), value=str(threshold), validate={ 'validator': 'real', 'min': 1.0, 'max': 100.0 }) self.middle_formlayout.add_widget_to_align(self.dist_threshold_enf) # Interaction center selection. self.interaction_center_rds = PyMod_radioselect_qt( label_text="Interaction Center", buttons=self.interaction_centers_list) self.interaction_center_rds.setvalue(self.interaction_centers_list[0]) self.middle_formlayout.add_widget_to_align(self.interaction_center_rds) # Reference structure combobox. if len(self.protocol.target_sequences) > 1: structures_list = [ element.my_header for element in self.protocol.target_sequences ] self.reference_combobox = PyMod_combobox_qt( label_text="Reference Structure", items=structures_list) self.reference_combobox.combobox.setCurrentIndex(0) self.middle_formlayout.add_widget_to_align(self.reference_combobox) self.middle_formlayout.set_input_widgets_width("auto", padding=10)
class Contact_map_options_window_qt(PyMod_protocol_window_qt): """ Window with options for the contact map analysis. """ feature_types_list = [ "Contacts", "Distances", "Distance Difference", "Distance Mean", "Distance Std" ] interaction_centers_list = ["Carbon alpha", "Carbon beta"] default_contact_threshold = 8.0 default_distance_threshold = 18.0 default_distance_diff_threshold = 3.5 default_distance_mean_threshold = default_distance_threshold default_distance_std_threshold = default_distance_diff_threshold def build_protocol_middle_frame(self): # Feature type selection. if len(self.protocol.target_sequences) == 1: # Contact and distance. sel_option_idx = 0 all_options_idx = (0, 1) elif len(self.protocol.target_sequences ) == 2: # Distance difference, mean and std. sel_option_idx = 2 all_options_idx = (2, 3, 4) elif len(self.protocol.target_sequences) > 2: # Distance mean and std. sel_option_idx = 3 all_options_idx = (3, 4) features_buttons = [ self.feature_types_list[i] for i in all_options_idx ] self.feature_type_rds = PyMod_radioselect_qt(label_text="Feature Type", buttons=features_buttons) for button, option_idx in zip(self.feature_type_rds.get_buttons(), all_options_idx): button.clicked.connect( lambda e=None, o=option_idx: self.feature_type_state(o)) self.feature_type_rds.setvalue(self.feature_types_list[sel_option_idx]) self.middle_formlayout.add_widget_to_align(self.feature_type_rds) # Distance threshold for contacts. if len(self.protocol.target_sequences) == 1: threshold = self.default_contact_threshold elif len(self.protocol.target_sequences) == 2: threshold = self.default_distance_diff_threshold else: threshold = self.default_distance_threshold self.dist_threshold_enf = PyMod_entryfield_qt( label_text="Contact Threshold (%s)" % ("\u212B"), value=str(threshold), validate={ 'validator': 'real', 'min': 1.0, 'max': 100.0 }) self.middle_formlayout.add_widget_to_align(self.dist_threshold_enf) # Interaction center selection. self.interaction_center_rds = PyMod_radioselect_qt( label_text="Interaction Center", buttons=self.interaction_centers_list) self.interaction_center_rds.setvalue(self.interaction_centers_list[0]) self.middle_formlayout.add_widget_to_align(self.interaction_center_rds) # Reference structure combobox. if len(self.protocol.target_sequences) > 1: structures_list = [ element.my_header for element in self.protocol.target_sequences ] self.reference_combobox = PyMod_combobox_qt( label_text="Reference Structure", items=structures_list) self.reference_combobox.combobox.setCurrentIndex(0) self.middle_formlayout.add_widget_to_align(self.reference_combobox) self.middle_formlayout.set_input_widgets_width("auto", padding=10) def feature_type_state(self, state): """ Launched when the user changes the "Feature Type" in the Options window. """ # Change the status of the Tkinter checkbutton. if state == 0: self.feature_type_rds.setvalue(self.feature_types_list[0]) self.dist_threshold_enf.setvalue( str(self.default_contact_threshold)) elif state == 1: self.feature_type_rds.setvalue(self.feature_types_list[1]) self.dist_threshold_enf.setvalue( str(self.default_distance_threshold)) elif state == 2: self.feature_type_rds.setvalue(self.feature_types_list[2]) self.dist_threshold_enf.setvalue( str(self.default_distance_diff_threshold)) elif state == 3: self.feature_type_rds.setvalue(self.feature_types_list[3]) self.dist_threshold_enf.setvalue( str(self.default_distance_mean_threshold)) elif state == 4: self.feature_type_rds.setvalue(self.feature_types_list[4]) self.dist_threshold_enf.setvalue( str(self.default_distance_std_threshold)) else: KeyError(state) def get_feature_type(self): feature_type = self.feature_type_rds.getvalue() if feature_type == self.feature_types_list[0]: return "contact", feature_type elif feature_type == self.feature_types_list[1]: return "distance", feature_type elif feature_type == self.feature_types_list[2]: return "distances_difference", feature_type elif feature_type == self.feature_types_list[3]: return "distances_mean", feature_type elif feature_type == self.feature_types_list[4]: return "distances_std", feature_type else: raise KeyError(feature_type) def get_interaction_center(self): int_center_string = self.interaction_center_rds.getvalue() if int_center_string == self.interaction_centers_list[0]: return "ca" elif int_center_string == self.interaction_centers_list[1]: return "cb" else: raise KeyError(int_center_string) def get_reference_id(self): try: return self.reference_combobox.get_index() except: print("- WARNING: could not obtain the reference structure id.") return 0
def add_middle_frame_widgets(self): # Units combobox. self.unit_combobox = PyMod_combobox_qt(label_text="Unit Selection", items=self.protocol.units_list) self.unit_combobox.combobox.setCurrentIndex(0) self.middle_formlayout.add_widget_to_align(self.unit_combobox) # Color combobox. self.color_combobox = PyMod_combobox_qt(label_text="Color Scheme Selection", items=self.protocol.colorscheme_list) self.color_combobox.combobox.setCurrentIndex(5) self.middle_formlayout.add_widget_to_align(self.color_combobox) # Sub-frame created to display entries for Logo Range option. self.range_subframe = PyMod_hbox_option_qt(label_text="Logo Range") # Logo start position spinbox. self.logo_start = QtWidgets.QSpinBox() self.logo_start.setRange(1, self.protocol.ali_length) self.range_subframe.hbox.addWidget(self.logo_start) # Separator dash. self.logo_range_dash = QtWidgets.QLabel(" - ") self.range_subframe.hbox.addWidget(self.logo_range_dash) # Logo end position spinbox. self.logo_end = QtWidgets.QSpinBox() self.logo_end.setRange(1, self.protocol.ali_length) self.logo_end.setValue(self.protocol.ali_length) self.range_subframe.hbox.addWidget(self.logo_end) self.range_subframe.set_auto_input_widget_width() self.middle_formlayout.add_widget_to_align(self.range_subframe) # ADVANCED OPTIONS. self.show_advanced_button() # Logo format combobox. self.format_combobox = PyMod_combobox_qt(label_text='Logo Format', items=self.protocol.format_list) self.format_combobox.combobox.setCurrentIndex(0) self.middle_formlayout.add_widget_to_align(self.format_combobox, advanced_option=True) # LOGO title entry. self.logo_title_enf = PyMod_entryfield_qt(label_text="Logo Title", value='') self.middle_formlayout.add_widget_to_align(self.logo_title_enf, advanced_option=True) # Stacks per line entry. self.logo_stacks_enf = PyMod_entryfield_qt(label_text="Stacks per line", value='80', validate={'validator': 'integer', 'min': 0, 'max': 100}) self.middle_formlayout.add_widget_to_align(self.logo_stacks_enf, advanced_option=True) # Scale stacks width. self.scale_width_rds = PyMod_radioselect_qt(label_text="Scale stacks width", buttons=('Yes', 'No')) self.scale_width_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.scale_width_rds, advanced_option=True) # Show error bars. self.show_error_rds = PyMod_radioselect_qt(label_text="Show error bars", buttons=('Yes', 'No')) self.show_error_rds.setvalue("No") self.middle_formlayout.add_widget_to_align(self.show_error_rds, advanced_option=True) self.middle_formlayout.set_input_widgets_width(175)
def build_rmsd_option(self): # Decide whether to compute the RMSD matrix if the structural alignment. self.compute_rmsd_rds = PyMod_radioselect_qt(label_text="Compute RMSD Matrix", buttons=('Yes', 'No')) self.compute_rmsd_rds.setvalue("Yes") self.middle_formlayout.add_widget_to_align(self.compute_rmsd_rds)