add screen shot code

andrewbytecoder · andrewbytecoder · commit 0ef23b7fade7 · 2023-10-12T18:20:26.000+08:00
diff --git a/qt_screenshot/core/screen_list.cpp b/qt_screenshot/core/screen_list.cpp
@@ -0,0 +1,5 @@
+//
+// Created by wangyz38535 on 2023/10/12.
+//
+
+#include "screen_list.h"
diff --git a/qt_screenshot/core/screen_list.h b/qt_screenshot/core/screen_list.h
@@ -0,0 +1,57 @@
+//
+// Created by wangyz38535 on 2023/10/12.
+//
+
+#ifndef FIRESHOT_SCREEN_LIST_H
+#define FIRESHOT_SCREEN_LIST_H
+
+
+
+#include <QPainter>
+#include <QRect>
+
+#include <memory>
+#include <QScreen>
+#include <QPixmap>
+#include <QRect>
+
+struct ScreenInfo{
+    QScreen *object;
+    QRect boundary;
+    QPixmap pixmap;
+};
+
+/**
+ * @class : ScreenList
+ * @brief : 屏幕列表集合
+ * @note  : 屏幕列表集合基本功能
+ */
+class ScreenList
+{
+public:
+    ScreenList(QList<ScreenInfo> list);
+
+    QRect allBoundary(bool isGlobal=false) const;
+    QPixmap allPixMap() const;
+
+    QRect boundaryAt(int screenIndex,bool isGlobal=false);
+    QScreen *screenAt(int screenIndex);
+    QPixmap pixmapAt(int screenIndex);
+
+    QRect toLocal(QRect rect);
+    float scale();
+
+    void draw(QPainter &painter);
+    void draw(QPainter &painter,QPainterPath maskPath,QBrush maskBrush);
+
+private:
+    QRect m_allBoundary;
+    std::shared_ptr<QPixmap> m_allPixMap;
+    QList<ScreenInfo> m_List;
+
+    void initParams();
+};
+
+
+
+#endif //FIRESHOT_SCREEN_LIST_H
diff --git a/qt_screenshot/core/screen_shot.cpp b/qt_screenshot/core/screen_shot.cpp
@@ -1,5 +1,27 @@
 //
 // Created by wangyz38535 on 2023/10/12.
 //
-
+#include "utils/spdlog_wrapper.h"
 #include "screen_shot.h"
+#include "fireshot.h"
+
+ScreenShot::ScreenShot(FireShot *lpFireShot) : QObject(lpFireShot), m_lpFireShot(lpFireShot)
+{
+
+}
+
+ScreenShot::~ScreenShot() {
+    SPD_FUNCTION();
+}
+
+void ScreenShot::cleanup() {
+
+}
+
+void ScreenShot::rasie() {
+
+}
+
+void ScreenShot::init(QWidget *parent) {
+
+}
diff --git a/qt_screenshot/core/screen_shot.h b/qt_screenshot/core/screen_shot.h
@@ -5,9 +5,39 @@
 #ifndef FIRESHOT_SCREEN_SHOT_H
 #define FIRESHOT_SCREEN_SHOT_H
 
+#include <QObject>
+#include <QWidget>
+#include <QList>
+//#include "widget.h"
 
-class screen_shot {
+class FireShot;
 
+class ScreenShot : public QObject {
+    Q_OBJECT
+public:
+    explicit ScreenShot(FireShot *lpFireShot);
+    ~ScreenShot() override;
+
+    void init(QWidget* parent);
+
+    void cleanup();
+
+    void rasie();
+
+private:
+    //QList<Widget*>* m_widgets;
+    //QList<Widget*>* m_unused_widgets;
+
+signals:
+    void ShotDone(Starter*);
+    void ShotResult(std::shared_ptr<QPixmap> pixmap);
+
+private slots:
+    void finishShot(int code);
+    void finishConfirmArea();
+
+private:
+    FireShot *m_lpFireShot;
 };
 
 
diff --git a/qt_screenshot/core/screen_widget.cpp b/qt_screenshot/core/screen_widget.cpp
@@ -0,0 +1,14 @@
+//
+// Created by wangyz38535 on 2023/10/12.
+//
+
+
+#include "screen_widget.h"
+
+ScreenWidgets::ScreenWidgets(QWidget *parent) : QWidget(parent) {
+
+}
+
+ScreenWidgets::~ScreenWidgets() {
+
+}
diff --git a/qt_screenshot/core/screen_widget.h b/qt_screenshot/core/screen_widget.h
@@ -0,0 +1,27 @@
+//
+// Created by wangyz38535 on 2023/10/12.
+//
+
+#ifndef FIRESHOT_SCREEN_WIDGET_H
+#define FIRESHOT_SCREEN_WIDGET_H
+
+#include <QWidget>
+#include <QPixmap>
+#include <QString>
+#include <QPainter>
+
+
+class ScreenWidgets : public QWidget {
+    Q_OBJECT
+public:
+    explicit ScreenWidgets(QWidget *parent);
+    ~ScreenWidgets() override;
+
+
+
+private:
+
+};
+
+
+#endif //FIRESHOT_SCREEN_WIDGET_H
diff --git a/qt_screenshot/core/widget.cpp b/qt_screenshot/core/widget.cpp
@@ -0,0 +1,163 @@
+//
+// Created by wangyz38535 on 2023/10/12.
+//
+
+#include "widget.h"
+#include <QPainter>
+#include <QBitmap>
+#include "utils/spdlog_wrapper.h"
+
+#define UNVISIABLE_X -819200
+#define UNVISIABLE_Y -819200
+
+Widget::Widget(QWidget* parent)
+        : QWidget(parent)
+        , m_workspace(new Workspace(this))
+        , m_screenlist(nullptr)
+        , m_status("unknown")
+{
+#ifndef USE_SPDLOG_
+    setWindowFlags(Qt::Tool);
+    setAttribute(Qt::WA_TranslucentBackground);
+#endif // USE_SPDLOG_
+}
+
+Widget::~Widget()
+{
+    SPD_ERROR("{0}", __FUNCTION__);
+}
+
+void Widget::start(std::shared_ptr<ScreenList> list, int index)
+{
+    SPD_FUNCTION();
+
+    QRect geometry = list->allBoundary(true);
+    setGeometry(geometry);
+
+    showFullScreen();
+
+    raise();
+    activateWindow();
+
+    setCursor(Qt::ArrowCursor);
+
+    m_screenlist = list;
+    m_workspace->start(m_screenlist, index);
+}
+
+void Widget::cleanup()
+{
+    SPD_FUNCTION();
+    hide();
+
+    SPD_TRACE("is visable = {0} && w: {1}, h: {2} @ {3}", this->isVisible(), size().width(), size().height(), __FUNCTION__);
+
+    m_status = "unknown";
+    m_workspace->cleanup();
+
+    update();
+}
+
+Workspace* Widget::workspace() const
+{
+    return m_workspace;
+}
+
+void Widget::finishConfirmArea()
+{
+    if (m_workspace->areaConfirmed() == true)
+        m_status = "active";
+    else
+        m_status = "giveup";
+}
+
+void Widget::showEvent(QShowEvent* event)
+{
+    SPD_FUNCTION();
+
+    raise();
+    activateWindow();
+
+    setMouseTracking(true);
+
+    QWidget::showEvent(event);
+    SPD_WARN("is visable = {0} && x: {1}, y: {2}, w: {3}, h: {4} @ {5} --- {6}"
+            , this->isVisible(), pos().x(), pos().y(), size().width(), size().height(), m_status.toStdString().c_str()
+            , __FUNCTION__);
+}
+
+void Widget::hideEvent(QHideEvent* event)
+{
+    setMouseTracking(false);
+
+    resize(0, 0);
+
+    QWidget::hideEvent(event);
+
+    SPD_INFO("is visable = {0} && x: {1}, y: {2}, w: {3}, h: {4} @ {5} --- {6}"
+            , this->isVisible(), pos().x(), pos().y(), size().width(), size().height(), m_status.toStdString().c_str()
+            , __FUNCTION__);
+}
+
+void Widget::closeEvent(QCloseEvent* event)
+{
+    event->ignore();
+
+    SPD_TRACE("{0} stat = {1}", __FUNCTION__, m_status.toStdString().c_str());
+
+    hide();
+}
+
+void Widget::mousePressEvent(QMouseEvent* event)
+{
+    if (m_status != "giveup")
+        m_workspace->onMousePress(event);
+}
+
+void Widget::mouseMoveEvent(QMouseEvent* event)
+{
+    SPD_TRACE("{0} stat = {1}", __FUNCTION__, m_status.toStdString().c_str());
+    if (m_status != "giveup")
+        m_workspace->onMouseMove(event);
+}
+
+void Widget::mouseReleaseEvent(QMouseEvent* event)
+{
+    if (m_status != "giveup")
+        m_workspace->onMouseRelease(event);
+}
+
+void Widget::keyPressEvent(QKeyEvent* event)
+{
+    m_workspace->onKeyPress(event);
+}
+
+void Widget::paintEvent(QPaintEvent* event)
+{
+    SPD_FUNCTION();
+
+    Q_UNUSED(event);
+
+    QPainter painter(this);
+
+    m_workspace->draw(painter);
+}
+
+void Widget::enterEvent(QEvent* event)
+{
+    SPD_FUNCTION();
+    QWidget::enterEvent(event);
+}
+
+void Widget::leaveEvent(QEvent* event)
+{
+    SPD_FUNCTION();
+    //     if (m_workspace->areaConfirmed() == false)
+    //     {
+    //         m_workspace->setAreaBoundary(QRect(0,0,0,0));
+    //         update();
+    //     }
+
+    QWidget::leaveEvent(event);
+}
+
diff --git a/qt_screenshot/core/widget.h b/qt_screenshot/core/widget.h
@@ -0,0 +1,52 @@
+//
+// Created by wangyz38535 on 2023/10/12.
+//
+
+#ifndef FIRESHOT_WIDGET_H
+#define FIRESHOT_WIDGET_H
+
+
+
+#include <QWidget>
+#include <screen/workspace.h>
+#include <QPixmap>
+#include <QPainter>
+
+#include "core/screenlist.h"
+
+#include <memory>
+
+class Widget : public QWidget
+{
+Q_OBJECT
+
+public:
+    Widget(QWidget *parent = nullptr);
+    ~Widget();
+
+    void start(std::shared_ptr<ScreenList> list, int index);
+    void cleanup();
+
+    void finishConfirmArea();
+
+protected:
+    void showEvent(QShowEvent* event) override;
+    void hideEvent(QHideEvent* event) override;
+    void closeEvent(QCloseEvent* event) override;
+
+    void mousePressEvent(QMouseEvent *) override;
+    void mouseReleaseEvent(QMouseEvent *) override;
+    void mouseMoveEvent(QMouseEvent *) override;
+    void keyPressEvent(QKeyEvent *) override;
+    void paintEvent(QPaintEvent *) override;
+    void enterEvent(QEvent* event) override;
+    void leaveEvent(QEvent *event) override;
+
+private:
+    QString m_status;  //unknown, active, giveup
+    Workspace* m_workspace;
+    std::shared_ptr<ScreenList> m_screenlist;
+};
+
+
+#endif //FIRESHOT_WIDGET_H
diff --git a/qt_screenshot/fireshot.cpp b/qt_screenshot/fireshot.cpp
diff --git a/qt_screenshot/fireshot.h b/qt_screenshot/fireshot.h

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +//
 +// Created by wangyz38535 on 2023/10/12.
 +//
++
 +#include "screen_list.h"