【QGIS二次开发】地图编辑-04

QgsMapToolEditPoint::QgsMapToolEditPoint(QgsMapCanvas* mapcanvas, QgsVectorLayer* currentlayer) : QgsMapTool(mapcanvas), m_mapCanvas(mapcanvas), m_currentLayer(currentlayer)  
{  m_mapCanvas = mapcanvas;  m_currentLayer = currentlayer;  
}  QgsMapToolEditPoint::~QgsMapToolEditPoint() {};  void  QgsMapToolEditPoint::canvasPressEvent(QgsMapMouseEvent* e) {  if (!m_currentLayer)  {  // 处理图层无效的情况  return;  }  // 从函数体内获取点击的点坐标  QgsMapCanvas* canvas = this->canvas();  if (!canvas)  {  // 处理无效的画布  return;  }  QgsPointXY point = canvas->getCoordinateTransform()->toMapCoordinates(canvas->mouseLastXY());  m_currentLayer->startEditing();  QgsFeature feature;  // 创建新要素，传递用户点击的点坐标  feature.setGeometry(QgsGeometry::fromPointXY(point));  QgsFields fields = m_currentLayer->fields();  feature.setFields(fields);  QDialog inputDialog;  QFormLayout formLayout(&inputDialog);  QList<QString> attributeNames;  QList<QVariant> attributeValues;  for (const QgsField& field : fields)  {  QString attributeName = field.name();  QVariant::Type attributeType = field.type();  QString labelText = "输入属性 " + attributeName + " 的属性值:";  QString defaultValue = "";  // Set default value if needed  QLineEdit* inputLineEdit = new QLineEdit(&inputDialog);  inputLineEdit->setText(defaultValue);  formLayout.addRow(labelText, inputLineEdit);  attributeNames.append(attributeName);  }  QPushButton* okButton = new QPushButton("OK", &inputDialog);  QPushButton* cancelButton = new QPushButton("CANCEL", &inputDialog);  formLayout.addRow(okButton);  formLayout.addRow(cancelButton);  QObject::connect(okButton, &QPushButton::clicked, [&]() {  QMap<QString, QVariant> attributeMap;  for (int i = 0; i < attributeNames.size(); ++i)  {  QLineEdit* inputLineEdit = qobject_cast<QLineEdit*>(formLayout.itemAt(i, QFormLayout::FieldRole)->widget());  if (inputLineEdit)  {  QString inputValue = inputLineEdit->text().trimmed();  // Trim leading and trailing whitespaces  if (inputValue == "") continue;  QVariant::Type attributeType = fields[i].type();  QVariant convertedValue;  bool conversionOk;  switch (attributeType)  {  case QVariant::Int:  convertedValue = inputValue.toInt(&conversionOk);  break;  case QVariant::Double:  convertedValue = inputValue.toDouble(&conversionOk);  break;  // 其他类型的转换可以根据需要继续添加  default:  convertedValue = inputValue;  conversionOk = true;  }  if (!conversionOk)  {  // 用户取消输入，中断操作  return;  }  attributeMap[attributeNames[i]] = convertedValue;  }  }  QgsAttributes attributes;  for (const QString& attributeName : attributeNames)  {  QVariant value = attributeMap.value(attributeName, QVariant());  // Get value from map, default to QVariant()  attributes << value;  }  feature.setAttributes(attributes);  m_currentLayer->dataProvider()->addFeature(feature);  m_currentLayer->commitChanges();  m_mapCanvas->refresh();  inputDialog.accept();  // Close the dialog after processing  });  QObject::connect(cancelButton, &QPushButton::clicked, [&]() {  m_currentLayer->removeSelection();  m_currentLayer->dataProvider()->deleteFeatures({ feature.id() });  inputDialog.reject();  // Close the dialog without processing  return;  });  inputDialog.setWindowTitle("输入属性值");  inputDialog.exec();  }  

QgsMapToolAddLine::QgsMapToolAddLine(QgsMapCanvas* canvas, QgsVectorLayer* currentlayer) :QgsMapTool(canvas), pMapCanvas(canvas), m_currentLayer(currentlayer)  
{  pMapCanvas = canvas;  m_currentLayer = currentlayer;  //设置QgsRubberBand对象,绘制折线  pRubBand = new QgsRubberBand(pMapCanvas);  mColor = QColor(255, 0, 0);  LineWidth = 2;  ButtonClickFlag = false;  
}  QgsMapToolAddLine::~QgsMapToolAddLine(void)  
{  /*if(pRubBand){ delete pRubBand ; }*/  
}  
//重载鼠标移动事件  
void QgsMapToolAddLine::canvasMoveEvent(QgsMapMouseEvent* e)  
{  m_currentLayer->startEditing();  int xc, yc;  //如果鼠标左键没有双击  if (!ButtonClickFlag) {  return;  }  pRubBand->setColor(mColor);  pRubBand->setWidth(LineWidth);  xc = e->x();  yc = e->y();  //得到当前坐标变换对象  const QgsMapToPixel* pTransform = pMapCanvas->getCoordinateTransform();  //转换成地图坐标  QgsPoint mPoint(pTransform->toMapCoordinates(xc, yc));  if (pRubBand->numberOfVertices() > 1) {  pRubBand->removeLastPoint();  }  //把当前点添加到QgsRubberBand对象中用于绘制  pRubBand->addPoint(mPoint);  
}  
//重载鼠标单击事件  
void  QgsMapToolAddLine::canvasPressEvent(QgsMapMouseEvent* e)  
{  // 创建要素  QgsFeature feature;  int xc, yc;  pRubBand->setColor(mColor);  pRubBand->setWidth(LineWidth);  //得到当前坐标变换对象  const QgsMapToPixel* pTransform = pMapCanvas->getCoordinateTransform();  //得到产生事件的按钮信息  Qt::MouseButton mButton = e->button();  xc = e->x();  yc = e->y();  QgsPoint  mPoint(pTransform->toMapCoordinates(xc, yc));  //如果是左按钮  if (mButton == Qt::MouseButton::LeftButton) {  ButtonClickFlag = true;  //把当前点添加到QgsRubberBand对象中用于绘制  pRubBand->addPoint(mPoint);  mPointSet.append(mPoint);  }  else if (mButton == Qt::MouseButton::RightButton) {  pRubBand->addPoint(mPoint);  mPointSet.append(mPoint);  ButtonClickFlag = false;  //转化为几何体  QgsGeometry geometry = pRubBand->asGeometry();  feature.setGeometry(geometry);  if (feature.hasGeometry())  {  QgsFields fields = m_currentLayer->fields();  feature.setFields(fields);  QDialog inputDialog;  QFormLayout formLayout(&inputDialog);  QList<QString> attributeNames;  QList<QVariant> attributeValues;  for (const QgsField& field : fields)  {  QString attributeName = field.name();  QVariant::Type attributeType = field.type();  QString labelText = "输入属性 " + attributeName + " 的属性值:";  QString defaultValue = "";  // Set default value if needed  QLineEdit* inputLineEdit = new QLineEdit(&inputDialog);  inputLineEdit->setText(defaultValue);  formLayout.addRow(labelText, inputLineEdit);  attributeNames.append(attributeName);  }  QPushButton* okButton = new QPushButton("OK", &inputDialog);  QPushButton* cancelButton = new QPushButton("CANCEL", &inputDialog);  formLayout.addRow(okButton);  formLayout.addRow(cancelButton);  QObject::connect(okButton, &QPushButton::clicked, [&]() {  QMap<QString, QVariant> attributeMap;  for (int i = 0; i < attributeNames.size(); ++i)  {  QLineEdit* inputLineEdit = qobject_cast<QLineEdit*>(formLayout.itemAt(i, QFormLayout::FieldRole)->widget());  if (inputLineEdit)  {  QString inputValue = inputLineEdit->text().trimmed();  // Trim leading and trailing whitespaces  if (inputValue == "") continue;  QVariant::Type attributeType = fields[i].type();  QVariant convertedValue;  bool conversionOk;  switch (attributeType)  {  case QVariant::Int:  convertedValue = inputValue.toInt(&conversionOk);  break;  case QVariant::Double:  convertedValue = inputValue.toDouble(&conversionOk);  break;  // 其他类型的转换可以根据需要继续添加  default:  convertedValue = inputValue;  conversionOk = true;  }  if (!conversionOk)  {  // 用户取消输入，中断操作  return;  }  attributeMap[attributeNames[i]] = convertedValue;  }  }  QgsAttributes attributes;  for (const QString& attributeName : attributeNames)  {  QVariant value = attributeMap.value(attributeName, QVariant());  // Get value from map, default to QVariant()  attributes << value;  }  feature.setAttributes(attributes);  m_currentLayer->dataProvider()->addFeature(feature);  m_currentLayer->commitChanges();  pMapCanvas->refresh();  inputDialog.accept();  // Close the dialog after processing  });  QObject::connect(cancelButton, &QPushButton::clicked, [&]() {  pRubBand->reset();  mPointSet.clear();  inputDialog.reject();  // Close the dialog without processing  });  inputDialog.setWindowTitle("输入属性值");  inputDialog.exec();  pMapCanvas->refresh();  pRubBand->reset();  mPointSet.clear();  }  }  }  
//设置绘制颜色和线宽  
void QgsMapToolAddLine::SetColorAndWidth(QColor color, int nWidth)  
{  mColor = color;  LineWidth = nWidth;  
}  //返回构成线段的数据点数  
int QgsMapToolAddLine::GetVertexCount()  
{  return mPointSet.size();  
}  
//得到点的坐标  
bool QgsMapToolAddLine::GetCoord(int index, double& x, double& y)  
{  QgsPoint mPoint;  if (index >= mPointSet.size()) {  return true;  }  mPoint = mPointSet.at(index);  x = mPoint.x();  y = mPoint.y();  
}  

void DataViewer::on_actionAddPolygons_triggered()    
{    currentLayer = m_layerTreeView->currentLayer();    QgsVectorLayer* layer = qobject_cast<QgsVectorLayer*>(currentLayer);    //面编辑——添加要素    layer->startEditing();    m_mapCanvas->setLayers(QList<QgsMapLayer*>() << layer);    m_mapCanvas->setCurrentLayer(layer);    QgsMapToolAddFeature* addFeaturepolygonTool = new QgsMapToolAddFeature(m_mapCanvas, QgsMapToolCapture::CapturePolygon);    m_mapCanvas->setMapTool(addFeaturepolygonTool);    }