Example #1
0
 def save(self, fname, format, draft):
     if is_text_string(fname):
         if format == "pdf":
             self.app = guidata.qapplication()
             if draft:
                 mode = QPrinter.ScreenResolution
             else:
                 mode = QPrinter.HighResolution
             printer = QPrinter(mode)
             printer.setOutputFormat(QPrinter.PdfFormat)
             printer.setOrientation(QPrinter.Landscape)
             printer.setOutputFileName(fname)
             printer.setCreator("guiqwt.pyplot")
             self.print_(printer)
         else:
             if self.win is None:
                 self.show()
             if PYQT5:
                 pixmap = self.win.centralWidget().grab()
             else:
                 pixmap = QPixmap.grabWidget(self.win.centralWidget())
             pixmap.save(fname, format.upper())
     else:
         # Buffer
         fd = fname
         assert hasattr(fd, "write"), "object is not file-like as expected"
         if self.win is None:
             self.show()
         pixmap = QPixmap.grabWidget(self.win.centralWidget())
         buff = QBuffer()
         buff.open(QIODevice.ReadWrite)
         pixmap.save(buff, format.upper())
         fd.write(buff.data())
         buff.close()
         fd.seek(0)
Example #2
0
 def copy_to_clipboard(self):
     """Copy widget's window to clipboard"""
     clipboard = QApplication.clipboard()
     if PYQT5:
         pixmap = self.grab()
     else:
         pixmap = QPixmap.grabWidget(self)
     clipboard.setPixmap(pixmap)
Example #3
0
 def mousePressEvent(self, event):
     if event.button() == Qt.LeftButton:
         if PYQT5:
             pm = self.grab()
         else:
             pm = QPixmap.grabWidget(self)
         color = QColor()
         color.setRgb(pm.toImage().pixel(event.x(), event.y()))
         self.colorSelected.emit(color)
         event.accept()
Example #4
0
def take_screenshot(widget):
    """Take screenshot and save it to the data folder"""
    if PYQT5:
        pixmap = widget.grab()
    else:
        pixmap = QPixmap.grabWidget(widget)
    bname = (widget.objectName().lower() + ".png").replace("window", "")
    bname = bname.replace("plot", "").replace("widget", "")
    pixmap.save(osp.join(TEST_PATH, "data", bname))
    QTimer.singleShot(0, QApplication.instance().quit)
Example #5
0
def copy(fig=None):
    if fig is None:
        canvas = plt.gcf().canvas
    else:
        canvas = fig.canvas

    canvas.draw()
    try:
        pixmap = QPixmap.grabWidget(canvas)
    except:  # PyQt5+
        pixmap = canvas.grab()
    QApplication.clipboard().setPixmap(pixmap)
    print("Image copied to clipboard.")
Example #6
0
def take_picture(widget: QWidget, filename: str) -> str:
    """
    Takes a screenshot and saves it to the
    filename given, ensuring the call is processed
    through a slot if the call is from a separate
    thread
    """
    if hasattr(widget, 'grab'):
        pix = widget.grab()
    else:
        pix = QPixmap.grabWidget(widget)
    success = pix.save(filename)
    if not success:
        raise RuntimeError(f"Error saving widget image to file '{filename}'")
Example #7
0
 def save_widget(self, fname):
     """Grab widget's window and save it to filename (\*.png, \*.pdf)"""
     fname = to_text_string(fname)
     if fname.lower().endswith(".pdf"):
         printer = QPrinter()
         printer.setOutputFormat(QPrinter.PdfFormat)
         printer.setOrientation(QPrinter.Landscape)
         printer.setOutputFileName(fname)
         printer.setCreator("guidata")
         self.print_(printer)
     elif fname.lower().endswith(".png"):
         if PYQT5:
             pixmap = self.grab()
         else:
             pixmap = QPixmap.grabWidget(self)
         pixmap.save(fname, "PNG")
     else:
         raise RuntimeError(_("Unknown file extension"))
Example #8
0
    def setPaintAttribute(self, attribute, on=True):
        """
        Changing the paint attributes

        Paint attributes:
        
            * `QwtPlotCanvas.BackingStore`
            * `QwtPlotCanvas.Opaque`
            * `QwtPlotCanvas.HackStyledBackground`
            * `QwtPlotCanvas.ImmediatePaint`
        
        :param int attribute: Paint attribute
        :param bool on: On/Off
        
        .. seealso::
        
            :py:meth:`testPaintAttribute()`, :py:meth:`backingStore()`
        """
        if bool(self.__data.paintAttributes & attribute) == on:
            return
        if on:
            self.__data.paintAttributes |= attribute
        else:
            self.__data.paintAttributes &= ~attribute
        if attribute == self.BackingStore:
            if on:
                if self.__data.backingStore is None:
                    self.__data.backingStore = QPixmap()
                if self.isVisible():
                    if QT_MAJOR_VERSION >= 5:
                        self.__data.backingStore = self.grab(self.rect())
                    else:
                        self.__data.backingStore = QPixmap.grabWidget(
                            self, self.rect())
            else:
                self.__data.backingStore = None
        elif attribute == self.Opaque:
            if on:
                self.setAttribute(Qt.WA_OpaquePaintEvent, True)
        elif attribute in (self.HackStyledBackground, self.ImmediatePaint):
            pass