118 lines
3.6 KiB
C++
118 lines
3.6 KiB
C++
|
#include "draglinehandler.h"
|
||
|
|
|
||
|
|
DragLineHandler::DragLineHandler(QCustomPlot *plot, QCPItemStraightLine *line1,
|
||
|
|
QCPItemStraightLine *line2, QCPGraph*graph,
|
||
|
|
QObject *parent)
|
||
|
|
: QObject(parent), mPlot(plot), mLine1(line1), mLine2(line2),mGraph(graph)
|
||
|
|
,_enableFlag(false)
|
||
|
|
{
|
||
|
|
mPlot->setInteractions(QCP::iSelectItems);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool DragLineHandler::eventFilter(QObject *obj, QEvent *event)
|
||
|
|
{
|
||
|
|
if(!_enableFlag){
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (obj == mPlot)
|
||
|
|
{
|
||
|
|
if (event->type() == QEvent::MouseMove)
|
||
|
|
{
|
||
|
|
QMouseEvent *mouseEvent = static_cast<QMouseEvent*>(event);
|
||
|
|
QPoint mousePos = mouseEvent->pos();
|
||
|
|
qDebug()<<"x:"<<mousePos.x();
|
||
|
|
|
||
|
|
bool nearLine1 = isNearLine(mLine1, mousePos);
|
||
|
|
bool nearLine2 = isNearLine(mLine2, mousePos);
|
||
|
|
|
||
|
|
if (nearLine1 || nearLine2)
|
||
|
|
{
|
||
|
|
mPlot->setCursor(Qt::SplitHCursor);
|
||
|
|
|
||
|
|
qDebug()<<"mIconLabel visiable.";
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
qDebug()<<"mIconLabel not visiable.";
|
||
|
|
|
||
|
|
mPlot->setCursor(Qt::ArrowCursor);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (mDraggingLine)
|
||
|
|
{
|
||
|
|
double x = mPlot->xAxis->pixelToCoord(mousePos.x());
|
||
|
|
mDraggingLine->point1->setCoords(x, mPlot->yAxis->range().lower);
|
||
|
|
mDraggingLine->point2->setCoords(x, mPlot->yAxis->range().upper);
|
||
|
|
mPlot->replot();
|
||
|
|
updateSelectedRegion();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (event->type() == QEvent::MouseButtonPress)
|
||
|
|
{
|
||
|
|
QMouseEvent *mouseEvent = static_cast<QMouseEvent*>(event);
|
||
|
|
if (mouseEvent->button() == Qt::LeftButton)
|
||
|
|
{
|
||
|
|
QPoint mousePos = mouseEvent->pos();
|
||
|
|
if (isNearLine(mLine1, mousePos))
|
||
|
|
{
|
||
|
|
mDraggingLine = mLine1;
|
||
|
|
mLine1->setSelected(true);
|
||
|
|
}
|
||
|
|
else if (isNearLine(mLine2, mousePos))
|
||
|
|
{
|
||
|
|
mDraggingLine = mLine2;
|
||
|
|
mLine2->setSelected(true);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (event->type() == QEvent::MouseButtonRelease)
|
||
|
|
{
|
||
|
|
if (mDraggingLine)
|
||
|
|
{
|
||
|
|
mDraggingLine->setSelected(false);
|
||
|
|
mDraggingLine = nullptr;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return QObject::eventFilter(obj, event);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool DragLineHandler::isNearLine(QCPItemStraightLine *line, const QPoint &pos)
|
||
|
|
{
|
||
|
|
double lineX = line->point1->coords().x();
|
||
|
|
int linePixelX = mPlot->xAxis->coordToPixel(lineX);
|
||
|
|
return qAbs(pos.x() - linePixelX) < 5;
|
||
|
|
}
|
||
|
|
|
||
|
|
void DragLineHandler::updateSelectedRegion()
|
||
|
|
{
|
||
|
|
double x1 = mLine1->point1->coords().x();
|
||
|
|
double x2 = mLine2->point1->coords().x();
|
||
|
|
if (x1 > x2)
|
||
|
|
std::swap(x1, x2);
|
||
|
|
|
||
|
|
// 获取曲线数据
|
||
|
|
QVector<double> xData, yData;
|
||
|
|
// mGraph->data()->data(xData, yData);
|
||
|
|
auto dataContainer = mGraph->data();
|
||
|
|
for (auto it = dataContainer->begin(); it != dataContainer->end(); ++it)
|
||
|
|
{
|
||
|
|
xData.append(it->key);
|
||
|
|
yData.append(it->value);
|
||
|
|
}
|
||
|
|
|
||
|
|
// 查找对应x位置
|
||
|
|
int index1 = findClosestIndex(xData, x1);
|
||
|
|
int index2 = findClosestIndex(xData, x2);
|
||
|
|
|
||
|
|
qDebug() << "Selected region: [" << xData[index1] << ", " << xData[index2] << "]";
|
||
|
|
}
|
||
|
|
|
||
|
|
int DragLineHandler::findClosestIndex(const QVector<double> &data, double target)
|
||
|
|
{
|
||
|
|
auto it = std::min_element(data.begin(), data.end(), [target](double a, double b) {
|
||
|
|
return qAbs(a - target) < qAbs(b - target);
|
||
|
|
});
|
||
|
|
return std::distance(data.begin(), it);
|
||
|
|
}
|