#include "QgsMapToolIdentifyBox.h"
#include <qgsvertexmarker.h>
#include <qgssymbollayer.h>
#include "qgsapplication.h"
#include "QgsMessageLog.h"
QgsMapToolIdentifyAllFeaturesEX::QgsMapToolIdentifyAllFeaturesEX(QgsMapCanvas* canvas)
: QgsMapToolIdentify(canvas)
{
mToolName = tr("Identify All Features");
// set cursor
mCursor = QCursor(Qt::CrossCursor);
}
void QgsMapToolIdentifyAllFeaturesEX::canvasReleaseEvent(QgsMapMouseEvent* e)
{
QPoint point = e->pos() - mInitDragPos;
//点
if (!mSelectionActive || (point.manhattanLength() < QApplication::startDragDistance()))
{
mSelectionActive = false;
mSelectGeometry = QgsGeometry::fromPointXY(toMapCoordinates(e->pos()));
identifyFromGeometry();
}
//矩形
if (mpRubberBand && mSelectionActive)
{
mSelectGeometry = mpRubberBand->asGeometry();
mpRubberBand.reset();
identifyFromGeometry();
}
mSelectionActive = false;
}
void QgsMapToolIdentifyAllFeaturesEX::identifyFromGeometry()
{
if (mCanvas)
{
mCanvas->setSelectionColor(Qt::red);//设置颜色
QList< QgsMapLayer* > layers = mCanvas->layers();
foreach(QgsMapLayer * l, layers)
{
QgsVectorLayer* l1 = qobject_cast<QgsVectorLayer*>(l);
l1->removeSelection();
}
}
//返回选中的结果
QDateTime start = QDateTime::currentDateTime();
QList<IdentifyResult> results = QgsMapToolIdentify::identify(mSelectGeometry, m_mod, LayerType::enum_type::AllLayers);
QDateTime end = QDateTime::currentDateTime();
//QgsMessageLog::logMessage("耗时:"+QString::number((end - start).seconds()), "Messages");
//QList<IdentifyResult> results = QgsMapToolIdentify::identify(mSelectGeometry, m_mod, LayerType::enum_type::AllLayers);
//选择的Features集合
QMap<QString, QList<QgsFeature>> maplist;
//用于显示
QMap<QString,QgsFeatureIds> ids;
for (IdentifyResult var : results)
{
QgsFeature _Feature = var.mFeature;
if (ids.contains(var.mLayer->id())) {
QgsFeatureIds ids1 = ids.value(var.mLayer->id());
ids1.insert(_Feature.id());
ids.insert(var.mLayer->id(), ids1);
}
else {
QgsFeatureIds ids1;
ids1.insert(_Feature.id());
ids.insert(var.mLayer->id(), ids1);
}
}
for (int i = 0; i < results.count(); ++i)
{
QgsVectorLayer* layer = qobject_cast<QgsVectorLayer*>(results.at(i).mLayer);
//只针对开启编辑图层
/*if (!layer->isEditable()) {
continue;
}*/
if (ids.count() > 0)
layer->selectByIds(ids.value(layer->id()));
QgsFeature feature = results.at(i).mFeature;
if (maplist.contains(layer->id())) {
QList<QgsFeature> sfeatures = maplist.value(layer->id());
sfeatures.append(feature);
maplist.insert(layer->id(), sfeatures);
}
else {
QList<QgsFeature> selectFeatures;
selectFeatures.append(feature);
maplist.insert(layer->id(), selectFeatures);
}
}
//发出选中的Feature信息信号
emit AllfeatureIdentified(maplist);
}
void QgsMapToolIdentifyAllFeaturesEX::clearRubberBand()
{
}
bool QgsMapToolIdentifyAllFeaturesEX::init()
{
open();
return true;
}
void QgsMapToolIdentifyAllFeaturesEX::setMode(IdentifyMode mode)
{
m_mod = mode;
}
void QgsMapToolIdentifyAllFeaturesEX::keyPressEvent(QKeyEvent* e)
{
if (e->key() == Qt::Key_Escape)
{
close();
mCanvas->unsetMapTool(this);
clearRubberBand();
}
emit keyPressEventEx(e);
}
void QgsMapToolIdentifyAllFeaturesEX::keyReleaseEvent(QKeyEvent* e)
{
emit keyReleaseEventEX(e);
}
void QgsMapToolIdentifyAllFeaturesEX::canvasPressEvent(QgsMapMouseEvent* e)
{
if (!mpRubberBand) {
mpRubberBand.reset(new QgsRubberBand(mCanvas, Qgis::GeometryType::Polygon));
/*mpRubberBand->setFillColor(mFillColor);
mpRubberBand->setStrokeColor(mStrokeColor);*/
}
mInitDragPos = e->pos();
}
void QgsMapToolIdentifyAllFeaturesEX::canvasMoveEvent(QgsMapMouseEvent* e)
{
if (e->buttons() != Qt::LeftButton)
return;
QRect rect;
if (!mSelectionActive)
{
mSelectionActive = true;
rect = QRect(e->pos(), e->pos());
}
else
{
rect = QRect(e->pos(), mInitDragPos);
}
if (mpRubberBand)
mpRubberBand->setToCanvasRectangle(rect);
}
