select to import button added. only X11 supported yet

Author: sulincix

data/images/page/screenshot-crop.svg

@@ -12,6 +12,7 @@ src = [
     'src/widgets/FloatingSettings.cpp',
     'src/widgets/WhiteBoard.cpp',
     'src/widgets/Background.cpp',
+    'src/widgets/ScreenshotWidget.cpp',
     'src/widgets/Button.cpp',
     'src/tools/update.cpp',
     'src/tools/icon.cpp',

meson.build

@@ -157,7 +157,11 @@ void setupWidgets(){
 
     desktopWidget->num_of_rows = 1;
     desktopWidget->addWidget(toolButtons[UNMINIFY]);
-
+#ifdef screenshot
+    if(!is_wayland && !force_xwayland){ // X11 only feature (yet)
+        desktopWidget->addWidget(toolButtons[SCREENSHOT_CROP]);
+    }
+#endif
 /********** desktop menu done **********/
 
 

src/SetupWidgets.cpp

@@ -53,3 +53,4 @@
 #define CURSOR 74
 #define SHAPEMENU 75
 #define UNMINIFY 76
+#define SCREENSHOT_CROP 77

src/constants.h

@@ -15,6 +15,7 @@
 bool is_wayland;
 bool is_etap = false;
 extern void mainWindowInit();
+bool force_xwayland = false;
 int history;
 
 static bool detect_etap(){
@@ -52,7 +53,6 @@ int main(int argc, char *argv[]) {
     is_wayland = (getenv("WAYLAND_DISPLAY") != NULL);
     // gnome wayland fullscreen compositor is buggy.
     // Force prefer Xwayland for fix this issue.
-    bool force_xwayland = false;
     if(getenv("XDG_CURRENT_DESKTOP")){
         force_xwayland = strncmp(getenv("XDG_CURRENT_DESKTOP"), "gnome", 5) || \
             strncmp(getenv("XDG_CURRENT_DESKTOP"), "GNOME", 5);

src/main.cpp

@@ -1,5 +1,6 @@
 
 #include "widgets/DrawingWidget.h"
+#include "widgets/ScreenshotWidget.h"
 #include "widgets/FloatingWidget.h"
 #include "widgets/FloatingSettings.h"
 #include "widgets/WhiteBoard.h"
@@ -65,6 +66,7 @@ extern QWidget *bgMenu;
 extern float scale;
 extern int history;
 extern bool is_wayland;
+extern bool force_xwayland;
 extern bool is_etap;
 
 extern QSlider *scrollHSlider;

src/tools.h

@@ -74,6 +74,8 @@ char* get_icon_by_id(int id){
             return (char*)":images/page/close.svg";
         case SCREENSHOT:
             return (char*)":images/page/screenshot.svg";
+        case SCREENSHOT_CROP:
+            return (char*)":images/page/screenshot-crop.svg";
         case CURSOR:
             return (char*)":images/cursor.svg";
         case MINIFY:

src/tools/icon.cpp

@@ -1,6 +1,22 @@
 #include "../tools.h"
+
 #include <QTimer>
 
+#ifdef screenshot
+
+static void setCropScreenShot(QPixmap pix){
+    drawing->setOverlay(pix.toImage(), drawing->getPageNum());
+    board->setType(WHITE);
+    board->setOverlayType(CUSTOM);
+    board->rotates[drawing->getPageNum()] = 0;
+    board->ratios[drawing->getPageNum()] = pix.width() * 100 / board->geometry().width();
+    board->updateTransform();
+    setHideMainWindow(false);
+
+    printf("%s\n", "crop done");
+}
+#endif
+
 void setupScreenShot(){
     #ifdef screenshot
     QTimer *ssTimer = new QTimer();
@@ -17,5 +33,21 @@ void setupScreenShot(){
 
     });
     set_shortcut(toolButtons[SCREENSHOT], Qt::Key_F2, 0);
+
+    ScreenshotWidget* widget = new ScreenshotWidget();
+    widget->crop_signal = setCropScreenShot;
+    QTimer *sscTimer = new QTimer();
+    QObject::connect(sscTimer, &QTimer::timeout, [=](){
+        sscTimer->stop();
+        widget->showFullScreen();
+        floatingWidget->show();
+
+    });
+    toolButtons[SCREENSHOT_CROP] = create_button(SCREENSHOT_CROP, [=](){
+        floatingWidget->hide();
+        floatingSettings->setHide();
+        sscTimer->start(500);
+    });
+    set_shortcut(toolButtons[SCREENSHOT_CROP], Qt::Key_F2, Qt::ControlModifier);
     #endif
 }

src/tools/screenshot.cpp

@@ -0,0 +1,51 @@
+#include <QApplication>
+#include <QScreen>
+#include <QPainter>
+#include <QDebug>
+#include <QRect>
+
+#include "ScreenshotWidget.h"
+
+
+ScreenshotWidget::ScreenshotWidget(QWidget *parent)
+    : QWidget(parent), selecting(false) {
+    setWindowFlags(Qt::FramelessWindowHint | Qt::WindowStaysOnTopHint);
+    setCursor(Qt::CrossCursor);
+    hide();
+}
+
+void ScreenshotWidget::mousePressEvent(QMouseEvent *event) {
+    startPos = event->pos();
+}
+
+void ScreenshotWidget::mouseMoveEvent(QMouseEvent *event) {
+    endPos = event->pos();
+    update();
+}
+
+void ScreenshotWidget::mouseReleaseEvent(QMouseEvent *event) {
+    endPos = event->pos();
+    cropScreenshot();
+    startPos = QPoint(-1, -1);
+    endPos = QPoint(-1, -1);
+    update();
+}
+
+void ScreenshotWidget::paintEvent(QPaintEvent *) {
+    QPainter painter(this);
+    painter.setPen(Qt::blue);
+    painter.setBrush(QColor(0, 0, 255, 50)); // Semi-transparent blue
+    QRect rect(startPos, endPos);
+    painter.drawRect(rect.normalized());
+}
+
+void ScreenshotWidget::cropScreenshot() {
+    QScreen *screen = QGuiApplication::primaryScreen();
+    QPixmap originalPixmap = screen->grabWindow(0);
+
+    QRect cropRect(startPos, endPos);
+    QPixmap croppedPixmap = originalPixmap.copy(cropRect.normalized());
+    crop_signal(croppedPixmap);
+    hide();
+}
+

src/widgets/ScreenshotWidget.cpp

@@ -0,0 +1,28 @@
+#ifndef SCREENSHOTWIDGET_H
+#define SCREENSHOTWIDGET_H
+
+#include <QWidget>
+#include <QPixmap>
+#include <QMouseEvent>
+
+class ScreenshotWidget : public QWidget {
+
+public:
+    ScreenshotWidget(QWidget *parent = nullptr);
+    void (*crop_signal)(QPixmap pixmap);
+
+protected:
+    void mousePressEvent(QMouseEvent *event) override;
+    void mouseMoveEvent(QMouseEvent *event) override;
+    void mouseReleaseEvent(QMouseEvent *event) override;
+    void paintEvent(QPaintEvent *event) override;
+
+private:
+    void cropScreenshot();
+
+    QPoint startPos;
+    QPoint endPos;
+    bool selecting;
+};
+
+#endif