コード例 #1
0
ファイル: Qt4_TreeBrick.py プロジェクト: jan-meyer/mxcube
class Qt4_TreeBrick(BlissWidget):
    """
    Descript. :
    """

    def __init__(self, *args):
        """
        Descript. :
        """
        BlissWidget.__init__(self, *args)

        # Hardware objects ----------------------------------------------------
        self.beamline_config_hwobj = None
        self.session_hwobj = None
        self.lims_hwobj = None
        self.sample_changer_one_hwobj = None
        self.sample_changer_two_hwobj = None
        self.queue_hwobj = None

        # Internal variables --------------------------------------------------
        #self.current_cpos = None
        self.__collection_stopped = False 
        self.current_view = None

        # Properties ---------------------------------------------------------- 
        self.addProperty("holderLengthMotor", "string", "")
        self.addProperty("queue", "string", "/Qt4_queue")
        self.addProperty("queue_model", "string", "/Qt4_queue-model")
        self.addProperty("beamline_setup", "string", "/Qt4_beamline-setup-break")
        self.addProperty("xml_rpc_server", "string", "/Qt4_xml_rpc_server")
        #names of sample changers could come from hwobj
        self.addProperty("scOneName", "string", "Sample changer")
        self.addProperty("scTwoName", "string", "Plate")

        # Signals ------------------------------------------------------------
        self.defineSignal("enable_hutch_menu", ())
        self.defineSignal("enable_command_menu", ())
        self.defineSignal("enable_task_toolbox", ())

        # Hiding and showing the tabs
        self.defineSignal("hide_sample_tab", ())
        self.defineSignal("hide_dc_parameters_tab", ())
        self.defineSignal("hide_sample_centring_tab", ())
        self.defineSignal("hide_dcg_tab", ())
        self.defineSignal("hide_sample_changer_one_tab", ())
        self.defineSignal("hide_sample_changer_two_tab", ())
        self.defineSignal("hide_char_parameters_tab", ())
        self.defineSignal("hide_energy_scan_tab",())
        self.defineSignal("hide_xrf_spectrum_tab",())
        self.defineSignal("hide_workflow_tab", ())
        self.defineSignal("hide_advanced_tab", ())

        # Populating the tabs with data
        self.defineSignal("populate_dc_parameter_widget", ())
        self.defineSignal("populate_dc_group_widget", ())
        self.defineSignal("populate_char_parameter_widget",())
        self.defineSignal("populate_sample_details",())
        self.defineSignal("populate_energy_scan_widget", ())
        self.defineSignal("populate_xrf_spectrum_widget", ())
        self.defineSignal("populate_workflow_tab", ())
        self.defineSignal("populate_advanced_widget", ())

        # Handle selection
        self.defineSignal("selection_changed",())
        self.defineSignal("set_directory", ())
        self.defineSignal("set_prefix", ())
        self.defineSignal("set_sample", ())

        # Slots ---------------------------------------------------------------
        self.defineSlot("logged_in", ())
        self.defineSlot("status_msg_changed", ())
        self.defineSlot("sample_load_state_changed", ())
        self.defineSlot("set_session", ())

        self.defineSlot("get_tree_brick",())
        self.defineSlot("get_selected_samples", ())

        #self.defineSlot("get_mounted_sample", ())
        #self.defineSlot("new_centred_position", ())
        #self.defineSlot("add_dcg", ())
        #self.defineSlot("add_data_collection", ())
        #self.defineSlot("set_session", ())

        # Graphic elements ----------------------------------------------------
        self.sample_changer_widget = uic.loadUi(os.path.join(\
             os.path.dirname(__file__),
             "widgets/ui_files/Qt4_sample_changer_widget_layout.ui"))

        #self.refresh_pixmap = Qt4_Icons.load("Refresh2.png")
        #self.sample_changer_widget.synch_button.setIcon(QtGui.QIcon(self.refresh_pixmap))
        #self.sample_changer_widget.synch_button.setText("Synch ISPyB")

        self.dc_tree_widget = DataCollectTree(self)
        self.dc_tree_widget.selection_changed_cb = self.selection_changed
        self.dc_tree_widget.run_cb = self.run
        #self.dc_tree_widget.clear_centred_positions_cb = \
        #    self.clear_centred_positions
        self.tree_options_dialog = TreeOptionsDialog(self, 'Tree options Dialog')
        self.tree_options_dialog.setModal(True)

        # Layout --------------------------------------------------------------
        main_layout = QtGui.QVBoxLayout(self)
        main_layout.addWidget(self.sample_changer_widget)
        main_layout.addWidget(self.dc_tree_widget)
        main_layout.setSpacing(0)
        main_layout.setContentsMargins(0, 0, 0, 0) 

        # SizePolicies --------------------------------------------------------
        self.sample_changer_widget.setSizePolicy(\
             QtGui.QSizePolicy.MinimumExpanding, QtGui.QSizePolicy.Fixed)

        # Qt signal/slot connections ------------------------------------------
        self.sample_changer_widget.details_button.clicked.connect(\
             self.toggle_sample_changer_tab)
        self.sample_changer_widget.filter_cbox.activated.connect(\
             self.mount_mode_combo_changed)
        self.sample_changer_widget.centring_cbox.activated.connect(\
             self.dc_tree_widget.set_centring_method)
        self.sample_changer_widget.synch_ispyb_button.clicked.connect(\
             self.refresh_sample_list)
        #self.sample_changer_widget.tree_options_button.clicked.connect(\
        #     self.open_tree_options_dialog)
        self.sample_changer_widget.filter_combo.activated.connect(\
             self.filter_combo_changed)
        self.sample_changer_widget.filter_ledit.textChanged.connect(\
             self.filter_text_changed)

        # Other --------------------------------------------------------------- 
        self.enable_collect(False)
        #self.setFixedWidth(315) 
        #self.sample_changer_widget.child('centring_cbox').setCurrentItem(1)
        self.dc_tree_widget.set_centring_method(1)

    # Framework 2 method
    def run(self):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    # Framework 2 method
    def propertyChanged(self, property_name, old_value, new_value):
        """
        Descript. :
        """
        if property_name == 'holder_length_motor':
            self.dc_tree_widget.hl_motor_hwobj = self.getHardwareObject(new_value)

        elif property_name == 'queue':            
            self.queue_hwobj = self.getHardwareObject(new_value)
            self.dc_tree_widget.queue_hwobj = self.queue_hwobj
            self.connect(self.queue_hwobj, 'show_workflow_tab',
                         self.show_workflow_tab_from_model)

            self.connect(self.queue_hwobj, 'queue_entry_execute',
                         self.dc_tree_widget.queue_entry_execution_started)

            self.connect(self.queue_hwobj, 'queue_paused', 
                         self.dc_tree_widget.queue_paused_handler)

            self.connect(self.queue_hwobj, 'queue_execution_finished', 
                         self.dc_tree_widget.queue_execution_completed)

            self.connect(self.queue_hwobj, 'queue_stopped', 
                         self.dc_tree_widget.queue_stop_handler)
        elif property_name == 'queue_model':
            self.queue_model_hwobj = self.getHardwareObject(new_value)

            self.dc_tree_widget.queue_model_hwobj = self.queue_model_hwobj
            self.dc_tree_widget.confirm_dialog.queue_model_hwobj = self.queue_model_hwobj
            self.connect(self.queue_model_hwobj, 'child_added',
                         self.dc_tree_widget.add_to_view)
        elif property_name == 'beamline_setup':
            bl_setup = self.getHardwareObject(new_value)
            self.dc_tree_widget.beamline_setup_hwobj = bl_setup
            self.sample_changer_one_hwobj = bl_setup.sample_changer_one_hwobj
            self.sample_changer_two_hwobj = bl_setup.sample_changer_two_hwobj
            self.session_hwobj = bl_setup.session_hwobj
            self.lims_hwobj = bl_setup.lims_client_hwobj

            if self.sample_changer_one_hwobj is not None:
                self.connect(self.sample_changer_one_hwobj, SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.sample_changer_one_hwobj, SampleChanger.INFO_CHANGED_EVENT, 
                             self.set_sample_pin_icon)
            if self.sample_changer_two_hwobj is not None:
                self.connect(self.sample_changer_two_hwobj, SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.sample_changer_two_hwobj, SampleChanger.INFO_CHANGED_EVENT,
                             self.set_sample_pin_icon)

            has_shutter_less = bl_setup.detector_has_shutterless()

            if has_shutter_less:
                self.dc_tree_widget.confirm_dialog.disable_dark_current_cbx()
        elif property_name == 'xml_rpc_server':
            xml_rpc_server_hwobj = self.getHardwareObject(new_value)

            if xml_rpc_server_hwobj:
                self.connect(xml_rpc_server_hwobj, 'add_to_queue',
                             self.add_to_queue)

                self.connect(xml_rpc_server_hwobj, 'start_queue',
                             self.dc_tree_widget.collect_items)
        elif property_name == 'scOneName':
              self.sample_changer_widget.filter_cbox.setItemText(1, new_value)
        elif property_name == 'scTwoName':
              self.sample_changer_widget.filter_cbox.setItemText(2, new_value) 
        else:
            BlissWidget.propertyChanged(self, property_name, old_value, new_value)

    def set_session(self, session_id, t_prop_code = None, prop_number = None,
                    prop_id = None, start_date = None, prop_code = None,
                    is_inhouse = None):
        """
        Descript. :
        """
        self.session_hwobj.set_session_start_date(start_date)

    def logged_in(self, logged_in):
        """
        Descript. :Connected to the signal loggedIn of ProposalBrick2.
                   The signal is emitted when a user was succesfully logged in.
                   At first free-pin mode is created
                   Then it tries to initialize two sample changers and create
                   two associated queue models.
        """
        self.enable_collect(logged_in)
        
        if not logged_in:
            self.dc_tree_widget.sample_mount_method = 0
            self.dc_tree_widget.populate_free_pin()
          
            if self.sample_changer_one_hwobj: 
                sc_basket_content, sc_sample_content = self.get_sc_content(\
                    self.sample_changer_one_hwobj)
                if sc_basket_content and sc_sample_content:
                    sc_basket_list, sc_sample_list = self.dc_tree_widget.\
                         samples_from_sc_content(sc_basket_content, sc_sample_content)
                    self.dc_tree_widget.sample_mount_method = 1
                    self.dc_tree_widget.populate_tree_widget(sc_basket_list, sc_sample_list, 
                         self.dc_tree_widget.sample_mount_method)
   
            if self.sample_changer_two_hwobj:        
                sc_basket_content, sc_sample_content = self.get_sc_content(\
                    self.sample_changer_two_hwobj)
                if sc_basket_content and sc_sample_content:
                    sc_basket_list, sc_sample_list = self.dc_tree_widget.\
                         samples_from_sc_content(sc_basket_content, sc_sample_content)
                    self.dc_tree_widget.sample_mount_method = 2
                    self.dc_tree_widget.populate_tree_widget(sc_basket_list, sc_sample_list, 
                         self.dc_tree_widget.sample_mount_method)    

            #if self.dc_tree_widget.sample_mount_method > 0:
            #self.sample_changer_widget.synch_combo.setEnabled()
            #self.sample_changer_widget.synch_label.setEnabled(True)

            self.sample_changer_widget.filter_cbox.setCurrentIndex(\
                 self.dc_tree_widget.sample_mount_method)
            self.dc_tree_widget.filter_sample_list(\
                 self.dc_tree_widget.sample_mount_method)

        self.dc_tree_widget.sample_tree_widget_selection()
        self.dc_tree_widget.set_sample_pin_icon()

    def enable_collect(self, state):
        """
        Enables the collect controls.

        :param state: Enable if state is True and disable if False
        :type state: bool

        :returns: None
        """
        self.dc_tree_widget.enable_collect(state)

    def enable_hutch_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_hutch_menu"), state)

    def enable_command_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_command_menu"), state)

    def enable_task_toolbox(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_task_toolbox"), state)

    def get_tree_brick(self, tree_brick):
        """
        Gets the reference to the tree brick. Used to get a reference from
        another brick via the signal get_tree_brick. The attribute tree_brick
        of the passed dictionary will contain the reference.

        :param tree_brick: A dictonary to contain the reference.
        :type tree_brick: dict

        :returns: None
        """
        tree_brick['tree_brick'] = self

    def samples_from_lims(self, samples):
        """
        Descript. :
        """
        barcode_samples, location_samples = self.dc_tree_widget.samples_from_lims(samples)
        l_samples = dict()            
   
        # TODO: add test for sample changer type, here code is for Robodiff only
        for location, l_sample in location_samples.iteritems():
          if l_sample.lims_location != (None, None):
            basket, sample = l_sample.lims_location
            cell = int(round((basket+0.5)/3.0))
            puck = basket-3*(cell-1)
            new_location = (cell, puck, sample)
            l_sample.lims_location = new_location
            l_samples[new_location] = l_sample
            name = l_sample.get_name()
            l_sample.init_from_sc_sample([new_location])
            l_sample.set_name(name)

        return barcode_samples, l_samples

    def refresh_sample_list(self):
        """
        Retrives sample information from ISPyB and populates the sample list
        accordingly.
        """
        if True:
            lims_client = self.lims_hwobj
            samples = lims_client.get_samples(self.session_hwobj.proposal_id,
                                              self.session_hwobj.session_id)
            basket_list = []
            sample_list = []
          
            if samples:
                sample_changer = None
                if self.dc_tree_widget.sample_mount_method == 1:
                    sample_changer = self.sample_changer_one_hwobj
                else:
                    sample_changer = self.sample_changer_two_hwobj     
 
                (barcode_samples, location_samples) = \
                    self.dc_tree_widget.samples_from_lims(samples)
                sc_basket_content, sc_sample_content = self.get_sc_content(\
                  sample_changer)
                sc_basket_list, sc_sample_list = self.dc_tree_widget.\
                  samples_from_sc_content(sc_basket_content, sc_sample_content)

                basket_list = sc_basket_list
           
                for sc_sample in sc_sample_list:
                    # Get the sample in lims with the barcode
                    # sc_sample.code
                    lims_sample = barcode_samples.get(sc_sample.code)

                    # There was a sample with that barcode
                    if lims_sample:
                        if lims_sample.lims_location == sc_sample.location:
                            logging.getLogger("user_level_log").\
                                warning("Found sample in ISPyB for location %s" % str(sc_sample.location))
                            sample_list.append(lims_sample)
                        else:
                            logging.getLogger("user_level_log").\
                                warning("The sample with the barcode (%s) exists"+\
                                        " in LIMS but the location does not mat" +\
                                        "ch. Sample changer location: %s, LIMS " +\
                                        "location %s" % (sc_sample.code,
                                                         sc_sample.location,
                                                         lims_sample.lims_location))
                            sample_list.append(sc_sample)
                    else: # No sample with that barcode, continue with location
                        lims_sample = location_samples.get(sc_sample.location)
                        if lims_sample:
                            if lims_sample.lims_code:
                                logging.getLogger("user_level_log").\
                                    warning("The sample has a barcode in LIMS, but "+\
                                            "the SC has no barcode information for "+\
                                            "this sample. For location: %s" % str(sc_sample.location))
                                sample_list.append(lims_sample)
                            else:
                                logging.getLogger("user_level_log").\
                                    warning("Found sample in ISPyB for location %s" % str(sc_sample.location))
                                sample_list.append(lims_sample)
                        else:
                            if lims_sample:
                                if lims_sample.lims_location != None:
                                    logging.getLogger("user_level_log").\
                                        warning("No barcode was provided in ISPyB "+\
                                                "which makes it impossible to verify if"+\
                                                "the locations are correct, assuming "+\
                                                "that the positions are correct.")
                                    sample_list.append(lims_sample)
                            else:
                                logging.getLogger("user_level_log").\
                                    warning("No sample in ISPyB for location %s" % str(sc_sample.location))
                                sample_list.append(sc_sample)
            self.dc_tree_widget.populate_tree_widget(basket_list, sample_list, 
                 self.dc_tree_widget.sample_mount_method)

    def open_tree_options_dialog(self):
        self.tree_options_dialog.set_filter_lists(\
             self.dc_tree_widget.sample_tree_widget)
        self.tree_options_dialog.show()

    def get_sc_content(self, sample_changer):
        """
        Gets the 'raw' data from the sample changer.
        
        :returns: A list with tuples, containing the sample information.
        """
        sc_basket_content = []
        sc_sample_content = []
        for basket in sample_changer.getBasketList():
            basket_index = basket.getIndex()
            basket_code = basket.getID() or ""
            basket_name = basket.getName()
            is_present = basket.isPresent()
            sc_basket_content.append((basket_index+1, basket, basket_name))
             
        for sample in sample_changer.getSampleList():
            matrix = sample.getID() or ""
            basket_index = sample.getContainer().getIndex()
            sample_index = sample.getIndex()
            basket_code = sample.getContainer().getID() or ""
            sample_name = sample.getName()
            sc_sample_content.append((matrix, basket_index + 1, sample_index + 1, sample_name))
        return sc_basket_content, sc_sample_content


    def status_msg_changed(self, msg, color):
        """
        Status message from the SampleChangerBrick.
        
        :param msg: The message
        :type msg: str

        :returns: None
        """
        logging.getLogger("user_level_log").info(msg)

    def set_sample_pin_icon(self):
        """
        Updates the location of the sample pin when the
        matrix code information changes. The matrix code information
        is updated, but not exclusively, when a sample is changed.
        """
        self.dc_tree_widget.set_sample_pin_icon()

    def sample_load_state_changed(self, state, *args):
        """
        The state in the sample loading procedure changed.
        Ie from Loading to mounted

        :param state: str (Enumerable)
        :returns: None
        """
        s_color = SC_STATE_COLOR.get(state, "UNKNOWN")
        Qt4_widget_colors.set_widget_color(self.sample_changer_widget.details_button,
                                           QtGui.QColor(s_color))

    def show_sample_centring_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def show_sample_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def show_dcg_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def populate_dc_parameters_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_parameter_widget"), item)
        
    def show_datacollection_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_dc_parameters_tab(item)

    def populate_dc_group_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_group_widget"), item)

    def show_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_char_parameters_tab(item)

    def populate_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_char_parameter_widget"), item)

    def show_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_energy_scan_tab(item)

    def populate_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_energy_scan_widget"), item)

    def show_xrf_spectrum_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_xrf_spectrum_tab(item)

    def populate_xrf_spectrum_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_spectrum_widget"), item)

    def show_advanced_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), False)
        self.populate_advanced_widget(item)

    def populate_advanced_widget(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_advanced_widget"), item)

    def show_workflow_tab_from_model(self):
        """
        Descript. :
        """
        self.show_workflow_tab(None)
        
    def show_workflow_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), False)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

        running = self.queue_hwobj.is_executing() 
        self.populate_workflow_tab(item, running=running)

    def populate_workflow_tab(self, item, running = False):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_workflow_tab"), item, running)

    def mount_mode_combo_changed(self, index):
        self.dc_tree_widget.filter_sample_list(index)
        
    def toggle_sample_changer_tab(self): 
        """
        Descript. :
        """
        if self.current_view == self.sample_changer_widget:
            self.current_view = None
            self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), True)
            self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), True)
            self.dc_tree_widget.sample_tree_widget_selection()
            self.sample_changer_widget.details_button.setText("Show SC-details")
        else:
            self.current_view = self.sample_changer_widget
            self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
            self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
            if self.dc_tree_widget.sample_mount_method == 1:
                self.emit(QtCore.SIGNAL("hide_sample_changer_one_tab"), False)
            else:
                self.emit(QtCore.SIGNAL("hide_sample_changer_two_tab"), False)
            self.sample_changer_widget.details_button.setText("Hide SC-details")
            self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        
    def selection_changed(self, items):
        """
        Descript. :
        """
        if len(items) == 1:
            item = items[0]
            if isinstance(item, Qt4_queue_item.SampleQueueItem):
                self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
                self.emit_set_sample(item)
                self.emit_set_directory()
                self.emit_set_prefix(item)
                #self.populate_edna_parameter_widget(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                self.populate_dc_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.CharacterisationQueueItem):
                self.populate_char_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
                self.populate_energy_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.XRFSpectrumQueueItem):
                self.populate_xrf_spectrum_tab(item)
            elif isinstance(item, Qt4_queue_item.GenericWorkflowQueueItem):
                self.populate_workflow_tab(item)
            elif isinstance(item, Qt4_queue_item.AdvancedQueueItem):
                self.populate_advanced_widget(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionGroupQueueItem):
                self.populate_dc_group_tab(item)

        self.emit(QtCore.SIGNAL("selection_changed"), items)

    def emit_set_directory(self):
        """
        Descript. :
        """
        directory = self.session_hwobj.get_base_image_directory()
        self.emit(QtCore.SIGNAL("set_directory"), directory)

    def emit_set_prefix(self, item):
        """
        Descript. :
        """
        prefix = self.session_hwobj.get_default_prefix(item.get_model())
        self.emit(QtCore.SIGNAL("set_prefix"), prefix)

    def emit_set_sample(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("set_sample"), item)

    def get_selected_items(self):
        """
        Descript. :
        """
        items = self.dc_tree_widget.get_selected_items()
        return items

    def add_to_queue(self, task_list, parent_tree_item = None, set_on = True):
        """
        Descript. :
        """
        if not parent_tree_item :
            parent_tree_item = self.dc_tree_widget.get_mounted_sample_item()
        
        self.dc_tree_widget.add_to_queue(task_list, parent_tree_item, set_on)

    def select_last_added_item(self):
        self.dc_tree_widget.select_last_added_item()

    def filter_combo_changed(self, filter_index):
        """Filters sample treewidget based on the selected filter criteria:
           0 : No filter
           1 : Sample name
           2 : Protein name
           3 : Basket index
           4 : Executed
           5 : Not executed
           6 : OSC
           7 : Helical
           8 : Characterisation
           9 : Energy Scan
           10: XRF spectrum            
           11: Advanced

        """
        self.sample_changer_widget.filter_ledit.setEnabled(\
             filter_index in (1, 2, 3))
        self.clear_filter()
        if filter_index == 0:
            self.clear_filter() 
        else:
            item_iterator = QtGui.QTreeWidgetItemIterator(\
                  self.dc_tree_widget.sample_tree_widget)
            item = item_iterator.value()
            while item:
                  hide = False
                  item_model = item.get_model() 
                  if filter_index == 4:
                      hide = not item_model.is_executed()
                  elif filter_index == 5:
                      hide = item_model.is_executed()
                  elif filter_index == 6:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = item_model.is_helical()
                  elif filter_index == 7:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = not item_model.is_helical()
                  elif filter_index == 8:
                      hide = not isinstance(item, Qt4_queue_item.CharacterisationQueueItem)
                  item.set_hidden(hide)
                  item_iterator += 1
                  item = item_iterator.value()

        self.hide_empty_baskets()     

    def filter_text_changed(self, new_text):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget) 
        item = item_iterator.value()
        while item:
              hide = False
              new_text = str(new_text)
              if self.sample_changer_widget.filter_combo.currentIndex() == 1:
                  if isinstance(item, Qt4_queue_item.SampleQueueItem):
                      hide = not new_text in item.text(0)
              elif self.sample_changer_widget.filter_combo.currentIndex() == 2:
                  if isinstance(item, Qt4_queue_item.SampleQueueItem):
                      hide = not new_text in item.get_model().crystals[0].protein_acronym
              item.set_hidden(hide) 
              item_iterator += 1
              item = item_iterator.value()

        self.hide_empty_baskets()
        
    def clear_filter(self):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget)
        item = item_iterator.value()
        while item:
              item.setHidden(False)
              item_iterator += 1
              item = item_iterator.value() 

    def hide_empty_baskets(self):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget) 
        item = item_iterator.value()
        while item:
              hide = True
              
              if type(item) in(Qt4_queue_item.BasketQueueItem,
                               Qt4_queue_item.DataCollectionGroupQueueItem): 
                  for index in range(item.childCount()):
                      if not item.child(index).isHidden():
                          hide = False
                          break
                  item.setHidden(hide) 
                 
              item_iterator += 1
              item = item_iterator.value()
コード例 #2
0
ファイル: Qt4_TreeBrick.py プロジェクト: jordiandreu/mxcube
class Qt4_TreeBrick(BlissWidget):
    """
    Descript. :
    """

    def __init__(self, *args):
        """
        Descript. :
        """
        BlissWidget.__init__(self, *args)

        # Hardware objects ----------------------------------------------------
        self.diffractometer_hwobj = None
        self.session_hwobj = None
        self.lims_hwobj = None
        self.sample_changer_hwobj = None
        self.plate_manipulator_hwobj = None
        self.queue_hwobj = None

        # Internal variables --------------------------------------------------
        self.enable_collect_conditions = {}
        self.current_view = None
        self.lims_samples = None
        self.filtered_lims_samples = None

        # Properties ---------------------------------------------------------- 
        self.addProperty("holderLengthMotor", "string", "")
        self.addProperty("queue", "string", "/queue")
        self.addProperty("queue_model", "string", "/queue-model")
        self.addProperty("beamline_setup", "string", "/beamline-setup")
        self.addProperty("xml_rpc_server", "string", "/xml_rpc_server")
        self.addProperty("useFilterWidget", "boolean", True)
        self.addProperty("useSampleWidget", "boolean", True)

        # Signals ------------------------------------------------------------
        self.defineSignal("enable_hutch_menu", ())
        self.defineSignal("enable_command_menu", ())
        self.defineSignal("enable_task_toolbox", ())
        self.defineSignal("queue_is_executing", ())

        # Hiding and showing the tabs
        self.defineSignal("hide_sample_tab", ())
        self.defineSignal("hide_dc_parameters_tab", ())
        self.defineSignal("hide_sample_centring_tab", ())
        self.defineSignal("hide_dcg_tab", ())
        self.defineSignal("hide_sample_changer_tab", ())
        self.defineSignal("hide_plate_manipulator_tab", ())
        self.defineSignal("hide_char_parameters_tab", ())
        self.defineSignal("hide_energy_scan_tab",())
        self.defineSignal("hide_xrf_spectrum_tab",())
        self.defineSignal("hide_workflow_tab", ())
        self.defineSignal("hide_advanced_tab", ())

        # Populating the tabs with data
        self.defineSignal("populate_dc_parameter_widget", ())
        self.defineSignal("populate_dc_group_widget", ())
        self.defineSignal("populate_char_parameter_widget",())
        self.defineSignal("populate_sample_details",())
        self.defineSignal("populate_energy_scan_widget", ())
        self.defineSignal("populate_xrf_spectrum_widget", ())
        self.defineSignal("populate_workflow_tab", ())
        self.defineSignal("populate_advanced_widget", ())

        # Handle selection
        self.defineSignal("selection_changed",())
        self.defineSignal("set_directory", ())
        self.defineSignal("set_prefix", ())
        self.defineSignal("set_sample", ())

        # Slots ---------------------------------------------------------------
        self.defineSlot("logged_in", ())
        self.defineSlot("status_msg_changed", ())
        self.defineSlot("sample_load_state_changed", ())
        self.defineSlot("set_session", ())

        self.defineSlot("get_tree_brick",())
        self.defineSlot("get_selected_samples", ())

        #self.defineSlot("get_mounted_sample", ())
        #self.defineSlot("new_centred_position", ())
        #self.defineSlot("add_dcg", ())
        #self.defineSlot("add_data_collection", ())
        #self.defineSlot("set_session", ())

        # Graphic elements ----------------------------------------------------
        self.sample_changer_widget = uic.loadUi(os.path.join(\
             os.path.dirname(__file__),
             "widgets/ui_files/Qt4_sample_changer_widget_layout.ui"))

        #self.refresh_pixmap = Qt4_Icons.load("Refresh2.png")
        #self.sample_changer_widget.synch_button.setIcon(QtGui.QIcon(self.refresh_pixmap))
        #self.sample_changer_widget.synch_button.setText("Synch ISPyB")

        self.dc_tree_widget = DataCollectTree(self)
        self.dc_tree_widget.selection_changed_cb = self.selection_changed
        self.dc_tree_widget.run_cb = self.run
        #self.dc_tree_widget.clear_centred_positions_cb = \
        #    self.clear_centred_positions
        self.tree_options_dialog = TreeOptionsDialog(self, 'Tree options Dialog')
        self.tree_options_dialog.setModal(True)

        # Layout --------------------------------------------------------------
        main_layout = QtGui.QVBoxLayout(self)
        main_layout.addWidget(self.sample_changer_widget)
        main_layout.addWidget(self.dc_tree_widget)
        main_layout.setSpacing(0)
        main_layout.setContentsMargins(0, 0, 0, 0) 

        # SizePolicies --------------------------------------------------------

        # Qt signal/slot connections ------------------------------------------
        self.sample_changer_widget.details_button.clicked.connect(\
             self.toggle_sample_changer_tab)
        self.sample_changer_widget.filter_cbox.activated.connect(\
             self.mount_mode_combo_changed)
        self.sample_changer_widget.centring_cbox.activated.connect(\
             self.dc_tree_widget.set_centring_method)
        self.sample_changer_widget.synch_ispyb_button.clicked.connect(\
             self.refresh_sample_list)
        #self.sample_changer_widget.tree_options_button.clicked.connect(\
        #     self.open_tree_options_dialog)
        self.sample_changer_widget.filter_combo.activated.connect(\
             self.filter_combo_changed)
        self.sample_changer_widget.filter_ledit.textChanged.connect(\
             self.filter_text_changed)
        self.sample_changer_widget.sample_combo.activated.connect(\
             self.sample_combo_changed)

        # Other --------------------------------------------------------------- 
        self.enable_collect(False)
        #self.setFixedWidth(315) 
        self.dc_tree_widget.set_centring_method(1)

    # Framework 2 method
    def run(self):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    # Framework 2 method
    def propertyChanged(self, property_name, old_value, new_value):
        """
        Descript. :
        """
        if property_name == 'holder_length_motor':
            self.dc_tree_widget.hl_motor_hwobj = self.getHardwareObject(new_value)
        elif property_name == "useFilterWidget":
            self.sample_changer_widget.filter_label.setVisible(new_value)
            self.sample_changer_widget.filter_ledit.setVisible(new_value)
            self.sample_changer_widget.filter_combo.setVisible(new_value)
        elif property_name == "useSampleWidget":
            self.sample_changer_widget.sample_label.setVisible(new_value)
            self.sample_changer_widget.sample_combo.setVisible(new_value)
        elif property_name == 'queue':            
            self.queue_hwobj = self.getHardwareObject(new_value)
            self.dc_tree_widget.queue_hwobj = self.queue_hwobj
            self.connect(self.queue_hwobj, 'show_workflow_tab',
                         self.show_workflow_tab_from_model)

            self.connect(self.queue_hwobj, 'queue_entry_execute_started',
                         self.queue_entry_execution_started)

            self.connect(self.queue_hwobj, 'queue_entry_execute_finished',
                         self.queue_entry_execution_finished)

            self.connect(self.queue_hwobj, 'queue_paused', 
                         self.queue_paused_handler)

            self.connect(self.queue_hwobj, 'queue_execution_finished', 
                         self.queue_execution_finished)

            self.connect(self.queue_hwobj, 'queue_stopped', 
                         self.queue_stop_handler)
        elif property_name == 'queue_model':
            self.queue_model_hwobj = self.getHardwareObject(new_value)

            self.dc_tree_widget.queue_model_hwobj = self.queue_model_hwobj
            self.dc_tree_widget.confirm_dialog.queue_model_hwobj = self.queue_model_hwobj
            self.connect(self.queue_model_hwobj, 'child_added',
                         self.dc_tree_widget.add_to_view)
        elif property_name == 'beamline_setup':
            bl_setup = self.getHardwareObject(new_value)
            self.dc_tree_widget.beamline_setup_hwobj = bl_setup
            self.diffractometer_hwobj = bl_setup.diffractometer_hwobj
            self.session_hwobj = bl_setup.session_hwobj
            self.lims_hwobj = bl_setup.lims_client_hwobj

            try:
               self.sample_changer_hwobj = bl_setup.sample_changer_hwobj
            except AttributeError:
               logging.getLogger("GUI").\
                    debug("Qt4_TreeBrick: sample changer hwobj not defined.")
               self.sample_changer_hwobj = None
            try:
               self.plate_manipulator_hwobj = bl_setup.plate_manipulator_hwobj
               logging.getLogger("GUI").\
                    debug("Qt4_TreeBrick: plate manipulator hwobj not defined.")
            except:
               self.plate_manipulator_hwobj = None 

            if self.sample_changer_hwobj is not None:
                self.connect(self.sample_changer_hwobj, 
                             SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.sample_changer_hwobj,
                             SampleChanger.INFO_CHANGED_EVENT, 
                             self.set_sample_pin_icon)
            if self.plate_manipulator_hwobj is not None:
                self.connect(self.plate_manipulator_hwobj,
                             SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.plate_manipulator_hwobj,
                             SampleChanger.INFO_CHANGED_EVENT,
                             self.set_sample_pin_icon)

            # This is to enable/disable collection button
            # Could be much more generic
            if hasattr(bl_setup, "diffractometer_hwobj"):
                self.connect(bl_setup.diffractometer_hwobj, 
                             "minidiffPhaseChanged",
                             self.diffractometer_phase_changed)
                self.diffractometer_phase_changed(\
                     bl_setup.diffractometer_hwobj.get_current_phase())

            if hasattr(bl_setup, "ppu_control_hwobj"):
                self.connect(bl_setup.ppu_control_hwobj,
                             'ppuStatusChanged',
                             self.ppu_status_changed)
                bl_setup.ppu_control_hwobj.update_values()
            if hasattr(bl_setup, "safety_shutter_hwobj"):
                self.connect(bl_setup.safety_shutter_hwobj,
                             'shutterStateChanged',
                             self.shutter_state_changed)
                self.shutter_state_changed(bl_setup.safety_shutter_hwobj.getShutterState())

            has_shutter_less = bl_setup.detector_has_shutterless()
            if has_shutter_less:
                self.dc_tree_widget.confirm_dialog.disable_dark_current_cbx()
        elif property_name == 'xml_rpc_server':
            xml_rpc_server_hwobj = self.getHardwareObject(new_value)

            if xml_rpc_server_hwobj:
                self.connect(xml_rpc_server_hwobj,
                             'add_to_queue',
                             self.add_to_queue)
                self.connect(xml_rpc_server_hwobj,
                             'start_queue',
                             self.dc_tree_widget.collect_items)
        else:
            BlissWidget.propertyChanged(self, property_name, old_value, new_value)

    def set_session(self, session_id, t_prop_code = None, prop_number = None,
                    prop_id = None, start_date = None, prop_code = None,
                    is_inhouse = None):
        """
        Descript. :
        """
        self.session_hwobj.set_session_start_date(start_date)

    def logged_in(self, logged_in):
        """
        Descript. :Connected to the signal loggedIn of ProposalBrick2.
                   The signal is emitted when a user was succesfully logged in.
                   At first free-pin mode is created
                   Then it tries to initialize two sample changers and create
                   two associated queue models.
        """
        self.enable_collect(logged_in)
        if not logged_in:
            self.dc_tree_widget.sample_mount_method = 0
            self.dc_tree_widget.populate_free_pin()
         
            if self.sample_changer_hwobj is not None and \
               self.diffractometer_hwobj.use_sample_changer():
                sc_basket_content, sc_sample_content = self.get_sc_content()
                if sc_basket_content and sc_sample_content:
                    sc_basket_list, sc_sample_list = self.dc_tree_widget.\
                         samples_from_sc_content(sc_basket_content, sc_sample_content)
                    self.dc_tree_widget.sample_mount_method = 1
                    self.dc_tree_widget.populate_tree_widget(sc_basket_list, sc_sample_list, 
                         self.dc_tree_widget.sample_mount_method)
                    self.sample_changer_widget.details_button.setText("Show SC-details")
   
            if self.plate_manipulator_hwobj is not None and \
               self.diffractometer_hwobj.in_plate_mode():
                plate_row_content, plate_sample_content = self.get_plate_content()
                self.dc_tree_widget.beamline_setup_hwobj.set_plate_mode(True)
                if plate_sample_content:
                     plate_row_list, plate_sample_list = self.dc_tree_widget.\
                        samples_from_sc_content(plate_row_content, plate_sample_content)
                     self.dc_tree_widget.sample_mount_method = 2
                     self.dc_tree_widget.populate_tree_widget(plate_row_list, 
                         plate_sample_list, self.dc_tree_widget.sample_mount_method)
                     self.sample_changer_widget.details_button.setText("Show Plate-details")

            self.sample_changer_widget.filter_cbox.setCurrentIndex(\
                 self.dc_tree_widget.sample_mount_method)
            self.dc_tree_widget.filter_sample_list(\
                 self.dc_tree_widget.sample_mount_method)

            if self.dc_tree_widget.sample_mount_method > 0:
                #Enable buttons related to sample changer
                self.sample_changer_widget.filter_cbox.setEnabled(True)
                self.sample_changer_widget.details_button.setEnabled(True)
            if self.dc_tree_widget.sample_mount_method < 2:
                self.sample_changer_widget.synch_ispyb_button.setEnabled(True)

        self.dc_tree_widget.sample_tree_widget_selection()
        self.dc_tree_widget.set_sample_pin_icon()

    def enable_collect(self, state):
        """
        Enables the collect controls.

        :param state: Enable if state is True and disable if False
        :type state: bool

        :returns: None
        """
        self.dc_tree_widget.enable_collect(state)

    def enable_hutch_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_hutch_menu"), state)

    def enable_command_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_command_menu"), state)

    def enable_task_toolbox(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_task_toolbox"), state)

    def get_tree_brick(self, tree_brick):
        """
        Gets the reference to the tree brick. Used to get a reference from
        another brick via the signal get_tree_brick. The attribute tree_brick
        of the passed dictionary will contain the reference.

        :param tree_brick: A dictonary to contain the reference.
        :type tree_brick: dict

        :returns: None
        """
        tree_brick['tree_brick'] = self

    def queue_entry_execution_started(self, queue_entry):
        self.emit(QtCore.SIGNAL("queue_is_executing"), True)
        self.dc_tree_widget.queue_entry_execution_started(queue_entry)

    def queue_entry_execution_finished(self, queue_entry, status):
        self.dc_tree_widget.queue_entry_execution_finished(queue_entry, status)

    def queue_paused_handler(self, status):
        self.dc_tree_widget.queue_paused_handler(status)

    def queue_execution_finished(self, status):
        self.emit(QtCore.SIGNAL("queue_is_executing"), False)
        self.dc_tree_widget.queue_execution_completed(status)

    def queue_stop_handler(self, status):
        self.emit(QtCore.SIGNAL("queue_is_executing"), False)
        self.dc_tree_widget.queue_stop_handler(status)

    def samples_from_lims(self, samples):
        """
        Descript. :
        """
        barcode_samples, location_samples = \
            self.dc_tree_widget.samples_from_lims(samples)
        l_samples = dict()            
   
        # TODO: add test for sample changer type, here code is for Robodiff only
        for location, l_sample in location_samples.iteritems():
            if l_sample.lims_location != (None, None):
                basket, sample = l_sample.lims_location
                cell = int(round((basket + 0.5) / 3.0))
                puck = basket - 3 * (cell - 1)
                new_location = (cell, puck, sample)
                l_sample.lims_location = new_location
                l_samples[new_location] = l_sample
                name = l_sample.get_name()
                l_sample.init_from_sc_sample([new_location])
                l_sample.set_name(name)

        return barcode_samples, l_samples

    def refresh_sample_list(self):
        """
        Retrives sample information from ISPyB and populates the sample list
        accordingly.
        """
        lims_client = self.lims_hwobj
        log = logging.getLogger("GUI") 

        self.lims_samples = lims_client.get_samples(\
             self.session_hwobj.proposal_id,
             self.session_hwobj.session_id)

        basket_list = []
        sample_list = []        
        self.filtered_lims_samples = [] 
        sample_changer = None

        self.sample_changer_widget.sample_combo.clear()
        for sample in self.lims_samples:
            if sample.containerSampleChangerLocation:
                self.filtered_lims_samples.append(sample)
                item_text = "%s-%s" %(sample.proteinAcronym, sample.sampleName)
                if sample.code:
                    item_text += " (%s)" % sample.code
                self.sample_changer_widget.sample_combo.addItem(item_text)
        self.sample_changer_widget.sample_combo.setEnabled(True)
        self.sample_changer_widget.sample_combo.setCurrentIndex(-1)
                 
        if self.dc_tree_widget.sample_mount_method == 1:
            sample_changer = self.sample_changer_hwobj
        elif self.dc_tree_widget.sample_mount_method == 2:
            sample_changer = self.plate_manipulator_hwobj
     
        if self.lims_samples and sample_changer is not None:
            (barcode_samples, location_samples) = \
             self.dc_tree_widget.samples_from_lims(self.lims_samples)
            sc_basket_content, sc_sample_content = self.get_sc_content()
            sc_basket_list, sc_sample_list = self.dc_tree_widget.\
              samples_from_sc_content(sc_basket_content, sc_sample_content)

            basket_list = sc_basket_list
            for sc_sample in sc_sample_list:
                # Get the sample in lims with the barcode
                # sc_sample.code
                lims_sample = barcode_samples.get(sc_sample.code)
                # There was a sample with that barcode
                if lims_sample:
                    if lims_sample.lims_location == sc_sample.location:
                        log.debug("Found sample in ISPyB for location %s" % \
                                  str(sc_sample.location))
                        sample_list.append(lims_sample)
                    else:
                        log.warning("The sample with the barcode (%s) exists" + \
                                    " in LIMS but the location does not mat"  + \
                                    "ch. Sample changer location: %s, LIMS "  + \
                                    "location %s" % (sc_sample.code,
                                                     sc_sample.location,
                                                     lims_sample.lims_location))
                        sample_list.append(sc_sample)
                else: # No sample with that barcode, continue with location
                    lims_sample = location_samples.get(sc_sample.location)
                    if lims_sample:
                        if lims_sample.lims_code:
                            log.warning("The sample has a barcode in LIMS, but " + \
                                        "the SC has no barcode information for " + \
                                        "this sample. For location: %s" % \
                                        str(sc_sample.location))
                            sample_list.append(lims_sample)
                        else:
                            log.debug("Found sample in ISPyB for location %s" % \
                                      str(sc_sample.location))
                            sample_list.append(lims_sample)
                    else:
                        if lims_sample:
                            if lims_sample.lims_location != None:
                                log.warning("No barcode was provided in ISPyB " + \
                                            "which makes it impossible to verify if" + \
                                            "the locations are correct, assuming " + \
                                            "that the positions are correct.")
                                sample_list.append(lims_sample)
                        else:
                            log.warning("No sample in ISPyB for location %s" % \
                                        str(sc_sample.location))
                            sample_list.append(sc_sample)
            self.dc_tree_widget.populate_tree_widget(basket_list, sample_list, 
                 self.dc_tree_widget.sample_mount_method)
            self.dc_tree_widget.de_select_items()

    def sample_combo_changed(self, index):
        """
        Assigns lims sample to manually-mounted sample
        """
        self.dc_tree_widget.filter_sample_list(0)
        root_model = self.queue_model_hwobj.get_model_root()
        sample_model = root_model.get_children()[0]

        sample_model.init_from_lims_object(self.filtered_lims_samples[index])
        self.dc_tree_widget.sample_tree_widget.clear()
        self.dc_tree_widget.populate_free_pin(sample_model)

    def get_sc_content(self):
        """
        Gets the 'raw' data from the sample changer.
        
        :returns: A list with tuples, containing the sample information.
        """
        sc_basket_content = []
        sc_sample_content = []

        for basket in self.sample_changer_hwobj.getBasketList():
            basket_index = basket.getIndex()
            basket_code = basket.getID() or ""
            basket_name = basket.getName()
            is_present = basket.isPresent()
            sc_basket_content.append((basket_index+1, basket, basket_name))
             
        for sample in self.sample_changer_hwobj.getSampleList():
            matrix = sample.getID() or ""
            basket_index = sample.getContainer().getIndex()
            sample_index = sample.getIndex()
            basket_code = sample.getContainer().getID() or ""
            sample_name = sample.getName()
            sc_sample_content.append((matrix, basket_index + 1, 
                                      sample_index + 1, sample_name))
        return sc_basket_content, sc_sample_content

    def get_plate_content(self):
        """
        """
        plate_row_content = []
        plate_sample_content = []

        for row in self.plate_manipulator_hwobj.getBasketList():
            row_index = row.getIndex() 
            row_code = row.getID() or ""
            row_name = row.getName()
            is_present = row.isPresent()
            plate_row_content.append((row_index, row, row_name))

        for sample in self.plate_manipulator_hwobj.getSampleList():
            row_index = sample.getCell().getRowIndex()
            col_index = sample.getCell().getCol()
            drop_index = sample.getDrop().getWellNo()
            sample_name = sample.getName()
            coords = sample.getCoords()
            matrix = sample.getID() or ""
            #vial_index = ":".join(map(str, coords[1:]))
            plate_sample_content.append((matrix, coords[0], 
                                         coords[1], sample_name))
 
        return plate_row_content, plate_sample_content

    def status_msg_changed(self, msg, color):
        """
        Status message from the SampleChangerBrick.
        
        :param msg: The message
        :type msg: str

        :returns: None
        """
        logging.getLogger("GUI").info(msg)

    def set_sample_pin_icon(self):
        """
        Updates the location of the sample pin when the
        matrix code information changes. The matrix code information
        is updated, but not exclusively, when a sample is changed.
        """
        self.dc_tree_widget.set_sample_pin_icon()

    def sample_load_state_changed(self, state, *args):
        """
        The state in the sample loading procedure changed.
        Ie from Loading to mounted

        :param state: str (Enumerable)
        :returns: None
        """
        s_color = SC_STATE_COLOR.get(state, "UNKNOWN")
        Qt4_widget_colors.set_widget_color(self.sample_changer_widget.details_button,
                                           QtGui.QColor(s_color))

    def show_sample_centring_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def show_sample_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def show_dcg_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def populate_dc_parameters_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_parameter_widget"), item)
        
    def show_datacollection_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_dc_parameters_tab(item)

    def populate_dc_group_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_group_widget"), item)

    def show_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_char_parameters_tab(item)

    def populate_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_char_parameter_widget"), item)

    def show_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_energy_scan_tab(item)

    def populate_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_energy_scan_widget"), item)

    def show_xrf_spectrum_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_xrf_spectrum_tab(item)

    def populate_xrf_spectrum_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_spectrum_widget"), item)

    def show_advanced_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), False)
        self.populate_advanced_widget(item)

    def populate_advanced_widget(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_advanced_widget"), item)

    def show_workflow_tab_from_model(self):
        """
        Descript. :
        """
        self.show_workflow_tab(None)
        
    def show_workflow_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), False)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

        running = self.queue_hwobj.is_executing() 
        self.populate_workflow_tab(item, running=running)

    def populate_workflow_tab(self, item, running = False):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_workflow_tab"), item, running)

    def mount_mode_combo_changed(self, index):
        self.dc_tree_widget.filter_sample_list(index)
        self.sample_changer_widget.details_button.setEnabled(index > 0) 
        self.sample_changer_widget.synch_ispyb_button.setEnabled(index < 2)
        self.sample_changer_widget.sample_label.setEnabled(index == 0)
        self.sample_changer_widget.sample_combo.setEnabled(index == 0)
        if index == 0:
            self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
            self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        
    def toggle_sample_changer_tab(self): 
        """
        Descript. :
        """
        if self.current_view == self.sample_changer_widget:
            self.current_view = None
            if self.dc_tree_widget.sample_mount_method == 1:
                self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
                self.sample_changer_widget.details_button.setText("Show SC-details")
            else:
                self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
                self.sample_changer_widget.details_button.setText("Show Plate-details")
            self.dc_tree_widget.sample_tree_widget_selection()
        else:
            self.current_view = self.sample_changer_widget
            self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
            self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
            if self.dc_tree_widget.sample_mount_method == 1:
                self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), False)
                self.sample_changer_widget.details_button.setText("Hide SC-details")
            else:
                self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), False)
                self.sample_changer_widget.details_button.setText("Hide Plate-details")
            self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        
    def selection_changed(self, items):
        """
        Descript. :
        """
        if len(items) == 1:
            item = items[0]
            if isinstance(item, Qt4_queue_item.SampleQueueItem):
                self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
                self.emit_set_sample(item)
                self.emit_set_directory()
                self.emit_set_prefix(item)
                #self.populate_edna_parameter_widget(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                data_collection = item.get_model()
                if data_collection.is_mesh():
                    self.populate_advanced_widget(item)
                else:
                    self.populate_dc_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.CharacterisationQueueItem):
                self.populate_char_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
                self.populate_energy_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.XRFSpectrumQueueItem):
                self.populate_xrf_spectrum_tab(item)
            elif isinstance(item, Qt4_queue_item.GenericWorkflowQueueItem):
                self.populate_workflow_tab(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionGroupQueueItem):
                self.populate_dc_group_tab(item)

        self.emit(QtCore.SIGNAL("selection_changed"), items)

    def emit_set_directory(self):
        """
        Descript. :
        """
        directory = self.session_hwobj.get_base_image_directory()
        self.emit(QtCore.SIGNAL("set_directory"), directory)

    def emit_set_prefix(self, item):
        """
        Descript. :
        """
        prefix = self.session_hwobj.get_default_prefix(item.get_model())
        self.emit(QtCore.SIGNAL("set_prefix"), prefix)

    def emit_set_sample(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("set_sample"), item)

    def get_selected_items(self):
        """
        Descript. :
        """
        items = self.dc_tree_widget.get_selected_items()
        return items

    def add_to_queue(self, task_list, parent_tree_item = None, set_on = True):
        """
        Descript. :
        """
        if not parent_tree_item :
            parent_tree_item = self.dc_tree_widget.get_mounted_sample_item()
        self.dc_tree_widget.add_to_queue(task_list, parent_tree_item, set_on)

    def select_last_added_item(self):
        self.dc_tree_widget.select_last_added_item()

    def filter_combo_changed(self, filter_index):
        """Filters sample treewidget based on the selected filter criteria:
           0 : No filter
           1 : Sample name
           2 : Protein name
           3 : Basket index
           4 : Executed
           5 : Not executed
           6 : OSC
           7 : Helical
           8 : Characterisation
           9 : Energy Scan
           10: XRF spectrum            
        """
        self.sample_changer_widget.filter_ledit.setEnabled(\
             filter_index in (1, 2, 3))
        self.clear_filter()
        if filter_index == 0:
            self.clear_filter() 
        else:
            item_iterator = QtGui.QTreeWidgetItemIterator(\
                  self.dc_tree_widget.sample_tree_widget)
            item = item_iterator.value()
            while item:
                  hide = False
                  item_model = item.get_model() 
                  if filter_index == 4:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = not item_model.is_executed()
                  elif filter_index == 5:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = item_model.is_executed()
                  elif filter_index == 6:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = item_model.is_helical()
                      else:
                          hide = True
                  elif filter_index == 7:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = not item_model.is_helical()
                      else:
                          hide = True
                  elif filter_index == 8:
                      hide = not isinstance(item, Qt4_queue_item.CharacterisationQueueItem)
                  elif filter_index == 9:
                      hide = not isinstance(item, Qt4_queue_item.EnergyScanQueueItem)
                  elif filter_index == 10:
                      hide = not isinstance(item, Qt4_queue_item.XRFSpectrumQueueItem)
                  #elif filter_index == 11:
                  #    hide = not isinstance(item, Qt4_queue_item.AdvancedQueueItem)

                  if isinstance(item, Qt4_queue_item.TaskQueueItem):
                      item.set_hidden(hide)
                  item_iterator += 1
                  item = item_iterator.value()

        self.dc_tree_widget.hide_empty_baskets()     

    def filter_text_changed(self, new_text):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget) 
        item = item_iterator.value()
        filter_index = self.sample_changer_widget.filter_combo.currentIndex()
        while item:
              hide = False
              new_text = str(new_text)
              if filter_index == 1:
                  if isinstance(item, Qt4_queue_item.SampleQueueItem):
                      hide = not new_text in item.text(0)
              elif filter_index == 2:
                  if isinstance(item, Qt4_queue_item.SampleQueueItem):
                      hide = not new_text in item.get_model().crystals[0].protein_acronym
              elif filter_index == 3:
                  if isinstance(item, Qt4_queue_item.BasketQueueItem):
                      if new_text.isdigit(): 
                          # Display one basket
                          hide = item.get_model().location != int(new_text)
                      else: 
                          # Display several baskets. Separated with ","
                          enable_baskat_list = new_text.split(',')
                          if len(enable_baskat_list) > 1:
                              hide = str(item.get_model().location) not in enable_baskat_list
              item.set_hidden(hide) 
              item_iterator += 1
              item = item_iterator.value()

        if filter_index != 3:
            self.dc_tree_widget.hide_empty_baskets()
        
    def clear_filter(self):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget)
        item = item_iterator.value()
        while item:
              item.setHidden(False)
              item_iterator += 1
              item = item_iterator.value() 

    def diffractometer_phase_changed(self, phase):
        self.enable_collect_conditions["diffractometer"] = \
             phase != "BeamLocation"
        self.update_enable_collect()

    def ppu_status_changed(self, in_error, status_msg):
        self.enable_collect_conditions["ppu"] = in_error != True
        self.update_enable_collect()

    def shutter_state_changed(self, state):
        if state == "closed":
            self.enable_collect_conditions["shutter"] = False
        else:
            self.enable_collect_conditions["shutter"] = True
        self.update_enable_collect()

    def update_enable_collect(self):
        enable_collect = all(item == True for item in self.enable_collect_conditions.values())
        if enable_collect != self.dc_tree_widget.enable_collect_condition:
            if enable_collect:
                logging.getLogger("GUI").info("Data collection is enabled")    
            else:
                logging.getLogger("GUI").warning("Data collect is disabled")
                for key, value in self.enable_collect_conditions.iteritems():
                    if value == False:
                        if key == "diffractometer":
                            logging.getLogger("GUI").info("Diffractometer is in beam location phase")
                        elif key == "shutter":
                            logging.getLogger("GUI").info("Safety shutter is closed")
                        elif key == "ppu":
                            logging.getLogger("GUI").error("PPU is in error state")
            self.dc_tree_widget.enable_collect_condition = enable_collect
            self.dc_tree_widget.sample_tree_widget_selection()
コード例 #3
0
class Qt4_TreeBrick(BlissWidget):
    """
    Descript. :
    """
    def __init__(self, *args):
        """
        Descript. :
        """
        BlissWidget.__init__(self, *args)

        # Hardware objects ----------------------------------------------------
        self.beamline_config_hwobj = None
        self.lims_hwobj = None
        self.sample_changer_hwobj = None
        self.queue_hwobj = None

        # Internal variables --------------------------------------------------
        self.current_cpos = None
        self.__collection_stopped = False
        self.current_view = None

        # Properties ----------------------------------------------------------
        self.addProperty("holderLengthMotor", "string", "")
        self.addProperty("queue", "string", "/Qt4_queue")
        self.addProperty("queue_model", "string", "/Qt4_queue-model")
        self.addProperty("beamline_setup", "string",
                         "/Qt4_beamline-setup-break")
        self.addProperty("xml_rpc_server", "string", "/Qt4_xml_rpc_server")

        # Signals ------------------------------------------------------------
        self.defineSignal("enable_hutch_menu", ())
        self.defineSignal("enable_command_menu", ())
        self.defineSignal("enable_task_toolbox", ())

        # Hiding and showing the tabs
        self.defineSignal("hide_sample_tab", ())
        self.defineSignal("hide_dc_parameters_tab", ())
        self.defineSignal("hide_sample_centring_tab", ())
        self.defineSignal("hide_dcg_tab", ())
        self.defineSignal("hide_sample_changer_tab", ())
        self.defineSignal("hide_char_parameters_tab", ())
        self.defineSignal("hide_energy_scan_tab", ())
        self.defineSignal("hide_xrf_scan_tab", ())
        self.defineSignal("hide_workflow_tab", ())
        self.defineSignal("hide_advanced_scan_tab", ())

        # Populating the tabs with data
        self.defineSignal("populate_dc_parameter_widget", ())
        self.defineSignal("populate_char_parameter_widget", ())
        self.defineSignal("populate_sample_details", ())
        self.defineSignal("populate_energy_scan_widget", ())
        self.defineSignal("populate_xrf_scan_widget", ())
        self.defineSignal("populate_workflow_tab", ())
        self.defineSignal("populate_advanced_scan_tab", ())

        # Handle selection
        self.defineSignal("selection_changed", ())
        self.defineSignal("set_directory", ())
        self.defineSignal("set_prefix", ())
        self.defineSignal("set_sample", ())

        # Slots ---------------------------------------------------------------
        self.defineSlot("logged_in", ())
        self.defineSlot("status_msg_changed", ())
        self.defineSlot("sample_load_state_changed", ())
        self.defineSlot("set_session", ())

        self.defineSlot("get_tree_brick", ())
        self.defineSlot("get_selected_samples", ())

        #self.defineSlot("get_mounted_sample", ())
        #self.defineSlot("new_centred_position", ())
        #self.defineSlot("add_dcg", ())
        #self.defineSlot("add_data_collection", ())
        #self.defineSlot("set_session", ())

        # Graphic elements ----------------------------------------------------
        self.sample_changer_widget = uic.loadUi(
            os.path.join(
                os.path.dirname(__file__),
                "widgets/ui_files/Qt4_sample_changer_widget_layout.ui"))

        self.refresh_pixmap = Qt4_Icons.load("Refresh2.png")
        self.sample_changer_widget.synch_button.setIcon(
            QtGui.QIcon(self.refresh_pixmap))
        self.sample_changer_widget.synch_button.setText("Synch ISPyB")

        self.dc_tree_widget = DataCollectTree(self)
        self.dc_tree_widget.selection_changed_cb = self.selection_changed
        self.dc_tree_widget.run_cb = self.run
        #self.dc_tree_widget.clear_centred_positions_cb = \
        #    self.clear_centred_positions

        # Layout --------------------------------------------------------------
        main_layout = QtGui.QVBoxLayout(self)
        main_layout.addWidget(self.sample_changer_widget)
        main_layout.addWidget(self.dc_tree_widget)
        main_layout.setSpacing(0)
        main_layout.setContentsMargins(0, 0, 0, 0)
        self.setLayout(main_layout)

        # SizePolicies --------------------------------------------------------
        self.sample_changer_widget.setSizePolicy(
            QtGui.QSizePolicy.MinimumExpanding, QtGui.QSizePolicy.Fixed)

        # Qt signal/slot connections ------------------------------------------
        self.sample_changer_widget.details_button.clicked.\
             connect(self.toggle_sample_changer_tab)

        self.sample_changer_widget.filter_cbox.activated.\
             connect(self.dc_tree_widget.filter_sample_list)

        self.sample_changer_widget.centring_cbox.activated.\
             connect(self.dc_tree_widget.set_centring_method)

        self.sample_changer_widget.synch_button.clicked.\
             connect(self.refresh_sample_list)

        # Other ---------------------------------------------------------------
        self.enable_collect(False)
        #self.setFixedWidth(315)
        #self.sample_changer_widget.child('centring_cbox').setCurrentItem(1)
        self.dc_tree_widget.set_centring_method(1)

    # Framework 2 method
    def run(self):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    # Framework 2 method
    def propertyChanged(self, property_name, old_value, new_value):
        """
        Descript. :
        """
        if property_name == 'holder_length_motor':
            self.dc_tree_widget.hl_motor_hwobj = self.getHardwareObject(
                new_value)

        elif property_name == 'queue':
            self.queue_hwobj = self.getHardwareObject(new_value)
            self.dc_tree_widget.queue_hwobj = self.queue_hwobj
            self.connect(self.queue_hwobj, 'show_workflow_tab',
                         self.show_workflow_tab_from_model)

            self.connect(self.queue_hwobj, 'queue_paused',
                         self.dc_tree_widget.queue_paused_handler)

            self.connect(self.queue_hwobj, 'queue_execution_finished',
                         self.dc_tree_widget.queue_execution_completed)

            self.connect(self.queue_hwobj, 'queue_stopped',
                         self.dc_tree_widget.queue_stop_handler)
        elif property_name == 'queue_model':
            self.queue_model_hwobj = self.getHardwareObject(new_value)

            self.dc_tree_widget.queue_model_hwobj = self.queue_model_hwobj
            self.dc_tree_widget.confirm_dialog.queue_model_hwobj = self.queue_model_hwobj
            self.connect(self.queue_model_hwobj, 'child_added',
                         self.dc_tree_widget.add_to_view)
        elif property_name == 'beamline_setup':
            bl_setup = self.getHardwareObject(new_value)
            self.dc_tree_widget.beamline_setup_hwobj = bl_setup
            self.sample_changer_hwobj = bl_setup.sample_changer_hwobj
            self.dc_tree_widget.sample_changer_hwobj = self.sample_changer_hwobj
            self.session_hwobj = bl_setup.session_hwobj
            self.lims_hwobj = bl_setup.lims_client_hwobj

            if self.sample_changer_hwobj is not None:
                self.connect(self.sample_changer_hwobj,
                             SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.sample_changer_hwobj,
                             SampleChanger.INFO_CHANGED_EVENT,
                             self.set_sample_pin_icon)
            has_shutter_less = bl_setup.detector_has_shutterless()

            if has_shutter_less:
                self.dc_tree_widget.confirm_dialog.disable_dark_current_cbx()
        elif property_name == 'xml_rpc_server':
            xml_rpc_server_hwobj = self.getHardwareObject(new_value)

            if xml_rpc_server_hwobj:
                self.connect(xml_rpc_server_hwobj, 'add_to_queue',
                             self.add_to_queue)

                self.connect(xml_rpc_server_hwobj, 'start_queue',
                             self.dc_tree_widget.collect_items)

    def set_session(self,
                    session_id,
                    t_prop_code=None,
                    prop_number=None,
                    prop_id=None,
                    start_date=None,
                    prop_code=None,
                    is_inhouse=None):
        """
        Descript. :
        """
        self.session_hwobj.set_session_start_date(start_date)

    def logged_in(self, logged_in):
        """
        Descript. :Connected to the signal loggedIn of ProposalBrick2.
                   The signal is emitted when a user was succesfully logged in.
        """
        self.enable_collect(logged_in)

        if not logged_in:
            self.dc_tree_widget.populate_free_pin()
            sc_basket_content, sc_sample_content = self.get_sc_content()
            if sc_basket_content and sc_sample_content:
                sc_basket_list, sc_sample_list = \
                 self.dc_tree_widget.samples_from_sc_content(\
                 sc_basket_content, sc_sample_content)
                self.dc_tree_widget.populate_tree_widget(sc_basket_list, \
                 sc_sample_list)
            self.sample_changer_widget.filter_cbox.setCurrentIndex(0)

        if not self.sample_changer_hwobj.hasLoadedSample():
            self.dc_tree_widget.filter_sample_list(2)
            self.sample_changer_widget.filter_cbox.setCurrentIndex(2)

        self.dc_tree_widget.sample_tree_widget_selection()

    def enable_collect(self, state):
        """
        Enables the collect controls.

        :param state: Enable if state is True and disable if False
        :type state: bool

        :returns: None
        """
        self.dc_tree_widget.enable_collect(state)

    def enable_hutch_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_hutch_menu"), state)

    def enable_command_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_command_menu"), state)

    def enable_task_toolbox(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_task_toolbox"), state)

    def get_tree_brick(self, tree_brick):
        """
        Gets the reference to the tree brick. Used to get a reference from
        another brick via the signal get_tree_brick. The attribute tree_brick
        of the passed dictionary will contain the reference.

        :param tree_brick: A dictonary to contain the reference.
        :type tree_brick: dict

        :returns: None
        """
        tree_brick['tree_brick'] = self

    def samples_from_lims(self, samples):
        """
        Descript. :
        """
        barcode_samples, location_samples = self.dc_tree_widget.samples_from_lims(
            samples)
        l_samples = dict()

        # TODO: add test for sample changer type, here code is for Robodiff only
        for location, l_sample in location_samples.iteritems():
            if l_sample.lims_location != (None, None):
                basket, sample = l_sample.lims_location
                cell = int(round((basket + 0.5) / 3.0))
                puck = basket - 3 * (cell - 1)
                new_location = (cell, puck, sample)
                l_sample.lims_location = new_location
                l_samples[new_location] = l_sample
                name = l_sample.get_name()
                l_sample.init_from_sc_sample([new_location])
                l_sample.set_name(name)

        #import pdb;pdb.set_trace()
        return barcode_samples, l_samples

    def refresh_sample_list(self):
        """
        Retrives sample information from ISPyB and populates the sample list
        accordingly.
        """
        lims_client = self.lims_hwobj
        samples = lims_client.get_samples(self.session_hwobj.proposal_id,
                                          self.session_hwobj.session_id)
        sample_list = []

        if samples:
            (barcode_samples, location_samples) = \
                self.dc_tree_widget.samples_from_lims(samples)
            """sc_content = self.get_sc_content()
            sc_sample_list = self.dc_tree_widget.\
                             samples_from_sc_content(sc_content)"""
            sc_basket_content, sc_sample_content = self.get_sc_content()
            sc_basket_list, sc_sample_list = self.dc_tree_widget.\
             samples_from_sc_content(sc_basket_content, sc_sample_content)

            #IK TODO
            basket_list = sc_basket_list

            for sc_sample in sc_sample_list:
                # Get the sample in lims with the barcode
                # sc_sample.code
                lims_sample = barcode_samples.get(sc_sample.code)

                # There was a sample with that barcode
                if lims_sample:
                    if lims_sample.lims_location == sc_sample.location:
                        logging.getLogger("user_level_log").\
                            warning("Found sample in ISPyB for location %s" % str(sc_sample.location))
                        sample_list.append(lims_sample)
                    else:
                        logging.getLogger("user_level_log").\
                            warning("The sample with the barcode (%s) exists"+\
                                    " in LIMS but the location does not mat" +\
                                    "ch. Sample changer location: %s, LIMS " +\
                                    "location %s" % (sc_sample.code,
                                                     sc_sample.location,
                                                     lims_sample.lims_location))
                        sample_list.append(sc_sample)
                else:  # No sample with that barcode, continue with location
                    lims_sample = location_samples.get(sc_sample.location)
                    if lims_sample:
                        if lims_sample.lims_code:
                            logging.getLogger("user_level_log").\
                                warning("The sample has a barcode in LIMS, but "+\
                                        "the SC has no barcode information for "+\
                                        "this sample. For location: %s" % str(sc_sample.location))
                            sample_list.append(lims_sample)
                        else:
                            logging.getLogger("user_level_log").\
                                warning("Found sample in ISPyB for location %s" % str(sc_sample.location))
                            sample_list.append(lims_sample)
                    else:
                        if lims_sample:
                            if lims_sample.lims_location != None:
                                logging.getLogger("user_level_log").\
                                    warning("No barcode was provided in ISPyB "+\
                                            "which makes it impossible to verify if"+\
                                            "the locations are correct, assuming "+\
                                            "that the positions are correct.")
                                sample_list.append(lims_sample)
                        else:
                            logging.getLogger("user_level_log").\
                                warning("No sample in ISPyB for location %s" % str(sc_sample.location))
                            sample_list.append(sc_sample)

            self.dc_tree_widget.populate_tree_widget(basket_list, sample_list)

    def get_sc_content(self):
        """
        Gets the 'raw' data from the sample changer.
        
        :returns: A list with tuples, containing the sample information.
        """
        sc_basket_content = []
        sc_sample_content = []

        #try:
        if True:
            for basket in self.sample_changer_hwobj.getBasketList():
                basket_index = basket.getIndex()
                basket_code = basket.getID() or ""
                is_present = basket.isPresent()
                sc_basket_content.append(
                    (basket_index + 1, basket))  #basket_code, is_present))

            for sample in self.sample_changer_hwobj.getSampleList():
                matrix = sample.getID() or ""
                basket_index = sample.getContainer().getIndex()
                vial_index = sample.getIndex()
                basket_code = sample.getContainer().getID() or ""

                sc_sample_content.append(
                    (matrix, basket_index + 1, vial_index + 1, basket_code, 0))
        else:
            #except Exception:
            logging.getLogger("user_level_log").\
                info("Could not connect to sample changer,"  + \
                     " unable to list contents. Make sure that" + \
                     " the sample changer is turned on. Using free pin mode")
            sc_basket_content = [(0, '', False)]
            sc_sample_content = [('', -1, -1, '', 1)]

        return sc_basket_content, sc_sample_content

    def status_msg_changed(self, msg, color):
        """
        Status message from the SampleChangerBrick.
        
        :param msg: The message
        :type msg: str

        :returns: None
        """
        logging.getLogger("user_level_log").info(msg)

    def set_sample_pin_icon(self):
        """
        Updates the location of the sample pin when the
        matrix code information changes. The matrix code information
        is updated, but not exclusively, when a sample is changed.
        """
        self.dc_tree_widget.set_sample_pin_icon()

    def sample_load_state_changed(self, state, *args):
        """
        The state in the sample loading procedure changed.
        Ie from Loading to mounted

        :param state: str (Enumerable)
        :returns: None
        """
        s_color = SC_STATE_COLOR.get(state, "UNKNOWN")
        Qt4_widget_colors.set_widget_color(
            self.sample_changer_widget.details_button.QtGui.QColor(s_color))

    def show_sample_centring_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    def show_sample_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    def show_dcg_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    def populate_dc_parameters_tab(self, item=None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_parameter_widget"), item)

    def show_datacollection_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_dc_parameters_tab(item)

    def show_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_char_parameters_tab(item)

    def populate_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_char_parameter_widget"), item)

    def show_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_energy_scan_tab(item)

    def populate_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_energy_scan_widget"), item)

    def show_xrf_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_xrf_scan_tab(item)

    def populate_xrf_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_scan_widget"), item)

    def show_advanced_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), False)
        self.populate_xrf_scan_tab(item)

    def populate_xrf_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_scan_widget"), item)

    def show_workflow_tab_from_model(self):
        """
        Descript. :
        """
        self.show_workflow_tab(None)

    def show_workflow_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), False)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

        running = self.queue_hwobj.is_executing()
        self.populate_workflow_tab(item, running=running)

    def populate_workflow_tab(self, item, running=False):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_workflow_tab"), item, running)

    def toggle_sample_changer_tab(self):
        """
        Descript. :
        """
        if self.current_view == self.sample_changer_widget:
            self.current_view = None
            self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
            self.dc_tree_widget.sample_tree_widget_selection()
            self.sample_changer_widget.details_button.setText(
                "Show SC-details")
        else:
            self.current_view = self.sample_changer_widget
            self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
            self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
            self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), False)
            self.sample_changer_widget.details_button.setText(
                "Hide SC-details")
            self.emit(QtCore.SIGNAL("hide_sample_tab"), True)

    def selection_changed(self, items):
        """
        Descript. :
        """
        if len(items) == 1:
            item = items[0]
            if isinstance(item, Qt4_queue_item.SampleQueueItem):
                self.emit(QtCore.SIGNAL("populate_sample_details"),
                          item.get_model())
                self.emit_set_sample(item)
                self.emit_set_directory()
                self.emit_set_prefix(item)
                #self.populate_edna_parameter_widget(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                self.populate_dc_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.CharacterisationQueueItem):
                self.populate_char_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
                self.populate_energy_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.XRFScanQueueItem):
                self.populate_xrf_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.GenericWorkflowQueueItem):
                self.populate_workflow_tab(item)

        self.emit(QtCore.SIGNAL("selection_changed"), items)

    def emit_set_directory(self):
        """
        Descript. :
        """
        directory = self.session_hwobj.get_base_image_directory()
        self.emit(QtCore.SIGNAL("set_directory"), directory)

    def emit_set_prefix(self, item):
        """
        Descript. :
        """
        prefix = self.session_hwobj.get_default_prefix(item.get_model())
        self.emit(QtCore.SIGNAL("set_prefix"), prefix)

    def emit_set_sample(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("set_sample"), item)

    def get_selected_items(self):
        """
        Descript. :
        """
        items = self.dc_tree_widget.get_selected_items()
        return items

    def add_to_queue(self, task_list, parent_tree_item=None, set_on=True):
        """
        Descript. :
        """
        if not parent_tree_item:
            parent_tree_item = self.dc_tree_widget.get_mounted_sample_item()

        self.dc_tree_widget.add_to_queue(task_list, parent_tree_item, set_on)
コード例 #4
0
ファイル: Qt4_TreeBrick.py プロジェクト: jlmuir/mxcube
class Qt4_TreeBrick(BlissWidget):
    """
    Descript. :
    """

    def __init__(self, *args):
        """
        Descript. :
        """
        BlissWidget.__init__(self, *args)

        # Hardware objects ----------------------------------------------------
        self.diffractometer_hwobj = None
        self.session_hwobj = None
        self.lims_hwobj = None
        self.sample_changer_hwobj = None
        self.plate_manipulator_hwobj = None
        self.queue_hwobj = None

        # Internal variables --------------------------------------------------
        self.enable_collect_conditions = {}
        self.current_view = None
        self.lims_samples = None
        self.filtered_lims_samples = None

        # Properties ---------------------------------------------------------- 
        self.addProperty("holderLengthMotor", "string", "")
        self.addProperty("queue", "string", "/queue")
        self.addProperty("queue_model", "string", "/queue-model")
        self.addProperty("beamline_setup", "string", "/beamline-setup")
        self.addProperty("xml_rpc_server", "string", "/xml_rpc_server")
        self.addProperty("useFilterWidget", "boolean", True)
        self.addProperty("useSampleWidget", "boolean", True)
        self.addProperty("scOneName", "string", "Sample changer")
        self.addProperty("scTwoName", "string", "Plate")
        self.addProperty("usePlateNavigator", "boolean", False)

        # Signals ------------------------------------------------------------
        self.defineSignal("enable_hutch_menu", ())
        self.defineSignal("enable_command_menu", ())
        self.defineSignal("enable_task_toolbox", ())
        self.defineSignal("queue_is_executing", ())

        # Hiding and showing the tabs
        self.defineSignal("hide_sample_tab", ())
        self.defineSignal("hide_dc_parameters_tab", ())
        self.defineSignal("hide_sample_centring_tab", ())
        self.defineSignal("hide_dcg_tab", ())
        self.defineSignal("hide_sample_changer_tab", ())
        self.defineSignal("hide_plate_manipulator_tab", ())
        self.defineSignal("hide_char_parameters_tab", ())
        self.defineSignal("hide_energy_scan_tab",())
        self.defineSignal("hide_xrf_spectrum_tab",())
        self.defineSignal("hide_workflow_tab", ())
        self.defineSignal("hide_advanced_tab", ())

        # Populating the tabs with data
        self.defineSignal("populate_dc_parameter_widget", ())
        self.defineSignal("populate_dc_group_widget", ())
        self.defineSignal("populate_char_parameter_widget",())
        self.defineSignal("populate_sample_details",())
        self.defineSignal("populate_energy_scan_widget", ())
        self.defineSignal("populate_xrf_spectrum_widget", ())
        self.defineSignal("populate_workflow_tab", ())
        self.defineSignal("populate_advanced_widget", ())

        # Handle selection
        self.defineSignal("selection_changed",())
        self.defineSignal("set_directory", ())
        self.defineSignal("set_prefix", ())
        self.defineSignal("set_sample", ())

        # Slots ---------------------------------------------------------------
        self.defineSlot("logged_in", ())
        self.defineSlot("status_msg_changed", ())
        self.defineSlot("sample_load_state_changed", ())
        self.defineSlot("set_session", ())

        self.defineSlot("get_tree_brick",())
        self.defineSlot("get_selected_samples", ())

        #self.defineSlot("get_mounted_sample", ())
        #self.defineSlot("new_centred_position", ())
        #self.defineSlot("add_dcg", ())
        #self.defineSlot("add_data_collection", ())
        #self.defineSlot("set_session", ())

        # Graphic elements ----------------------------------------------------
        self.sample_changer_widget = uic.loadUi(os.path.join(\
             os.path.dirname(__file__),
             "widgets/ui_files/Qt4_sample_changer_widget_layout.ui"))

        #self.refresh_pixmap = Qt4_Icons.load("Refresh2.png")
        #self.sample_changer_widget.synch_button.setIcon(QtGui.QIcon(self.refresh_pixmap))
        #self.sample_changer_widget.synch_button.setText("Synch ISPyB")

        self.dc_tree_widget = DataCollectTree(self)
        self.dc_tree_widget.selection_changed_cb = self.selection_changed
        self.dc_tree_widget.run_cb = self.run
        #self.dc_tree_widget.clear_centred_positions_cb = \
        #    self.clear_centred_positions
        self.tree_options_dialog = TreeOptionsDialog(self, 'Tree options Dialog')
        self.tree_options_dialog.setModal(True)

        # Layout --------------------------------------------------------------
        main_layout = QtGui.QVBoxLayout(self)
        main_layout.addWidget(self.sample_changer_widget)
        main_layout.addWidget(self.dc_tree_widget)
        main_layout.setSpacing(0)
        main_layout.setContentsMargins(0, 0, 0, 0) 

        # SizePolicies --------------------------------------------------------

        # Qt signal/slot connections ------------------------------------------
        self.sample_changer_widget.details_button.clicked.connect(\
             self.toggle_sample_changer_tab)
        self.sample_changer_widget.filter_cbox.activated.connect(\
             self.mount_mode_combo_changed)
        self.sample_changer_widget.centring_cbox.activated.connect(\
             self.dc_tree_widget.set_centring_method)
        self.sample_changer_widget.synch_ispyb_button.clicked.connect(\
             self.refresh_sample_list)
        #self.sample_changer_widget.tree_options_button.clicked.connect(\
        #     self.open_tree_options_dialog)
        self.sample_changer_widget.filter_combo.activated.connect(\
             self.filter_combo_changed)
        self.sample_changer_widget.filter_ledit.textChanged.connect(\
             self.filter_text_changed)
        self.sample_changer_widget.sample_combo.activated.connect(\
             self.sample_combo_changed)

        # Other --------------------------------------------------------------- 
        self.enable_collect(False)
        #self.setFixedWidth(315) 
        self.dc_tree_widget.set_centring_method(1)

    # Framework 2 method
    def run(self):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    # Framework 2 method
    def propertyChanged(self, property_name, old_value, new_value):
        """
        Descript. :
        """
        if property_name == 'holder_length_motor':
            self.dc_tree_widget.hl_motor_hwobj = self.getHardwareObject(new_value)
        elif property_name == "useFilterWidget":
            self.sample_changer_widget.filter_label.setVisible(new_value)
            self.sample_changer_widget.filter_ledit.setVisible(new_value)
            self.sample_changer_widget.filter_combo.setVisible(new_value)
        elif property_name == "useSampleWidget":
            self.sample_changer_widget.sample_label.setVisible(new_value)
            self.sample_changer_widget.sample_combo.setVisible(new_value)
        elif property_name == 'queue':            
            self.queue_hwobj = self.getHardwareObject(new_value)
            self.dc_tree_widget.queue_hwobj = self.queue_hwobj
            self.connect(self.queue_hwobj, 'show_workflow_tab',
                         self.show_workflow_tab_from_model)

            self.connect(self.queue_hwobj, 'queue_entry_execute_started',
                         self.queue_entry_execution_started)

            self.connect(self.queue_hwobj, 'queue_entry_execute_finished',
                         self.queue_entry_execution_finished)

            self.connect(self.queue_hwobj, 'queue_paused', 
                         self.queue_paused_handler)

            self.connect(self.queue_hwobj, 'queue_execution_finished', 
                         self.queue_execution_finished)

            self.connect(self.queue_hwobj, 'queue_stopped', 
                         self.queue_stop_handler)
        elif property_name == 'queue_model':
            self.queue_model_hwobj = self.getHardwareObject(new_value)

            self.dc_tree_widget.queue_model_hwobj = self.queue_model_hwobj
            self.dc_tree_widget.confirm_dialog.queue_model_hwobj = self.queue_model_hwobj
            self.connect(self.queue_model_hwobj, 'child_added',
                         self.dc_tree_widget.add_to_view)
        elif property_name == 'beamline_setup':
            bl_setup = self.getHardwareObject(new_value)
            self.dc_tree_widget.beamline_setup_hwobj = bl_setup
            self.diffractometer_hwobj = bl_setup.diffractometer_hwobj
            self.session_hwobj = bl_setup.session_hwobj
            self.lims_hwobj = bl_setup.lims_client_hwobj

            try:
               self.sample_changer_hwobj = bl_setup.sample_changer_hwobj
            except AttributeError:
               logging.getLogger("GUI").\
                    debug("Qt4_TreeBrick: sample changer hwobj not defined.")
               self.sample_changer_hwobj = None

            #try:
            if True:
               self.plate_manipulator_hwobj = bl_setup.plate_manipulator_hwobj
               self.dc_tree_widget.init_plate_navigator(self.plate_manipulator_hwobj)
               logging.getLogger("GUI").\
                    debug("Qt4_TreeBrick: plate manipulator hwobj not defined.")
            #except:
            #   self.plate_manipulator_hwobj = None 

            if self.sample_changer_hwobj is not None:
                self.connect(self.sample_changer_hwobj, 
                             SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.sample_changer_hwobj,
                             SampleChanger.INFO_CHANGED_EVENT, 
                             self.set_sample_pin_icon)
            if self.plate_manipulator_hwobj is not None:
                self.connect(self.plate_manipulator_hwobj,
                             SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.plate_manipulator_hwobj,
                             SampleChanger.INFO_CHANGED_EVENT,
                             self.plate_info_changed)

            # This is to enable/disable collection button
            # Could be much more generic
            if hasattr(bl_setup, "diffractometer_hwobj"):
                self.connect(bl_setup.diffractometer_hwobj, 
                             "minidiffPhaseChanged",
                             self.diffractometer_phase_changed)
                self.diffractometer_phase_changed(\
                     bl_setup.diffractometer_hwobj.get_current_phase())

            if hasattr(bl_setup, "ppu_control_hwobj"):
                self.connect(bl_setup.ppu_control_hwobj,
                             'ppuStatusChanged',
                             self.ppu_status_changed)
                bl_setup.ppu_control_hwobj.update_values()
            if hasattr(bl_setup, "safety_shutter_hwobj"):
                self.connect(bl_setup.safety_shutter_hwobj,
                             'shutterStateChanged',
                             self.shutter_state_changed)
                self.shutter_state_changed(bl_setup.safety_shutter_hwobj.getShutterState())

            has_shutter_less = bl_setup.detector_has_shutterless()
            if has_shutter_less:
                self.dc_tree_widget.confirm_dialog.disable_dark_current_cbx()
        elif property_name == 'xml_rpc_server':
            xml_rpc_server_hwobj = self.getHardwareObject(new_value)

            if xml_rpc_server_hwobj:
                self.connect(xml_rpc_server_hwobj,
                             'add_to_queue',
                             self.add_to_queue)
                self.connect(xml_rpc_server_hwobj,
                             'start_queue',
                             self.dc_tree_widget.collect_items)
        elif property_name == 'scOneName':
              self.sample_changer_widget.filter_cbox.setItemText(1, new_value)
        elif property_name == 'scTwoName':
              self.sample_changer_widget.filter_cbox.setItemText(2, new_value) 
        #elif property_name == 'usePlateNavigator':
        #      self.dc_tree_widget.show_plate_navigator_cbox.setVisible(new_value)
        else:
            BlissWidget.propertyChanged(self, property_name, old_value, new_value)

    def set_session(self, session_id, t_prop_code = None, prop_number = None,
                    prop_id = None, start_date = None, prop_code = None,
                    is_inhouse = None):
        """
        Descript. :
        """
        self.session_hwobj.set_session_start_date(start_date)

    def logged_in(self, logged_in):
        """
        Descript. :Connected to the signal loggedIn of ProposalBrick2.
                   The signal is emitted when a user was succesfully logged in.
                   At first free-pin mode is created
                   Then it tries to initialize two sample changers and create
                   two associated queue models.
        """
        self.enable_collect(logged_in)
        if not logged_in:
            self.dc_tree_widget.sample_mount_method = 0
            self.dc_tree_widget.populate_free_pin()
            self.dc_tree_widget.plate_navigator_cbox.setVisible(False)

            if self.sample_changer_hwobj is not None and \
               self.diffractometer_hwobj.use_sample_changer():
                sc_basket_content, sc_sample_content = self.get_sc_content()
                if sc_basket_content and sc_sample_content:
                    sc_basket_list, sc_sample_list = self.dc_tree_widget.\
                         samples_from_sc_content(sc_basket_content, sc_sample_content)
                    self.dc_tree_widget.sample_mount_method = 1
                    self.dc_tree_widget.populate_tree_widget(sc_basket_list, sc_sample_list, 
                         self.dc_tree_widget.sample_mount_method)
                    self.sample_changer_widget.details_button.setText("Show SC-details")
 
            if self.plate_manipulator_hwobj is not None and \
               self.diffractometer_hwobj.in_plate_mode():
                if self['usePlateNavigator']:
                    self.dc_tree_widget.plate_navigator_cbox.setVisible(True)
                plate_row_content, plate_sample_content = self.get_plate_content()
                self.dc_tree_widget.beamline_setup_hwobj.set_plate_mode(True)
                if plate_sample_content:
                     plate_row_list, plate_sample_list = self.dc_tree_widget.\
                        samples_from_sc_content(plate_row_content, plate_sample_content)
                     self.dc_tree_widget.sample_mount_method = 2
                     self.dc_tree_widget.populate_tree_widget(plate_row_list, 
                         plate_sample_list, self.dc_tree_widget.sample_mount_method)
                     self.sample_changer_widget.details_button.setText("Show Plate-details")

            self.sample_changer_widget.filter_cbox.setCurrentIndex(\
                 self.dc_tree_widget.sample_mount_method)
            self.dc_tree_widget.filter_sample_list(\
                 self.dc_tree_widget.sample_mount_method)

            if self.dc_tree_widget.sample_mount_method > 0:
                #Enable buttons related to sample changer
                self.sample_changer_widget.filter_cbox.setEnabled(True)
                self.sample_changer_widget.details_button.setEnabled(True)
            if self.dc_tree_widget.sample_mount_method < 2:
                self.sample_changer_widget.synch_ispyb_button.setEnabled(True)

        self.dc_tree_widget.sample_tree_widget_selection()
        self.dc_tree_widget.set_sample_pin_icon()

    def enable_collect(self, state):
        """
        Enables the collect controls.

        :param state: Enable if state is True and disable if False
        :type state: bool

        :returns: None
        """
        self.dc_tree_widget.enable_collect(state)

    def enable_hutch_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_hutch_menu"), state)

    def enable_command_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_command_menu"), state)

    def enable_task_toolbox(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_task_toolbox"), state)

    def get_tree_brick(self, tree_brick):
        """
        Gets the reference to the tree brick. Used to get a reference from
        another brick via the signal get_tree_brick. The attribute tree_brick
        of the passed dictionary will contain the reference.

        :param tree_brick: A dictonary to contain the reference.
        :type tree_brick: dict

        :returns: None
        """
        tree_brick['tree_brick'] = self

    def queue_entry_execution_started(self, queue_entry):
        self.emit(QtCore.SIGNAL("queue_is_executing"), True)
        self.dc_tree_widget.queue_entry_execution_started(queue_entry)

    def queue_entry_execution_finished(self, queue_entry, status):
        self.dc_tree_widget.queue_entry_execution_finished(queue_entry, status)

    def queue_paused_handler(self, status):
        self.emit(QtCore.SIGNAL("queue_is_executing"), False)
        self.dc_tree_widget.queue_paused_handler(status)

    def queue_execution_finished(self, status):
        self.emit(QtCore.SIGNAL("queue_is_executing"), False)
        self.dc_tree_widget.queue_execution_completed(status)

    def queue_stop_handler(self, status):
        self.emit(QtCore.SIGNAL("queue_is_executing"), False)
        self.dc_tree_widget.queue_stop_handler(status)

    def samples_from_lims(self, samples):
        """
        Descript. :
        """
        barcode_samples, location_samples = \
            self.dc_tree_widget.samples_from_lims(samples)
        l_samples = dict()            
   
        # TODO: add test for sample changer type, here code is for Robodiff only
        for location, l_sample in location_samples.iteritems():
            if l_sample.lims_location != (None, None):
                basket, sample = l_sample.lims_location
                cell = int(round((basket + 0.5) / 3.0))
                puck = basket - 3 * (cell - 1)
                new_location = (cell, puck, sample)
                l_sample.lims_location = new_location
                l_samples[new_location] = l_sample
                name = l_sample.get_name()
                l_sample.init_from_sc_sample([new_location])
                l_sample.set_name(name)

        return barcode_samples, l_samples

    def refresh_sample_list(self):
        """
        Retrives sample information from ISPyB and populates the sample list
        accordingly.
        """
        lims_client = self.lims_hwobj
        log = logging.getLogger("GUI") 

        self.lims_samples = lims_client.get_samples(\
             self.session_hwobj.proposal_id,
             self.session_hwobj.session_id)

        basket_list = []
        sample_list = []        
        self.filtered_lims_samples = [] 
        sample_changer = None

        self.sample_changer_widget.sample_combo.clear()
        for sample in self.lims_samples:
            if sample.containerSampleChangerLocation:
                self.filtered_lims_samples.append(sample)
                item_text = "%s-%s" %(sample.proteinAcronym, sample.sampleName)
                if sample.code:
                    item_text += " (%s)" % sample.code
                self.sample_changer_widget.sample_combo.addItem(item_text)
        self.sample_changer_widget.sample_combo.setEnabled(True)
        self.sample_changer_widget.sample_combo.setCurrentIndex(-1)
                 
        if self.dc_tree_widget.sample_mount_method == 1:
            sample_changer = self.sample_changer_hwobj
        elif self.dc_tree_widget.sample_mount_method == 2:
            sample_changer = self.plate_manipulator_hwobj
     
        if self.lims_samples and sample_changer is not None:
            (barcode_samples, location_samples) = \
             self.dc_tree_widget.samples_from_lims(self.lims_samples)
            sc_basket_content, sc_sample_content = self.get_sc_content()
            sc_basket_list, sc_sample_list = self.dc_tree_widget.\
              samples_from_sc_content(sc_basket_content, sc_sample_content)

            basket_list = sc_basket_list
            for sc_sample in sc_sample_list:
                # Get the sample in lims with the barcode
                # sc_sample.code
                lims_sample = barcode_samples.get(sc_sample.code)
                # There was a sample with that barcode
                if lims_sample:
                    if lims_sample.lims_location == sc_sample.location:
                        log.debug("Found sample in ISPyB for location %s" % \
                                  str(sc_sample.location))
                        sample_list.append(lims_sample)
                    else:
                        log.warning("The sample with the barcode (%s) exists" + \
                                    " in LIMS but the location does not mat"  + \
                                    "ch. Sample changer location: %s, LIMS "  + \
                                    "location %s" % (sc_sample.code,
                                                     sc_sample.location,
                                                     lims_sample.lims_location))
                        sample_list.append(sc_sample)
                else: # No sample with that barcode, continue with location
                    lims_sample = location_samples.get(sc_sample.location)
                    if lims_sample:
                        if lims_sample.lims_code:
                            log.warning("The sample has a barcode in LIMS, but " + \
                                        "the SC has no barcode information for " + \
                                        "this sample. For location: %s" % \
                                        str(sc_sample.location))
                            sample_list.append(lims_sample)
                        else:
                            log.debug("Found sample in ISPyB for location %s" % \
                                      str(sc_sample.location))
                            sample_list.append(lims_sample)
                    else:
                        if lims_sample:
                            if lims_sample.lims_location != None:
                                log.warning("No barcode was provided in ISPyB " + \
                                            "which makes it impossible to verify if" + \
                                            "the locations are correct, assuming " + \
                                            "that the positions are correct.")
                                sample_list.append(lims_sample)
                        else:
                            log.warning("No sample in ISPyB for location %s" % \
                                        str(sc_sample.location))
                            sample_list.append(sc_sample)
            self.dc_tree_widget.populate_tree_widget(basket_list, sample_list, 
                 self.dc_tree_widget.sample_mount_method)
            self.dc_tree_widget.de_select_items()

    def sample_combo_changed(self, index):
        """
        Assigns lims sample to manually-mounted sample
        """
        self.dc_tree_widget.filter_sample_list(0)
        root_model = self.queue_model_hwobj.get_model_root()
        sample_model = root_model.get_children()[0]

        sample_model.init_from_lims_object(self.filtered_lims_samples[index])
        self.dc_tree_widget.sample_tree_widget.clear()
        self.dc_tree_widget.populate_free_pin(sample_model)

    def get_sc_content(self):
        """
        Gets the 'raw' data from the sample changer.
        
        :returns: A list with tuples, containing the sample information.
        """
        sc_basket_content = []
        sc_sample_content = []

        for basket in self.sample_changer_hwobj.getBasketList():
            basket_index = basket.getIndex()
            basket_code = basket.getID() or ""
            basket_name = basket.getName()
            is_present = basket.isPresent()
            sc_basket_content.append((basket_index+1, basket, basket_name))
             
        for sample in self.sample_changer_hwobj.getSampleList():
            matrix = sample.getID() or ""
            basket_index = sample.getContainer().getIndex()
            sample_index = sample.getIndex()
            basket_code = sample.getContainer().getID() or ""
            sample_name = sample.getName()
            sc_sample_content.append((matrix, basket_index + 1, 
                                      sample_index + 1, sample_name))
        return sc_basket_content, sc_sample_content

    def get_plate_content(self):
        """
        """
        plate_row_content = []
        plate_sample_content = []

        for row in self.plate_manipulator_hwobj.getBasketList():
            row_index = row.getIndex() 
            row_code = row.getID() or ""
            row_name = row.getName()
            is_present = row.isPresent()
            plate_row_content.append((row_index, row, row_name))

        for sample in self.plate_manipulator_hwobj.getSampleList():
            row_index = sample.getCell().getRowIndex()
            col_index = sample.getCell().getCol()
            drop_index = sample.getDrop().getWellNo()
            sample_name = sample.getName()
            coords = sample.getCoords()
            matrix = sample.getID() or ""
            #vial_index = ":".join(map(str, coords[1:]))
            plate_sample_content.append((matrix, coords[0], 
                                         coords[1], sample_name))
 
        return plate_row_content, plate_sample_content

    def status_msg_changed(self, msg, color):
        """
        Status message from the SampleChangerBrick.
        
        :param msg: The message
        :type msg: str

        :returns: None
        """
        logging.getLogger("GUI").info(msg)

    def set_sample_pin_icon(self):
        """
        Updates the location of the sample pin when the
        matrix code information changes. The matrix code information
        is updated, but not exclusively, when a sample is changed.
        """
        self.dc_tree_widget.set_sample_pin_icon()

    def sample_load_state_changed(self, state, *args):
        """
        The state in the sample loading procedure changed.
        Ie from Loading to mounted

        :param state: str (Enumerable)
        :returns: None
        """
        s_color = SC_STATE_COLOR.get(state, "UNKNOWN")
        Qt4_widget_colors.set_widget_color(self.sample_changer_widget.details_button,
                                           QtGui.QColor(s_color))

    def plate_info_changed(self):
        self.set_sample_pin_icon()
        self.dc_tree_widget.plate_navigator_widget.refresh_plate_location() 

    def show_sample_centring_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def show_sample_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def show_dcg_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

    def populate_dc_parameters_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_parameter_widget"), item)
        
    def show_datacollection_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_dc_parameters_tab(item)

    def populate_dc_group_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_group_widget"), item)

    def show_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_char_parameters_tab(item)

    def populate_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_char_parameter_widget"), item)

    def show_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_energy_scan_tab(item)

    def populate_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_energy_scan_widget"), item)

    def show_xrf_spectrum_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)
        self.populate_xrf_spectrum_tab(item)

    def populate_xrf_spectrum_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_spectrum_widget"), item)

    def show_advanced_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), False)
        self.populate_advanced_widget(item)

    def populate_advanced_widget(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_advanced_widget"), item)

    def show_workflow_tab_from_model(self):
        """
        Descript. :
        """
        self.show_workflow_tab(None)
        
    def show_workflow_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.details_button.setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_spectrum_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), False)
        self.emit(QtCore.SIGNAL("hide_advanced_tab"), True)

        running = self.queue_hwobj.is_executing() 
        self.populate_workflow_tab(item, running=running)

    def populate_workflow_tab(self, item, running = False):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_workflow_tab"), item, running)

    def mount_mode_combo_changed(self, index):
        self.dc_tree_widget.filter_sample_list(index)
        self.sample_changer_widget.details_button.setEnabled(index > 0) 
        self.sample_changer_widget.synch_ispyb_button.setEnabled(index < 2)
        self.sample_changer_widget.sample_label.setEnabled(index == 0)
        self.sample_changer_widget.sample_combo.setEnabled(index == 0)
        if index == 0:
            self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
            self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
        
    def toggle_sample_changer_tab(self): 
        """
        Descript. :
        """
        if self.current_view == self.sample_changer_widget:
            self.current_view = None
            if self.dc_tree_widget.sample_mount_method == 1:
                self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
                self.sample_changer_widget.details_button.setText("Show SC-details")
            else:
                self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), True)
                self.sample_changer_widget.details_button.setText("Show Plate-details")
            self.dc_tree_widget.sample_tree_widget_selection()
        else:
            self.current_view = self.sample_changer_widget
            self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
            self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
            if self.dc_tree_widget.sample_mount_method == 1:
                self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), False)
                self.sample_changer_widget.details_button.setText("Hide SC-details")
            else:
                self.emit(QtCore.SIGNAL("hide_plate_manipulator_tab"), False)
                self.sample_changer_widget.details_button.setText("Hide Plate-details")
            self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        
    def selection_changed(self, items):
        """
        Descript. :
        """
        if len(items) == 1:
            item = items[0]
            if isinstance(item, Qt4_queue_item.SampleQueueItem):
                self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
                self.emit_set_sample(item)
                self.emit_set_directory()
                self.emit_set_prefix(item)
                #self.populate_edna_parameter_widget(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                data_collection = item.get_model()
                if data_collection.is_mesh():
                    self.populate_advanced_widget(item)
                else:
                    self.populate_dc_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.CharacterisationQueueItem):
                self.populate_char_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
                self.populate_energy_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.XRFSpectrumQueueItem):
                self.populate_xrf_spectrum_tab(item)
            elif isinstance(item, Qt4_queue_item.GenericWorkflowQueueItem):
                self.populate_workflow_tab(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionGroupQueueItem):
                self.populate_dc_group_tab(item)

        self.emit(QtCore.SIGNAL("selection_changed"), items)

    def emit_set_directory(self):
        """
        Descript. :
        """
        directory = self.session_hwobj.get_base_image_directory()
        self.emit(QtCore.SIGNAL("set_directory"), directory)

    def emit_set_prefix(self, item):
        """
        Descript. :
        """
        prefix = self.session_hwobj.get_default_prefix(item.get_model())
        self.emit(QtCore.SIGNAL("set_prefix"), prefix)

    def emit_set_sample(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("set_sample"), item)

    def get_selected_items(self):
        """
        Descript. :
        """
        items = self.dc_tree_widget.get_selected_items()
        return items

    def add_to_queue(self, task_list, parent_tree_item = None, set_on = True):
        """
        Descript. :
        """
        if not parent_tree_item :
            parent_tree_item = self.dc_tree_widget.get_mounted_sample_item()
        self.dc_tree_widget.add_to_queue(task_list, parent_tree_item, set_on)

    def select_last_added_item(self):
        self.dc_tree_widget.select_last_added_item()

    def filter_combo_changed(self, filter_index):
        """Filters sample treewidget based on the selected filter criteria:
           0 : No filter
           1 : Sample name
           2 : Protein name
           3 : Basket index
           4 : Executed
           5 : Not executed
           6 : OSC
           7 : Helical
           8 : Characterisation
           9 : Energy Scan
           10: XRF spectrum            
        """
        self.sample_changer_widget.filter_ledit.setEnabled(\
             filter_index in (1, 2, 3))
        self.clear_filter()
        if filter_index == 0:
            self.clear_filter() 
        else:
            item_iterator = QtGui.QTreeWidgetItemIterator(\
                  self.dc_tree_widget.sample_tree_widget)
            item = item_iterator.value()
            while item:
                  hide = False
                  item_model = item.get_model() 
                  if filter_index == 4:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = not item_model.is_executed()
                  elif filter_index == 5:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = item_model.is_executed()
                  elif filter_index == 6:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = item_model.is_helical()
                      else:
                          hide = True
                  elif filter_index == 7:
                      if isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                          hide = not item_model.is_helical()
                      else:
                          hide = True
                  elif filter_index == 8:
                      hide = not isinstance(item, Qt4_queue_item.CharacterisationQueueItem)
                  elif filter_index == 9:
                      hide = not isinstance(item, Qt4_queue_item.EnergyScanQueueItem)
                  elif filter_index == 10:
                      hide = not isinstance(item, Qt4_queue_item.XRFSpectrumQueueItem)
                  #elif filter_index == 11:
                  #    hide = not isinstance(item, Qt4_queue_item.AdvancedQueueItem)

                  if isinstance(item, Qt4_queue_item.TaskQueueItem):
                      item.set_hidden(hide)
                  item_iterator += 1
                  item = item_iterator.value()

        self.dc_tree_widget.hide_empty_baskets()     

    def filter_text_changed(self, new_text):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget) 
        item = item_iterator.value()
        filter_index = self.sample_changer_widget.filter_combo.currentIndex()
        while item:
              hide = False
              new_text = str(new_text)
              if filter_index == 1:
                  if isinstance(item, Qt4_queue_item.SampleQueueItem):
                      hide = not new_text in item.text(0)
              elif filter_index == 2:
                  if isinstance(item, Qt4_queue_item.SampleQueueItem):
                      hide = not new_text in item.get_model().crystals[0].protein_acronym
              elif filter_index == 3:
                  if isinstance(item, Qt4_queue_item.BasketQueueItem):
                      if new_text.isdigit(): 
                          # Display one basket
                          hide = item.get_model().location != int(new_text)
                      else: 
                          # Display several baskets. Separated with ","
                          enable_baskat_list = new_text.split(',')
                          if len(enable_baskat_list) > 1:
                              hide = str(item.get_model().location) not in enable_baskat_list
              item.set_hidden(hide) 
              item_iterator += 1
              item = item_iterator.value()

        if filter_index != 3:
            self.dc_tree_widget.hide_empty_baskets()
        
    def clear_filter(self):
        item_iterator = QtGui.QTreeWidgetItemIterator(\
             self.dc_tree_widget.sample_tree_widget)
        item = item_iterator.value()
        while item:
              item.setHidden(False)
              item_iterator += 1
              item = item_iterator.value() 

    def diffractometer_phase_changed(self, phase):
        self.enable_collect_conditions["diffractometer"] = \
             phase != "BeamLocation"
        self.update_enable_collect()

    def ppu_status_changed(self, in_error, status_msg):
        self.enable_collect_conditions["ppu"] = in_error != True
        self.update_enable_collect()

    def shutter_state_changed(self, state):
        if state == "closed":
            self.enable_collect_conditions["shutter"] = False
        else:
            self.enable_collect_conditions["shutter"] = True
        self.update_enable_collect()

    def update_enable_collect(self):
        enable_collect = all(item == True for item in self.enable_collect_conditions.values())
        if enable_collect != self.dc_tree_widget.enable_collect_condition:
            if enable_collect:
                logging.getLogger("GUI").info("Data collection is enabled")    
            else:
                logging.getLogger("GUI").warning("Data collect is disabled")
                for key, value in self.enable_collect_conditions.iteritems():
                    if value == False:
                        if key == "diffractometer":
                            logging.getLogger("GUI").info("Diffractometer is in beam location phase")
                        elif key == "shutter":
                            logging.getLogger("GUI").info("Safety shutter is closed")
                        elif key == "ppu":
                            logging.getLogger("GUI").error("PPU is in error state")
            self.dc_tree_widget.enable_collect_condition = enable_collect
            self.dc_tree_widget.sample_tree_widget_selection()
コード例 #5
0
ファイル: Qt4_TreeBrick.py プロジェクト: olofsvensson/mxcube
class Qt4_TreeBrick(BlissWidget):
    """
    Descript. :
    """

    def __init__(self, *args):
        """
        Descript. :
        """
        BlissWidget.__init__(self, *args)

        # Hardware objects ----------------------------------------------------
        self.beamline_config_hwobj = None
        self.lims_hwobj = None
        self.sample_changer_hwobj = None
        self.queue_hwobj = None

        # Internal variables --------------------------------------------------
        self.current_cpos = None
        self.__collection_stopped = False 
        self.current_view = None

        # Properties ---------------------------------------------------------- 
        self.addProperty("holderLengthMotor", "string", "")
        self.addProperty("queue", "string", "/Qt4_queue")
        self.addProperty("queue_model", "string", "/Qt4_queue-model")
        self.addProperty("beamline_setup", "string", "/Qt4_beamline-setup")
        self.addProperty("xml_rpc_server", "string", "/xml_rpc_server")

        # Signals ------------------------------------------------------------
        self.defineSignal("enable_hutch_menu", ())
        self.defineSignal("enable_command_menu", ())
        self.defineSignal("enable_task_toolbox", ())

        # Hiding and showing the tabs
        self.defineSignal("hide_sample_tab", ())
        self.defineSignal("hide_dc_parameters_tab", ())
        self.defineSignal("hide_sample_centring_tab", ())
        self.defineSignal("hide_dcg_tab", ())
        self.defineSignal("hide_sample_changer_tab", ())
        self.defineSignal("hide_char_parameters_tab", ())
        self.defineSignal("hide_energy_scan_tab",())
        self.defineSignal("hide_xrf_scan_tab",())
        self.defineSignal("hide_workflow_tab", ())
        self.defineSignal("hide_advanced_scan_tab", ())

        # Populating the tabs with data
        self.defineSignal("populate_dc_parameter_widget", ())
        self.defineSignal("populate_char_parameter_widget",())
        self.defineSignal("populate_sample_details",())
        self.defineSignal("populate_energy_scan_widget", ())
        self.defineSignal("populate_xrf_scan_widget", ())
        self.defineSignal("populate_workflow_tab", ())
        self.defineSignal("populate_advanced_scan_tab", ())

        # Handle selection
        self.defineSignal("selection_changed",())
        self.defineSignal("set_directory", ())
        self.defineSignal("set_prefix", ())
        self.defineSignal("set_sample", ())

        # Slots ---------------------------------------------------------------
        self.defineSlot("logged_in", ())
        self.defineSlot("status_msg_changed", ())
        self.defineSlot("sample_load_state_changed", ())
        self.defineSlot("set_session", ())

        self.defineSlot("get_tree_brick",())
        self.defineSlot("get_selected_samples", ())

        
        #self.defineSlot("get_mounted_sample", ())
        #self.defineSlot("new_centred_position", ())
        #self.defineSlot("add_dcg", ())
        #self.defineSlot("add_data_collection", ())
        #self.defineSlot("set_session", ())

        # Graphic elements ----------------------------------------------------
        self.sample_changer_widget = uic.loadUi(os.path.join(os.path.dirname(__file__),
                                     "widgets/ui_files/Qt4_sample_changer_widget_layout.ui"))

        self.refresh_pixmap = Qt4_Icons.load("Refresh2.png")
        self.sample_changer_widget.findChild(QtGui.QPushButton, 
            'synch_button').setIcon(QtGui.QIcon(self.refresh_pixmap))
        self.sample_changer_widget.findChild(QtGui.QPushButton, 
            'synch_button').setText("Synch ISPyB")

        self.dc_tree_widget = DataCollectTree(self)
        self.dc_tree_widget.selection_changed_cb = self.selection_changed
        self.dc_tree_widget.run_cb = self.run
        #self.dc_tree_widget.clear_centred_positions_cb = \
        #    self.clear_centred_positions

        # Layout --------------------------------------------------------------
        main_layout = QtGui.QVBoxLayout(self)
        main_layout.addWidget(self.sample_changer_widget)
        main_layout.addWidget(self.dc_tree_widget)
        main_layout.setSpacing(0)
        main_layout.setContentsMargins(0, 0, 0, 0) 
        self.setLayout(main_layout)

        # SizePolicies --------------------------------------------------------
        self.sample_changer_widget.setSizePolicy(QtGui.QSizePolicy.MinimumExpanding,
                                                 QtGui.QSizePolicy.Fixed)
  

        # Qt signal/slot connections ------------------------------------------
        self.connect(self.sample_changer_widget.findChild(QtGui.QPushButton, 
                     'details_button'),QtCore.SIGNAL("clicked()"),
                     self.toggle_sample_changer_tab)

        self.connect(self.sample_changer_widget.findChild(QtGui.QComboBox,
                     'filter_cbox'), QtCore.SIGNAL("activated(int)"),
                     self.dc_tree_widget.filter_sample_list)
        
        self.connect(self.sample_changer_widget.findChild(QtGui.QComboBox,
                     'centring_cbox'), QtCore.SIGNAL("activated(int)"),
                     self.dc_tree_widget.set_centring_method)

        self.connect(self.sample_changer_widget.findChild(QtGui.QPushButton,
                     'synch_button'), QtCore.SIGNAL("clicked()"),
                     self.refresh_sample_list)

        # Other --------------------------------------------------------------- 
        self.enable_collect(False)
        #self.setFixedWidth(315) 
        #self.sample_changer_widget.child('centring_cbox').setCurrentItem(1)
        self.dc_tree_widget.set_centring_method(1)

    # Framework 2 method
    def run(self):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    # Framework 2 method
    def propertyChanged(self, property_name, old_value, new_value):
        """
        Descript. :
        """
        if property_name == 'holder_length_motor':
            self.dc_tree_widget.hl_motor_hwobj = self.getHardwareObject(new_value)

        elif property_name == 'queue':            
            self.queue_hwobj = self.getHardwareObject(new_value)
            self.dc_tree_widget.queue_hwobj = self.queue_hwobj
            self.connect(self.queue_hwobj, 'show_workflow_tab',
                         self.show_workflow_tab_from_model)

            self.connect(self.queue_hwobj, 'queue_paused', 
                         self.dc_tree_widget.queue_paused_handler)

            self.connect(self.queue_hwobj, 'queue_execution_finished', 
                         self.dc_tree_widget.queue_execution_completed)

            self.connect(self.queue_hwobj, 'queue_stopped', 
                         self.dc_tree_widget.queue_stop_handler)
        elif property_name == 'queue_model':
            self.queue_model_hwobj = self.getHardwareObject(new_value)

            self.dc_tree_widget.queue_model_hwobj = self.queue_model_hwobj
            self.dc_tree_widget.confirm_dialog.queue_model_hwobj = self.queue_model_hwobj
            self.connect(self.queue_model_hwobj, 'child_added',
                         self.dc_tree_widget.add_to_view)

        elif property_name == 'beamline_setup':
            bl_setup = self.getHardwareObject(new_value)
            self.dc_tree_widget.beamline_setup_hwobj = bl_setup
            self.sample_changer_hwobj = bl_setup.sample_changer_hwobj
            self.dc_tree_widget.sample_changer_hwobj = self.sample_changer_hwobj
            self.session_hwobj = bl_setup.session_hwobj
            self.lims_hwobj = bl_setup.lims_client_hwobj

            if self.sample_changer_hwobj is not None:
                self.connect(self.sample_changer_hwobj, SampleChanger.STATE_CHANGED_EVENT,
                             self.sample_load_state_changed)
                self.connect(self.sample_changer_hwobj, SampleChanger.INFO_CHANGED_EVENT, 
                             self.set_sample_pin_icon)

            has_shutter_less = bl_setup.detector_has_shutterless()

            if has_shutter_less:
                self.dc_tree_widget.confirm_dialog.disable_dark_current_cbx()

        elif property_name == 'xml_rpc_server':
            xml_rpc_server_hwobj = self.getHardwareObject(new_value)

            if xml_rpc_server_hwobj:
                self.connect(xml_rpc_server_hwobj, 'add_to_queue',
                             self.add_to_queue)

                self.connect(xml_rpc_server_hwobj, 'start_queue',
                             self.dc_tree_widget.collect_items)

    def set_session(self, session_id, t_prop_code = None, prop_number = None,
                    prop_id = None, start_date = None, prop_code = None,
                    is_inhouse = None):
        """
        Descript. :
        """
        self.session_hwobj.set_session_start_date(start_date)

    def logged_in(self, logged_in):
        """
        Descript. :Connected to the signal loggedIn of ProposalBrick2.
                   The signal is emitted when a user was succesfully logged in.
        """
        self.enable_collect(logged_in)
        
        if not logged_in:
            self.dc_tree_widget.populate_free_pin()
            sc_basket_content, sc_sample_content = self.get_sc_content()
            if sc_basket_content and sc_sample_content:
                sc_basket_list, sc_sample_list = \
                 self.dc_tree_widget.samples_from_sc_content(\
                 sc_basket_content, sc_sample_content)
                self.dc_tree_widget.populate_tree_widget(sc_basket_list, \
                 sc_sample_list)
            self.sample_changer_widget.findChild(QtGui.QComboBox, 
                 'filter_cbox').setCurrentIndex(0)

        if not self.sample_changer_hwobj.hasLoadedSample():
            self.dc_tree_widget.filter_sample_list(2)
            self.sample_changer_widget.findChild(QtGui.QComboBox,
                   'filter_cbox').setCurrentIndex(2)

        self.dc_tree_widget.sample_tree_widget_selection()

    def enable_collect(self, state):
        """
        Enables the collect controls.

        :param state: Enable if state is True and disable if False
        :type state: bool

        :returns: None
        """
        self.dc_tree_widget.enable_collect(state)

    def enable_hutch_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_hutch_menu"), state)

    def enable_command_menu(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_command_menu"), state)

    def enable_task_toolbox(self, state):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("enable_task_toolbox"), state)

    def get_tree_brick(self, tree_brick):
        """
        Gets the reference to the tree brick. Used to get a reference from
        another brick via the signal get_tree_brick. The attribute tree_brick
        of the passed dictionary will contain the reference.

        :param tree_brick: A dictonary to contain the reference.
        :type tree_brick: dict

        :returns: None
        """
        tree_brick['tree_brick'] = self

    def samples_from_lims(self, samples):
        """
        Descript. :
        """
        barcode_samples, location_samples = self.dc_tree_widget.samples_from_lims(samples)
        l_samples = dict()            
   
        # TODO: add test for sample changer type, here code is for Robodiff only
        for location, l_sample in location_samples.iteritems():
          if l_sample.lims_location != (None, None):
            basket, sample = l_sample.lims_location
            cell = int(round((basket+0.5)/3.0))
            puck = basket-3*(cell-1)
            new_location = (cell, puck, sample)
            l_sample.lims_location = new_location
            l_samples[new_location] = l_sample
            name = l_sample.get_name()
            l_sample.init_from_sc_sample([new_location])
            l_sample.set_name(name)

        #import pdb;pdb.set_trace()
        return barcode_samples, l_samples

    def refresh_sample_list(self):
        """
        Retrives sample information from ISPyB and populates the sample list
        accordingly.
        """
        lims_client = self.lims_hwobj
        samples = lims_client.get_samples(self.session_hwobj.proposal_id,
                                          self.session_hwobj.session_id)
        sample_list = []
        
        if samples:
            (barcode_samples, location_samples) = \
                self.dc_tree_widget.samples_from_lims(samples)

            """sc_content = self.get_sc_content()
            sc_sample_list = self.dc_tree_widget.\
                             samples_from_sc_content(sc_content)"""
            sc_basket_content, sc_sample_content = self.get_sc_content()
            sc_basket_list, sc_sample_list = self.dc_tree_widget.\
             samples_from_sc_content(sc_basket_content, sc_sample_content)

            #IK TODO
            basket_list = sc_basket_list
           
            for sc_sample in sc_sample_list:
                # Get the sample in lims with the barcode
                # sc_sample.code
                lims_sample = barcode_samples.get(sc_sample.code)

                # There was a sample with that barcode
                if lims_sample:
                    if lims_sample.lims_location == sc_sample.location:
                        logging.getLogger("user_level_log").\
                            warning("Found sample in ISPyB for location %s" % str(sc_sample.location))
                        sample_list.append(lims_sample)
                    else:
                        logging.getLogger("user_level_log").\
                            warning("The sample with the barcode (%s) exists"+\
                                    " in LIMS but the location does not mat" +\
                                    "ch. Sample changer location: %s, LIMS " +\
                                    "location %s" % (sc_sample.code,
                                                     sc_sample.location,
                                                     lims_sample.lims_location))
                        sample_list.append(sc_sample)
                else: # No sample with that barcode, continue with location
                    lims_sample = location_samples.get(sc_sample.location)
                    if lims_sample:
                        if lims_sample.lims_code:
                            logging.getLogger("user_level_log").\
                                warning("The sample has a barcode in LIMS, but "+\
                                        "the SC has no barcode information for "+\
                                        "this sample. For location: %s" % str(sc_sample.location))
                            sample_list.append(lims_sample)
                        else:
                            logging.getLogger("user_level_log").\
                                warning("Found sample in ISPyB for location %s" % str(sc_sample.location))
                            sample_list.append(lims_sample)
                    else:
                        if lims_sample:
                            if lims_sample.lims_location != None:
                                logging.getLogger("user_level_log").\
                                    warning("No barcode was provided in ISPyB "+\
                                            "which makes it impossible to verify if"+\
                                            "the locations are correct, assuming "+\
                                            "that the positions are correct.")
                                sample_list.append(lims_sample)
                        else:
                            logging.getLogger("user_level_log").\
                                warning("No sample in ISPyB for location %s" % str(sc_sample.location))
                            sample_list.append(sc_sample)

            self.dc_tree_widget.populate_tree_widget(basket_list, sample_list)

    def get_sc_content(self):
        """
        Gets the 'raw' data from the sample changer.
        
        :returns: A list with tuples, containing the sample information.
        """
        sc_basket_content = []
        sc_sample_content = []

        #try:
        if True:
            for basket in self.sample_changer_hwobj.getBasketList():
                basket_index = basket.getIndex()
                basket_code = basket.getID() or ""
                is_present = basket.isPresent()
                sc_basket_content.append((basket_index+1, basket)) #basket_code, is_present))

            for sample in self.sample_changer_hwobj.getSampleList():
                matrix = sample.getID() or ""
                basket_index = sample.getContainer().getIndex()
                vial_index = sample.getIndex()
                basket_code = sample.getContainer().getID() or ""

                sc_sample_content.append((matrix, basket_index+1, vial_index+1, basket_code, 0))
        else:
        #except Exception:
            logging.getLogger("user_level_log").\
                info("Could not connect to sample changer,"  + \
                     " unable to list contents. Make sure that" + \
                     " the sample changer is turned on. Using free pin mode")
            sc_basket_content = [(0, '', False)]
            sc_sample_content = [('', -1, -1, '', 1)]


        return sc_basket_content, sc_sample_content


    def status_msg_changed(self, msg, color):
        """
        Status message from the SampleChangerBrick.
        
        :param msg: The message
        :type msg: str

        :returns: None
        """
        logging.getLogger("user_level_log").info(msg)

    def set_sample_pin_icon(self):
        """
        Updates the location of the sample pin when the
        matrix code information changes. The matrix code information
        is updated, but not exclusively, when a sample is changed.
        """
        self.dc_tree_widget.set_sample_pin_icon()

    def sample_load_state_changed(self, state, *args):
        """
        The state in the sample loading procedure changed.
        Ie from Loading to mounted

        :param state: str (Enumerable)
        :returns: None
        """
        s_color = SC_STATE_COLOR.get(state, "UNKNOWN")
        self.sample_changer_widget.child('details_button').\
            setPaletteBackgroundColor(qt.QColor(s_color))

    def show_sample_centring_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton, 
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    def show_sample_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_centring_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    def show_dcg_tab(self):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

    def populate_dc_parameters_tab(self, item = None):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_dc_parameter_widget"), item)
        
    def show_datacollection_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_dc_parameters_tab(item)

    def show_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), False)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_char_parameters_tab(item)

    def populate_char_parameters_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_char_parameter_widget"), item)

    def show_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_energy_scan_tab(item)

    def populate_energy_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_energy_scan_widget"), item)

    def show_xrf_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton, 
             'details_button').setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)
        self.populate_xrf_scan_tab(item)

    def populate_xrf_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_scan_widget"), item)

    def show_advanced_scan_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), True)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), False)
        self.populate_xrf_scan_tab(item)

    def populate_xrf_scan_tab(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_xrf_scan_widget"), item)

    def show_workflow_tab_from_model(self):
        """
        Descript. :
        """
        self.show_workflow_tab(None)
        
    def show_workflow_tab(self, item):
        """
        Descript. :
        """
        self.sample_changer_widget.findChild(QtGui.QPushButton,
             'details_button').setText("Show SC-details")
        self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
        self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
        self.emit(QtCore.SIGNAL("hide_char_parameters_tab"), True)
        self.emit(QtCore.SIGNAL("hide_sample_tab"), True) 
        self.emit(QtCore.SIGNAL("hide_energy_scan_tab"), True)
        self.emit(QtCore.SIGNAL("hide_xrf_scan_tab"), False)
        self.emit(QtCore.SIGNAL("hide_workflow_tab"), False)
        self.emit(QtCore.SIGNAL("hide_advanced_scan_tab"), True)

        running = self.queue_hwobj.is_executing() 
        self.populate_workflow_tab(item, running=running)

    def populate_workflow_tab(self, item, running = False):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("populate_workflow_tab"), item, running)
        
    def toggle_sample_changer_tab(self): 
        """
        Descript. :
        """
        if self.current_view == self.sample_changer_widget:
            self.current_view = None
            self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), True)
            self.dc_tree_widget.sample_tree_widget_selection()
            self.sample_changer_widget.findChild(QtGui.QPushButton,
                 'details_button').setText("Show SC-details")
        else:
            self.current_view = self.sample_changer_widget
            self.emit(QtCore.SIGNAL("hide_dc_parameters_tab"), True)
            self.emit(QtCore.SIGNAL("hide_dcg_tab"), True)
            self.emit(QtCore.SIGNAL("hide_sample_changer_tab"), False)
            self.sample_changer_widget.findChild(QtGui.QPushButton,
                 'details_button').setText("Hide SC-details")
            self.emit(QtCore.SIGNAL("hide_sample_tab"), True)
        
    def selection_changed(self, items):
        """
        Descript. :
        """
        if len(items) == 1:
            item = items[0]
            if isinstance(item, Qt4_queue_item.SampleQueueItem):
                self.emit(QtCore.SIGNAL("populate_sample_details"), item.get_model())
                self.emit_set_sample(item)
                self.emit_set_directory()
                self.emit_set_prefix(item)
                #self.populate_edna_parameter_widget(item)
            elif isinstance(item, Qt4_queue_item.DataCollectionQueueItem):
                self.populate_dc_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.CharacterisationQueueItem):
                self.populate_char_parameters_tab(item)
            elif isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
                self.populate_energy_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.XRFScanQueueItem):
                self.populate_xrf_scan_tab(item)
            elif isinstance(item, Qt4_queue_item.GenericWorkflowQueueItem):
                self.populate_workflow_tab(item)

        self.emit(QtCore.SIGNAL("selection_changed"), items)

    def emit_set_directory(self):
        """
        Descript. :
        """
        directory = self.session_hwobj.get_base_image_directory()
        self.emit(QtCore.SIGNAL("set_directory"), directory)

    def emit_set_prefix(self, item):
        """
        Descript. :
        """
        prefix = self.session_hwobj.get_default_prefix(item.get_model())
        self.emit(QtCore.SIGNAL("set_prefix"), prefix)

    def emit_set_sample(self, item):
        """
        Descript. :
        """
        self.emit(QtCore.SIGNAL("set_sample"), item)

    def get_selected_items(self):
        """
        Descript. :
        """
        items = self.dc_tree_widget.get_selected_items()
        return items

    def add_to_queue(self, task_list, parent_tree_item = None, set_on = True):
        """
        Descript. :
        """
        if not parent_tree_item :
            parent_tree_item = self.dc_tree_widget.get_mounted_sample_item()
        
        self.dc_tree_widget.add_to_queue(task_list, parent_tree_item, set_on)