JSBTDieBondH/View/die-bonder-ui/ImageView.cpp

#include "ImageView.h"
#include <QPainter>
#include <QDebug>
#include <QGraphicsPixmapItem>
#include <QScrollBar>
#include <QMenu>
#include <QFileDialog>
#include <QDateTime>
#include <QMessageBox>
#include <QApplication>

#define VIEW_CENTER viewport()->rect().center()
#define VIEW_WIDTH  viewport()->rect().width()
#define VIEW_HEIGHT viewport()->rect().height()


ImageView::ImageView(QWidget* parent) :
    QGraphicsView(parent),
    m_imageOffset(0, 0),
    m_bMouseTranslate(false),
    m_isDrawing(false),
    m_curDrawing(false),
    m_isCreating(false),
    m_currentPathItem(nullptr)

{
    //    m_rubberBand = nullptr;
    m_bRulerState = false;
    m_pRuleLine = nullptr;
    setScene(new QGraphicsScene(this));
    setTransformationAnchor(QGraphicsView::AnchorUnderMouse);
    this->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
    this->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
    //设置视图属性
    //setRenderHint(QPainter::SmoothPixmapTransform);
    setRenderHint(QPainter::Antialiasing);
    setMouseTracking(true);
    setDragMode(RubberBandDrag);
}

ImageView::~ImageView()
{
}

void ImageView::setPixmap(const QPixmap& newPixmap) {
    this->m_pixmap = newPixmap;
    scene()->clear();
    m_pixmapItem = scene()->addPixmap(m_pixmap);
    m_pixmapItem->setZValue(1);
    QPointF pixmapCenter(m_pixmap.width() / 2.0, m_pixmap.height() / 2.0);
    m_pixmapItem->setPos(-pixmapCenter);
    QRectF sceneRect(-pixmapCenter.x(), -pixmapCenter.y(), m_pixmap.width(), m_pixmap.height());
    scene()->setSceneRect(sceneRect);
    setSceneRect(scene()->itemsBoundingRect());
    this->resize(m_pixmap.width(), m_pixmap.height());
    m_imageOffset = QPoint(0, 0); // 重置图片偏移量为(0, 0)
    setCursor(Qt::ArrowCursor);
    update(); // 触发重绘
}

void ImageView::setCurPixmap(const QPixmap& newPixmap) {
    if (this->m_pixmap.isNull()) {
        this->m_pixmap = newPixmap;
        scene()->clear();
        m_pixmapItem = scene()->addPixmap(m_pixmap);
        m_pixmapItem->setZValue(1);
        QPointF pixmapCenter(m_pixmap.width() / 2.0, m_pixmap.height() / 2.0);
        m_pixmapItem->setPos(-pixmapCenter);
        QRectF sceneRect(-pixmapCenter.x(), -pixmapCenter.y(), m_pixmap.width(), m_pixmap.height());
        scene()->setSceneRect(sceneRect);
        setSceneRect(scene()->itemsBoundingRect());
        //this->resize(m_pixmap.width(), m_pixmap.height());
        m_imageOffset = QPoint(0, 0); // 重置图片偏移量为(0, 0)
        setCursor(Qt::ArrowCursor);
        update(); // 触发重绘
    }
    else {
        this->m_pixmap = newPixmap;
        m_pixmapItem->setPixmap(m_pixmap);
    }
}

void ImageView::addRuler()
{
    m_bRulerState = true;
}

void ImageView::cancelRuler()
{
    m_bRulerState = false;
}

void ImageView::paintEvent(QPaintEvent* event) {

    QGraphicsView::paintEvent(event);
    QPainter painter(viewport());
    painter.setRenderHint(QPainter::Antialiasing);
    drawCrosshair(painter);
}

void ImageView::mousePressEvent(QMouseEvent* event) {
    if (event->button() == Qt::LeftButton) {
        // 当光标底下没有 item 时，才能移动
        QPointF point = mapToScene(event->pos());
        if (scene()->itemAt(point, transform()) != nullptr
            && scene()->itemAt(point, transform())->type() == QGraphicsPixmapItem::Type) {
            setCursor(Qt::OpenHandCursor);
            m_bMouseTranslate = true;
            m_lastMousePos = event->pos();
        }
    }
    else if (event->button() == Qt::RightButton) {
        m_rightClickPressPos = event->pos();
        m_rightButtonPressed = true;

        if (m_bRulerState) {
            if (!m_isCreating) {
                // 第一次点击：开始创建
                m_isCreating = true;
                m_startPos = mapToScene(event->pos());
                m_pTempLine = scene()->addLine(QLineF(m_startPos, m_startPos),
                    QPen(Qt::gray, 1, Qt::DashLine));
                m_pTempLine->setZValue(2);
            }
            else {
                // 第二次点击：完成创建
                m_isCreating = false;
                QPointF endPos = mapToScene(event->pos());
                scene()->removeItem(m_pTempLine);
                delete m_pTempLine;

                RulerLineItem* ruler = new RulerLineItem(QLineF(m_startPos, endPos));
                ruler->setRulerText(tr("test ruler"));
                ruler->setZValue(2);
                scene()->addItem(ruler);
            }
        }
        event->accept();
    }
    QGraphicsView::mousePressEvent(event);
}

void ImageView::mouseMoveEvent(QMouseEvent* event) {
    if (m_isCreating && m_pTempLine) {
        QPointF currentPos = mapToScene(event->pos());
        m_pTempLine->setLine(QLineF(m_startPos, currentPos));
    }

    if (m_bMouseTranslate) {
        // 计算鼠标当前位置与上次位置的差值
        QPointF delta = event->pos() - m_lastMousePos;
        // 平移视图
        horizontalScrollBar()->setValue(horizontalScrollBar()->value() - delta.x());
        verticalScrollBar()->setValue(verticalScrollBar()->value() - delta.y());
        m_lastMousePos = event->pos();  // 更新鼠标位置
    }
    else if ((event->buttons() & Qt::RightButton) && m_isDrawing && m_rightButtonPressed) {
        if (!m_curDrawing && (event->pos() - m_rightClickPressPos).manhattanLength() > 5) {
            m_curDrawing = true;
            // 根据初始位置开始绘制路径
            QPointF scenePos = mapToScene(m_rightClickPressPos);
            m_currentPath = QPainterPath(scenePos);
            m_currentPathItem = new QGraphicsPathItem();
            m_currentPathItem->setPen(QPen(Qt::red, 2));
            m_currentPathItem->setPath(m_currentPath);
            m_currentPathItem->setZValue(2);
            scene()->addItem(m_currentPathItem);
            m_drawnPaths.append(m_currentPathItem);
        }
        // 如果已经处于绘制状态，则继续添加点
        if (m_curDrawing && m_currentPathItem) {
            QPointF scenePos = mapToScene(event->pos());
            m_currentPath.lineTo(scenePos);
            m_currentPathItem->setPath(m_currentPath);
        }
        event->accept();
        return;
    }
    QGraphicsView::mouseMoveEvent(event);
}

void ImageView::mouseReleaseEvent(QMouseEvent* event) {
    if (event->button() == Qt::LeftButton) {
        setCursor(Qt::ArrowCursor);  // 松开时恢复为箭头
        m_bMouseTranslate = false;
    }
    else if (event->button() == Qt::RightButton && m_rightButtonPressed) {
        if (!m_curDrawing) {
            if (!m_bRulerState) {
                // 右键点击（无拖动），弹出菜单
                QMenu menu(this);
                QAction* act1 = menu.addAction(tr("save picture"));
                QAction* act2 = menu.addAction(tr("clear"));

                connect(act1, &QAction::triggered, this, [this]() {
                    saveImage();
                    });
                connect(act2, &QAction::triggered, this, [this]() {
                    clearDrawnLines();
                    });
                menu.exec(event->globalPos());
            }
        }
        else {
            // 绘制结束
            m_currentPathItem = nullptr;
            m_curDrawing = false;
        }
        // 恢复状态
        m_rightButtonPressed = false;
        event->accept();
        return;
    }
    QGraphicsView::mouseReleaseEvent(event);
}

void ImageView::wheelEvent(QWheelEvent* event)
{
    if (event->orientation() == Qt::Vertical) {
        event->ignore(); // 忽略竖直滚轮事件（即禁用滚动条滑动）
        return;
    }
    event->accept();
}

void ImageView::resizeEvent(QResizeEvent* event)
{
    QGraphicsView::resizeEvent(event);
    viewport()->update();  // 窗口大小改变时重绘十字线
}

void ImageView::scrollContentsBy(int dx, int dy)
{
    QGraphicsView::scrollContentsBy(dx, dy);
    viewport()->update();  // 滚动时更新十字线
}

void ImageView::drawCrosshair(QPainter& painter)
{
    const QRect viewRect = viewport()->rect();
    int resolutionWidth = viewRect.width();
    int resolutionHeight = viewRect.height();
    painter.setPen(QPen(QColor(139, 105, 20, 255), 1));
    // 绘制垂直线
    painter.drawLine(resolutionWidth / 2, 0, resolutionWidth / 2, resolutionHeight);
    // 绘制水平线
    painter.drawLine(0, resolutionHeight / 2, resolutionWidth, resolutionHeight / 2);

    int centerX = resolutionWidth / 2;
    int centerY = resolutionHeight / 2;

    int rulerLength = viewRect.width(); // 量尺长度
    int tickSpacing = 5; // 刻度间距

    // 绘制水平中心线上的量尺
    int rulerStartXHorizontal = centerX;
    int rulerStartYHorizontal = centerY;

    for (int i = 0; i <= rulerLength / 2; i += tickSpacing) {
        int tickHeight = 0;
        if (i % (tickSpacing * 10) == 0) { // 长刻度
            tickHeight = 15;
        }
        else if (i % (tickSpacing * 5) == 0) { // 中等刻度
            tickHeight = 10;
        }
        else { // 短刻度
            tickHeight = 5;
        }

        painter.drawLine(rulerStartXHorizontal + i, rulerStartYHorizontal, rulerStartXHorizontal + i, rulerStartYHorizontal - tickHeight); // 绘制刻度线
        painter.drawLine(rulerStartXHorizontal - i, rulerStartYHorizontal, rulerStartXHorizontal - i, rulerStartYHorizontal - tickHeight); // 绘制刻度线
    }

    // 绘制垂直中心线上的量尺
    int rulerStartXVertical = centerX; 
    int rulerStartYVertical = centerY;

    for (int i = 0; i <= rulerLength / 2; i += tickSpacing) {
        int tickWidth = 0;
        if (i % (tickSpacing * 10) == 0) { // 长刻度
            tickWidth = 15;
        }
        else if (i % (tickSpacing * 5) == 0) { // 中等刻度
            tickWidth = 10;
        }
        else { // 短刻度
            tickWidth = 5;
        }
        painter.drawLine(rulerStartXVertical, rulerStartYVertical - i, rulerStartXVertical - tickWidth, rulerStartYVertical - i);
        painter.drawLine(rulerStartXVertical, rulerStartYVertical + i, rulerStartXVertical - tickWidth, rulerStartYVertical + i); 
    }
}

void ImageView::saveImage()
{
    // 使用当前时间生成默认文件名
    QString defaultFileName = QDateTime::currentDateTime().toString("yyyy-MM-dd_HH-mm-ss") + ".png";

    QString fileName = QFileDialog::getSaveFileName(
        this,
        tr("保存图片"),
        QDir::homePath() + "/" + defaultFileName, // 默认路径为用户目录，文件名为当前时间
        tr("PNG 文件 (*.png);;JPEG 文件 (*.jpg);;所有文件 (*)")
    );
    if (!fileName.isEmpty()) {
        QImage image(viewport()->size(), QImage::Format_ARGB32);
        image.fill(Qt::white);
        QPainter painter(&image);
        this->render(&painter);
        if (!image.save(fileName)) {
            QMessageBox::warning(this, tr("保存失败"), tr("无法保存图片到指定路径。"));
        }
        else {
            QMessageBox::information(this, tr("保存成功"), tr("图片已成功保存到：") + fileName);
        }
        painter.end();
    }
}


void ImageView::clearDrawnLines()
{
    qDeleteAll(m_drawnPaths);
    m_drawnPaths.clear();
    foreach(QGraphicsItem * item, scene()->selectedItems()) {
        if (nullptr == item)
            continue;
        scene()->removeItem(item);
    }
}