psychopy 2024.2.1__py3-none-any.whl → 2024.2.4__py3-none-any.whl

psychopy/app/colorpicker/__init__.py.orig

@@ -0,0 +1,411 @@
+# -*- coding: utf-8 -*-
+"""Classes for the color picker."""
+
+# Part of the PsychoPy library
+# Copyright (C) 2002-2018 Jonathan Peirce (C) 2019 Open Science Tools Ltd.
+# Distributed under the terms of the GNU General Public License (GPL).
+
+import wx
+from .panels import ColorPresets, ColorPreview
+from .pages import ColorPickerPageHSV, ColorPickerPageRGB
+from psychopy.colors import Color
+
+LAST_COLOR = Color((0, 0, 0, 1), space='rgba')
+LAST_OUTPUT_SPACE = 0
+
+from psychopy.localization import _translate
+
+
+class PsychoColorPicker(wx.Dialog):
+    """Class for the color picker dialog.
+
+    This dialog is used to standardize color selection across platforms. It also
+    supports PsychoPy's RGB representation directly.
+
+    Parameters
+    ----------
+    parent : object
+        Reference to a :class:`~wx.Frame` which owns this dialog.
+
+    """
+    def __init__(self, parent):
+<<<<<<< HEAD
+        wx.Dialog.__init__(
+            self,
+            parent,
+            id=wx.ID_ANY,
+            title=u"Color Picker",
+            pos=wx.DefaultPosition,
+            size=wx.Size(640, 480),
+            style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)
+
+        self.SetSizeHints(wx.DefaultSize, wx.DefaultSize)
+        self.SetMinSize(wx.Size(600, 480))
+
+        # current output color, should be displayed in the preview
+        global LAST_COLOR
+        self._color = LAST_COLOR.copy()
+
+        # output spaces mapped to the `cboOutputSpace` object
+        self._spaces = {
+            0: 'rgba',
+            1: 'rgb',
+            2: 'rgba1',
+            3: 'rgb1',
+            4: 'rgba255',
+            5: 'rgb255',
+            6: 'hex',
+            7: 'hsva',
+            8: 'hsv'
+        }
+        # what to show in the output selection, maps to the spaces above
+        self._outputChoices = [
+            u'PsychoPy RGBA (rgba)',
+            u'PsychoPy RGB (rgb)',
+            u'Normalized RGBA (rgba1)',
+            u'Normalized RGB (rgb1)',
+            u'8-bit RGBA (rgba255)',
+            u'8-bit RGBA (rgb255)',
+            u'Hex/HTML (hex)',
+            u'Hue-Saturation-Value-Alpha (hsva)',
+            u'Hue-Saturation-Value (hsv)'
+        ]
+
+        # initialize the window controls
+        self._setupUI()
+=======
+        wx.Dialog.__init__(self, parent, id=wx.ID_ANY, title=_translate("Color Picker"), pos=wx.DefaultPosition,
+                           style=wx.DEFAULT_DIALOG_STYLE)
+        # Set main params
+        self.color = Color((0,0,0,1), 'rgba')
+        self.sizer = wx.GridBagSizer()
+        # Add colourful top bar
+        self.preview = ColorPreview(color=self.color, parent=self)
+        self.sizer.Add(self.preview, pos=(0,0), span=wx.GBSpan(2,1), border=5, flag=wx.RIGHT | wx.EXPAND)
+        # Add notebook of controls
+        self.ctrls = aui.AuiNotebook(self, wx.ID_ANY, size=wx.Size(400, 400), agwStyle=0)
+        self.sizer.Add(self.ctrls, pos=(0,1), border=5, flag=wx.ALL)
+        self.ctrls.AddPage(ColorPage(self.ctrls, self, 'rgba'), 'RGB (-1 to 1)')
+        self.ctrls.AddPage(ColorPage(self.ctrls, self, 'rgba1'), 'RGB (0 to 1)')
+        self.ctrls.AddPage(ColorPage(self.ctrls, self, 'rgba255'), 'RGB (0 to 255)')
+        self.ctrls.AddPage(ColorPage(self.ctrls, self, 'hsva'), 'HSV')
+        self.ctrls.AddPage(ColorPage(self.ctrls, self, 'hex'), 'Hex')
+        # Add array of named colours
+        self.presets = ColorPresets(parent=self)
+        self.sizer.Add(self.presets, pos=(0,2), border=5, flag=wx.ALL)
+        # Add buttons
+        self.buttons = wx.BoxSizer(wx.HORIZONTAL)
+        self.closeButton = wx.Button(self, label=_translate("Close"))
+        self.closeButton.Bind(wx.EVT_BUTTON, self.Close)
+        self.buttons.Add(self.closeButton, border=5, flag=wx.ALL)
+        # Add insert buttons
+        # self.insertValueButton = wx.Button(self, label="Insert As Value")
+        # self.insertValueButton.Bind(wx.EVT_BUTTON, self.insertValue)
+        # self.buttons.Add(self.insertValueButton, border=5, flag=wx.ALL)
+        # self.insertObjectButton = wx.Button(self, label="Insert As Object")
+        # self.insertObjectButton.Bind(wx.EVT_BUTTON, self.insertObject)
+        # self.buttons.Add(self.insertObjectButton, border=5, flag=wx.ALL)
+        # Add copy buttons
+        self.copyValueButton = wx.Button(self, label=_translate("Copy As Value"))
+        self.copyValueButton.Bind(wx.EVT_BUTTON, self.copyValue)
+        self.buttons.Add(self.copyValueButton, border=5, flag=wx.ALL)
+        self.copyObjectButton = wx.Button(self, label=_translate("Copy As Object"))
+        self.copyObjectButton.Bind(wx.EVT_BUTTON, self.copyObject)
+        self.buttons.Add(self.copyObjectButton, border=5, flag=wx.ALL)
+
+        self.sizer.Add(self.buttons, pos=(1,1), span=wx.GBSpan(1,2), border=5, flag=wx.ALL | wx.ALIGN_RIGHT)
+
+        # Configure sizer
+        self.sizer.AddGrowableRow(0)
+        self.sizer.AddGrowableCol(1)
+        self.SetSizerAndFit(self.sizer)
+        self._applyAppTheme()
+        self._applyAppTheme(self.ctrls)
+>>>>>>> hsogo/feature-localization
+
+    @property
+    def color(self):
+        """Current color the user has specified. Should be reflected in the
+        preview area. Value has type :class:`psychopy.colors.Color`.
+
+        This is the primary setter for the dialog's color value. It will
+        automatically update all color space pages to reflect this color value.
+        Pages should absolutely not attempt to set other page's values directly
+        or risk blowing up the call stack.
+
+        """
+        return self._color
+
+    @color.setter
+    def color(self, value):
+        if value is None:
+            value = Color('none', space='named')
+
+        global LAST_COLOR
+        self.pnlColorPreview.color = self._color = value
+        LAST_COLOR = self._color.copy()
+        self._updateColorSpacePage()
+
+    def _updateColorSpacePage(self):
+        """Update the current colorspace page to reflect the current color being
+        specified by the dialog. Called only on the current page or when the
+        page has been changed. Pointless to update all pages at once.
+
+        """
+        page = self.nbColorSpaces.GetCurrentPage()
+
+        if hasattr(page, 'updateChannels'):
+            page.updateChannels()
+
+        if hasattr(page, 'updateHex'):
+            page.updateHex()
+
+    def _addColorSpacePages(self):
+        """Add pages for each supported color space.
+
+        In the future this will be modified to support plugging in additional
+        color spaces. Right now the pages are hard-coded in.
+
+        """
+        self.pnlRGB = ColorPickerPageRGB(self.nbColorSpaces)
+        self.pnlHSV = ColorPickerPageHSV(self.nbColorSpaces)
+
+        self.nbColorSpaces.AddPage(self.pnlRGB, u"RGB", True)
+        self.nbColorSpaces.AddPage(self.pnlHSV, u"HSV/HSB", False)
+
+    def _setupUI(self):
+        """Setup the UI for the color picker dialog box.
+
+        """
+        szrMain = wx.BoxSizer(wx.VERTICAL)
+
+        # Color area panel. Shows the preview, colorspace pages and presets
+        # panels at the top portion of the dialog.
+        #
+        self.pnlColorArea = wx.Panel(
+            self,
+            wx.ID_ANY,
+            wx.DefaultPosition,
+            wx.DefaultSize,
+            wx.TAB_TRAVERSAL)
+
+        szrColorArea = wx.BoxSizer(wx.HORIZONTAL)
+
+        # color preview panel
+        self.pnlColorPreview = ColorPreview(
+            self.pnlColorArea, self._color)
+        self.pnlColorPreview.SetMinSize(wx.Size(100, -1))
+        self.pnlColorPreview.SetMaxSize(wx.Size(100, -1))
+
+        # color space notebook area
+        self.nbColorSpaces = wx.Notebook(
+            self.pnlColorArea,
+            wx.ID_ANY,
+            wx.DefaultPosition,
+            wx.DefaultSize,
+            0)
+
+        self._addColorSpacePages()  # add the pages to the notebook
+
+        # preset panel
+        self.colorPresets = ColorPresets(self.pnlColorArea)
+        self.colorPresets.SetMinSize(wx.Size(140, -1))
+        self.colorPresets.SetMaxSize(wx.Size(140, -1))
+
+        szrColorArea.Add(self.pnlColorPreview, 0, wx.EXPAND | wx.ALL, 5)
+        szrColorArea.Add(
+            self.nbColorSpaces, 1, wx.EXPAND | wx.TOP | wx.BOTTOM, 5)
+        szrColorArea.Add(self.colorPresets, 0, wx.EXPAND | wx.ALL, 5)
+
+        self.pnlColorArea.SetSizer(szrColorArea)
+        self.pnlColorArea.Layout()
+        szrColorArea.Fit(self.pnlColorArea)
+        szrMain.Add(self.pnlColorArea, 1, wx.ALL| wx.EXPAND, 5)
+
+        # line to divide the color area from the dialog controls, looks neat
+        self.stlMain = wx.StaticLine(
+            self,
+            wx.ID_ANY,
+            wx.DefaultPosition,
+            wx.DefaultSize,
+            wx.LI_HORIZONTAL)
+
+        szrMain.Add(self.stlMain, 0, wx.EXPAND | wx.LEFT | wx.RIGHT, 10)
+
+        # Dialog controls area. Shows buttons for closing the dialog or copying
+        # colors. Also shows the output format to use when copying to the
+        # clipboard.
+        #
+        szrDlgButtonArea = wx.BoxSizer(wx.VERTICAL)
+        szrDlgCtrls = wx.FlexGridSizer(1, 5, 0, 5)
+        szrDlgCtrls.AddGrowableCol(1)
+        szrDlgCtrls.SetFlexibleDirection(wx.BOTH)
+        szrDlgCtrls.SetNonFlexibleGrowMode(wx.FLEX_GROWMODE_SPECIFIED)
+
+        self.lblResult = wx.StaticText(
+            self,
+            wx.ID_ANY,
+            u"Output Space:",
+            wx.DefaultPosition,
+            wx.DefaultSize,
+            0)
+        self.lblResult.Wrap(-1)
+
+        szrDlgCtrls.Add(
+            self.lblResult,
+            0,
+            wx.ALIGN_RIGHT | wx.ALIGN_CENTER_VERTICAL,
+            5)
+
+        self.cboOutputSpace = wx.Choice(
+            self,
+            wx.ID_ANY,
+            wx.DefaultPosition,
+            wx.DefaultSize,
+            choices=self._outputChoices)
+        self.cboOutputSpace.SetSelection(LAST_OUTPUT_SPACE)
+
+        szrDlgCtrls.Add(self.cboOutputSpace, 1, wx.ALIGN_CENTER_VERTICAL, 5)
+
+        self.cmdCopyObject = wx.Button(
+            self, wx.ID_ANY, u"Copy as &Object", wx.DefaultPosition,
+            wx.DefaultSize, 0)
+        self.cmdCopy = wx.Button(
+            self, wx.ID_ANY, u"Copy as &Value", wx.DefaultPosition,
+            wx.DefaultSize, 0)
+        self.cmdClose = wx.Button(
+            self, wx.ID_ANY, u"&Cancel", wx.DefaultPosition, wx.DefaultSize, 0)
+
+        self.cmdCopyObject.SetToolTip(
+            u"Copy as a `psychopy.colors.Color` object.")
+        self.cmdCopy.SetToolTip(
+            u"Copy as a value (tuple).")
+
+        szrDlgCtrls.Add(self.cmdCopyObject, 0, wx.EXPAND, 5)
+        szrDlgCtrls.Add(self.cmdCopy, 0, wx.EXPAND, 5)
+        szrDlgCtrls.Add(self.cmdClose, 0, wx.EXPAND, 5)
+
+        szrDlgButtonArea.Add(szrDlgCtrls, 1, wx.ALL | wx.EXPAND, 5)
+
+        szrMain.Add(szrDlgButtonArea, 0, wx.EXPAND | wx.ALL, 5)
+
+        self.SetSizer(szrMain)
+        self.Layout()
+
+        self.Centre(wx.BOTH)
+
+        # Connect Events
+        self.nbColorSpaces.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.onPageChanged)
+        self.cmdCopyObject.Bind(wx.EVT_BUTTON, self.onCopyObject)
+        self.cmdCopy.Bind(wx.EVT_BUTTON, self.onCopyValue)
+        self.cboOutputSpace.Bind(
+            wx.EVT_CHOICE, self.onOutputSpaceChanged)
+        self.cmdClose.Bind(wx.EVT_BUTTON, self.onClose)
+
+    def _copyToClipboard(self, text):
+        """Copy text to the clipboard.
+
+        Shows an error dialog if the clipboard cannot be opened to inform the
+        user the values have not been copied.
+
+        Parameters
+        ----------
+        text : str
+            Text to copy to the clipboard.
+
+        """
+        # copy the value to the clipboard
+        if wx.TheClipboard.Open():
+            wx.TheClipboard.SetData(wx.TextDataObject(str(text)))
+            wx.TheClipboard.Close()
+        else:
+            # Raised if the clipboard fails to open, warns the user the value
+            # wasn't copied.
+            errDlg = wx.MessageDialog(
+                self,
+                'Failed to open the clipboard, output value not copied.',
+                'Clipboard Error',
+                wx.OK | wx.ICON_ERROR)
+            errDlg.ShowModal()
+            errDlg.Destroy()
+
+    def getOutputValue(self):
+        """Get the string value using the specified output format.
+
+        Returns
+        -------
+        str
+            Color value using the current output format.
+
+        """
+        outputSpace = self.cboOutputSpace.GetSelection()
+        dlgCol = self.GetTopLevelParent().color
+        if outputSpace == 0:  # RGBA
+            colorOut = '({:.4f}, {:.4f}, {:.4f}, {:.4f})'.format(*dlgCol.rgba)
+        elif outputSpace == 1:  # RGB
+            colorOut = '({:.4f}, {:.4f}, {:.4f})'.format(*dlgCol.rgb)
+        elif outputSpace == 2:  # RGBA1
+            colorOut = '({:.4f}, {:.4f}, {:.4f}, {:.4f})'.format(*dlgCol.rgba1)
+        elif outputSpace == 3:  # RGB1
+            colorOut = '({:.4f}, {:.4f}, {:.4f})'.format(*dlgCol.rgb1)
+        elif outputSpace == 4:  # RGBA255
+            colorOut = '({:d}, {:d}, {:d}, {:d})'.format(
+                *[int(i) for i in dlgCol.rgba255])
+        elif outputSpace == 5:  # RGB255
+            colorOut = '({:d}, {:d}, {:d})'.format(
+                *[int(i) for i in dlgCol.rgb255])
+        elif outputSpace == 6:  # Hex
+            colorOut = "'{}'".format(dlgCol.hex)
+        elif outputSpace == 7:  # HSVA
+            colorOut = '({:.4f}, {:.4f}, {:.4f}, {:.4f})'.format(*dlgCol.hsva)
+        elif outputSpace == 8:  # HSV
+            colorOut = '({:.4f}, {:.4f}, {:.4f})'.format(*dlgCol.hsv)
+        else:
+            raise ValueError(
+                "Invalid output color space selection. Have you added any "
+                "more choices to `cboOutputSpace`?")
+
+        return colorOut
+
+    def onOutputSpaceChanged(self, event):
+        """Set when the output space choicebox is changed."""
+        global LAST_OUTPUT_SPACE
+        LAST_OUTPUT_SPACE = event.GetSelection()
+        event.Skip()
+
+    def onPageChanged(self, event):
+        """Called when the color space notebook is changed. Updates the current
+        page to show values appropriate for the current color specified by the
+        dialog.
+
+        """
+        self._updateColorSpacePage()
+        event.Skip()
+
+    def onCopyObject(self, event):
+        """Event to copy the color to the clipboard as a an object.
+
+        """
+        outputSpace = self.cboOutputSpace.GetSelection()
+        toReturn = "colors.Color({}, space='{}')".format(
+            self.getOutputValue(), self._spaces[outputSpace])
+        self._copyToClipboard(toReturn)
+        self.Close()
+        event.Skip()
+
+    def onCopyValue(self, event):
+        """Event to copy the color to the clipboard as a value.
+
+        """
+        self._copyToClipboard(self.getOutputValue())  # copy out to clipboard
+        self.Close()
+        event.Skip()
+
+    def onClose(self, event):
+        """Called when the cancel button is clicked."""
+        self.Close()
+        event.Skip()
+
+
+if __name__ == "__main__":
+    pass

psychopy/app/jobs.py

@@ -101,7 +101,14 @@ class PipeReader(Thread):
     @property
     def isAvailable(self):
         """Are there bytes available to be read (`bool`)?"""
-        return self._queue.full()
+        if self._queue.full():
+            return True
+        elif self._overflowBuffer:  # have leftover bytes
+            self._queue.put("".join(self._overflowBuffer))
+            self._overflowBuffer = []  # clear the overflow buffer
+            return True
+        else:
+            return False
 
     def read(self):
         """Read all bytes enqueued by the thread coming off the pipe. This is