Esempio n. 1
0
    def init(self):
        Qt.QVBoxLayout(self.brick_widget)
        self.frameDisplay = QubDataImageDisplay(self.brick_widget, noAction = True, forcePopupSubWindow = True)
        self.timer = QtCore.QTimer(self.brick_widget)
        self.timer.timeout.connect(self.timerEvent)

        # layout
        #TODO:  Next line will always have an error in Eclipse, even if it is correct.
        self.frameDisplay.setSizePolicy(Qt.QSizePolicy.MinimumExpanding, Qt.QSizePolicy.MinimumExpanding)
        self.brick_widget.layout().addWidget(self.frameDisplay)
        # No foreground color change - all other left in
        action_enum = [QubDataImageDisplay.QUICK_SCROLL,
                      QubDataImageDisplay.SUB_DATA_VIEW,
                      QubDataImageDisplay.PRINT_PREVIEW,
                      QubDataImageDisplay.SAVE_IMAGE,
                      QubDataImageDisplay.STAT_ACTION,
                      QubDataImageDisplay.HEADER_INFO,
                      QubDataImageDisplay.ZOOM,
                      QubDataImageDisplay.COLORMAP,
                      QubDataImageDisplay.HORIZONTAL_SELECTION,
                      QubDataImageDisplay.VERTICAL_SELECTION,
                      QubDataImageDisplay.LINE_SELECTION,
                      QubDataImageDisplay.POSITION_AND_VALUE]
        self.frameDisplay.addStdAction(action_enum, zoomValList = None)
        # Add the monkey dialog box for masks
        openDialog = QubOpenDialogAction(parent = self.frameDisplay,
                                         name = 'mask', iconName = 'mask',
                                         group = 'Mask')
        dialog = QubMaskToolsDialog(None)
        dialog.setGraphicsView(self.frameDisplay.getDrawingView())
        openDialog.setDialog(dialog)
        openDialog._dialog = dialog
        self.frameDisplay.addDataAction(openDialog, dialog)
        # Reference the openDialog to make sure the Garbage Collector does not remove it (_openD could be _dummy)
        self.frameDisplay._openD = openDialog
Esempio n. 2
0
    def __init__(self, *args):
        BlissWidget.__init__(self, *args)

        actions = []

        self.__zoomList = QubZoomListAction(
            place="toolbar",
            zoomValList=[0.1, 0.25, 0.5, 1, 1.5, 2, 2.5, 3, 3.5, 4],
            show=1,
            group="zoom",
        )
        actions.append(self.__zoomList)

        self.__zoomFitOrFill = QubZoomAction(
            place="toolbar", keepROI=True, group="zoom"
        )
        self.__zoomFitOrFill.setList(self.__zoomList)
        self.__zoomList.setActionZoomMode(self.__zoomFitOrFill)
        actions.append(self.__zoomFitOrFill)

        zoomAction = QubZoomRectangle(
            label="Zoom Crop",
            place="toolbar",
            show=1,
            group="zoom",
            activate_click_drag=True,
            drawingObjectLayer=2 ** 31,
            unactiveActionWhenDub=True,
        )
        qt.QObject.connect(
            zoomAction, qt.PYSIGNAL("RectangleSelected"), self.__rectangleZoomChanged
        )
        actions.append(zoomAction)

        ####### CHANGE FOREGROUND COLOR #######
        fcoloraction = QubForegroundColorAction(name="color", group="image")
        actions.append(fcoloraction)

        ####### MEASURE #######
        measureAction = QubOpenDialogAction(
            parent=self,
            name="measure",
            iconName="measure",
            label="Measure",
            group="Tools",
        )
        measureAction.setConnectCallBack(self._measure_dialog_new)
        actions.append(measureAction)

        self.__mainView = QubMosaicView(self, actions=actions)

        layout = qt.QHBoxLayout(self)
        layout.addWidget(self.__mainView)

        ####### SLOT #######
        self.defineSlot("getView", ())
Esempio n. 3
0
    def __init__(self, *args):
        BlissWidget.__init__(self, *args)

        actions = []

        self.__zoomList = QubZoomListAction(
            place="toolbar",
            zoomValList=[0.1, 0.25, 0.5, 1, 1.5, 2, 2.5, 3, 3.5, 4],
            show=1,
            group="zoom",
        )
        actions.append(self.__zoomList)

        self.__zoomFitOrFill = QubZoomAction(place="toolbar",
                                             keepROI=True,
                                             group="zoom")
        self.__zoomFitOrFill.setList(self.__zoomList)
        self.__zoomList.setActionZoomMode(self.__zoomFitOrFill)
        actions.append(self.__zoomFitOrFill)

        zoomAction = QubZoomRectangle(
            label="Zoom Crop",
            place="toolbar",
            show=1,
            group="zoom",
            activate_click_drag=True,
            drawingObjectLayer=2**31,
            unactiveActionWhenDub=True,
        )
        qt.QObject.connect(zoomAction, qt.PYSIGNAL("RectangleSelected"),
                           self.__rectangleZoomChanged)
        actions.append(zoomAction)

        ####### CHANGE FOREGROUND COLOR #######
        fcoloraction = QubForegroundColorAction(name="color", group="image")
        actions.append(fcoloraction)

        ####### MEASURE #######
        measureAction = QubOpenDialogAction(
            parent=self,
            name="measure",
            iconName="measure",
            label="Measure",
            group="Tools",
        )
        measureAction.setConnectCallBack(self._measure_dialog_new)
        actions.append(measureAction)

        self.__mainView = QubMosaicView(self, actions=actions)

        layout = qt.QHBoxLayout(self)
        layout.addWidget(self.__mainView)

        ####### SLOT #######
        self.defineSlot("getView", ())
Esempio n. 4
0
    def __addActions(self):
        updateAction = QubToolbarToggleButtonAction(
            name='update',
            iconName='update',
            initState=True,
            group='image',
            tooltip='Start/Stop image refresh')
        Qt.QObject.connect(updateAction, Qt.SIGNAL('StateChanged'),
                           self.__setRefresh)
        self.addAction(updateAction)
        # a bug in Qub forces to set the state after adding the action
        updateAction.setState(True)
        self._updateAction = updateAction

        if EdfFile:
            saveAction = QubOpenDialogAction(
                name='save edf',
                iconName='save',
                label='Save data',
                group='admin',
                tooltip='Save image data in EDF format')

            saveDialog = Qt.QFileDialog(self, 'Save image as EDF')
            saveDialog.setAcceptMode(Qt.QFileDialog.AcceptSave)
            saveDialog.setDefaultSuffix('edf')
            nameFilters = list(saveDialog.nameFilters())
            nameFilters.insert(0, 'EDF Files (*.edf)')
            saveDialog.setNameFilters(nameFilters)
            Qt.QObject.connect(saveDialog,
                               Qt.SIGNAL('fileSelected(const QString &)'),
                               self.__saveEDF)
            Qt.QObject.connect(saveDialog,
                               Qt.SIGNAL('filesSelected(const QStringList &)'),
                               self.__saveEDFs)
            saveAction.setDialog(saveDialog)
            self._saveDataAction = saveAction

            plug = _EDFFileSavePlug()
            self.addDataAction(saveAction, plug)
            self._saveEDFPlug = plug
Esempio n. 5
0
    def __init__(self, *args):
        BlissWidget.__init__(self, *args)

        self.camera = None
        self.update_disabled = False
        self.__cameraName = None

        self.__zoomList = [1.5, 2, 2.5, 3, 3.5, 4]
        self.__initZoom = 2

        self.__fixWidth = -1
        self.__fixHeight = -1

        self.__swapRgb = True
        ####### PRINT #######
        self.__printWidget = QubPrintPreview(self)
        self.__printWidget.resize(400, 500)
        printer = qt.QPrinter()
        printer.setOutputToFile(1)
        printer.setOutputFileName('/tmp/print_file.ps')
        self.__printWidget.setPrinter(printer)
        self.__printWidget.update()

        self.__beamAction = None
        self.__scaleAction = None
        self.__chosenActions = {}
        self.__wholeActions = []
        self.__measureDialog = None

        ####### PRINT ACTION #######
        printAction = QubPrintPreviewAction(
            name="print", group="admin", withVectorMenu=True)
        printAction.previewConnect(self.__printWidget)
        self.__wholeActions.append(printAction)

        ####### SAVE IMAGE #######
        self.__saveAction = QubSaveImageAction(parent=self, label='Save falcon image',
                                               name="save", group="admin")
        self.__saveAction.setConnectCallBack(self._save_dialog_new)
        self.__wholeActions.append(self.__saveAction)
        self.__defaultSavePath = '/tmp'

        ####### UPDATE #######
        update = QubToggleAction(name="update", group="image", initState=True)
        self.connect(update, qt.PYSIGNAL("StateChanged"), self.__cameraUpdate)
        self.__wholeActions.append(update)

        ####### Start Camera ######
        startCamera = QubToggleAction(
            name="startCamera",
            group="image",
            iconName='bright-cont',
            initState=True)
        self.connect(startCamera, qt.PYSIGNAL("StateChanged"), self.__cameraStart)
        self.__wholeActions.append(startCamera)

        ####### BRIGHTNESS/CONTRAST #######
        self.__brightcount = QubOpenDialogAction(name="bright-cont",
                                                 iconName='bright-cont', group="image")
        self.__brightcount.setDialog(QubBrightnessContrastDialog(self))
        self.__wholeActions.append(self.__brightcount)

        ###### Grid TOOL ######
        self.__gridToolAction = QubOpenDialogAction(parent=self, name='grid_tool',
                                                    iconName='rectangle', label='Grid tool',
                                                    group="Tools")  # place="contextmenu")
        self.__gridDialog = GridDialog(
            self, "Grid Dialog", flags=qt.Qt.WStyle_StaysOnTop)
        self.__gridToolAction.setConnectCallBack(self._grid_dialog_connect_hdlr)
        self.__wholeActions.append(self.__gridToolAction)

        self.__previous_pos_dict = {}
        self.__beamWidth = 0
        self.__beamHeight = 0

        ####### BEAM ACTION #######
        self.__beamAction = QubBeamAction(name="beam", group="Tools")
        self.__wholeActions.append(self.__beamAction)
        self.connect(self.__beamAction, qt.PYSIGNAL("BeamSelected"),
                     self.beamSelection)

        ####### SCALE #######
        self.__scaleAction = QubScaleAction(name='scale', group='Tools')
        self.__wholeActions.append(self.__scaleAction)
        self.__wholeActions.extend(self.__creatStdActions())

        ####### ACTION INFO #######
        actionInfo = QubInfoAction(name="actionInfo", group="image", place="statusbar")
        self.__wholeActions.append(actionInfo)

        ####### CHANGE FOREGROUND COLOR #######
        self.__fcoloraction = QubForegroundColorAction(name="color", group="image")
        self.__wholeActions.append(self.__fcoloraction)

        ####### MEASURE #######
        self.__measureAction = QubOpenDialogAction(parent=self, name='measure',
                                                   iconName='measure', label='Measure',
                                                   group="Tools")
        self.__measureAction.setConnectCallBack(self._measure_dialog_new)
        self.__wholeActions.append(self.__measureAction)

        # ###### POSITION TOOL ######
        # self.__posToolAction = QubOpenDialogAction(parent=self, name='pos_tool',
        #                                            iconName='circle', label='Position tool',
        #                                            group="Tools")
        # self.__posToolAction.setConnectCallBack(self._line_dialog_new)
        # self.__wholeActions.append(self.__posToolAction)

        ####### ZOOM LIST #######
        zoomActionList = QubZoomListAction(place="toolbar",
                                           initZoom=1, zoomValList=[0.1, 0.25, 0.5, 0.75, 1, 1.5, 2],
                                           show=1, group="zoom")
        self.__wholeActions.append(zoomActionList)

        ####### ZOOM Action #######
        self.__zoomFitOrFill = QubZoomAction(place="toolbar", group="zoom")
        self.__wholeActions.append(self.__zoomFitOrFill)

        ####### LINK ZOOM ACTION #######
        self.__zoomFitOrFill.setList(zoomActionList)
        zoomActionList.setActionZoomMode(self.__zoomFitOrFill)

        ####### ZOOM WINDOW #######
        self.__zoomAction = QubZoomRectangle(
            label='Zoom Crop', place="toolbar", show=1, group="zoom")
        self.connect(self.__zoomAction, qt.PYSIGNAL("Actif"), self.__hide_show_zoom)
        self.__wholeActions.append(self.__zoomAction)

        self.__splitter = qt.QSplitter(qt.Qt.Horizontal, self)
        self.__splitter.show()

        self.__mainVideo = QubPixmapDisplayView(self.__splitter)
        self.__mainVideo.show()
        self.__mainVideo.setScrollbarMode('Auto')
        self.__mainPlug = _MainVideoPlug(self.__mainVideo, self.__zoomAction)

        actions = self.__creatStdActions()

        ####### ZOOM LIST #######
        self.__zoomActionList = QubZoomListAction(place="toolbar", keepROI=True,
                                                  initZoom=self.__initZoom, zoomValList=self.__zoomList,
                                                  show=1, group="zoom")
        actions.insert(0, self.__zoomActionList)

        ####### ZOOM Action #######
        zoomFitOrFill = QubZoomAction(place="toolbar", keepROI=True, group="zoom")
        zoomFitOrFill.setList(self.__zoomActionList)
        self.__zoomActionList.setActionZoomMode(zoomFitOrFill)
        actions.append(zoomFitOrFill)

        self.__zoomVideo = QubPixmapDisplayView(self.__splitter, None, actions)
        self.__zoomVideo.hide()
        self.__zoomPlug = _ZoomPlug(self.__zoomVideo)
        self.__zoomPlug.zoom().setZoom(2, 2)
        self.__cbk = _rectangleZoom(self.__zoomAction, self.__zoomPlug)

        layout = qt.QHBoxLayout(self, 0, 0, "layout")
        layout.addWidget(self.__splitter)

        self.__image2Pixmap = QubImage2Pixmap()
        self.__image2Pixmap.plug(self.__mainPlug)
        self.__zoomPlug.setPoller(self.__image2Pixmap)

        self.__jpegDecompress = QubStdData2Image()
        self.__jpegDecompress.setSwapRGB(True)
        self.__jpeg2image = None

        ####### PROPERTY #######
        self.addProperty('camera', 'string', '')
        self.addProperty('zoom list', 'string', ','.join(
            [str(x) for x in self.__zoomList]))
        self.addProperty('init zoom', 'integer', self.__initZoom)
        self.addProperty('swap rgb', 'boolean', True)

        self.addProperty('fix : width', 'integer', -1)
        self.addProperty('fix : height', 'integer', -1)

        self.addProperty('action : print', 'boolean', True)
        self.addProperty('action : save image', 'boolean', True)
        self.addProperty('action : update', 'boolean', True)
        self.addProperty('action : startCamera', 'boolean', True)
        self.addProperty('action : brightness contrast', 'boolean', True)
        self.addProperty('action : beam', 'boolean', True)
        self.addProperty('action : scale', 'boolean', True)
        self.addProperty('action : change foreground color', 'boolean', True)
        self.addProperty('action : measure', 'boolean', True)
        self.addProperty('action : measure (place)', 'combo',
                         ('toolbar', 'contextmenu'), 'toolbar')
        self.addProperty('action : zoom window', 'boolean', True)
        self.addProperty('action : zoom fit or fill', 'boolean', True)
        self.addProperty('action : zoom list', 'boolean', True)
        self.addProperty('action : x,y coordinates', 'boolean', True)
        self.addProperty('action : default color', 'combo',
                         ('black', 'red', 'green'), 'black')

        self.addProperty('action : save image (place)', "combo",
                         ('toolbar', 'contextmenu'), 'toolbar')
        self.addProperty('action : save image (default path)', "string", '/tmp')
        self.addProperty('action : save image (show always configure)', "boolean", True)

        self.addProperty("diffractometer", "string", "")
        self.diffractometerHwobj = None

        ####### SIGNAL #######
        self.defineSignal("BeamPositionChanged", ())

        ####### SLOT #######
        self.defineSlot("changeBeamPosition", ())
        self.defineSlot("changePixelScale", ())
        self.defineSlot('getView', ())
        self.defineSlot('getImage', ())

        ####### LINK VIEW AND SUB VIEW #######
        mainView = self.__mainVideo.view()
        zoomView = self.__zoomVideo.view()
        mainView.addEventMgrLink(zoomView,
                                 mainView.canvas(), zoomView.canvas(),
                                 mainView.matrix(), zoomView.matrix())

        self.imageReceivedConnected = None
Esempio n. 6
0
class CameraBrick(BlissWidget):
    def __init__(self, *args):
        BlissWidget.__init__(self, *args)

        self.camera = None
        self.update_disabled = False
        self.__cameraName = None

        self.__zoomList = [1.5, 2, 2.5, 3, 3.5, 4]
        self.__initZoom = 2

        self.__fixWidth = -1
        self.__fixHeight = -1

        self.__swapRgb = True
        ####### PRINT #######
        self.__printWidget = QubPrintPreview(self)
        self.__printWidget.resize(400, 500)
        printer = qt.QPrinter()
        printer.setOutputToFile(1)
        printer.setOutputFileName('/tmp/print_file.ps')
        self.__printWidget.setPrinter(printer)
        self.__printWidget.update()

        self.__beamAction = None
        self.__scaleAction = None
        self.__chosenActions = {}
        self.__wholeActions = []
        self.__measureDialog = None

        ####### PRINT ACTION #######
        printAction = QubPrintPreviewAction(
            name="print", group="admin", withVectorMenu=True)
        printAction.previewConnect(self.__printWidget)
        self.__wholeActions.append(printAction)

        ####### SAVE IMAGE #######
        self.__saveAction = QubSaveImageAction(parent=self, label='Save falcon image',
                                               name="save", group="admin")
        self.__saveAction.setConnectCallBack(self._save_dialog_new)
        self.__wholeActions.append(self.__saveAction)
        self.__defaultSavePath = '/tmp'

        ####### UPDATE #######
        update = QubToggleAction(name="update", group="image", initState=True)
        self.connect(update, qt.PYSIGNAL("StateChanged"), self.__cameraUpdate)
        self.__wholeActions.append(update)

        ####### Start Camera ######
        startCamera = QubToggleAction(
            name="startCamera",
            group="image",
            iconName='bright-cont',
            initState=True)
        self.connect(startCamera, qt.PYSIGNAL("StateChanged"), self.__cameraStart)
        self.__wholeActions.append(startCamera)

        ####### BRIGHTNESS/CONTRAST #######
        self.__brightcount = QubOpenDialogAction(name="bright-cont",
                                                 iconName='bright-cont', group="image")
        self.__brightcount.setDialog(QubBrightnessContrastDialog(self))
        self.__wholeActions.append(self.__brightcount)

        ###### Grid TOOL ######
        self.__gridToolAction = QubOpenDialogAction(parent=self, name='grid_tool',
                                                    iconName='rectangle', label='Grid tool',
                                                    group="Tools")  # place="contextmenu")
        self.__gridDialog = GridDialog(
            self, "Grid Dialog", flags=qt.Qt.WStyle_StaysOnTop)
        self.__gridToolAction.setConnectCallBack(self._grid_dialog_connect_hdlr)
        self.__wholeActions.append(self.__gridToolAction)

        self.__previous_pos_dict = {}
        self.__beamWidth = 0
        self.__beamHeight = 0

        ####### BEAM ACTION #######
        self.__beamAction = QubBeamAction(name="beam", group="Tools")
        self.__wholeActions.append(self.__beamAction)
        self.connect(self.__beamAction, qt.PYSIGNAL("BeamSelected"),
                     self.beamSelection)

        ####### SCALE #######
        self.__scaleAction = QubScaleAction(name='scale', group='Tools')
        self.__wholeActions.append(self.__scaleAction)
        self.__wholeActions.extend(self.__creatStdActions())

        ####### ACTION INFO #######
        actionInfo = QubInfoAction(name="actionInfo", group="image", place="statusbar")
        self.__wholeActions.append(actionInfo)

        ####### CHANGE FOREGROUND COLOR #######
        self.__fcoloraction = QubForegroundColorAction(name="color", group="image")
        self.__wholeActions.append(self.__fcoloraction)

        ####### MEASURE #######
        self.__measureAction = QubOpenDialogAction(parent=self, name='measure',
                                                   iconName='measure', label='Measure',
                                                   group="Tools")
        self.__measureAction.setConnectCallBack(self._measure_dialog_new)
        self.__wholeActions.append(self.__measureAction)

        # ###### POSITION TOOL ######
        # self.__posToolAction = QubOpenDialogAction(parent=self, name='pos_tool',
        #                                            iconName='circle', label='Position tool',
        #                                            group="Tools")
        # self.__posToolAction.setConnectCallBack(self._line_dialog_new)
        # self.__wholeActions.append(self.__posToolAction)

        ####### ZOOM LIST #######
        zoomActionList = QubZoomListAction(place="toolbar",
                                           initZoom=1, zoomValList=[0.1, 0.25, 0.5, 0.75, 1, 1.5, 2],
                                           show=1, group="zoom")
        self.__wholeActions.append(zoomActionList)

        ####### ZOOM Action #######
        self.__zoomFitOrFill = QubZoomAction(place="toolbar", group="zoom")
        self.__wholeActions.append(self.__zoomFitOrFill)

        ####### LINK ZOOM ACTION #######
        self.__zoomFitOrFill.setList(zoomActionList)
        zoomActionList.setActionZoomMode(self.__zoomFitOrFill)

        ####### ZOOM WINDOW #######
        self.__zoomAction = QubZoomRectangle(
            label='Zoom Crop', place="toolbar", show=1, group="zoom")
        self.connect(self.__zoomAction, qt.PYSIGNAL("Actif"), self.__hide_show_zoom)
        self.__wholeActions.append(self.__zoomAction)

        self.__splitter = qt.QSplitter(qt.Qt.Horizontal, self)
        self.__splitter.show()

        self.__mainVideo = QubPixmapDisplayView(self.__splitter)
        self.__mainVideo.show()
        self.__mainVideo.setScrollbarMode('Auto')
        self.__mainPlug = _MainVideoPlug(self.__mainVideo, self.__zoomAction)

        actions = self.__creatStdActions()

        ####### ZOOM LIST #######
        self.__zoomActionList = QubZoomListAction(place="toolbar", keepROI=True,
                                                  initZoom=self.__initZoom, zoomValList=self.__zoomList,
                                                  show=1, group="zoom")
        actions.insert(0, self.__zoomActionList)

        ####### ZOOM Action #######
        zoomFitOrFill = QubZoomAction(place="toolbar", keepROI=True, group="zoom")
        zoomFitOrFill.setList(self.__zoomActionList)
        self.__zoomActionList.setActionZoomMode(zoomFitOrFill)
        actions.append(zoomFitOrFill)

        self.__zoomVideo = QubPixmapDisplayView(self.__splitter, None, actions)
        self.__zoomVideo.hide()
        self.__zoomPlug = _ZoomPlug(self.__zoomVideo)
        self.__zoomPlug.zoom().setZoom(2, 2)
        self.__cbk = _rectangleZoom(self.__zoomAction, self.__zoomPlug)

        layout = qt.QHBoxLayout(self, 0, 0, "layout")
        layout.addWidget(self.__splitter)

        self.__image2Pixmap = QubImage2Pixmap()
        self.__image2Pixmap.plug(self.__mainPlug)
        self.__zoomPlug.setPoller(self.__image2Pixmap)

        self.__jpegDecompress = QubStdData2Image()
        self.__jpegDecompress.setSwapRGB(True)
        self.__jpeg2image = None

        ####### PROPERTY #######
        self.addProperty('camera', 'string', '')
        self.addProperty('zoom list', 'string', ','.join(
            [str(x) for x in self.__zoomList]))
        self.addProperty('init zoom', 'integer', self.__initZoom)
        self.addProperty('swap rgb', 'boolean', True)

        self.addProperty('fix : width', 'integer', -1)
        self.addProperty('fix : height', 'integer', -1)

        self.addProperty('action : print', 'boolean', True)
        self.addProperty('action : save image', 'boolean', True)
        self.addProperty('action : update', 'boolean', True)
        self.addProperty('action : startCamera', 'boolean', True)
        self.addProperty('action : brightness contrast', 'boolean', True)
        self.addProperty('action : beam', 'boolean', True)
        self.addProperty('action : scale', 'boolean', True)
        self.addProperty('action : change foreground color', 'boolean', True)
        self.addProperty('action : measure', 'boolean', True)
        self.addProperty('action : measure (place)', 'combo',
                         ('toolbar', 'contextmenu'), 'toolbar')
        self.addProperty('action : zoom window', 'boolean', True)
        self.addProperty('action : zoom fit or fill', 'boolean', True)
        self.addProperty('action : zoom list', 'boolean', True)
        self.addProperty('action : x,y coordinates', 'boolean', True)
        self.addProperty('action : default color', 'combo',
                         ('black', 'red', 'green'), 'black')

        self.addProperty('action : save image (place)', "combo",
                         ('toolbar', 'contextmenu'), 'toolbar')
        self.addProperty('action : save image (default path)', "string", '/tmp')
        self.addProperty('action : save image (show always configure)', "boolean", True)

        self.addProperty("diffractometer", "string", "")
        self.diffractometerHwobj = None

        ####### SIGNAL #######
        self.defineSignal("BeamPositionChanged", ())

        ####### SLOT #######
        self.defineSlot("changeBeamPosition", ())
        self.defineSlot("changePixelScale", ())
        self.defineSlot('getView', ())
        self.defineSlot('getImage', ())

        ####### LINK VIEW AND SUB VIEW #######
        mainView = self.__mainVideo.view()
        zoomView = self.__zoomVideo.view()
        mainView.addEventMgrLink(zoomView,
                                 mainView.canvas(), zoomView.canvas(),
                                 mainView.matrix(), zoomView.matrix())

        self.imageReceivedConnected = None

    def safeConnect(self):
        if not self.imageReceivedConnected and self.camera is not None:
            #print "CONNECTING 0"
            self.camera.connect('imageReceived', self.__imageReceived)
            # self.connect(self.camera,qt.PYSIGNAL('imageReceived'),self.__imageReceived)
            self.imageReceivedConnected = False

    def safeDisconnect(self):
        if self.imageReceivedConnected and self.camera is not None:
            #print "DISCONNECTING 3"
            self.camera.disconnect('imageReceived', self.__imageReceived)
            # self.disconnect(self.camera,qt.PYSIGNAL('imageReceived'),self.__imageReceived)
            self.imageReceivedConnected = False

    def disable_update(self):
        # self.__imageReceived(DISABLED_JPEG,DISABLED_WIDTH,DISABLED_HEIGHT)
        self.update_disabled = True

    def enable_update(self):
        self.update_disabled = False

    def __imageReceived(self, image, width, height, force_update=False):
        if self.update_disabled:
            return

        if not force_update:
            if not self.isVisible():
                return

        if isinstance(image, qt.QImage):
            if self.__swapRgb:
                image = image.swapRGB()
            self.__jpeg2image.setImage(image, image)
        else:
            self.__jpegDecompress.putData(image, width, height)

    def propertyChanged(self, property, oldValue, newValue):
        if property == 'camera':
            # if self.camera is not None and not self.isRunning():
            #    self.safeDisconnect()
            self.safeDisconnect()
            self.camera = self.getHardwareObject(newValue)
            self.__hwoName = newValue

            if self.camera is not None:
                try:
                    self.__tangoName = self.camera.tangoname
                except BaseException:
                    self.__tangoName = ""

                try:
                    camera_role = self.camera.getDeviceByRole('camera')
                    if camera_role is not None:
                        self.camera = camera_role
                except AttributeError:
                    pass

                self.__brightcount.setCamera(self.camera)

        elif property == 'zoom list':
            zoomList = [float(x) for x in newValue.split(',')]
            if zoomList != self.__zoomList:
                self.__zoomList = zoomList
                self.__zoomActionList.changeZoomList(zoomList, self.__initZoom)
        elif property == 'init zoom':
            if newValue != self.__initZoom:
                self.__initZoom = newValue
                self.__zoomActionList.changeZoomList(self.__zoomList, newValue)
        elif property == 'swap rgb':
            self.__jpegDecompress.setSwapRGB(newValue)
            self.__swapRgb = newValue
        elif property == 'action : print':
            self.__chosenActions['print'] = newValue
        elif property == 'action : save image':
            self.__chosenActions['save'] = newValue
        elif property == 'action : update':
            self.__chosenActions['update'] = newValue
        elif property == 'action : startCamera':
            self.__chosenActions['startCamera'] = newValue
        elif property == 'action : brightness contrast':
            self.__chosenActions['bright-cont'] = newValue
        elif property == 'action : beam':
            self.__chosenActions['beam'] = newValue
        elif property == 'action : scale':
            self.__chosenActions['scale'] = newValue
        elif property == 'action : change foreground color':
            self.__chosenActions['color'] = newValue
        elif property == 'action : measure':
            self.__chosenActions['measure'] = newValue
        elif property == 'action : zoom window':
            self.__chosenActions['zoomrect'] = newValue
        elif property == 'action : zoom fit or fill':
            self.__chosenActions['Zoom tools'] = newValue
        elif property == 'action : zoom list':
            self.__chosenActions['zoomlist'] = newValue
        elif property == 'action : x,y coordinates':
            self.__chosenActions['position'] = newValue
        elif property == 'action : default color':
            self.__chosenDefaultColor = newValue
        elif property == 'action : save image (place)':
            self.__saveAction.setPlace(newValue, True)
        elif property == 'action : measure (place)':
            self.__measureAction.setPlace(newValue, True)
        elif property == 'action : save image (default path)':
            self.__defaultSavePath = newValue
        elif property == 'action : save image (show always configure)':
            self.__saveAction.setConfigureOnClick(newValue)

        elif property == 'fix : width':
            self.__fixWidth = newValue
        elif property == 'fix : height':
            self.__fixHeight = newValue
        elif property == "diffractometer":
            self.diffractometerHwobj = self.getHardwareObject(newValue)
            if self.diffractometerHwobj is not None:
                self.__previous_pos_dict = self.diffractometerHwobj.getPositions()
                self.diffractometerHwobj.connect("minidiffStateChanged",
                                                 self.diffractometerChanged)
                if self.diffractometerHwobj.zoomMotor is not None:
                    zoom = self.diffractometerHwobj.zoomMotor.getPosition()
                    xSize, ySize = self.diffractometerHwobj.getCalibrationData(zoom)
                    #xSize,ySize = self.__scaleAction.xPixelSize(), self.__scaleAction.yPixelSize()
                    self.diffractometerHwobj.getBeamInfo(self.__getBeamInfo)
                    beam_pos_x = self.diffractometerHwobj.getBeamPosX()
                    beam_pos_y = self.diffractometerHwobj.getBeamPosY()

                    self.__gridDialog.set_x_pixel_size(xSize)
                    self.__gridDialog.set_y_pixel_size(ySize)
                    self.__gridDialog.set_beam_position(beam_pos_x, beam_pos_y,
                                                        self.__beamWidth, self.__beamHeight)

    def run(self):
        chosenActions = []

        for action in self.__wholeActions:
            if self.__chosenActions.get(action.name(), True):
                chosenActions.append(action)

        self.__mainVideo.addAction(chosenActions)

        if not self.__chosenActions.get('update', False):
            self.__cameraUpdate(True)

        if not self.__chosenActions.get('startCamera', False):
            self.__cameraStart(True)

        if self.__fixWidth != -1 and self.__fixHeight != -1:
            self.__mainVideo.view().setFixedSize(self.__fixWidth + 4, self.__fixHeight + 4)

        self.__startIdle = qt.QTimer(self)
        qt.QObject.connect(self.__startIdle, qt.SIGNAL('timeout()'), self.__idleRun)
        self.__startIdle.start(0)

    def __idleRun(self):
        self.__startIdle.stop()
        self.__fcoloraction.setColor(self.__chosenDefaultColor)
        self.__mainVideo.view().setForegroundColor(qt.QColor(self.__chosenDefaultColor))

    def stop(self):
        self.safeDisconnect()

    def beamSelection(self, x, y):
        self.emit(qt.PYSIGNAL("BeamPositionChanged"), (x, y))
        self.changeBeamPosition(x, y)

    def changeBeamPosition(self, x, y, beam_width=None, beam_height=None):
        self.__beamAction.setBeamPosition(x, y)
        try:
            if not beam_width:
                beam_width = self.__beamWidth

            if not beam_height:
                beam_height = self.__beamHeight

            self.__gridDialog.set_beam_position(x, y, beam_width, beam_height)
        except BaseException:
            pass

    def changePixelScale(self, sizex, sizey):
        self.__scaleAction.setXPixelSize(sizex)
        self.__scaleAction.setYPixelSize(sizey)
        if self.__measureDialog:
            self.__measureDialog.setXPixelSize(sizex)
            self.__measureDialog.setYPixelSize(sizey)

        self.__gridDialog.set_x_pixel_size(sizex)
        self.__gridDialog.set_y_pixel_size(sizey)

    def getView(self, key):
        try:
            key['drawing'] = self.__mainVideo.view()
            key['view'] = self.__mainVideo
            key['hwname'] = self.__hwoName
            key['dsname'] = self.__tangoName
        except BaseException:
            pass

    def getImage(self, key):
        if self.__jpeg2image is not None:
            try:
                key['image'] = self.__jpeg2image.getLastImage()
            except BaseException:
                pass

    def __creatStdActions(self):
        actions = []
        ####### MOUSE POSITION #######
        posaction = QubPositionAction(name="position", group="image", place="statusbar")
        actions.append(posaction)

        return actions

    def __cameraUpdate(self, onoff):
        if onoff:
            self.__jpeg2image = _Jpeg2ImagePlug(self.__image2Pixmap)
            self.__jpegDecompress.plug(self.__jpeg2image)
            if self.camera is not None:
                imageInfo = self.camera.imageType()
            else:
                imageInfo = None
            imType = self.__jpegDecompress.STANDARD
            if imageInfo:
                if imageInfo.type() == 'bayer':
                    imType = self.__jpegDecompress.BAYER_RG
                elif imageInfo.type() == 'raw':
                    imType = self.__jpegDecompress.RAW
            self.__jpegDecompress.setImageType(imType)
            self.safeConnect()
        else:
            self.safeDisconnect()
            self.__jpeg2image.setEnd()

    def __cameraStart(self, onoff):
        if onoff:
            print "********************************************************Set the camera ON"
            if self.camera is not None:
                self.camera.setLive(True)
            else:
                print "camera is none"
        else:
            print "****************************************************Set the camera off"
            if self.camera is not None:
                print "camera :"
                print self.camera
                self.camera.setLive(False)
            else:
                print "camera is none"

    def __hide_show_zoom(self, actif):
        if actif:
            if self.__chosenActions['Zoom tools']:
                self.__zoomFitOrFill.setState(True)
            else:
                self.__mainVideo.setScrollbarMode('Fit2Screen')
            self.__zoomPlug.start()
            self.__zoomVideo.show()
            splitterWidth = self.__splitter.width()
            self.__splitter.setSizes(
                [splitterWidth / 3, splitterWidth - splitterWidth / 3])
        else:
            if self.__chosenActions['Zoom tools']:
                self.__zoomFitOrFill.setState(False)
            else:
                self.__mainVideo.setScrollbarMode('Auto')
            self.__zoomPlug.stop()
            self.__zoomVideo.hide()

    def _measure_dialog_new(self, openDialogAction, aQubImage):
        try:
            self.__measureDialog = QubMeasureListDialog(self,
                                                        canvas=aQubImage.canvas(),
                                                        matrix=aQubImage.matrix(),
                                                        eventMgr=aQubImage)
            xSize, ySize = self.__scaleAction.xPixelSize(), self.__scaleAction.yPixelSize()
            self.__measureDialog.setXPixelSize(xSize)
            self.__measureDialog.setYPixelSize(ySize)
            self.__measureDialog.connect(aQubImage, qt.PYSIGNAL("ForegroundColorChanged"),
                                         self.__measureDialog.setDefaultColor)
            openDialogAction.setDialog(self.__measureDialog)
        except BaseException:
            import traceback
            traceback.print_exc()

    def _grid_dialog_connect_hdlr(self, openDialogAction, aQubImage):
        try:
            self.__gridDialog.set_qub_event_mgr(aQubImage)
            openDialogAction.setDialog(self.__gridDialog)
        except BaseException:
            import traceback
            traceback.print_exc()

    def __getBeamInfo(self, ret):
        self.__beamWidth = float(ret["size_x"])
        self.__beamHeight = float(ret["size_y"])
        #self.__beamShape = ret["shape"]

    def _save_dialog_new(self, openDialogAction, aQubImage):
        saveDialog = QubSaveImageDialog(
            self, matrix=aQubImage.matrix(), canvas=aQubImage.canvas())
        saveDialog.setImage2Pixmap(self.__image2Pixmap)
        saveDialog.setSavePath(self.__defaultSavePath)
        openDialogAction.setDialog(saveDialog)

    def instanceMirrorChanged(self, mirror):
        #print "INSTANCEMIRRORCHANGED",mirror,BlissWidget.isInstanceMirrorAllow()
        return
        if BlissWidget.isInstanceModeSlave():
            if BlissWidget.isInstanceMirrorAllow():
                self.safeConnect()
            else:
                self.safeDisconnect()

    def diffractometerChanged(self, *args):
        """
        Handles diffractometer change events, connected to the signal
        minidiffStateChanged of the diffractometer hardware object.
        """

        if self.diffractometerHwobj.isReady():
            pos_dict = self.diffractometerHwobj.getPositions()
            p1 = self.diffractometerHwobj.motor_positions_to_screen(
                self.__previous_pos_dict)
            p2 = (self.diffractometerHwobj.getBeamPosX(),
                  self.diffractometerHwobj.getBeamPosY())

            dx = p2[0] - p1[0]
            dy = p2[1] - p1[1]

            if dy != 0:
                self.__gridDialog.move_grid_ver(-dy)

            if dx != 0:
                self.__gridDialog.move_grid_hor(-dx)

            #print dx, dy

            self.__previous_pos_dict = pos_dict
Esempio n. 7
0
    def configureAction(self):
        """
        Move To action
        """
        if self.movetoMode is not None:
            if self.movetoAction is None:
                """
                create action
                """
                self.movetoAction = QubSelectPointAction(
                    name='Move to Beam',
                    place=self.movetoMode,
                    actionInfo='Move to Beam',
                    group='Tools')
                self.connect(self.movetoAction, qt.PYSIGNAL("StateChanged"),
                             self.movetoStateChanged)
                self.connect(self.movetoAction, qt.PYSIGNAL("PointSelected"),
                             self.pointSelected)

                if self.view is not None:
                    self.view.addAction(self.movetoAction)
                    self.oldMoveToActionColor = self.movetoAction.paletteBackgroundColor(
                    )

        else:
            if self.movetoAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.movetoAction)
                """
                del action from view
                """
                self.disconnect(self.movetoAction,
                                qt.PYSIGNAL("PointSelected"),
                                self.pointSelected)
                self.movetoAction = None
        """
        Limits action
        """
        if self.limitsMode is not None:
            if self.limitsAction is None:
                """
                create action
                """
                self.limitsAction = QubRulerAction(name='Motor Limits',
                                                   place=self.limitsMode,
                                                   group='Tools')

                if self.view is not None:
                    self.view.addAction(self.limitsAction)
            """
            configure action
            """
            if self.horMotHwo is not None:
                mne = self.horMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.HORIZONTAL, 0, mne)

            if self.verMotHwo is not None:
                mne = self.verMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.VERTICAL, 0, mne)

        else:
            if self.limitsAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.limitsAction)
                """
                del action from view
                """
                self.limitsAction = None

        if self.measureMode is not None:
            if self.measureAction is None:
                self.measureAction = QubOpenDialogAction(
                    parent=self,
                    name='measure',
                    iconName='measure',
                    label='Measure',
                    group='Tools',
                    place=self.measureMode)
                self.measureAction.setConnectCallBack(self._measure_dialog_new)
                logging.getLogger().info("setting measure mode")
                if self.view is not None:
                    logging.getLogger().info("adding action")
                    self.view.addAction(self.measureAction)
        else:
            if self.measureAction is not None:
                if self.view is not None:
                    self.view.delAction(self.measureAction)
                self.measureAction = None

        if self.movetoMode is not None:
            if self.__movetoActionMosaic is not None:
                self.__mosaicView.delAction(self.__movetoActionMosaic)
                self.disconnect(self.__movetoActionMosaic,
                                qt.PYSIGNAL("PointSelected"),
                                self.__mosaicPointSelected)
                self.diconnect(self.__movetoActionMosaic,
                               qt.PYSIGNAL("StateChanged"),
                               self.__movetoMosaicStateChanged)
                self.__movetoActionMosaic = None

            if self.__mosaicView is not None:
                self.__movetoActionMosaic = QubSelectPointAction(
                    name='Move to Beam',
                    place=self.movetoMode,
                    actionInfo='Move to Beam',
                    mosaicMode=True,
                    residualMode=True,
                    group='Tools')
                self.connect(self.__movetoActionMosaic,
                             qt.PYSIGNAL("PointSelected"),
                             self.__mosaicPointSelected)
                self.connect(self.__movetoActionMosaic,
                             qt.PYSIGNAL("StateChanged"),
                             self.__movetoMosaicStateChanged)

                self.__mosaicView.addAction(self.__movetoActionMosaic)
                self.__oldMoveToMosaicActionColor = self.__movetoActionMosaic.paletteBackgroundColor(
                )

        if self.focusMode is not None:
            if self.focusAction is None:
                self.focusAction = QubToggleAction(label='Autofocus',
                                                   name='autofocus',
                                                   place=self.focusMode,
                                                   group='Tools',
                                                   autoConnect=True)
                qt.QObject.connect(self.focusAction,
                                   qt.PYSIGNAL('StateChanged'),
                                   self.showFocusGrab)

            if self.view and self.drawing:
                self.focusDrawingRectangle, _ = QubAddDrawing(
                    self.drawing, QubPointDrawingMgr,
                    QubCanvasHomotheticRectangle)
                self.focusDrawingRectangle.setDrawingEvent(
                    QubMoveNPressed1Point)
                self.focusDrawingRectangle.setKeyPressedCallBack(
                    self.focusRawKeyPressed)

                qt.QObject.connect(self.drawing,
                                   qt.PYSIGNAL("ForegroundColorChanged"),
                                   self.focusDrawingRectangle.setColor)
                self.focusDrawingRectangle.setEndDrawCallBack(
                    self.setFocusPointSelected)
                self.focusRectangleSize = 12
                self.focusDrawingRectangle.setWidthNHeight(
                    self.focusRectangleSize, self.focusRectangleSize)
                self.view.addAction(self.focusAction)

        elif self.view is not None:
            self.view.delAction(self.focusAction)
            self.focusDrawingRectangle = None
Esempio n. 8
0
class CameraMotorToolsBrick(BlissWidget):
    def __init__(self, parent, name):
        BlissWidget.__init__(self, parent, name)
        """
        variables
        """
        self.YBeam = None
        self.ZBeam = None
        self.YSize = None
        self.ZSize = None

        self.view = None
        self.firstCall = True
        self.motorArrived = 0

        self.__mosaicView, self.__mosaicDraw = None, None
        """
        property
        """
        self.addProperty("Hor. Motor", "string", "")
        self.horMotHwo = None

        self.addProperty("Vert. Motor", "string", "")
        self.verMotHwo = None

        #self.addProperty("Focus Motor","string","")
        #self.focusMotHwo = None

        #self.addProperty("Focus min step","float",0.05)
        #self.focusMinStep = None

        self.addProperty("Move To", "combo", ("none", "toolbar", "popup"),
                         "none")
        self.movetoMode = None
        self.movetoAction = None

        self.addProperty("Move to color when activate", "combo",
                         ("none", "blue", "green", "yellow", "red", "orange"),
                         "none")
        self.movetoColorWhenActivate = None

        self.addProperty("Limits", "combo", ("none", "toolbar", "popup"),
                         "none")
        self.limitsMode = None
        self.limitsAction = None

        self.addProperty("Measure", "combo", ("none", "toolbar", "popup"),
                         "none")
        self.measureMode = None
        self.measureAction = None

        self.addProperty("Focus", "combo", ("none", "toolbar", "popup"),
                         "none")
        self.focusMode = None
        self.focusAction = None
        self.focusState = None
        self.focusDrawingRectangle = None
        self.focusPointSelected = None

        self.__movetoActionMosaic = None
        """
        Signal
        """
        self.defineSignal('getView', ())
        self.defineSignal('getMosaicView', ())
        self.defineSignal('getCalibration', ())
        self.defineSignal('getBeamPosition', ())
        self.defineSignal('getImage', ())
        self.defineSignal('moveDone', ())
        self.defineSignal('mosaicImageSelected', ())
        """
        Slot
        """
        self.defineSlot('beamPositionChanged', ())
        self.defineSlot('pixelCalibrationChanged', ())
        self.defineSlot('setMoveToMode', ())
        self.defineSlot('setLimitsDisplay', ())
        self.defineSlot('setMoveToState', ())
        """
        widgets - NO APPEARANCE
        """
        self.setFixedSize(0, 0)

    def propertyChanged(self, prop, oldValue, newValue):
        if prop == "Hor. Motor":
            if self.horMotHwo is not None:
                self.disconnect(self.horMotHwo, qt.PYSIGNAL('positionChanged'),
                                self.setHorizontalPosition)
                self.disconnect(self.horMotHwo, qt.PYSIGNAL("limitsChanged"),
                                self.setHorizontalLimits)

            self.horMotHwo = self.getHardwareObject(newValue)

            if self.horMotHwo is not None:
                self.connect(self.horMotHwo, qt.PYSIGNAL('positionChanged'),
                             self.setHorizontalPosition)
                self.connect(self.horMotHwo, qt.PYSIGNAL("limitsChanged"),
                             self.setHorizontalLimits)

        elif prop == "Vert. Motor":
            if self.verMotHwo is not None:
                self.disconnect(self.verMotHwo, qt.PYSIGNAL('positionChanged'),
                                self.setVerticalPosition)
                self.disconnect(self.verMotHwo, qt.PYSIGNAL("limitsChanged"),
                                self.setVerticalLimits)

            self.verMotHwo = self.getHardwareObject(newValue)

            if self.verMotHwo is not None:
                self.connect(self.verMotHwo, qt.PYSIGNAL('positionChanged'),
                             self.setVerticalPosition)
                self.connect(self.verMotHwo, qt.PYSIGNAL("limitsChanged"),
                             self.setVerticalLimits)
        #elif prop == "Focus Motor":
        #    if self.focusMotHwo is not None:
        #        self.disconnect(self.focusMotHwo,qt.PYSIGNAL('positionChanged'),
        #                        self.setFocusPosition)
        #        self.disconnect(self.focusMotHwo,qt.PYSIGNAL('limitsChanged'),
        #                        self.setFocusLimits)

        #    self.focusMotHwo = self.getHardwareObject(newValue)
        #    self.focusState = FocusState(self,self.focusMotHwo)

        #    if self.focusMotHwo is not None:
        #        self.connect(self.focusMotHwo,qt.PYSIGNAL('positionChanged'),
        #                     self.setFocusPosition)
        #        self.connect(self.focusMotHwo,qt.PYSIGNAL('limitsChanged'),
        #                     self.setFocusLimits)
        #        self.connect(self.focusMotHwo, qt.PYSIGNAL("moveDone"),
        #                     self.focusMoveFinished)
        elif prop == "Move To":
            self.movetoMode = _enumTranslate[newValue]
        elif prop == "Limits":
            self.limitsMode = _enumTranslate[newValue]
        elif prop == "Focus":
            self.focusMode = _enumTranslate[newValue]
        elif prop == "Measure":
            self.measureMode = _enumTranslate[newValue]
        elif prop == "Focus min step":
            self.focusMinStep = newValue
        elif prop == "Move to color when activate":
            if newValue == "none":
                self.movetoColorWhenActivate = None
            else:
                self.movetoColorWhenActivate = newValue
        if not self.firstCall:
            self.configureAction()

    def run(self):
        """
        get view
        """
        view = {}
        self.emit(qt.PYSIGNAL("getView"), (view, ))
        try:
            self.drawing = view["drawing"]
            self.view = view["view"]
        except:
            print "No View"
        """
        get calibration
        """
        calib = {}
        self.emit(qt.PYSIGNAL("getCalibration"), (calib, ))
        try:
            # in all this brick we work with pixel calibration in mm
            self.YSize = calib["ycalib"]
            self.ZSize = calib["zcalib"]
            if calib["ycalib"] is not None and calib["zcalib"] is not None:
                self.YSize = self.YSize * 1000
                self.ZSize = self.ZSize * 1000
        except:
            print "No Calibration"
        """
        get beam position
        """
        position = {}
        self.emit(qt.PYSIGNAL("getBeamPosition"), (position, ))
        try:
            self.YBeam = position["ybeam"]
            self.ZBeam = position["zbeam"]
        except:
            print "No Beam Position"
        """
        get mosaic view
        """
        mosaicView = {}
        self.emit(qt.PYSIGNAL('getMosaicView'), (mosaicView, ))
        try:
            self.__mosaicView = mosaicView['view']
            self.__mosaicDraw = mosaicView['drawing']
        except KeyError:
            self.__mosaicView, self.__mosaicDraw = None, None

        self.configureAction()

        self.firstCall = False

    def configureAction(self):
        """
        Move To action
        """
        if self.movetoMode is not None:
            if self.movetoAction is None:
                """
                create action
                """
                self.movetoAction = QubSelectPointAction(
                    name='Move to Beam',
                    place=self.movetoMode,
                    actionInfo='Move to Beam',
                    group='Tools')
                self.connect(self.movetoAction, qt.PYSIGNAL("StateChanged"),
                             self.movetoStateChanged)
                self.connect(self.movetoAction, qt.PYSIGNAL("PointSelected"),
                             self.pointSelected)

                if self.view is not None:
                    self.view.addAction(self.movetoAction)
                    self.oldMoveToActionColor = self.movetoAction.paletteBackgroundColor(
                    )

        else:
            if self.movetoAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.movetoAction)
                """
                del action from view
                """
                self.disconnect(self.movetoAction,
                                qt.PYSIGNAL("PointSelected"),
                                self.pointSelected)
                self.movetoAction = None
        """
        Limits action
        """
        if self.limitsMode is not None:
            if self.limitsAction is None:
                """
                create action
                """
                self.limitsAction = QubRulerAction(name='Motor Limits',
                                                   place=self.limitsMode,
                                                   group='Tools')

                if self.view is not None:
                    self.view.addAction(self.limitsAction)
            """
            configure action
            """
            if self.horMotHwo is not None:
                mne = self.horMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.HORIZONTAL, 0, mne)

            if self.verMotHwo is not None:
                mne = self.verMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.VERTICAL, 0, mne)

        else:
            if self.limitsAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.limitsAction)
                """
                del action from view
                """
                self.limitsAction = None

        if self.measureMode is not None:
            if self.measureAction is None:
                self.measureAction = QubOpenDialogAction(
                    parent=self,
                    name='measure',
                    iconName='measure',
                    label='Measure',
                    group='Tools',
                    place=self.measureMode)
                self.measureAction.setConnectCallBack(self._measure_dialog_new)
                logging.getLogger().info("setting measure mode")
                if self.view is not None:
                    logging.getLogger().info("adding action")
                    self.view.addAction(self.measureAction)
        else:
            if self.measureAction is not None:
                if self.view is not None:
                    self.view.delAction(self.measureAction)
                self.measureAction = None

        if self.movetoMode is not None:
            if self.__movetoActionMosaic is not None:
                self.__mosaicView.delAction(self.__movetoActionMosaic)
                self.disconnect(self.__movetoActionMosaic,
                                qt.PYSIGNAL("PointSelected"),
                                self.__mosaicPointSelected)
                self.diconnect(self.__movetoActionMosaic,
                               qt.PYSIGNAL("StateChanged"),
                               self.__movetoMosaicStateChanged)
                self.__movetoActionMosaic = None

            if self.__mosaicView is not None:
                self.__movetoActionMosaic = QubSelectPointAction(
                    name='Move to Beam',
                    place=self.movetoMode,
                    actionInfo='Move to Beam',
                    mosaicMode=True,
                    residualMode=True,
                    group='Tools')
                self.connect(self.__movetoActionMosaic,
                             qt.PYSIGNAL("PointSelected"),
                             self.__mosaicPointSelected)
                self.connect(self.__movetoActionMosaic,
                             qt.PYSIGNAL("StateChanged"),
                             self.__movetoMosaicStateChanged)

                self.__mosaicView.addAction(self.__movetoActionMosaic)
                self.__oldMoveToMosaicActionColor = self.__movetoActionMosaic.paletteBackgroundColor(
                )

        if self.focusMode is not None:
            if self.focusAction is None:
                self.focusAction = QubToggleAction(label='Autofocus',
                                                   name='autofocus',
                                                   place=self.focusMode,
                                                   group='Tools',
                                                   autoConnect=True)
                qt.QObject.connect(self.focusAction,
                                   qt.PYSIGNAL('StateChanged'),
                                   self.showFocusGrab)

            if self.view and self.drawing:
                self.focusDrawingRectangle, _ = QubAddDrawing(
                    self.drawing, QubPointDrawingMgr,
                    QubCanvasHomotheticRectangle)
                self.focusDrawingRectangle.setDrawingEvent(
                    QubMoveNPressed1Point)
                self.focusDrawingRectangle.setKeyPressedCallBack(
                    self.focusRawKeyPressed)

                qt.QObject.connect(self.drawing,
                                   qt.PYSIGNAL("ForegroundColorChanged"),
                                   self.focusDrawingRectangle.setColor)
                self.focusDrawingRectangle.setEndDrawCallBack(
                    self.setFocusPointSelected)
                self.focusRectangleSize = 12
                self.focusDrawingRectangle.setWidthNHeight(
                    self.focusRectangleSize, self.focusRectangleSize)
                self.view.addAction(self.focusAction)

        elif self.view is not None:
            self.view.delAction(self.focusAction)
            self.focusDrawingRectangle = None

    def _measure_dialog_new(self, openDialogAction, aQubImage):
        if self.YSize is not None and self.ZSize is not None:
            self.__measureDialog = QubMeasureListDialog(
                self,
                canvas=aQubImage.canvas(),
                matrix=aQubImage.matrix(),
                eventMgr=aQubImage)
            self.__measureDialog.setXPixelSize(self.YSize / 1000.0)
            self.__measureDialog.setYPixelSize(self.ZSize / 1000.0)
            self.__measureDialog.connect(aQubImage,
                                         qt.PYSIGNAL("ForegroundColorChanged"),
                                         self.__measureDialog.setDefaultColor)
            openDialogAction.setDialog(self.__measureDialog)

    def setHorizontalPosition(self, newPosition):
        if self.limitsAction is not None:
            self.limitsAction.setCursorPosition(QubRulerAction.HORIZONTAL, 0,
                                                newPosition)

    def setHorizontalLimits(self, limit):
        if self.limitsAction is not None:
            self.limitsAction.setLimits(QubRulerAction.HORIZONTAL, 0, limit[0],
                                        limit[1])

    def setVerticalPosition(self, newPosition):
        if self.limitsAction is not None:
            self.limitsAction.setCursorPosition(QubRulerAction.VERTICAL, 0,
                                                newPosition)

    def setVerticalLimits(self, limit):
        if self.limitsAction is not None:
            self.limitsAction.setLimits(QubRulerAction.VERTICAL, 0, limit[0],
                                        limit[1])

    #def setFocusLimits(self,limit) :
    #    self.focusState.setLimit(limit)

    #def setFocusPosition(self,newPosition) :
    #    self.focusState.newPosition(newPosition)

    #def focusMoveFinished(self, ver, mne):
    #    self.focusState.endMovement(ver)

    #def focusRawKeyPressed(self,keyevent) :
    #    key = keyevent.key()
    #    if key == qt.Qt.Key_Plus:
    #        self.focusRectangleSize += 3
    #        if self.focusRectangleSize > 99:
    #            self.focusRectangleSize = 99
    #    elif key == qt.Qt.Key_Minus:
    #        self.focusRectangleSize -= 3
    #        if self.focusRectangleSize < 12:
    #            self.focusRectangleSize = 12
    #    else: return
    #
    #    self.focusDrawingRectangle.setWidthNHeight(self.focusRectangleSize,self.focusRectangleSize)

    #def showFocusGrab(self,state) :
    #    self.focusPointSelected = None
    #    if state:
    #        self.focusDrawingRectangle.startDrawing()
    #    else:
    #        self.focusDrawingRectangle.stopDrawing()
    #        self.focusDrawingRectangle.hide()
    #        self.focusPointSelected = None

    #def setFocusPointSelected(self,drawingMgr) :
    #    self.focusPointSelected = drawingMgr.point()
    #    if self.focusMotHwo is not None:
    #        self.focusState.start()

    def beamPositionChanged(self, beamy, beamz):
        self.YBeam = beamy
        self.ZBeam = beamz

    def pixelCalibrationChanged(self, sizey, sizez):
        if sizey is not None:
            self.YSize = sizey * 1000
            try:
                self.__measureDialog.setXPixelSize(sizey)
            except:
                pass
        else:
            self.YSize = None

        if sizez is not None:
            self.ZSize = sizez * 1000
            try:
                self.__measureDialog.setYPixelSize(sizez)
            except:
                pass
        else:
            self.ZSize = None

    def setMoveToState(self, state):
        if self.movetoAction is not None:
            self.movetoAction.setState(state)

    def movetoStateChanged(self, state):
        if self.movetoColorWhenActivate:
            if state:
                self.movetoAction.setPaletteBackgroundColor(
                    qt.QColor(self.movetoColorWhenActivate))
            else:
                self.movetoAction.setPaletteBackgroundColor(
                    self.oldMoveToActionColor)

    def pointSelected(self, drawingMgr):
        if self.horMotHwo is not None and self.verMotHwo is not None:
            if  self.YSize is not None and \
                self.ZSize is not None and \
                self.YBeam is not None and \
                self.ZBeam is not None :

                self.drawingMgr = drawingMgr

                (y, z) = drawingMgr.point()

                self.drawingMgr.stopDrawing()

                sign = 1
                if self.horMotHwo.unit < 0:
                    sign = -1
                movetoy = -sign * (self.YBeam - y) * self.YSize

                sign = 1
                if self.verMotHwo.unit < 0:
                    sign = -1
                movetoz = -sign * (self.ZBeam - z) * self.ZSize

                self.motorArrived = 0

                self.connect(self.horMotHwo, qt.PYSIGNAL("moveDone"),
                             self.moveFinished)
                self.connect(self.verMotHwo, qt.PYSIGNAL("moveDone"),
                             self.moveFinished)

                self.horMotHwo.moveRelative(movetoy)
                self.verMotHwo.moveRelative(movetoz)

    def __movetoMosaicStateChanged(self, state):
        if self.movetoColorWhenActivate:
            if state:
                self.__movetoActionMosaic.setPaletteBackgroundColor(
                    qt.QColor(self.movetoColorWhenActivate))
            else:
                self.__movetoActionMosaic.setPaletteBackgroundColor(
                    self.__oldMoveToMosaicActionColor)

    def __mosaicPointSelected(self, drawingMgr):
        point = drawingMgr.mosaicPoints()
        try:
            point = point[0]
            beamY, beamZ = point.refPoint
            YSize, ZSize = point.calibration
            horMotorPos, verMotorPos = point.absPoint
            y, z = point.point
            imageId = point.imageId
        except TypeError:
            return  # The click wasn't on image

        self.drawingMgr = drawingMgr

        drawingMgr.stopDrawing()

        sign = 1
        if self.horMotHwo.unit < 0:
            sign = -1
        movetoy = horMotorPos - sign * (beamY - y) * YSize

        sign = 1
        if self.verMotHwo.unit < 0:
            sign = -1
        movetoz = verMotorPos - sign * (beamZ - z) * ZSize

        self.motorArrived = 0

        self.connect(self.horMotHwo, qt.PYSIGNAL("moveDone"),
                     self.moveFinished)
        self.connect(self.verMotHwo, qt.PYSIGNAL("moveDone"),
                     self.moveFinished)

        self.horMotHwo.move(movetoy)
        self.verMotHwo.move(movetoz)

        self.emit(qt.PYSIGNAL("mosaicImageSelected"), (imageId, ))

    def moveFinished(self, ver, mne):
        if mne == self.horMotHwo.getMotorMnemonic():
            self.disconnect(self.horMotHwo, qt.PYSIGNAL("moveDone"),
                            self.moveFinished)
            self.motorArrived = self.motorArrived + 1

        if mne == self.verMotHwo.getMotorMnemonic():
            self.disconnect(self.verMotHwo, qt.PYSIGNAL("moveDone"),
                            self.moveFinished)
            self.motorArrived = self.motorArrived + 1

        if self.motorArrived == 2:
            self.drawingMgr.startDrawing()
            self.motorArrived = 0
            self.emit(qt.PYSIGNAL("moveDone"), (self.YBeam, self.ZBeam))
    def configureAction(self):
        """
        Move To action
        """
        if self.movetoMode is not None:
            if self.movetoAction is None:
                """
                create action
                """
                self.movetoAction = QubSelectPointAction(name='Move to Beam',
                                                         place=self.movetoMode,
                                                         actionInfo = 'Move to Beam',
                                                         group='Tools')
                self.connect(self.movetoAction,qt.PYSIGNAL("StateChanged"),self.movetoStateChanged)
                self.connect(self.movetoAction, qt.PYSIGNAL("PointSelected"),
                             self.pointSelected)
                
                if self.view is not None:
                    self.view.addAction(self.movetoAction)
                    self.oldMoveToActionColor = self.movetoAction.paletteBackgroundColor()

        else:
            if self.movetoAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.movetoAction)
                    
                """
                del action from view
                """
                self.disconnect(self.movetoAction, qt.PYSIGNAL("PointSelected"),
                                self.pointSelected)
                self.movetoAction = None
        """
        Limits action
        """
        if self.limitsMode is not None:
            if self.limitsAction is None:
                """
                create action
                """
                self.limitsAction = QubRulerAction(name='Motor Limits',
                                                   place=self.limitsMode,
                                                   group='Tools')
                    
                if self.view is not None:
                    self.view.addAction(self.limitsAction)
                
            """
            configure action
            """
            if self.horMotHwo is not None:
                mne = self.horMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.HORIZONTAL,0,mne)

            if self.verMotHwo is not None:
                mne = self.verMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.VERTICAL,0, mne)
                
        else:
            if self.limitsAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.limitsAction)
                    
                """
                del action from view
                """
                self.limitsAction = None

        if self.measureMode is not None:
                if self.measureAction is None:
                        self.measureAction = QubOpenDialogAction(parent=self, name='measure', iconName='measure', label='Measure', group='Tools', place=self.measureMode)
                        self.measureAction.setConnectCallBack(self._measure_dialog_new)
                        logging.getLogger().info("setting measure mode")
                        if self.view is not None:
                                logging.getLogger().info("adding action")
                                self.view.addAction(self.measureAction)
        else:
                if self.measureAction is not None:
                        if self.view is not None:
                                self.view.delAction(self.measureAction)
                        self.measureAction = None

        if self.movetoMode is not None:
            if self.__movetoActionMosaic is not None:
                self.__mosaicView.delAction(self.__movetoActionMosaic)
                self.disconnect(self.__movetoActionMosaic, qt.PYSIGNAL("PointSelected"),
                                self.__mosaicPointSelected)
                self.diconnect(self.__movetoActionMosaic,qt.PYSIGNAL("StateChanged"),
                               self.__movetoMosaicStateChanged)
                self.__movetoActionMosaic = None
                
            if self.__mosaicView is not None :
                self.__movetoActionMosaic = QubSelectPointAction(name='Move to Beam',
                                                                 place=self.movetoMode,
                                                                 actionInfo = 'Move to Beam',
                                                                 mosaicMode = True,
                                                                 residualMode = True,
                                                                 group='Tools')
                self.connect(self.__movetoActionMosaic, qt.PYSIGNAL("PointSelected"),
                             self.__mosaicPointSelected)
                self.connect(self.__movetoActionMosaic,qt.PYSIGNAL("StateChanged"),
                             self.__movetoMosaicStateChanged)

                self.__mosaicView.addAction(self.__movetoActionMosaic)
                self.__oldMoveToMosaicActionColor = self.__movetoActionMosaic.paletteBackgroundColor()

        if self.focusMode is not None:
            if self.focusAction is None:
                self.focusAction = QubToggleAction(label='Autofocus',name='autofocus',place=self.focusMode,
                                                   group='Tools',autoConnect = True)
                qt.QObject.connect(self.focusAction,qt.PYSIGNAL('StateChanged'),self.showFocusGrab)

            if self.view and self.drawing :
                self.focusDrawingRectangle,_ = QubAddDrawing(self.drawing,QubPointDrawingMgr,QubCanvasHomotheticRectangle)
                self.focusDrawingRectangle.setDrawingEvent(QubMoveNPressed1Point)
                self.focusDrawingRectangle.setKeyPressedCallBack(self.focusRawKeyPressed)

                qt.QObject.connect(self.drawing,qt.PYSIGNAL("ForegroundColorChanged"),self.focusDrawingRectangle.setColor)
                self.focusDrawingRectangle.setEndDrawCallBack(self.setFocusPointSelected)
                self.focusRectangleSize = 12
                self.focusDrawingRectangle.setWidthNHeight(self.focusRectangleSize,self.focusRectangleSize)
                self.view.addAction(self.focusAction)
                
        elif self.view is not None:
            self.view.delAction(self.focusAction)
            self.focusDrawingRectangle = None
Esempio n. 10
0
class CameraMotorToolsBrick(BlissWidget):
    def __init__(self, parent, name):
        BlissWidget.__init__(self, parent, name)
        
        """
        variables
        """
        self.YBeam = None
        self.ZBeam = None
        self.YSize = None
        self.ZSize = None
        
        self.view = None
        self.firstCall = True
        self.motorArrived = 0

        self.__mosaicView,self.__mosaicDraw = None,None

        """
        property
        """
        self.addProperty("Hor. Motor", "string", "")
        self.horMotHwo = None
        
        self.addProperty("Vert. Motor", "string", "")
        self.verMotHwo = None

        #self.addProperty("Focus Motor","string","")
        #self.focusMotHwo = None

        #self.addProperty("Focus min step","float",0.05)
        #self.focusMinStep = None
        
        self.addProperty("Move To", "combo",
                         ("none", "toolbar", "popup"), "none")
        self.movetoMode = None
        self.movetoAction = None

        self.addProperty("Move to color when activate","combo",
                         ("none","blue","green","yellow","red","orange"),"none")
        self.movetoColorWhenActivate = None
        
        self.addProperty("Limits", "combo",
                         ("none", "toolbar", "popup"), "none")
        self.limitsMode = None
        self.limitsAction = None

        self.addProperty("Measure", "combo", ("none", "toolbar", "popup"), "none")
        self.measureMode = None
        self.measureAction = None

        self.addProperty("Focus","combo",
                         ("none", "toolbar", "popup"), "none")
        self.focusMode = None
        self.focusAction = None
        self.focusState = None
        self.focusDrawingRectangle = None
        self.focusPointSelected = None
        
        self.__movetoActionMosaic = None
        
        """
        Signal
        """
        self.defineSignal('getView',())
        self.defineSignal('getMosaicView',())
        self.defineSignal('getCalibration',())
        self.defineSignal('getBeamPosition',())
        self.defineSignal('getImage',())
        self.defineSignal('moveDone', ())
        self.defineSignal('mosaicImageSelected', ())
        """
        Slot
        """
        self.defineSlot('beamPositionChanged',())
        self.defineSlot('pixelCalibrationChanged',())
        self.defineSlot('setMoveToMode',())
        self.defineSlot('setLimitsDisplay',())
        self.defineSlot('setMoveToState',())

        """
        widgets - NO APPEARANCE
        """
        self.setFixedSize(0, 0)
                        
               
    def propertyChanged(self, prop, oldValue, newValue):
        if prop == "Hor. Motor":
            if self.horMotHwo is not None:
                self.disconnect(self.horMotHwo, qt.PYSIGNAL('positionChanged'),
                                self.setHorizontalPosition)
                self.disconnect(self.horMotHwo, qt.PYSIGNAL("limitsChanged"),
                                self.setHorizontalLimits)

            self.horMotHwo = self.getHardwareObject(newValue)
            
            if self.horMotHwo is not None:
                self.connect(self.horMotHwo, qt.PYSIGNAL('positionChanged'),
                             self.setHorizontalPosition)
                self.connect(self.horMotHwo, qt.PYSIGNAL("limitsChanged"),
                             self.setHorizontalLimits)

        elif prop == "Vert. Motor":
            if self.verMotHwo is not None:
                self.disconnect(self.verMotHwo, qt.PYSIGNAL('positionChanged'),
                                self.setVerticalPosition)
                self.disconnect(self.verMotHwo, qt.PYSIGNAL("limitsChanged"),
                                self.setVerticalLimits)

            self.verMotHwo = self.getHardwareObject(newValue)
            
            if self.verMotHwo is not None:
                self.connect(self.verMotHwo, qt.PYSIGNAL('positionChanged'),
                             self.setVerticalPosition)
                self.connect(self.verMotHwo, qt.PYSIGNAL("limitsChanged"),
                             self.setVerticalLimits)
        #elif prop == "Focus Motor":
        #    if self.focusMotHwo is not None:
        #        self.disconnect(self.focusMotHwo,qt.PYSIGNAL('positionChanged'),
        #                        self.setFocusPosition)
        #        self.disconnect(self.focusMotHwo,qt.PYSIGNAL('limitsChanged'),
        #                        self.setFocusLimits)

        #    self.focusMotHwo = self.getHardwareObject(newValue)
        #    self.focusState = FocusState(self,self.focusMotHwo)
            
        #    if self.focusMotHwo is not None:
        #        self.connect(self.focusMotHwo,qt.PYSIGNAL('positionChanged'),
        #                     self.setFocusPosition)
        #        self.connect(self.focusMotHwo,qt.PYSIGNAL('limitsChanged'),
        #                     self.setFocusLimits)
        #        self.connect(self.focusMotHwo, qt.PYSIGNAL("moveDone"),
        #                     self.focusMoveFinished)
        elif prop == "Move To": self.movetoMode = _enumTranslate[newValue]
        elif prop == "Limits": self.limitsMode = _enumTranslate[newValue]
        elif prop == "Focus": self.focusMode = _enumTranslate[newValue]
        elif prop == "Measure": self.measureMode = _enumTranslate[newValue]
        elif prop == "Focus min step": self.focusMinStep = newValue
        elif prop == "Move to color when activate" :
            if newValue == "none" :
                self.movetoColorWhenActivate = None
            else:
                self.movetoColorWhenActivate = newValue
        if not self.firstCall:
            self.configureAction()
            
    def run(self):
        """
        get view
        """
        view = {}
        self.emit(qt.PYSIGNAL("getView"), (view,))
        try:
            self.drawing = view["drawing"]
            self.view = view["view"]        
        except:
            print "No View"
        
        """
        get calibration
        """
        calib = {}
        self.emit(qt.PYSIGNAL("getCalibration"), (calib,))
        try:
            # in all this brick we work with pixel calibration in mm
            self.YSize = calib["ycalib"]
            self.ZSize = calib["zcalib"]
            if calib["ycalib"] is not None and calib["zcalib"] is not None:
                self.YSize = self.YSize * 1000
                self.ZSize = self.ZSize * 1000
        except:
            print "No Calibration"
            
        """
        get beam position
        """
        position = {}
        self.emit(qt.PYSIGNAL("getBeamPosition"), (position,))
        try:
            self.YBeam = position["ybeam"]
            self.ZBeam = position["zbeam"]
        except:
            print "No Beam Position"

        """
        get mosaic view
        """
        mosaicView = {}
        self.emit(qt.PYSIGNAL('getMosaicView'),(mosaicView,))
        try:
            self.__mosaicView = mosaicView['view']
            self.__mosaicDraw = mosaicView['drawing']
        except KeyError:
            self.__mosaicView,self.__mosaicDraw = None,None
            
        self.configureAction()
        
        self.firstCall = False
    
    def configureAction(self):
        """
        Move To action
        """
        if self.movetoMode is not None:
            if self.movetoAction is None:
                """
                create action
                """
                self.movetoAction = QubSelectPointAction(name='Move to Beam',
                                                         place=self.movetoMode,
                                                         actionInfo = 'Move to Beam',
                                                         group='Tools')
                self.connect(self.movetoAction,qt.PYSIGNAL("StateChanged"),self.movetoStateChanged)
                self.connect(self.movetoAction, qt.PYSIGNAL("PointSelected"),
                             self.pointSelected)
                
                if self.view is not None:
                    self.view.addAction(self.movetoAction)
                    self.oldMoveToActionColor = self.movetoAction.paletteBackgroundColor()

        else:
            if self.movetoAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.movetoAction)
                    
                """
                del action from view
                """
                self.disconnect(self.movetoAction, qt.PYSIGNAL("PointSelected"),
                                self.pointSelected)
                self.movetoAction = None
        """
        Limits action
        """
        if self.limitsMode is not None:
            if self.limitsAction is None:
                """
                create action
                """
                self.limitsAction = QubRulerAction(name='Motor Limits',
                                                   place=self.limitsMode,
                                                   group='Tools')
                    
                if self.view is not None:
                    self.view.addAction(self.limitsAction)
                
            """
            configure action
            """
            if self.horMotHwo is not None:
                mne = self.horMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.HORIZONTAL,0,mne)

            if self.verMotHwo is not None:
                mne = self.verMotHwo.getMotorMnemonic()
                self.limitsAction.setLabel(QubRulerAction.VERTICAL,0, mne)
                
        else:
            if self.limitsAction is not None:
                """
                remove action
                """
                if self.view is not None:
                    self.view.delAction(self.limitsAction)
                    
                """
                del action from view
                """
                self.limitsAction = None

        if self.measureMode is not None:
                if self.measureAction is None:
                        self.measureAction = QubOpenDialogAction(parent=self, name='measure', iconName='measure', label='Measure', group='Tools', place=self.measureMode)
                        self.measureAction.setConnectCallBack(self._measure_dialog_new)
                        logging.getLogger().info("setting measure mode")
                        if self.view is not None:
                                logging.getLogger().info("adding action")
                                self.view.addAction(self.measureAction)
        else:
                if self.measureAction is not None:
                        if self.view is not None:
                                self.view.delAction(self.measureAction)
                        self.measureAction = None

        if self.movetoMode is not None:
            if self.__movetoActionMosaic is not None:
                self.__mosaicView.delAction(self.__movetoActionMosaic)
                self.disconnect(self.__movetoActionMosaic, qt.PYSIGNAL("PointSelected"),
                                self.__mosaicPointSelected)
                self.diconnect(self.__movetoActionMosaic,qt.PYSIGNAL("StateChanged"),
                               self.__movetoMosaicStateChanged)
                self.__movetoActionMosaic = None
                
            if self.__mosaicView is not None :
                self.__movetoActionMosaic = QubSelectPointAction(name='Move to Beam',
                                                                 place=self.movetoMode,
                                                                 actionInfo = 'Move to Beam',
                                                                 mosaicMode = True,
                                                                 residualMode = True,
                                                                 group='Tools')
                self.connect(self.__movetoActionMosaic, qt.PYSIGNAL("PointSelected"),
                             self.__mosaicPointSelected)
                self.connect(self.__movetoActionMosaic,qt.PYSIGNAL("StateChanged"),
                             self.__movetoMosaicStateChanged)

                self.__mosaicView.addAction(self.__movetoActionMosaic)
                self.__oldMoveToMosaicActionColor = self.__movetoActionMosaic.paletteBackgroundColor()

        if self.focusMode is not None:
            if self.focusAction is None:
                self.focusAction = QubToggleAction(label='Autofocus',name='autofocus',place=self.focusMode,
                                                   group='Tools',autoConnect = True)
                qt.QObject.connect(self.focusAction,qt.PYSIGNAL('StateChanged'),self.showFocusGrab)

            if self.view and self.drawing :
                self.focusDrawingRectangle,_ = QubAddDrawing(self.drawing,QubPointDrawingMgr,QubCanvasHomotheticRectangle)
                self.focusDrawingRectangle.setDrawingEvent(QubMoveNPressed1Point)
                self.focusDrawingRectangle.setKeyPressedCallBack(self.focusRawKeyPressed)

                qt.QObject.connect(self.drawing,qt.PYSIGNAL("ForegroundColorChanged"),self.focusDrawingRectangle.setColor)
                self.focusDrawingRectangle.setEndDrawCallBack(self.setFocusPointSelected)
                self.focusRectangleSize = 12
                self.focusDrawingRectangle.setWidthNHeight(self.focusRectangleSize,self.focusRectangleSize)
                self.view.addAction(self.focusAction)
                
        elif self.view is not None:
            self.view.delAction(self.focusAction)
            self.focusDrawingRectangle = None
    
    def _measure_dialog_new(self,openDialogAction,aQubImage) :
        if  self.YSize is not None and self.ZSize is not None:
            self.__measureDialog = QubMeasureListDialog(self,
                                                      canvas=aQubImage.canvas(),
                                                      matrix=aQubImage.matrix(),
                                                      eventMgr=aQubImage)
            self.__measureDialog.setXPixelSize(self.YSize/1000.0)
            self.__measureDialog.setYPixelSize(self.ZSize/1000.0)
            self.__measureDialog.connect(aQubImage, qt.PYSIGNAL("ForegroundColorChanged"),
                                       self.__measureDialog.setDefaultColor)
            openDialogAction.setDialog(self.__measureDialog)
        
    def setHorizontalPosition(self, newPosition):
        if self.limitsAction is not None:        
            self.limitsAction.setCursorPosition(QubRulerAction.HORIZONTAL, 0,
                                                newPosition)
          
    def setHorizontalLimits(self, limit):
        if self.limitsAction is not None:        
            self.limitsAction.setLimits(QubRulerAction.HORIZONTAL, 0,
                                        limit[0], limit[1])
             
    def setVerticalPosition(self, newPosition):
        if self.limitsAction is not None:        
            self.limitsAction.setCursorPosition(QubRulerAction.VERTICAL, 0,
                                                newPosition)

    def setVerticalLimits(self, limit):
        if self.limitsAction is not None:        
            self.limitsAction.setLimits(QubRulerAction.VERTICAL, 0,
                                        limit[0], limit[1])
         
    #def setFocusLimits(self,limit) :
    #    self.focusState.setLimit(limit)

    #def setFocusPosition(self,newPosition) :
    #    self.focusState.newPosition(newPosition)

    #def focusMoveFinished(self, ver, mne):
    #    self.focusState.endMovement(ver)

    #def focusRawKeyPressed(self,keyevent) :
    #    key = keyevent.key()
    #    if key == qt.Qt.Key_Plus:
    #        self.focusRectangleSize += 3
    #        if self.focusRectangleSize > 99:
    #            self.focusRectangleSize = 99
    #    elif key == qt.Qt.Key_Minus:
    #        self.focusRectangleSize -= 3
    #        if self.focusRectangleSize < 12:
    #            self.focusRectangleSize = 12
    #    else: return
    #    
    #    self.focusDrawingRectangle.setWidthNHeight(self.focusRectangleSize,self.focusRectangleSize)
    
    #def showFocusGrab(self,state) :
    #    self.focusPointSelected = None
    #    if state:
    #        self.focusDrawingRectangle.startDrawing()
    #    else:
    #        self.focusDrawingRectangle.stopDrawing()
    #        self.focusDrawingRectangle.hide()
    #        self.focusPointSelected = None
            
    #def setFocusPointSelected(self,drawingMgr) :
    #    self.focusPointSelected = drawingMgr.point()
    #    if self.focusMotHwo is not None:
    #        self.focusState.start()
            
    def beamPositionChanged(self, beamy, beamz):
        self.YBeam = beamy
        self.ZBeam = beamz
    
    def pixelCalibrationChanged(self, sizey, sizez):
        if sizey is not None:
            self.YSize = sizey * 1000
            try:
                self.__measureDialog.setXPixelSize(sizey)
            except:
                pass
        else:
            self.YSize = None
        
        if sizez is not None:
            self.ZSize = sizez * 1000
            try:
                self.__measureDialog.setYPixelSize(sizez)
            except:
                pass
        else:
            self.ZSize = None

    def setMoveToState(self,state):
        if self.movetoAction is not None:
            self.movetoAction.setState(state)

    def movetoStateChanged(self,state) :
        if self.movetoColorWhenActivate:
            if state:
                self.movetoAction.setPaletteBackgroundColor(qt.QColor(self.movetoColorWhenActivate))
            else:
                self.movetoAction.setPaletteBackgroundColor(self.oldMoveToActionColor)
        
    def pointSelected(self, drawingMgr):
        if self.horMotHwo is not None and self.verMotHwo is not None:
            if  self.YSize is not None and \
                self.ZSize is not None and \
                self.YBeam is not None and \
                self.ZBeam is not None :
                
                self.drawingMgr =  drawingMgr 
                   
                (y, z) = drawingMgr.point()
                
                self.drawingMgr.stopDrawing()

                sign = 1
                if self.horMotHwo.unit < 0:
                    sign = -1
                movetoy = - sign*(self.YBeam - y) * self.YSize

                sign = 1
                if self.verMotHwo.unit < 0:
                    sign = -1
                movetoz = - sign*(self.ZBeam - z) * self.ZSize
                
                self.motorArrived = 0
            
                self.connect(self.horMotHwo, qt.PYSIGNAL("moveDone"),
                             self.moveFinished)
                self.connect(self.verMotHwo, qt.PYSIGNAL("moveDone"),
                             self.moveFinished)

                self.horMotHwo.moveRelative(movetoy)
                self.verMotHwo.moveRelative(movetoz)
    def __movetoMosaicStateChanged(self,state):
        if self.movetoColorWhenActivate:
            if state:
                self.__movetoActionMosaic.setPaletteBackgroundColor(qt.QColor(self.movetoColorWhenActivate))
            else:
                self.__movetoActionMosaic.setPaletteBackgroundColor(self.__oldMoveToMosaicActionColor)

    def __mosaicPointSelected(self,drawingMgr) :
        point = drawingMgr.mosaicPoints()
        try:
            point = point[0]
            beamY,beamZ = point.refPoint
            YSize,ZSize = point.calibration
            horMotorPos,verMotorPos = point.absPoint
            y,z = point.point
            imageId = point.imageId
        except TypeError: return        # The click wasn't on image

        self.drawingMgr = drawingMgr
        
        drawingMgr.stopDrawing()

        sign = 1
        if self.horMotHwo.unit < 0:
            sign = -1
        movetoy = horMotorPos - sign*(beamY - y) * YSize

        sign = 1
        if self.verMotHwo.unit < 0:
            sign = -1
        movetoz = verMotorPos - sign*(beamZ - z) * ZSize

        self.motorArrived = 0
        
        self.connect(self.horMotHwo, qt.PYSIGNAL("moveDone"),
                     self.moveFinished)
        self.connect(self.verMotHwo, qt.PYSIGNAL("moveDone"),
                     self.moveFinished)

        self.horMotHwo.move(movetoy)
        self.verMotHwo.move(movetoz)

        self.emit(qt.PYSIGNAL("mosaicImageSelected"), (imageId,))


    def moveFinished(self, ver, mne):
        if mne == self.horMotHwo.getMotorMnemonic():
            self.disconnect(self.horMotHwo, qt.PYSIGNAL("moveDone"),
                            self.moveFinished)
            self.motorArrived = self.motorArrived + 1
            
        if mne == self.verMotHwo.getMotorMnemonic():
            self.disconnect(self.verMotHwo, qt.PYSIGNAL("moveDone"),
                            self.moveFinished)
            self.motorArrived = self.motorArrived + 1
                        
        if self.motorArrived == 2:
            self.drawingMgr.startDrawing()
            self.motorArrived = 0
            self.emit(qt.PYSIGNAL("moveDone"), (self.YBeam, self.ZBeam))
 def __init__(self,*args,**keys) :
     QubOpenDialogAction.__init__(self,*args,**keys)
Esempio n. 12
0
    def __init__(self, *args):
        BlissWidget.__init__(self, *args)

        self.camera       = None
        self.update_disabled = False
        self.__cameraName = None

        self.__zoomList = [1.5,2,2.5,3,3.5,4]
        self.__initZoom = 2

        self.__fixWidth  = -1
        self.__fixHeight = -1

        self.__swapRgb = True
                        ####### PRINT #######
        self.__printWidget = QubPrintPreview(self)
        self.__printWidget.resize(400,500)
        printer = qt.QPrinter()
        printer.setOutputToFile(1)
        printer.setOutputFileName('/tmp/print_file.ps')
        self.__printWidget.setPrinter(printer)
        self.__printWidget.update()

        self.__beamAction    = None
        self.__scaleAction   = None
        self.__chosenActions = {}
        self.__wholeActions  = []
        self.__measureDialog = None

        ####### PRINT ACTION #######
        printAction = QubPrintPreviewAction(name="print",group="admin",withVectorMenu=True)
        printAction.previewConnect(self.__printWidget)
        self.__wholeActions.append(printAction)

        ####### SAVE IMAGE #######
        self.__saveAction = QubSaveImageAction(parent=self, label='Save falcon image',
                                               name="save", group="admin")
        self.__saveAction.setConnectCallBack(self._save_dialog_new)
        self.__wholeActions.append(self.__saveAction)
        self.__defaultSavePath = '/tmp'

        ####### UPDATE #######
        update = QubToggleAction(name="update",group="image",initState=True)
        self.connect(update,qt.PYSIGNAL("StateChanged"),self.__cameraUpdate)
        self.__wholeActions.append(update)

        ####### Start Camera ######
        startCamera = QubToggleAction(name="startCamera", group="image", iconName = 'bright-cont', initState=True)
        self.connect(startCamera, qt.PYSIGNAL("StateChanged"), self.__cameraStart)
        self.__wholeActions.append(startCamera)


        ####### BRIGHTNESS/CONTRAST #######
        self.__brightcount = QubOpenDialogAction(name="bright-cont",
                                                 iconName = 'bright-cont', group="image")
        self.__brightcount.setDialog(QubBrightnessContrastDialog(self))
        self.__wholeActions.append(self.__brightcount)

        ###### Grid TOOL ######
        self.__gridToolAction = QubOpenDialogAction(parent=self, name='grid_tool',
                                                    iconName='rectangle', label='Grid tool',
                                                    group="Tools") #place="contextmenu")
        self.__gridDialog = GridDialog(self, "Grid Dialog", flags = qt.Qt.WStyle_StaysOnTop)
        self.__gridToolAction.setConnectCallBack(self._grid_dialog_connect_hdlr)
        self.__wholeActions.append(self.__gridToolAction)

        self.__previous_pos_dict = {}
        self.__beamWidth = 0
        self.__beamHeight = 0

        ####### BEAM ACTION #######
        self.__beamAction = QubBeamAction(name="beam", group="Tools")
        self.__wholeActions.append(self.__beamAction)
        self.connect(self.__beamAction,qt.PYSIGNAL("BeamSelected"),
                     self.beamSelection)

        ####### SCALE #######
        self.__scaleAction = QubScaleAction(name='scale',group='Tools')
        self.__wholeActions.append(self.__scaleAction)
        self.__wholeActions.extend(self.__creatStdActions())

        ####### ACTION INFO #######
        actionInfo = QubInfoAction(name="actionInfo", group="image",place="statusbar")
        self.__wholeActions.append(actionInfo)

        ####### CHANGE FOREGROUND COLOR #######
        self.__fcoloraction = QubForegroundColorAction(name="color", group="image")
        self.__wholeActions.append(self.__fcoloraction)

        ####### MEASURE #######
        self.__measureAction = QubOpenDialogAction(parent=self, name='measure',
                                            iconName='measure', label='Measure',
                                            group="Tools")
        self.__measureAction.setConnectCallBack(self._measure_dialog_new)
        self.__wholeActions.append(self.__measureAction)

        # ###### POSITION TOOL ######
        # self.__posToolAction = QubOpenDialogAction(parent=self, name='pos_tool',
        #                                            iconName='circle', label='Position tool',
        #                                            group="Tools")
        # self.__posToolAction.setConnectCallBack(self._line_dialog_new)
        # self.__wholeActions.append(self.__posToolAction)

        ####### ZOOM LIST #######
        zoomActionList = QubZoomListAction(place = "toolbar",
                                           initZoom = 1,zoomValList = [0.1,0.25,0.5,0.75,1,1.5,2],
                                           show = 1,group = "zoom")
        self.__wholeActions.append(zoomActionList)

        ####### ZOOM Action #######
        self.__zoomFitOrFill = QubZoomAction(place = "toolbar",group = "zoom")
        self.__wholeActions.append(self.__zoomFitOrFill)

        ####### LINK ZOOM ACTION #######
        self.__zoomFitOrFill.setList(zoomActionList)
        zoomActionList.setActionZoomMode(self.__zoomFitOrFill)

        ####### ZOOM WINDOW #######
        self.__zoomAction = QubZoomRectangle(label='Zoom Crop',place="toolbar", show=1, group="zoom")
        self.connect(self.__zoomAction,qt.PYSIGNAL("Actif"),self.__hide_show_zoom)
        self.__wholeActions.append(self.__zoomAction)

        self.__splitter = qt.QSplitter(qt.Qt.Horizontal,self)
        self.__splitter.show()

        self.__mainVideo = QubPixmapDisplayView(self.__splitter)
        self.__mainVideo.show()
        self.__mainVideo.setScrollbarMode('Auto')
        self.__mainPlug = _MainVideoPlug(self.__mainVideo,self.__zoomAction)

        actions = self.__creatStdActions()

        ####### ZOOM LIST #######
        self.__zoomActionList = QubZoomListAction(place = "toolbar",keepROI = True,
                                                  initZoom = self.__initZoom,zoomValList = self.__zoomList,
                                                  show = 1,group = "zoom")
        actions.insert(0,self.__zoomActionList)

        ####### ZOOM Action #######
        zoomFitOrFill = QubZoomAction(place = "toolbar",keepROI = True,group = "zoom")
        zoomFitOrFill.setList(self.__zoomActionList)
        self.__zoomActionList.setActionZoomMode(zoomFitOrFill)
        actions.append(zoomFitOrFill)

        self.__zoomVideo = QubPixmapDisplayView(self.__splitter,None,actions)
        self.__zoomVideo.hide()
        self.__zoomPlug = _ZoomPlug(self.__zoomVideo)
        self.__zoomPlug.zoom().setZoom(2,2)
        self.__cbk = _rectangleZoom(self.__zoomAction,self.__zoomPlug)

        layout = qt.QHBoxLayout(self,0,0,"layout")
        layout.addWidget(self.__splitter)

        self.__image2Pixmap = QubImage2Pixmap()
        self.__image2Pixmap.plug(self.__mainPlug)
        self.__zoomPlug.setPoller(self.__image2Pixmap)

        self.__jpegDecompress = QubStdData2Image()
        self.__jpegDecompress.setSwapRGB(True)
        self.__jpeg2image = None

        ####### PROPERTY #######
        self.addProperty('camera','string','')
        self.addProperty('zoom list','string',','.join([str(x) for x in self.__zoomList]))
        self.addProperty('init zoom','integer',self.__initZoom)
        self.addProperty('swap rgb','boolean',True)

        self.addProperty('fix : width','integer',-1)
        self.addProperty('fix : height','integer',-1)

        self.addProperty('action : print','boolean',True)
        self.addProperty('action : save image','boolean',True)
        self.addProperty('action : update','boolean',True)
        self.addProperty('action : startCamera','boolean',True)
        self.addProperty('action : brightness contrast','boolean',True)
        self.addProperty('action : beam','boolean',True)
        self.addProperty('action : scale','boolean',True)
        self.addProperty('action : change foreground color','boolean',True)
        self.addProperty('action : measure','boolean',True)
        self.addProperty('action : measure (place)', 'combo', ('toolbar','contextmenu'),'toolbar')
        self.addProperty('action : zoom window','boolean',True)
        self.addProperty('action : zoom fit or fill','boolean',True)
        self.addProperty('action : zoom list','boolean',True)
        self.addProperty('action : x,y coordinates','boolean',True)
        self.addProperty('action : default color','combo',('black','red','green'),'black')

        self.addProperty('action : save image (place)',"combo",('toolbar','contextmenu'),'toolbar')
        self.addProperty('action : save image (default path)',"string",'/tmp')
        self.addProperty('action : save image (show always configure)',"boolean",True)

        self.addProperty("diffractometer", "string", "")
        self.diffractometerHwobj = None


        ####### SIGNAL #######
        self.defineSignal("BeamPositionChanged", ())

        ####### SLOT #######
        self.defineSlot("changeBeamPosition", ())
        self.defineSlot("changePixelScale",())
        self.defineSlot('getView',())
        self.defineSlot('getImage',())

        ####### LINK VIEW AND SUB VIEW #######
        mainView = self.__mainVideo.view()
        zoomView = self.__zoomVideo.view()
        mainView.addEventMgrLink(zoomView,
                                 mainView.canvas(),zoomView.canvas(),
                                 mainView.matrix(),zoomView.matrix())

        self.imageReceivedConnected = None
Esempio n. 13
0
class CameraBrick(BlissWidget):
    def __init__(self, *args):
        BlissWidget.__init__(self, *args)

        self.camera       = None
        self.update_disabled = False
        self.__cameraName = None

        self.__zoomList = [1.5,2,2.5,3,3.5,4]
        self.__initZoom = 2

        self.__fixWidth  = -1
        self.__fixHeight = -1

        self.__swapRgb = True
                        ####### PRINT #######
        self.__printWidget = QubPrintPreview(self)
        self.__printWidget.resize(400,500)
        printer = qt.QPrinter()
        printer.setOutputToFile(1)
        printer.setOutputFileName('/tmp/print_file.ps')
        self.__printWidget.setPrinter(printer)
        self.__printWidget.update()

        self.__beamAction    = None
        self.__scaleAction   = None
        self.__chosenActions = {}
        self.__wholeActions  = []
        self.__measureDialog = None

        ####### PRINT ACTION #######
        printAction = QubPrintPreviewAction(name="print",group="admin",withVectorMenu=True)
        printAction.previewConnect(self.__printWidget)
        self.__wholeActions.append(printAction)

        ####### SAVE IMAGE #######
        self.__saveAction = QubSaveImageAction(parent=self, label='Save falcon image',
                                               name="save", group="admin")
        self.__saveAction.setConnectCallBack(self._save_dialog_new)
        self.__wholeActions.append(self.__saveAction)
        self.__defaultSavePath = '/tmp'

        ####### UPDATE #######
        update = QubToggleAction(name="update",group="image",initState=True)
        self.connect(update,qt.PYSIGNAL("StateChanged"),self.__cameraUpdate)
        self.__wholeActions.append(update)

        ####### Start Camera ######
        startCamera = QubToggleAction(name="startCamera", group="image", iconName = 'bright-cont', initState=True)
        self.connect(startCamera, qt.PYSIGNAL("StateChanged"), self.__cameraStart)
        self.__wholeActions.append(startCamera)


        ####### BRIGHTNESS/CONTRAST #######
        self.__brightcount = QubOpenDialogAction(name="bright-cont",
                                                 iconName = 'bright-cont', group="image")
        self.__brightcount.setDialog(QubBrightnessContrastDialog(self))
        self.__wholeActions.append(self.__brightcount)

        ###### Grid TOOL ######
        self.__gridToolAction = QubOpenDialogAction(parent=self, name='grid_tool',
                                                    iconName='rectangle', label='Grid tool',
                                                    group="Tools") #place="contextmenu")
        self.__gridDialog = GridDialog(self, "Grid Dialog", flags = qt.Qt.WStyle_StaysOnTop)
        self.__gridToolAction.setConnectCallBack(self._grid_dialog_connect_hdlr)
        self.__wholeActions.append(self.__gridToolAction)

        self.__previous_pos_dict = {}
        self.__beamWidth = 0
        self.__beamHeight = 0

        ####### BEAM ACTION #######
        self.__beamAction = QubBeamAction(name="beam", group="Tools")
        self.__wholeActions.append(self.__beamAction)
        self.connect(self.__beamAction,qt.PYSIGNAL("BeamSelected"),
                     self.beamSelection)

        ####### SCALE #######
        self.__scaleAction = QubScaleAction(name='scale',group='Tools')
        self.__wholeActions.append(self.__scaleAction)
        self.__wholeActions.extend(self.__creatStdActions())

        ####### ACTION INFO #######
        actionInfo = QubInfoAction(name="actionInfo", group="image",place="statusbar")
        self.__wholeActions.append(actionInfo)

        ####### CHANGE FOREGROUND COLOR #######
        self.__fcoloraction = QubForegroundColorAction(name="color", group="image")
        self.__wholeActions.append(self.__fcoloraction)

        ####### MEASURE #######
        self.__measureAction = QubOpenDialogAction(parent=self, name='measure',
                                            iconName='measure', label='Measure',
                                            group="Tools")
        self.__measureAction.setConnectCallBack(self._measure_dialog_new)
        self.__wholeActions.append(self.__measureAction)

        # ###### POSITION TOOL ######
        # self.__posToolAction = QubOpenDialogAction(parent=self, name='pos_tool',
        #                                            iconName='circle', label='Position tool',
        #                                            group="Tools")
        # self.__posToolAction.setConnectCallBack(self._line_dialog_new)
        # self.__wholeActions.append(self.__posToolAction)

        ####### ZOOM LIST #######
        zoomActionList = QubZoomListAction(place = "toolbar",
                                           initZoom = 1,zoomValList = [0.1,0.25,0.5,0.75,1,1.5,2],
                                           show = 1,group = "zoom")
        self.__wholeActions.append(zoomActionList)

        ####### ZOOM Action #######
        self.__zoomFitOrFill = QubZoomAction(place = "toolbar",group = "zoom")
        self.__wholeActions.append(self.__zoomFitOrFill)

        ####### LINK ZOOM ACTION #######
        self.__zoomFitOrFill.setList(zoomActionList)
        zoomActionList.setActionZoomMode(self.__zoomFitOrFill)

        ####### ZOOM WINDOW #######
        self.__zoomAction = QubZoomRectangle(label='Zoom Crop',place="toolbar", show=1, group="zoom")
        self.connect(self.__zoomAction,qt.PYSIGNAL("Actif"),self.__hide_show_zoom)
        self.__wholeActions.append(self.__zoomAction)

        self.__splitter = qt.QSplitter(qt.Qt.Horizontal,self)
        self.__splitter.show()

        self.__mainVideo = QubPixmapDisplayView(self.__splitter)
        self.__mainVideo.show()
        self.__mainVideo.setScrollbarMode('Auto')
        self.__mainPlug = _MainVideoPlug(self.__mainVideo,self.__zoomAction)

        actions = self.__creatStdActions()

        ####### ZOOM LIST #######
        self.__zoomActionList = QubZoomListAction(place = "toolbar",keepROI = True,
                                                  initZoom = self.__initZoom,zoomValList = self.__zoomList,
                                                  show = 1,group = "zoom")
        actions.insert(0,self.__zoomActionList)

        ####### ZOOM Action #######
        zoomFitOrFill = QubZoomAction(place = "toolbar",keepROI = True,group = "zoom")
        zoomFitOrFill.setList(self.__zoomActionList)
        self.__zoomActionList.setActionZoomMode(zoomFitOrFill)
        actions.append(zoomFitOrFill)

        self.__zoomVideo = QubPixmapDisplayView(self.__splitter,None,actions)
        self.__zoomVideo.hide()
        self.__zoomPlug = _ZoomPlug(self.__zoomVideo)
        self.__zoomPlug.zoom().setZoom(2,2)
        self.__cbk = _rectangleZoom(self.__zoomAction,self.__zoomPlug)

        layout = qt.QHBoxLayout(self,0,0,"layout")
        layout.addWidget(self.__splitter)

        self.__image2Pixmap = QubImage2Pixmap()
        self.__image2Pixmap.plug(self.__mainPlug)
        self.__zoomPlug.setPoller(self.__image2Pixmap)

        self.__jpegDecompress = QubStdData2Image()
        self.__jpegDecompress.setSwapRGB(True)
        self.__jpeg2image = None

        ####### PROPERTY #######
        self.addProperty('camera','string','')
        self.addProperty('zoom list','string',','.join([str(x) for x in self.__zoomList]))
        self.addProperty('init zoom','integer',self.__initZoom)
        self.addProperty('swap rgb','boolean',True)

        self.addProperty('fix : width','integer',-1)
        self.addProperty('fix : height','integer',-1)

        self.addProperty('action : print','boolean',True)
        self.addProperty('action : save image','boolean',True)
        self.addProperty('action : update','boolean',True)
        self.addProperty('action : startCamera','boolean',True)
        self.addProperty('action : brightness contrast','boolean',True)
        self.addProperty('action : beam','boolean',True)
        self.addProperty('action : scale','boolean',True)
        self.addProperty('action : change foreground color','boolean',True)
        self.addProperty('action : measure','boolean',True)
        self.addProperty('action : measure (place)', 'combo', ('toolbar','contextmenu'),'toolbar')
        self.addProperty('action : zoom window','boolean',True)
        self.addProperty('action : zoom fit or fill','boolean',True)
        self.addProperty('action : zoom list','boolean',True)
        self.addProperty('action : x,y coordinates','boolean',True)
        self.addProperty('action : default color','combo',('black','red','green'),'black')

        self.addProperty('action : save image (place)',"combo",('toolbar','contextmenu'),'toolbar')
        self.addProperty('action : save image (default path)',"string",'/tmp')
        self.addProperty('action : save image (show always configure)',"boolean",True)

        self.addProperty("diffractometer", "string", "")
        self.diffractometerHwobj = None


        ####### SIGNAL #######
        self.defineSignal("BeamPositionChanged", ())

        ####### SLOT #######
        self.defineSlot("changeBeamPosition", ())
        self.defineSlot("changePixelScale",())
        self.defineSlot('getView',())
        self.defineSlot('getImage',())

        ####### LINK VIEW AND SUB VIEW #######
        mainView = self.__mainVideo.view()
        zoomView = self.__zoomVideo.view()
        mainView.addEventMgrLink(zoomView,
                                 mainView.canvas(),zoomView.canvas(),
                                 mainView.matrix(),zoomView.matrix())

        self.imageReceivedConnected = None

    def safeConnect(self):
        if not self.imageReceivedConnected and self.camera is not None:
            #print "CONNECTING 0"
            self.camera.connect('imageReceived', self.__imageReceived)
            #self.connect(self.camera,qt.PYSIGNAL('imageReceived'),self.__imageReceived)
            self.imageReceivedConnected = False

    def safeDisconnect(self):
        if self.imageReceivedConnected and self.camera is not None:
            #print "DISCONNECTING 3"
            self.camera.disconnect('imageReceived', self.__imageReceived)
            #self.disconnect(self.camera,qt.PYSIGNAL('imageReceived'),self.__imageReceived)
            self.imageReceivedConnected = False

    def disable_update(self):
        #self.__imageReceived(DISABLED_JPEG,DISABLED_WIDTH,DISABLED_HEIGHT)
        self.update_disabled = True

    def enable_update(self):
        self.update_disabled = False

    def __imageReceived(self,image,width,height,force_update=False):
        if self.update_disabled:
            return

        if not force_update:
            if not self.isVisible():
                return
        
        if isinstance(image,qt.QImage):
            if self.__swapRgb:
                image = image.swapRGB()
            self.__jpeg2image.setImage(image,image)
        else:
            self.__jpegDecompress.putData(image,width,height)

    def propertyChanged(self,property,oldValue,newValue):
        if property=='camera':
            #if self.camera is not None and not self.isRunning():
            #    self.safeDisconnect()
            self.safeDisconnect()
            self.camera      = self.getHardwareObject(newValue)
            self.__hwoName   = newValue

            if self.camera is not None:
		try:
                    self.__tangoName = self.camera.tangoname
		except:
		    self.__tangoName = ""

                try:
                    camera_role=self.camera.getDeviceByRole('camera')
                    if camera_role is not None:
                        self.camera=camera_role
                except AttributeError:
                    pass

                self.__brightcount.setCamera(self.camera)

        elif property == 'zoom list' :
            zoomList = [float(x) for x in newValue.split(',')]
            if zoomList != self.__zoomList :
                self.__zoomList = zoomList
                self.__zoomActionList.changeZoomList(zoomList,self.__initZoom)
        elif property == 'init zoom' :
            if newValue != self.__initZoom :
                self.__initZoom = newValue
                self.__zoomActionList.changeZoomList(self.__zoomList,newValue)
        elif property == 'swap rgb' :
            self.__jpegDecompress.setSwapRGB(newValue)
            self.__swapRgb = newValue
        elif property == 'action : print' : self.__chosenActions['print'] = newValue
        elif property == 'action : save image' : self.__chosenActions['save'] = newValue
        elif property == 'action : update' : self.__chosenActions['update'] = newValue
        elif property == 'action : startCamera' : self.__chosenActions['startCamera'] = newValue
        elif property == 'action : brightness contrast' : self.__chosenActions['bright-cont'] = newValue
        elif property == 'action : beam' : self.__chosenActions['beam'] = newValue
        elif property == 'action : scale' : self.__chosenActions['scale'] = newValue
        elif property == 'action : change foreground color' : self.__chosenActions['color'] = newValue
        elif property == 'action : measure' : self.__chosenActions['measure'] = newValue
        elif property == 'action : zoom window' : self.__chosenActions['zoomrect'] = newValue
        elif property == 'action : zoom fit or fill' : self.__chosenActions['Zoom tools'] = newValue
        elif property == 'action : zoom list' : self.__chosenActions['zoomlist'] = newValue
        elif property == 'action : x,y coordinates' : self.__chosenActions['position'] = newValue
        elif property == 'action : default color' : self.__chosenDefaultColor = newValue
        elif property == 'action : save image (place)' :
            self.__saveAction.setPlace(newValue,True)
        elif property == 'action : measure (place)':
            self.__measureAction.setPlace(newValue, True)
        elif property == 'action : save image (default path)' :
            self.__defaultSavePath = newValue
        elif property == 'action : save image (show always configure)' :
            self.__saveAction.setConfigureOnClick(newValue)

        elif property == 'fix : width' : self.__fixWidth = newValue
        elif property == 'fix : height': self.__fixHeight = newValue
        elif property == "diffractometer":
            self.diffractometerHwobj = self.getHardwareObject(newValue)
            if self.diffractometerHwobj is not None:
                self.__previous_pos_dict = self.diffractometerHwobj.getPositions()
                self.diffractometerHwobj.connect("minidiffStateChanged",
                                                 self.diffractometerChanged)
                if self.diffractometerHwobj.zoomMotor is not None:    
                    zoom = self.diffractometerHwobj.zoomMotor.getPosition()
                    xSize,ySize = self.diffractometerHwobj.getCalibrationData(zoom)
                    #xSize,ySize = self.__scaleAction.xPixelSize(), self.__scaleAction.yPixelSize()
                    self.diffractometerHwobj.getBeamInfo(self.__getBeamInfo)
                    beam_pos_x = self.diffractometerHwobj.getBeamPosX()
                    beam_pos_y = self.diffractometerHwobj.getBeamPosY()


                    self.__gridDialog.set_x_pixel_size(xSize)
                    self.__gridDialog.set_y_pixel_size(ySize)
                    self.__gridDialog.set_beam_position(beam_pos_x, beam_pos_y,
                                                 self.__beamWidth, self.__beamHeight)

    def run(self) :
        chosenActions = []

        for action in self.__wholeActions :
            if self.__chosenActions.get(action.name(), True) :
                chosenActions.append(action)

        self.__mainVideo.addAction(chosenActions)

        if not self.__chosenActions.get('update', False) :
            self.__cameraUpdate(True)

        if not self.__chosenActions.get('startCamera', False) :
            self.__cameraStart(True)

        if self.__fixWidth != -1 and self.__fixHeight != -1 :
            self.__mainVideo.view().setFixedSize(self.__fixWidth + 4,self.__fixHeight + 4)

        self.__startIdle = qt.QTimer(self)
        qt.QObject.connect(self.__startIdle,qt.SIGNAL('timeout()'),self.__idleRun)
        self.__startIdle.start(0)

    def __idleRun(self) :
        self.__startIdle.stop()
        self.__fcoloraction.setColor(self.__chosenDefaultColor)
        self.__mainVideo.view().setForegroundColor(qt.QColor(self.__chosenDefaultColor))

    def stop(self) :
        self.safeDisconnect()

    def beamSelection(self,x,y) :
        self.emit(qt.PYSIGNAL("BeamPositionChanged"), (x,y))
        self.changeBeamPosition(x,y)

    def changeBeamPosition(self, x, y, beam_width=None, beam_height=None):
        self.__beamAction.setBeamPosition(x, y)
        try:
            if not beam_width:
                beam_width = self.__beamWidth

            if not beam_height:
                beam_height = self.__beamHeight

            self.__gridDialog.set_beam_position(x, y, beam_width, beam_height)
        except:
            pass

    def changePixelScale(self,sizex,sizey) :
        self.__scaleAction.setXPixelSize(sizex)
        self.__scaleAction.setYPixelSize(sizey)
        if self.__measureDialog :
            self.__measureDialog.setXPixelSize(sizex)
            self.__measureDialog.setYPixelSize(sizey)

        self.__gridDialog.set_x_pixel_size(sizex)
        self.__gridDialog.set_y_pixel_size(sizey)

    def getView(self,key):
        try:
            key['drawing'] = self.__mainVideo.view()
            key['view']    = self.__mainVideo
            key['hwname']  = self.__hwoName
            key['dsname']  = self.__tangoName
        except:
            pass

    def getImage(self,key) :
        if self.__jpeg2image is not None :
            try:
                key['image'] = self.__jpeg2image.getLastImage()
            except:
                pass

    def __creatStdActions(self) :
        actions = []
        ####### MOUSE POSITION #######
        posaction = QubPositionAction(name="position", group="image",place="statusbar")
        actions.append(posaction)

        return actions

    def __cameraUpdate(self,onoff) :
        if onoff :
            self.__jpeg2image = _Jpeg2ImagePlug(self.__image2Pixmap)
            self.__jpegDecompress.plug(self.__jpeg2image)
            if self.camera is not None:
                imageInfo = self.camera.imageType()
            else:
                imageInfo = None
            imType    = self.__jpegDecompress.STANDARD
            if imageInfo :
                if imageInfo.type() == 'bayer': imType = self.__jpegDecompress.BAYER_RG
                elif imageInfo.type() == 'raw': imType = self.__jpegDecompress.RAW
            self.__jpegDecompress.setImageType(imType)
            self.safeConnect()
        else:
            self.safeDisconnect()
            self.__jpeg2image.setEnd()

    def __cameraStart(self, onoff) :
        if onoff :
            print "********************************************************Set the camera ON"
            if self.camera is not None:
                self.camera.setLive(True)
            else:
                print "camera is none"
        else:
            print "****************************************************Set the camera off"
            if self.camera is not None:
                print "camera :"
                print self.camera
                self.camera.setLive(False)
            else:
                print "camera is none"

    def __hide_show_zoom(self,actif) :
        if actif :
            if self.__chosenActions['Zoom tools'] :
                self.__zoomFitOrFill.setState(True)
            else:
                self.__mainVideo.setScrollbarMode('Fit2Screen')
            self.__zoomPlug.start()
            self.__zoomVideo.show()
            splitterWidth = self.__splitter.width()
            self.__splitter.setSizes([splitterWidth/3,splitterWidth - splitterWidth/3])
        else:
            if self.__chosenActions['Zoom tools'] :
                self.__zoomFitOrFill.setState(False)
            else:
                self.__mainVideo.setScrollbarMode('Auto')
            self.__zoomPlug.stop()
            self.__zoomVideo.hide()

    def _measure_dialog_new(self,openDialogAction,aQubImage) :
        try :
            self.__measureDialog = QubMeasureListDialog(self,
                                                        canvas=aQubImage.canvas(),
                                                        matrix=aQubImage.matrix(),
                                                        eventMgr=aQubImage)
            xSize,ySize = self.__scaleAction.xPixelSize(),self.__scaleAction.yPixelSize()
            self.__measureDialog.setXPixelSize(xSize)
            self.__measureDialog.setYPixelSize(ySize)
            self.__measureDialog.connect(aQubImage, qt.PYSIGNAL("ForegroundColorChanged"),
                                         self.__measureDialog.setDefaultColor)
            openDialogAction.setDialog(self.__measureDialog)
        except:
            import traceback
            traceback.print_exc()

    def _grid_dialog_connect_hdlr(self,openDialogAction, aQubImage) :
        try :
            self.__gridDialog.set_qub_event_mgr(aQubImage)
            openDialogAction.setDialog(self.__gridDialog)
        except:
            import traceback
            traceback.print_exc()

    def __getBeamInfo(self, ret):
        self.__beamWidth = float(ret["size_x"])
        self.__beamHeight = float(ret["size_y"])
        #self.__beamShape = ret["shape"]

    def _save_dialog_new(self,openDialogAction,aQubImage) :
        saveDialog = QubSaveImageDialog(self,matrix=aQubImage.matrix(),canvas=aQubImage.canvas())
        saveDialog.setImage2Pixmap(self.__image2Pixmap)
        saveDialog.setSavePath(self.__defaultSavePath)
        openDialogAction.setDialog(saveDialog)

    def instanceMirrorChanged(self,mirror):
        #print "INSTANCEMIRRORCHANGED",mirror,BlissWidget.isInstanceMirrorAllow()
        return
        if BlissWidget.isInstanceModeSlave():
            if BlissWidget.isInstanceMirrorAllow():
                self.safeConnect()
            else:
                self.safeDisconnect()

    def diffractometerChanged(self, *args):
        """
        Handles diffractometer change events, connected to the signal 
        minidiffStateChanged of the diffractometer hardware object.
        """
        
        if self.diffractometerHwobj.isReady():
            pos_dict = self.diffractometerHwobj.getPositions()
            p1 = self.diffractometerHwobj.motor_positions_to_screen(self.__previous_pos_dict)
            p2 = (self.diffractometerHwobj.getBeamPosX(), self.diffractometerHwobj.getBeamPosY())

            dx = p2[0] - p1[0]
            dy = p2[1] - p1[1]

            if dy != 0:
                self.__gridDialog.move_grid_ver(-dy)

            if dx != 0:
                self.__gridDialog.move_grid_hor(-dx)

            #print dx, dy

            self.__previous_pos_dict = pos_dict
Esempio n. 14
0
 def __init__(self, *args, **keys):
     QubOpenDialogAction.__init__(self, *args, **keys)