def createWidget(self): if self.dialogType == DIALOG_STANDARD: return self._panel() else: w = QgsDoubleSpinBox() w.setShowClearButton(False) w.setMinimum(0) w.setMaximum(99999999999) w.setDecimals(6) w.setToolTip(self.tr('Resolution of each pixel in output raster, in layer units')) return w
class Ui_QgsComposerLegendWidgetBase(QtGui.QWidget): def __init__(self, parent): self.setObjectName(_fromUtf8("self")) self.resize(393, 995) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth(self.sizePolicy().hasHeightForWidth()) self.setSizePolicy(sizePolicy) self.verticalLayout = QtGui.QVBoxLayout(self) self.verticalLayout.setSpacing(0) self.verticalLayout.setMargin(0) self.verticalLayout.setObjectName(_fromUtf8("verticalLayout")) self.label_11 = QtGui.QLabel(self) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Maximum) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth( self.label_11.sizePolicy().hasHeightForWidth()) self.label_11.setSizePolicy(sizePolicy) self.label_11.setStyleSheet( _fromUtf8( "padding: 2px; font-weight: bold; background-color: rgb(200, 200, 200);" )) self.label_11.setObjectName(_fromUtf8("label_11")) self.verticalLayout.addWidget(self.label_11) self.scrollArea = QtGui.QScrollArea(self) self.scrollArea.setFrameShape(QtGui.QFrame.StyledPanel) self.scrollArea.setWidgetResizable(True) self.scrollArea.setObjectName(_fromUtf8("scrollArea")) self.scrollAreaWidgetContents = QtGui.QWidget() self.scrollAreaWidgetContents.setGeometry(QtCore.QRect( 0, 0, 375, 1478)) self.scrollAreaWidgetContents.setObjectName( _fromUtf8("scrollAreaWidgetContents")) self.mainLayout = QtGui.QVBoxLayout(self.scrollAreaWidgetContents) self.mainLayout.setObjectName(_fromUtf8("mainLayout")) self.mMainPropertiesColGroupBox = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mMainPropertiesColGroupBox.setFocusPolicy(QtCore.Qt.StrongFocus) self.mMainPropertiesColGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mMainPropertiesColGroupBox.setProperty("collapsed", False) self.mMainPropertiesColGroupBox.setObjectName( _fromUtf8("mMainPropertiesColGroupBox")) self.formLayout = QtGui.QFormLayout(self.mMainPropertiesColGroupBox) self.formLayout.setFieldGrowthPolicy( QtGui.QFormLayout.AllNonFixedFieldsGrow) self.formLayout.setLabelAlignment(QtCore.Qt.AlignLeading | QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.formLayout.setObjectName(_fromUtf8("formLayout")) self.mTitleLabel = QtGui.QLabel(self.mMainPropertiesColGroupBox) self.mTitleLabel.setObjectName(_fromUtf8("mTitleLabel")) self.formLayout.setWidget(0, QtGui.QFormLayout.LabelRole, self.mTitleLabel) self.mTitleLineEdit = QtGui.QLineEdit(self.mMainPropertiesColGroupBox) self.mTitleLineEdit.setObjectName(_fromUtf8("mTitleLineEdit")) self.formLayout.setWidget(0, QtGui.QFormLayout.FieldRole, self.mTitleLineEdit) self.mMapLabel = QtGui.QLabel(self.mMainPropertiesColGroupBox) self.mMapLabel.setObjectName(_fromUtf8("mMapLabel")) self.formLayout.setWidget(2, QtGui.QFormLayout.LabelRole, self.mMapLabel) self.mMapComboBox = QtGui.QComboBox(self.mMainPropertiesColGroupBox) self.mMapComboBox.setObjectName(_fromUtf8("mMapComboBox")) self.formLayout.setWidget(2, QtGui.QFormLayout.FieldRole, self.mMapComboBox) self.label = QtGui.QLabel(self.mMainPropertiesColGroupBox) self.label.setObjectName(_fromUtf8("label")) self.formLayout.setWidget(3, QtGui.QFormLayout.LabelRole, self.label) self.mWrapCharLineEdit = QtGui.QLineEdit( self.mMainPropertiesColGroupBox) self.mWrapCharLineEdit.setFrame(True) self.mWrapCharLineEdit.setObjectName(_fromUtf8("mWrapCharLineEdit")) self.formLayout.setWidget(3, QtGui.QFormLayout.FieldRole, self.mWrapCharLineEdit) self.label_15 = QtGui.QLabel(self.mMainPropertiesColGroupBox) self.label_15.setObjectName(_fromUtf8("label_15")) self.formLayout.setWidget(1, QtGui.QFormLayout.LabelRole, self.label_15) self.mTitleAlignCombo = QtGui.QComboBox( self.mMainPropertiesColGroupBox) self.mTitleAlignCombo.setObjectName(_fromUtf8("mTitleAlignCombo")) self.mTitleAlignCombo.addItem(_fromUtf8("")) self.mTitleAlignCombo.addItem(_fromUtf8("")) self.mTitleAlignCombo.addItem(_fromUtf8("")) self.formLayout.setWidget(1, QtGui.QFormLayout.FieldRole, self.mTitleAlignCombo) self.mainLayout.addWidget(self.mMainPropertiesColGroupBox) self.mLegendItemColGroupBox = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mLegendItemColGroupBox.setFocusPolicy(QtCore.Qt.StrongFocus) self.mLegendItemColGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mLegendItemColGroupBox.setProperty("collapsed", False) self.mLegendItemColGroupBox.setObjectName( _fromUtf8("mLegendItemColGroupBox")) self.verticalLayout_2 = QtGui.QVBoxLayout(self.mLegendItemColGroupBox) self.verticalLayout_2.setObjectName(_fromUtf8("verticalLayout_2")) self.horizontalLayout = QtGui.QHBoxLayout() self.horizontalLayout.setObjectName(_fromUtf8("horizontalLayout")) self.mCheckBoxAutoUpdate = QtGui.QCheckBox(self.mLegendItemColGroupBox) self.mCheckBoxAutoUpdate.setChecked(True) self.mCheckBoxAutoUpdate.setObjectName( _fromUtf8("mCheckBoxAutoUpdate")) self.horizontalLayout.addWidget(self.mCheckBoxAutoUpdate) spacerItem = QtGui.QSpacerItem(0, 20, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum) self.horizontalLayout.addItem(spacerItem) self.mUpdateAllPushButton = QtGui.QToolButton( self.mLegendItemColGroupBox) self.mUpdateAllPushButton.setObjectName( _fromUtf8("mUpdateAllPushButton")) self.horizontalLayout.addWidget(self.mUpdateAllPushButton) self.verticalLayout_2.addLayout(self.horizontalLayout) self.mItemTreeView = QgsLayerTreeView(self.mLegendItemColGroupBox) self.mItemTreeView.setMinimumSize(QtCore.QSize(0, 250)) self.mItemTreeView.setSelectionMode( QtGui.QAbstractItemView.ContiguousSelection) self.mItemTreeView.setHeaderHidden(True) self.mItemTreeView.setObjectName(_fromUtf8("mItemTreeView")) self.mItemTreeView.header().setVisible(False) self.verticalLayout_2.addWidget(self.mItemTreeView) self.horizontalLayout_2 = QtGui.QHBoxLayout() self.horizontalLayout_2.setSpacing(3) self.horizontalLayout_2.setObjectName(_fromUtf8("horizontalLayout_2")) self.mMoveDownToolButton = QtGui.QToolButton( self.mLegendItemColGroupBox) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth( self.mMoveDownToolButton.sizePolicy().hasHeightForWidth()) self.mMoveDownToolButton.setSizePolicy(sizePolicy) self.mMoveDownToolButton.setText(_fromUtf8("")) icon = QtGui.QIcon() icon.addPixmap( QtGui.QPixmap( _fromUtf8( "Resource/images/themes/default/mActionArrowDown.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mMoveDownToolButton.setIcon(icon) self.mMoveDownToolButton.setIconSize(QtCore.QSize(20, 20)) self.mMoveDownToolButton.setObjectName( _fromUtf8("mMoveDownToolButton")) self.horizontalLayout_2.addWidget(self.mMoveDownToolButton) self.mMoveUpToolButton = QtGui.QToolButton(self.mLegendItemColGroupBox) self.mMoveUpToolButton.setText(_fromUtf8("")) icon1 = QtGui.QIcon() icon1.addPixmap( QtGui.QPixmap( _fromUtf8( "Resource/images/themes/default/mActionArrowUp.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mMoveUpToolButton.setIcon(icon1) self.mMoveUpToolButton.setIconSize(QtCore.QSize(20, 20)) self.mMoveUpToolButton.setObjectName(_fromUtf8("mMoveUpToolButton")) self.horizontalLayout_2.addWidget(self.mMoveUpToolButton) self.mAddGroupToolButton = QtGui.QToolButton( self.mLegendItemColGroupBox) icon2 = QtGui.QIcon() icon2.addPixmap( QtGui.QPixmap( _fromUtf8( "Resource/images/themes/default/mActionAddGroup.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mAddGroupToolButton.setIcon(icon2) self.mAddGroupToolButton.setIconSize(QtCore.QSize(20, 20)) self.mAddGroupToolButton.setObjectName( _fromUtf8("mAddGroupToolButton")) self.horizontalLayout_2.addWidget(self.mAddGroupToolButton) self.mAddToolButton = QtGui.QToolButton(self.mLegendItemColGroupBox) self.mAddToolButton.setText(_fromUtf8("")) icon3 = QtGui.QIcon() icon3.addPixmap( QtGui.QPixmap( _fromUtf8("Resource/images/themes/default/symbologyAdd.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mAddToolButton.setIcon(icon3) self.mAddToolButton.setIconSize(QtCore.QSize(20, 20)) self.mAddToolButton.setObjectName(_fromUtf8("mAddToolButton")) self.horizontalLayout_2.addWidget(self.mAddToolButton) self.mRemoveToolButton = QtGui.QToolButton(self.mLegendItemColGroupBox) self.mRemoveToolButton.setText(_fromUtf8("")) icon4 = QtGui.QIcon() icon4.addPixmap( QtGui.QPixmap( _fromUtf8( "Resource/images/themes/default/symbologyRemove.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mRemoveToolButton.setIcon(icon4) self.mRemoveToolButton.setIconSize(QtCore.QSize(20, 20)) self.mRemoveToolButton.setObjectName(_fromUtf8("mRemoveToolButton")) self.horizontalLayout_2.addWidget(self.mRemoveToolButton) self.mEditPushButton = QtGui.QToolButton(self.mLegendItemColGroupBox) self.mEditPushButton.setText(_fromUtf8("")) icon5 = QtGui.QIcon() icon5.addPixmap( QtGui.QPixmap( _fromUtf8("Resource/images/themes/default/symbologyEdit.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mEditPushButton.setIcon(icon5) self.mEditPushButton.setIconSize(QtCore.QSize(20, 20)) self.mEditPushButton.setObjectName(_fromUtf8("mEditPushButton")) self.horizontalLayout_2.addWidget(self.mEditPushButton) self.mCountToolButton = QtGui.QToolButton(self.mLegendItemColGroupBox) self.mCountToolButton.setEnabled(False) self.mCountToolButton.setText(_fromUtf8("")) icon6 = QtGui.QIcon() icon6.addPixmap( QtGui.QPixmap( _fromUtf8("Resource/images/themes/default/mActionSum.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mCountToolButton.setIcon(icon6) self.mCountToolButton.setIconSize(QtCore.QSize(20, 20)) self.mCountToolButton.setCheckable(True) self.mCountToolButton.setObjectName(_fromUtf8("mCountToolButton")) self.horizontalLayout_2.addWidget(self.mCountToolButton) self.mFilterByMapToolButton = QtGui.QToolButton( self.mLegendItemColGroupBox) icon7 = QtGui.QIcon() icon7.addPixmap( QtGui.QPixmap( _fromUtf8( "Resource/images/themes/default/mActionFilter2.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mFilterByMapToolButton.setIcon(icon7) self.mFilterByMapToolButton.setIconSize(QtCore.QSize(20, 20)) self.mFilterByMapToolButton.setCheckable(True) self.mFilterByMapToolButton.setObjectName( _fromUtf8("mFilterByMapToolButton")) self.horizontalLayout_2.addWidget(self.mFilterByMapToolButton) self.mExpressionFilterButton = QgsLegendFilterButton( self.mLegendItemColGroupBox) self.mExpressionFilterButton.setText(_fromUtf8("")) icon8 = QtGui.QIcon() icon8.addPixmap( QtGui.QPixmap( _fromUtf8( "Resource/images/themes/default/mIconExpressionFilter.svg") ), QtGui.QIcon.Normal, QtGui.QIcon.Off) self.mExpressionFilterButton.setIcon(icon8) self.mExpressionFilterButton.setIconSize(QtCore.QSize(20, 20)) self.mExpressionFilterButton.setObjectName( _fromUtf8("mExpressionFilterButton")) self.horizontalLayout_2.addWidget(self.mExpressionFilterButton) spacerItem1 = QtGui.QSpacerItem(0, 20, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum) self.horizontalLayout_2.addItem(spacerItem1) self.verticalLayout_2.addLayout(self.horizontalLayout_2) self.mFilterLegendByAtlasCheckBox = QtGui.QCheckBox( self.mLegendItemColGroupBox) self.mFilterLegendByAtlasCheckBox.setObjectName( _fromUtf8("mFilterLegendByAtlasCheckBox")) self.verticalLayout_2.addWidget(self.mFilterLegendByAtlasCheckBox) self.mainLayout.addWidget(self.mLegendItemColGroupBox) self.mFontsColGroupBox = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mFontsColGroupBox.setFocusPolicy(QtCore.Qt.StrongFocus) self.mFontsColGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mFontsColGroupBox.setProperty("collapsed", True) self.mFontsColGroupBox.setObjectName(_fromUtf8("mFontsColGroupBox")) self.verticalLayout_3 = QtGui.QVBoxLayout(self.mFontsColGroupBox) self.verticalLayout_3.setObjectName(_fromUtf8("verticalLayout_3")) self.mTitleFontButton = QtGui.QPushButton(self.mFontsColGroupBox) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth( self.mTitleFontButton.sizePolicy().hasHeightForWidth()) self.mTitleFontButton.setSizePolicy(sizePolicy) self.mTitleFontButton.setObjectName(_fromUtf8("mTitleFontButton")) self.verticalLayout_3.addWidget(self.mTitleFontButton) self.mLayerFontButton = QtGui.QPushButton(self.mFontsColGroupBox) self.mLayerFontButton.setObjectName(_fromUtf8("mLayerFontButton")) self.verticalLayout_3.addWidget(self.mLayerFontButton) self.mGroupFontButton = QtGui.QPushButton(self.mFontsColGroupBox) self.mGroupFontButton.setObjectName(_fromUtf8("mGroupFontButton")) self.verticalLayout_3.addWidget(self.mGroupFontButton) self.mItemFontButton = QtGui.QPushButton(self.mFontsColGroupBox) self.mItemFontButton.setObjectName(_fromUtf8("mItemFontButton")) self.verticalLayout_3.addWidget(self.mItemFontButton) self.horizontalLayout_3 = QtGui.QHBoxLayout() self.horizontalLayout_3.setObjectName(_fromUtf8("horizontalLayout_3")) self.label_16 = QtGui.QLabel(self.mFontsColGroupBox) self.label_16.setObjectName(_fromUtf8("label_16")) self.horizontalLayout_3.addWidget(self.label_16) self.mFontColorButton = QgsColorButtonV2(self.mFontsColGroupBox) self.mFontColorButton.setMinimumSize(QtCore.QSize(120, 0)) self.mFontColorButton.setMaximumSize(QtCore.QSize(120, 16777215)) self.mFontColorButton.setText(_fromUtf8("")) self.mFontColorButton.setObjectName(_fromUtf8("mFontColorButton")) self.horizontalLayout_3.addWidget(self.mFontColorButton) spacerItem2 = QtGui.QSpacerItem(40, 20, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum) self.horizontalLayout_3.addItem(spacerItem2) self.verticalLayout_3.addLayout(self.horizontalLayout_3) self.mainLayout.addWidget(self.mFontsColGroupBox) self.mColumnsColGroupBox = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mColumnsColGroupBox.setFocusPolicy(QtCore.Qt.StrongFocus) self.mColumnsColGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mColumnsColGroupBox.setProperty("collapsed", True) self.mColumnsColGroupBox.setObjectName( _fromUtf8("mColumnsColGroupBox")) self.formLayout_2 = QtGui.QFormLayout(self.mColumnsColGroupBox) self.formLayout_2.setFieldGrowthPolicy( QtGui.QFormLayout.AllNonFixedFieldsGrow) self.formLayout_2.setLabelAlignment(QtCore.Qt.AlignLeading | QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.formLayout_2.setObjectName(_fromUtf8("formLayout_2")) self.label_2 = QtGui.QLabel(self.mColumnsColGroupBox) self.label_2.setObjectName(_fromUtf8("label_2")) self.formLayout_2.setWidget(1, QtGui.QFormLayout.LabelRole, self.label_2) self.mColumnCountSpinBox = QgsSpinBox(self.mColumnsColGroupBox) self.mColumnCountSpinBox.setPrefix(_fromUtf8("")) self.mColumnCountSpinBox.setMinimum(1) self.mColumnCountSpinBox.setMaximum(1000) self.mColumnCountSpinBox.setObjectName( _fromUtf8("mColumnCountSpinBox")) self.formLayout_2.setWidget(1, QtGui.QFormLayout.FieldRole, self.mColumnCountSpinBox) self.mEqualColumnWidthCheckBox = QtGui.QCheckBox( self.mColumnsColGroupBox) self.mEqualColumnWidthCheckBox.setObjectName( _fromUtf8("mEqualColumnWidthCheckBox")) self.formLayout_2.setWidget(3, QtGui.QFormLayout.SpanningRole, self.mEqualColumnWidthCheckBox) self.mSplitLayerCheckBox = QtGui.QCheckBox(self.mColumnsColGroupBox) self.mSplitLayerCheckBox.setObjectName( _fromUtf8("mSplitLayerCheckBox")) self.formLayout_2.setWidget(4, QtGui.QFormLayout.SpanningRole, self.mSplitLayerCheckBox) self.mainLayout.addWidget(self.mColumnsColGroupBox) self.mSymbolsColGroupBox = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mSymbolsColGroupBox.setFocusPolicy(QtCore.Qt.StrongFocus) self.mSymbolsColGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mSymbolsColGroupBox.setProperty("collapsed", True) self.mSymbolsColGroupBox.setObjectName( _fromUtf8("mSymbolsColGroupBox")) self.formLayout_3 = QtGui.QFormLayout(self.mSymbolsColGroupBox) self.formLayout_3.setFieldGrowthPolicy( QtGui.QFormLayout.AllNonFixedFieldsGrow) self.formLayout_3.setLabelAlignment(QtCore.Qt.AlignLeading | QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.formLayout_3.setObjectName(_fromUtf8("formLayout_3")) self.label_3 = QtGui.QLabel(self.mSymbolsColGroupBox) self.label_3.setObjectName(_fromUtf8("label_3")) self.formLayout_3.setWidget(0, QtGui.QFormLayout.LabelRole, self.label_3) self.mSymbolWidthSpinBox = QgsDoubleSpinBox(self.mSymbolsColGroupBox) self.mSymbolWidthSpinBox.setPrefix(_fromUtf8("")) self.mSymbolWidthSpinBox.setProperty("showClearButton", False) self.mSymbolWidthSpinBox.setObjectName( _fromUtf8("mSymbolWidthSpinBox")) self.formLayout_3.setWidget(0, QtGui.QFormLayout.FieldRole, self.mSymbolWidthSpinBox) self.label_4 = QtGui.QLabel(self.mSymbolsColGroupBox) self.label_4.setObjectName(_fromUtf8("label_4")) self.formLayout_3.setWidget(1, QtGui.QFormLayout.LabelRole, self.label_4) self.mSymbolHeightSpinBox = QgsDoubleSpinBox(self.mSymbolsColGroupBox) self.mSymbolHeightSpinBox.setPrefix(_fromUtf8("")) self.mSymbolHeightSpinBox.setProperty("showClearButton", False) self.mSymbolHeightSpinBox.setObjectName( _fromUtf8("mSymbolHeightSpinBox")) self.formLayout_3.setWidget(1, QtGui.QFormLayout.FieldRole, self.mSymbolHeightSpinBox) self.mRasterBorderGroupBox = QgsCollapsibleGroupBoxBasic( self.mSymbolsColGroupBox) self.mRasterBorderGroupBox.setCheckable(True) self.mRasterBorderGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mRasterBorderGroupBox.setProperty("collapsed", True) self.mRasterBorderGroupBox.setObjectName( _fromUtf8("mRasterBorderGroupBox")) self.gridLayout = QtGui.QGridLayout(self.mRasterBorderGroupBox) self.gridLayout.setObjectName(_fromUtf8("gridLayout")) self.label_17 = QtGui.QLabel(self.mRasterBorderGroupBox) self.label_17.setObjectName(_fromUtf8("label_17")) self.gridLayout.addWidget(self.label_17, 0, 0, 1, 1) self.horizontalLayout_6 = QtGui.QHBoxLayout() self.horizontalLayout_6.setObjectName(_fromUtf8("horizontalLayout_6")) self.mRasterBorderColorButton = QgsColorButtonV2( self.mRasterBorderGroupBox) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed, QtGui.QSizePolicy.Fixed) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth( self.mRasterBorderColorButton.sizePolicy().hasHeightForWidth()) self.mRasterBorderColorButton.setSizePolicy(sizePolicy) self.mRasterBorderColorButton.setMinimumSize(QtCore.QSize(120, 0)) self.mRasterBorderColorButton.setMaximumSize( QtCore.QSize(120, 16777215)) self.mRasterBorderColorButton.setText(_fromUtf8("")) self.mRasterBorderColorButton.setObjectName( _fromUtf8("mRasterBorderColorButton")) self.horizontalLayout_6.addWidget(self.mRasterBorderColorButton) spacerItem3 = QtGui.QSpacerItem(40, 20, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum) self.horizontalLayout_6.addItem(spacerItem3) self.gridLayout.addLayout(self.horizontalLayout_6, 0, 1, 1, 1) self.mOutlineWidthLabel = QtGui.QLabel(self.mRasterBorderGroupBox) self.mOutlineWidthLabel.setObjectName(_fromUtf8("mOutlineWidthLabel")) self.gridLayout.addWidget(self.mOutlineWidthLabel, 1, 0, 1, 1) self.mRasterBorderWidthSpinBox = QgsDoubleSpinBox( self.mRasterBorderGroupBox) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Fixed) sizePolicy.setHorizontalStretch(0) sizePolicy.setVerticalStretch(0) sizePolicy.setHeightForWidth( self.mRasterBorderWidthSpinBox.sizePolicy().hasHeightForWidth()) self.mRasterBorderWidthSpinBox.setSizePolicy(sizePolicy) self.mRasterBorderWidthSpinBox.setSingleStep(0.1) self.mRasterBorderWidthSpinBox.setProperty("showClearButton", False) self.mRasterBorderWidthSpinBox.setObjectName( _fromUtf8("mRasterBorderWidthSpinBox")) self.gridLayout.addWidget(self.mRasterBorderWidthSpinBox, 1, 1, 1, 1) self.formLayout_3.setWidget(2, QtGui.QFormLayout.SpanningRole, self.mRasterBorderGroupBox) self.mainLayout.addWidget(self.mSymbolsColGroupBox) self.mSymbolsColGroupBox_2 = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mSymbolsColGroupBox_2.setFocusPolicy(QtCore.Qt.StrongFocus) self.mSymbolsColGroupBox_2.setProperty("syncGroup", _fromUtf8("composeritem")) self.mSymbolsColGroupBox_2.setProperty("collapsed", True) self.mSymbolsColGroupBox_2.setObjectName( _fromUtf8("mSymbolsColGroupBox_2")) self.formLayout_5 = QtGui.QFormLayout(self.mSymbolsColGroupBox_2) self.formLayout_5.setFieldGrowthPolicy( QtGui.QFormLayout.AllNonFixedFieldsGrow) self.formLayout_5.setLabelAlignment(QtCore.Qt.AlignLeading | QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.formLayout_5.setObjectName(_fromUtf8("formLayout_5")) self.label_13 = QtGui.QLabel(self.mSymbolsColGroupBox_2) self.label_13.setObjectName(_fromUtf8("label_13")) self.formLayout_5.setWidget(0, QtGui.QFormLayout.LabelRole, self.label_13) self.mWmsLegendWidthSpinBox = QgsDoubleSpinBox( self.mSymbolsColGroupBox_2) self.mWmsLegendWidthSpinBox.setPrefix(_fromUtf8("")) self.mWmsLegendWidthSpinBox.setMaximum(9999.0) self.mWmsLegendWidthSpinBox.setProperty("showClearButton", False) self.mWmsLegendWidthSpinBox.setObjectName( _fromUtf8("mWmsLegendWidthSpinBox")) self.formLayout_5.setWidget(0, QtGui.QFormLayout.FieldRole, self.mWmsLegendWidthSpinBox) self.label_14 = QtGui.QLabel(self.mSymbolsColGroupBox_2) self.label_14.setObjectName(_fromUtf8("label_14")) self.formLayout_5.setWidget(1, QtGui.QFormLayout.LabelRole, self.label_14) self.mWmsLegendHeightSpinBox = QgsDoubleSpinBox( self.mSymbolsColGroupBox_2) self.mWmsLegendHeightSpinBox.setPrefix(_fromUtf8("")) self.mWmsLegendHeightSpinBox.setMaximum(9999.0) self.mWmsLegendHeightSpinBox.setProperty("showClearButton", False) self.mWmsLegendHeightSpinBox.setObjectName( _fromUtf8("mWmsLegendHeightSpinBox")) self.formLayout_5.setWidget(1, QtGui.QFormLayout.FieldRole, self.mWmsLegendHeightSpinBox) self.mainLayout.addWidget(self.mSymbolsColGroupBox_2) self.mSpacingColGroupBox = QgsCollapsibleGroupBoxBasic( self.scrollAreaWidgetContents) self.mSpacingColGroupBox.setFocusPolicy(QtCore.Qt.StrongFocus) self.mSpacingColGroupBox.setProperty("syncGroup", _fromUtf8("composeritem")) self.mSpacingColGroupBox.setProperty("collapsed", True) self.mSpacingColGroupBox.setObjectName( _fromUtf8("mSpacingColGroupBox")) self.formLayout_4 = QtGui.QFormLayout(self.mSpacingColGroupBox) self.formLayout_4.setFieldGrowthPolicy( QtGui.QFormLayout.AllNonFixedFieldsGrow) self.formLayout_4.setLabelAlignment(QtCore.Qt.AlignLeading | QtCore.Qt.AlignLeft | QtCore.Qt.AlignVCenter) self.formLayout_4.setObjectName(_fromUtf8("formLayout_4")) self.label_5 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_5.setObjectName(_fromUtf8("label_5")) self.formLayout_4.setWidget(1, QtGui.QFormLayout.LabelRole, self.label_5) self.mGroupSpaceSpinBox = QgsDoubleSpinBox(self.mSpacingColGroupBox) self.mGroupSpaceSpinBox.setPrefix(_fromUtf8("")) self.mGroupSpaceSpinBox.setObjectName(_fromUtf8("mGroupSpaceSpinBox")) self.formLayout_4.setWidget(1, QtGui.QFormLayout.FieldRole, self.mGroupSpaceSpinBox) self.label_6 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_6.setObjectName(_fromUtf8("label_6")) self.formLayout_4.setWidget(2, QtGui.QFormLayout.LabelRole, self.label_6) self.mLayerSpaceSpinBox = QgsDoubleSpinBox(self.mSpacingColGroupBox) self.mLayerSpaceSpinBox.setPrefix(_fromUtf8("")) self.mLayerSpaceSpinBox.setObjectName(_fromUtf8("mLayerSpaceSpinBox")) self.formLayout_4.setWidget(2, QtGui.QFormLayout.FieldRole, self.mLayerSpaceSpinBox) self.label_7 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_7.setObjectName(_fromUtf8("label_7")) self.formLayout_4.setWidget(3, QtGui.QFormLayout.LabelRole, self.label_7) self.mSymbolSpaceSpinBox = QgsDoubleSpinBox(self.mSpacingColGroupBox) self.mSymbolSpaceSpinBox.setPrefix(_fromUtf8("")) self.mSymbolSpaceSpinBox.setObjectName( _fromUtf8("mSymbolSpaceSpinBox")) self.formLayout_4.setWidget(3, QtGui.QFormLayout.FieldRole, self.mSymbolSpaceSpinBox) self.label_8 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_8.setObjectName(_fromUtf8("label_8")) self.formLayout_4.setWidget(4, QtGui.QFormLayout.LabelRole, self.label_8) self.mIconLabelSpaceSpinBox = QgsDoubleSpinBox( self.mSpacingColGroupBox) self.mIconLabelSpaceSpinBox.setPrefix(_fromUtf8("")) self.mIconLabelSpaceSpinBox.setObjectName( _fromUtf8("mIconLabelSpaceSpinBox")) self.formLayout_4.setWidget(4, QtGui.QFormLayout.FieldRole, self.mIconLabelSpaceSpinBox) self.label_9 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_9.setObjectName(_fromUtf8("label_9")) self.formLayout_4.setWidget(5, QtGui.QFormLayout.LabelRole, self.label_9) self.mBoxSpaceSpinBox = QgsDoubleSpinBox(self.mSpacingColGroupBox) self.mBoxSpaceSpinBox.setPrefix(_fromUtf8("")) self.mBoxSpaceSpinBox.setObjectName(_fromUtf8("mBoxSpaceSpinBox")) self.formLayout_4.setWidget(5, QtGui.QFormLayout.FieldRole, self.mBoxSpaceSpinBox) self.label_12 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_12.setObjectName(_fromUtf8("label_12")) self.formLayout_4.setWidget(6, QtGui.QFormLayout.LabelRole, self.label_12) self.mColumnSpaceSpinBox = QgsDoubleSpinBox(self.mSpacingColGroupBox) self.mColumnSpaceSpinBox.setPrefix(_fromUtf8("")) self.mColumnSpaceSpinBox.setObjectName( _fromUtf8("mColumnSpaceSpinBox")) self.formLayout_4.setWidget(6, QtGui.QFormLayout.FieldRole, self.mColumnSpaceSpinBox) self.label_10 = QtGui.QLabel(self.mSpacingColGroupBox) self.label_10.setObjectName(_fromUtf8("label_10")) self.formLayout_4.setWidget(0, QtGui.QFormLayout.LabelRole, self.label_10) self.mTitleSpaceBottomSpinBox = QgsDoubleSpinBox( self.mSpacingColGroupBox) self.mTitleSpaceBottomSpinBox.setObjectName( _fromUtf8("mTitleSpaceBottomSpinBox")) self.formLayout_4.setWidget(0, QtGui.QFormLayout.FieldRole, self.mTitleSpaceBottomSpinBox) self.mainLayout.addWidget(self.mSpacingColGroupBox) self.scrollArea.setWidget(self.scrollAreaWidgetContents) self.verticalLayout.addWidget(self.scrollArea) self.mTitleLabel.setBuddy(self.mTitleLineEdit) self.mOutlineWidthLabel.setBuddy(self.mRasterBorderWidthSpinBox) self.retranslateUi() QtCore.QMetaObject.connectSlotsByName(self) self.setTabOrder(self.scrollArea, self.mMainPropertiesColGroupBox) self.setTabOrder(self.mMainPropertiesColGroupBox, self.mTitleLineEdit) self.setTabOrder(self.mTitleLineEdit, self.mTitleAlignCombo) self.setTabOrder(self.mTitleAlignCombo, self.mMapComboBox) self.setTabOrder(self.mMapComboBox, self.mWrapCharLineEdit) self.setTabOrder(self.mWrapCharLineEdit, self.mLegendItemColGroupBox) self.setTabOrder(self.mLegendItemColGroupBox, self.mCheckBoxAutoUpdate) self.setTabOrder(self.mCheckBoxAutoUpdate, self.mUpdateAllPushButton) self.setTabOrder(self.mUpdateAllPushButton, self.mItemTreeView) self.setTabOrder(self.mItemTreeView, self.mMoveDownToolButton) self.setTabOrder(self.mMoveDownToolButton, self.mMoveUpToolButton) self.setTabOrder(self.mMoveUpToolButton, self.mAddGroupToolButton) self.setTabOrder(self.mAddGroupToolButton, self.mAddToolButton) self.setTabOrder(self.mAddToolButton, self.mRemoveToolButton) self.setTabOrder(self.mRemoveToolButton, self.mEditPushButton) self.setTabOrder(self.mEditPushButton, self.mCountToolButton) self.setTabOrder(self.mCountToolButton, self.mFilterByMapToolButton) self.setTabOrder(self.mFilterByMapToolButton, self.mExpressionFilterButton) self.setTabOrder(self.mExpressionFilterButton, self.mFilterLegendByAtlasCheckBox) self.setTabOrder(self.mFilterLegendByAtlasCheckBox, self.mFontsColGroupBox) self.setTabOrder(self.mFontsColGroupBox, self.mTitleFontButton) self.setTabOrder(self.mTitleFontButton, self.mLayerFontButton) self.setTabOrder(self.mLayerFontButton, self.mGroupFontButton) self.setTabOrder(self.mGroupFontButton, self.mItemFontButton) self.setTabOrder(self.mItemFontButton, self.mFontColorButton) self.setTabOrder(self.mFontColorButton, self.mColumnsColGroupBox) self.setTabOrder(self.mColumnsColGroupBox, self.mColumnCountSpinBox) self.setTabOrder(self.mColumnCountSpinBox, self.mEqualColumnWidthCheckBox) self.setTabOrder(self.mEqualColumnWidthCheckBox, self.mSplitLayerCheckBox) self.setTabOrder(self.mSplitLayerCheckBox, self.mSymbolsColGroupBox) self.setTabOrder(self.mSymbolsColGroupBox, self.mSymbolWidthSpinBox) self.setTabOrder(self.mSymbolWidthSpinBox, self.mSymbolHeightSpinBox) self.setTabOrder(self.mSymbolHeightSpinBox, self.mRasterBorderGroupBox) self.setTabOrder(self.mRasterBorderGroupBox, self.mRasterBorderColorButton) self.setTabOrder(self.mRasterBorderColorButton, self.mRasterBorderWidthSpinBox) self.setTabOrder(self.mRasterBorderWidthSpinBox, self.mSymbolsColGroupBox_2) self.setTabOrder(self.mSymbolsColGroupBox_2, self.mWmsLegendWidthSpinBox) self.setTabOrder(self.mWmsLegendWidthSpinBox, self.mWmsLegendHeightSpinBox) self.setTabOrder(self.mWmsLegendHeightSpinBox, self.mSpacingColGroupBox) self.setTabOrder(self.mSpacingColGroupBox, self.mTitleSpaceBottomSpinBox) self.setTabOrder(self.mTitleSpaceBottomSpinBox, self.mGroupSpaceSpinBox) self.setTabOrder(self.mGroupSpaceSpinBox, self.mLayerSpaceSpinBox) self.setTabOrder(self.mLayerSpaceSpinBox, self.mSymbolSpaceSpinBox) self.setTabOrder(self.mSymbolSpaceSpinBox, self.mIconLabelSpaceSpinBox) self.setTabOrder(self.mIconLabelSpaceSpinBox, self.mBoxSpaceSpinBox) self.setTabOrder(self.mBoxSpaceSpinBox, self.mColumnSpaceSpinBox) def retranslateUi(self): self.setWindowTitle( _translate("QgsComposerLegendWidgetBase", "Legend Options", None)) self.label_11.setText( _translate("QgsComposerLegendWidgetBase", "Legend", None)) self.mMainPropertiesColGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Main properties", None)) self.mTitleLabel.setText( _translate("QgsComposerLegendWidgetBase", "&Title", None)) self.mMapLabel.setText( _translate("QgsComposerLegendWidgetBase", "Map", None)) self.label.setText( _translate("QgsComposerLegendWidgetBase", "Wrap text on", None)) self.label_15.setText( _translate("QgsComposerLegendWidgetBase", "Title alignment:", None)) self.mTitleAlignCombo.setItemText( 0, _translate("QgsComposerLegendWidgetBase", "Left", None)) self.mTitleAlignCombo.setItemText( 1, _translate("QgsComposerLegendWidgetBase", "Center", None)) self.mTitleAlignCombo.setItemText( 2, _translate("QgsComposerLegendWidgetBase", "Right", None)) self.mLegendItemColGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Legend items", None)) self.mCheckBoxAutoUpdate.setText( _translate("QgsComposerLegendWidgetBase", "Auto update", None)) self.mUpdateAllPushButton.setToolTip( _translate( "QgsComposerLegendWidgetBase", "Update whole legend. Layers are added/removed according to main application legend. User defined labels will be deleted.", None)) self.mUpdateAllPushButton.setText( _translate("QgsComposerLegendWidgetBase", "Update all", None)) self.mAddGroupToolButton.setToolTip( _translate("QgsComposerLegendWidgetBase", "Add group", None)) self.mAddGroupToolButton.setText( _translate("QgsComposerLegendWidgetBase", "...", None)) self.mCountToolButton.setToolTip( _translate("QgsComposerLegendWidgetBase", "Show feature count for each class of vector layer.", None)) self.mFilterByMapToolButton.setToolTip( _translate("QgsComposerLegendWidgetBase", "Filter Legend By Map Content", None)) self.mFilterLegendByAtlasCheckBox.setToolTip( _translate( "QgsComposerLegendWidgetBase", "Filter out legend elements that lie outside the current atlas feature.", None)) self.mFilterLegendByAtlasCheckBox.setText( _translate("QgsComposerLegendWidgetBase", "Only show items inside current atlas feature", None)) self.mFontsColGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Fonts", None)) self.mTitleFontButton.setText( _translate("QgsComposerLegendWidgetBase", "Title font...", None)) self.mLayerFontButton.setText( _translate("QgsComposerLegendWidgetBase", "Subgroup font...", None)) self.mGroupFontButton.setText( _translate("QgsComposerLegendWidgetBase", "Group font...", None)) self.mItemFontButton.setText( _translate("QgsComposerLegendWidgetBase", "Item font...", None)) self.label_16.setText( _translate("QgsComposerLegendWidgetBase", "Font color", None)) self.mColumnsColGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Columns", None)) self.label_2.setText( _translate("QgsComposerLegendWidgetBase", "Count", None)) self.mEqualColumnWidthCheckBox.setText( _translate("QgsComposerLegendWidgetBase", "Equal column widths", None)) self.mSplitLayerCheckBox.setToolTip( _translate("QgsComposerLegendWidgetBase", "Allow splitting layer items into multiple columns.", None)) self.mSplitLayerCheckBox.setText( _translate("QgsComposerLegendWidgetBase", "Split layers", None)) self.mSymbolsColGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Symbol", None)) self.label_3.setText( _translate("QgsComposerLegendWidgetBase", "Symbol width", None)) self.mSymbolWidthSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_4.setText( _translate("QgsComposerLegendWidgetBase", "Symbol height", None)) self.mSymbolHeightSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.mRasterBorderGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Draw border for raster symbols", None)) self.label_17.setText( _translate("QgsComposerLegendWidgetBase", "Border color", None)) self.mOutlineWidthLabel.setText( _translate("QgsComposerLegendWidgetBase", "Thickness", None)) self.mRasterBorderWidthSpinBox.setSpecialValueText( _translate("QgsComposerLegendWidgetBase", "Hairline", None)) self.mRasterBorderWidthSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.mSymbolsColGroupBox_2.setTitle( _translate("QgsComposerLegendWidgetBase", "WMS LegendGraphic", None)) self.label_13.setText( _translate("QgsComposerLegendWidgetBase", "Legend width", None)) self.mWmsLegendWidthSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_14.setText( _translate("QgsComposerLegendWidgetBase", "Legend height", None)) self.mWmsLegendHeightSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.mSpacingColGroupBox.setTitle( _translate("QgsComposerLegendWidgetBase", "Spacing", None)) self.label_5.setToolTip( _translate("QgsComposerLegendWidgetBase", "Space above text using group style.", None)) self.label_5.setText( _translate("QgsComposerLegendWidgetBase", "Group Space", None)) self.mGroupSpaceSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_6.setToolTip( _translate("QgsComposerLegendWidgetBase", "Space above text using subgroup style.", None)) self.label_6.setText( _translate("QgsComposerLegendWidgetBase", "Subgroup space", None)) self.mLayerSpaceSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_7.setToolTip( _translate("QgsComposerLegendWidgetBase", "Space above symbol and symbol label.", None)) self.label_7.setText( _translate("QgsComposerLegendWidgetBase", "Symbol space", None)) self.mSymbolSpaceSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_8.setToolTip( _translate( "QgsComposerLegendWidgetBase", "Space between symbol icon and symbol label (symbol label left margin).", None)) self.label_8.setText( _translate("QgsComposerLegendWidgetBase", "Icon label space", None)) self.mIconLabelSpaceSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_9.setText( _translate("QgsComposerLegendWidgetBase", "Box space", None)) self.mBoxSpaceSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_12.setText( _translate("QgsComposerLegendWidgetBase", "Column space", None)) self.mColumnSpaceSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None)) self.label_10.setToolTip( _translate("QgsComposerLegendWidgetBase", "Space below title.", None)) self.label_10.setText( _translate("QgsComposerLegendWidgetBase", "Title space", None)) self.mTitleSpaceBottomSpinBox.setToolTip( _translate("QgsComposerLegendWidgetBase", "Space below title.", None)) self.mTitleSpaceBottomSpinBox.setSuffix( _translate("QgsComposerLegendWidgetBase", " mm", None))
class Serval(object): LINE_SELECTION = "line" POLYGON_SELECTION = "polygon" RGB = "RGB" SINGLE_BAND = "Single band" def __init__(self, iface): self.iface = iface self.canvas = self.iface.mapCanvas() self.plugin_dir = os.path.dirname(__file__) self.uc = UserCommunication(iface, 'Serval') self.load_settings() self.raster = None self.handler = None self.spin_boxes = None self.exp_dlg = None self.exp_builder = None self.block_pts_layer = None self.px, self.py = [0, 0] self.last_point = QgsPointXY(0, 0) self.rbounds = None self.changes = dict() # dict with rasters changes {raster_id: RasterChanges instance} self.project = QgsProject.instance() self.crs_transform = None self.all_touched = None self.selection_mode = None self.spatial_index_time = dict() # {layer_id: creation time} self.spatial_index = dict() # {layer_id: spatial index} self.selection_layers_count = 1 self.debug = DEBUG self.logger = get_logger() if self.debug else None self.menu = u'Serval' self.actions = [] self.actions_always_on = [] self.toolbar = self.iface.addToolBar(u'Serval Main Toolbar') self.toolbar.setObjectName(u'Serval Main Toolbar') self.toolbar.setToolTip(u'Serval Main Toolbar') self.sel_toolbar = self.iface.addToolBar(u'Serval Selection Toolbar') self.sel_toolbar.setObjectName(u'Serval Selection Toolbar') self.sel_toolbar.setToolTip(u'Serval Selection Toolbar') # Map tools self.probe_tool = QgsMapToolEmitPoint(self.canvas) self.probe_tool.setObjectName('ServalProbeTool') self.probe_tool.setCursor(QCursor(QPixmap(icon_path('probe_tool.svg')), hotX=2, hotY=22)) self.probe_tool.canvasClicked.connect(self.point_clicked) self.draw_tool = QgsMapToolEmitPoint(self.canvas) self.draw_tool.setObjectName('ServalDrawTool') self.draw_tool.setCursor(QCursor(QPixmap(icon_path('draw_tool.svg')), hotX=2, hotY=22)) self.draw_tool.canvasClicked.connect(self.point_clicked) self.selection_tool = RasterCellSelectionMapTool(self.iface, self.uc, self.raster, debug=self.debug) self.selection_tool.setObjectName('RasterSelectionTool') self.map_tool_btn = dict() # {map tool: button activating the tool} self.iface.currentLayerChanged.connect(self.set_active_raster) self.project.layersAdded.connect(self.set_active_raster) self.canvas.mapToolSet.connect(self.check_active_tool) self.register_exp_functions() def load_settings(self): """Return plugin settings dict - default values are overriden by user prefered values from QSettings.""" self.default_settings = { "undo_steps": {"value": 3, "vtype": int}, } self.settings = dict() s = QSettings() s.beginGroup("serval") for k, v in self.default_settings.items(): user_val = s.value(k, v["value"], v["vtype"]) self.settings[k] = user_val def edit_settings(self): """Open dialog with plugin settings.""" s = QSettings() s.beginGroup("serval") k = "undo_steps" cur_val = self.settings[k] val_type = self.default_settings[k]["vtype"] cur_steps = s.value(k, cur_val, val_type) label = 'Nr of Undo/Redo steps:' steps, ok = QInputDialog.getInt(None, "Serval Settings", label, cur_steps) if not ok: return if steps >= 0: s.setValue("undo_steps", steps) self.load_settings() self.uc.show_info("Some new settings may require QGIS restart.") def initGui(self): _ = self.add_action( 'serval_icon.svg', text=u'Show Serval Toolbars', add_to_menu=True, callback=self.show_toolbar, always_on=True, ) _ = self.add_action( 'serval_icon.svg', text=u'Hide Serval Toolbars', add_to_menu=True, callback=self.hide_toolbar, always_on=True, ) self.probe_btn = self.add_action( 'probe.svg', text="Probe raster", callback=self.activate_probing, add_to_toolbar=self.toolbar, checkable=True, ) self.map_tool_btn[self.probe_tool] = self.probe_btn self.color_btn = QgsColorButton() self.color_btn.setColor(Qt.gray) self.color_btn.setMinimumSize(QSize(40, 24)) self.color_btn.setMaximumSize(QSize(40, 24)) self.toolbar.addWidget(self.color_btn) self.color_picker_connection(connect=True) self.color_btn.setDisabled(True) self.toolbar.addWidget(QLabel("Band:")) self.bands_cbo = QComboBox() self.bands_cbo.addItem("1", 1) self.toolbar.addWidget(self.bands_cbo) self.bands_cbo.currentIndexChanged.connect(self.update_active_bands) self.bands_cbo.setDisabled(True) self.spin_boxes = BandBoxes() self.toolbar.addWidget(self.spin_boxes) self.spin_boxes.enter_hit.connect(self.apply_spin_box_values) self.draw_btn = self.add_action( 'draw.svg', text="Apply Value(s) To Single Cell", callback=self.activate_drawing, add_to_toolbar=self.toolbar, checkable=True, ) self.map_tool_btn[self.draw_tool] = self.draw_btn self.apply_spin_box_values_btn = self.add_action( 'apply_const_value.svg', text="Apply Value(s) to Selection", callback=self.apply_spin_box_values, add_to_toolbar=self.toolbar, ) self.gom_btn = self.add_action( 'apply_nodata_value.svg', text="Apply NoData to Selection", callback=self.apply_nodata_value, add_to_toolbar=self.toolbar, ) self.exp_dlg_btn = self.add_action( 'apply_expression_value.svg', text="Apply Expression Value To Selection", callback=self.define_expression, add_to_toolbar=self.toolbar, checkable=False, ) self.low_pass_filter_btn = self.add_action( 'apply_low_pass_filter.svg', text="Apply Low-Pass 3x3 Filter To Selection", callback=self.apply_low_pass_filter, add_to_toolbar=self.toolbar, checkable=False, ) self.undo_btn = self.add_action( 'undo.svg', text="Undo", callback=self.undo, add_to_toolbar=self.toolbar, ) self.redo_btn = self.add_action( 'redo.svg', text="Redo", callback=self.redo, add_to_toolbar=self.toolbar, ) self.set_nodata_btn = self.add_action( 'set_nodata.svg', text="Edit Raster NoData Values", callback=self.set_nodata, add_to_toolbar=self.toolbar, ) self.settings_btn = self.add_action( 'edit_settings.svg', text="Serval Settings", callback=self.edit_settings, add_to_toolbar=self.toolbar, always_on=True, ) self.show_help = self.add_action( 'help.svg', text="Help", add_to_menu=True, callback=self.show_website, add_to_toolbar=self.toolbar, always_on=True, ) # Selection Toolbar line_width_icon = QIcon(icon_path("line_width.svg")) line_width_lab = QLabel() line_width_lab.setPixmap(line_width_icon.pixmap(22, 12)) self.sel_toolbar.addWidget(line_width_lab) self.line_width_sbox = QgsDoubleSpinBox() self.line_width_sbox.setMinimumSize(QSize(50, 24)) self.line_width_sbox.setMaximumSize(QSize(50, 24)) # self.line_width_sbox.setButtonSymbols(QAbstractSpinBox.NoButtons) self.line_width_sbox.setValue(1) self.line_width_sbox.setMinimum(0.01) self.line_width_sbox.setShowClearButton(False) self.line_width_sbox.setToolTip("Selection Line Width") self.line_width_sbox.valueChanged.connect(self.update_selection_tool) self.width_unit_cbo = QComboBox() self.width_units = ("map units", "pixel width", "pixel height", "hairline",) for u in self.width_units: self.width_unit_cbo.addItem(u) self.width_unit_cbo.setToolTip("Selection Line Width Unit") self.sel_toolbar.addWidget(self.line_width_sbox) self.sel_toolbar.addWidget(self.width_unit_cbo) self.width_unit_cbo.currentIndexChanged.connect(self.update_selection_tool) self.line_select_btn = self.add_action( 'select_line.svg', text="Select Raster Cells by Line", callback=self.activate_line_selection, add_to_toolbar=self.sel_toolbar, checkable=True, ) self.polygon_select_btn = self.add_action( 'select_polygon.svg', text="Select Raster Cells by Polygon", callback=self.activate_polygon_selection, add_to_toolbar=self.sel_toolbar, checkable=True, ) self.selection_from_layer_btn = self.add_action( 'select_from_layer.svg', text="Create Selection From Layer", callback=self.selection_from_layer, add_to_toolbar=self.sel_toolbar, ) self.selection_to_layer_btn = self.add_action( 'selection_to_layer.svg', text="Create Memory Layer From Selection", callback=self.selection_to_layer, add_to_toolbar=self.sel_toolbar, ) self.clear_selection_btn = self.add_action( 'clear_selection.svg', text="Clear selection", callback=self.clear_selection, add_to_toolbar=self.sel_toolbar, ) self.toggle_all_touched_btn = self.add_action( 'all_touched.svg', text="Toggle All Touched Get Selected", callback=self.toggle_all_touched, checkable=True, checked=True, add_to_toolbar=self.sel_toolbar, ) self.all_touched = True self.enable_toolbar_actions(enable=False) self.check_undo_redo_btns() def add_action(self, icon_name, callback=None, text="", enabled_flag=True, add_to_menu=False, add_to_toolbar=None, status_tip=None, whats_this=None, checkable=False, checked=False, always_on=False): icon = QIcon(icon_path(icon_name)) action = QAction(icon, text, self.iface.mainWindow()) action.triggered.connect(callback) action.setEnabled(enabled_flag) action.setCheckable(checkable) action.setChecked(checked) if status_tip is not None: action.setStatusTip(status_tip) if whats_this is not None: action.setWhatsThis(whats_this) if add_to_toolbar is not None: add_to_toolbar.addAction(action) if add_to_menu: self.iface.addPluginToMenu(self.menu, action) self.actions.append(action) if always_on: self.actions_always_on.append(action) return action def unload(self): self.changes = None if self.selection_tool: self.selection_tool.reset() if self.spin_boxes is not None: self.spin_boxes.remove_spinboxes() for action in self.actions: self.iface.removePluginMenu('Serval', action) self.iface.removeToolBarIcon(action) del self.toolbar del self.sel_toolbar self.iface.actionPan().trigger() self.unregister_exp_functions() def show_toolbar(self): if self.toolbar: self.toolbar.show() self.sel_toolbar.show() def hide_toolbar(self): if self.toolbar: self.toolbar.hide() self.sel_toolbar.hide() @staticmethod def register_exp_functions(): QgsExpression.registerFunction(nearest_feature_attr_value) QgsExpression.registerFunction(nearest_pt_on_line_interpolate_z) QgsExpression.registerFunction(intersecting_features_attr_average) QgsExpression.registerFunction(interpolate_from_mesh) @staticmethod def unregister_exp_functions(): QgsExpression.unregisterFunction('nearest_feature_attr_value') QgsExpression.unregisterFunction('nearest_pt_on_line_interpolate_z') QgsExpression.unregisterFunction('intersecting_features_attr_average') QgsExpression.unregisterFunction('interpolate_from_mesh') def uncheck_all_btns(self): self.probe_btn.setChecked(False) self.draw_btn.setChecked(False) self.gom_btn.setChecked(False) self.line_select_btn.setChecked(False) self.polygon_select_btn.setChecked(False) def check_active_tool(self, cur_tool): self.uncheck_all_btns() if cur_tool in self.map_tool_btn: self.map_tool_btn[cur_tool].setChecked(True) if cur_tool == self.selection_tool: if self.selection_mode == self.LINE_SELECTION: self.line_select_btn.setChecked(True) else: self.polygon_select_btn.setChecked(True) def activate_probing(self): self.mode = 'probe' self.canvas.setMapTool(self.probe_tool) def define_expression(self): if not self.selection_tool.selected_geometries: self.uc.bar_warn("No selection for raster layer. Select some cells and retry...") return self.handler.select(self.selection_tool.selected_geometries, all_touched_cells=self.all_touched) self.handler.create_cell_pts_layer() if self.handler.cell_pts_layer.featureCount() == 0: self.uc.bar_warn("No selection for raster layer. Select some cells and retry...") return self.exp_dlg = QgsExpressionBuilderDialog(self.handler.cell_pts_layer) self.exp_builder = self.exp_dlg.expressionBuilder() self.exp_dlg.accepted.connect(self.apply_exp_value) self.exp_dlg.show() def apply_exp_value(self): if not self.exp_dlg.expressionText() or not self.exp_builder.isExpressionValid(): return QApplication.setOverrideCursor(Qt.WaitCursor) exp = self.exp_dlg.expressionText() idx = self.handler.cell_pts_layer.addExpressionField(exp, QgsField('exp_val', QVariant.Double)) self.handler.exp_field_idx = idx self.handler.write_block() QApplication.restoreOverrideCursor() self.raster.triggerRepaint() def activate_drawing(self): self.mode = 'draw' self.canvas.setMapTool(self.draw_tool) def get_cur_line_width(self): width_coef = { "map units": 1., "pixel width": self.raster.rasterUnitsPerPixelX(), "pixel height": self.raster.rasterUnitsPerPixelY(), "hairline": 0.000001, } return self.line_width_sbox.value() * width_coef[self.width_unit_cbo.currentText()] def set_selection_tool(self, mode): if self.raster is None: self.uc.bar_warn("Select a raster layer") return self.selection_mode = mode self.selection_tool.init_tool(self.raster, mode=self.selection_mode, line_width=self.get_cur_line_width()) self.selection_tool.set_prev_tool(self.canvas.mapTool()) self.canvas.setMapTool(self.selection_tool) def activate_line_selection(self): self.set_selection_tool(self.LINE_SELECTION) def activate_polygon_selection(self): self.set_selection_tool(self.POLYGON_SELECTION) def update_selection_tool(self): """Reactivate the selection tool with updated line width and units.""" if self.selection_mode == self.LINE_SELECTION: self.activate_line_selection() elif self.selection_mode == self.POLYGON_SELECTION: self.activate_polygon_selection() else: pass def apply_values(self, new_values): QApplication.setOverrideCursor(Qt.WaitCursor) self.handler.select(self.selection_tool.selected_geometries, all_touched_cells=self.all_touched) self.handler.write_block(new_values) QApplication.restoreOverrideCursor() self.raster.triggerRepaint() def apply_values_single_cell(self, new_vals): """Create single cell selection and apply the new values.""" cp = self.last_point if self.logger: self.logger.debug(f"Changing single cell for pt {cp}") col, row = self.handler.point_to_index([cp.x(), cp.y()]) px, py = self.handler.index_to_point(row, col, upper_left=False) d = 0.001 bbox = QgsRectangle(px - d, py - d, px + d, py + d) if self.logger: self.logger.debug(f"Changing single cell in {bbox}") QApplication.setOverrideCursor(Qt.WaitCursor) self.handler.select([QgsGeometry.fromRect(bbox)], all_touched_cells=False, transform=False) self.handler.write_block(new_vals) QApplication.restoreOverrideCursor() self.raster.triggerRepaint() def apply_spin_box_values(self): if not self.selection_tool.selected_geometries: return self.apply_values(self.spin_boxes.get_values()) def apply_nodata_value(self): if not self.selection_tool.selected_geometries: return self.apply_values(self.handler.nodata_values) def apply_low_pass_filter(self): QApplication.setOverrideCursor(Qt.WaitCursor) self.handler.select(self.selection_tool.selected_geometries, all_touched_cells=self.all_touched) self.handler.write_block(low_pass_filter=True) QApplication.restoreOverrideCursor() self.raster.triggerRepaint() def clear_selection(self): if self.selection_tool: self.selection_tool.clear_all_selections() def selection_from_layer(self): """Create a new selection from layer.""" self.selection_tool.init_tool(self.raster, mode=self.POLYGON_SELECTION, line_width=self.get_cur_line_width()) dlg = LayerSelectDialog() if not dlg.exec_(): return cur_layer = dlg.cbo.currentLayer() if not cur_layer.type() == QgsMapLayerType.VectorLayer: return self.selection_tool.selection_from_layer(cur_layer) def selection_to_layer(self): """Create a memory layer from current selection""" geoms = self.selection_tool.selected_geometries if geoms is None or not self.raster: return crs_str = self.raster.crs().toProj() nr = self.selection_layers_count self.selection_layers_count += 1 mlayer = QgsVectorLayer(f"Polygon?crs={crs_str}&field=fid:int", f"Raster selection {nr}", "memory") fields = mlayer.dataProvider().fields() features = [] for i, geom in enumerate(geoms): feat = QgsFeature(fields) feat["fid"] = i + 1 feat.setGeometry(geom) features.append(feat) mlayer.dataProvider().addFeatures(features) self.project.addMapLayer(mlayer) def toggle_all_touched(self): """Toggle selection mode.""" # button is toggled automatically when clicked, just update the attribute self.all_touched = self.toggle_all_touched_btn.isChecked() def point_clicked(self, point=None, button=None): if self.raster is None: self.uc.bar_warn("Choose a raster to work with...", dur=3) return if self.logger: self.logger.debug(f"Clicked point in canvas CRS: {point if point else self.last_point}") if point is None: ptxy_in_src_crs = self.last_point else: if self.crs_transform: if self.logger: self.logger.debug(f"Transforming clicked point {point}") try: ptxy_in_src_crs = self.crs_transform.transform(point) except QgsCsException as err: self.uc.show_warn( "Point coordinates transformation failed! Check the raster projection:\n\n{}".format(repr(err))) return else: ptxy_in_src_crs = QgsPointXY(point.x(), point.y()) if self.logger: self.logger.debug(f"Clicked point in raster CRS: {ptxy_in_src_crs}") self.last_point = ptxy_in_src_crs ident_vals = self.handler.provider.identify(ptxy_in_src_crs, QgsRaster.IdentifyFormatValue).results() cur_vals = list(ident_vals.values()) # check if the point is within active raster extent if not self.rbounds[0] <= ptxy_in_src_crs.x() <= self.rbounds[2]: self.uc.bar_info("Out of x bounds", dur=3) return if not self.rbounds[1] <= ptxy_in_src_crs.y() <= self.rbounds[3]: self.uc.bar_info("Out of y bounds", dur=3) return if self.mode == 'draw': new_vals = self.spin_boxes.get_values() if self.logger: self.logger.debug(f"Applying const value {new_vals}") self.apply_values_single_cell(new_vals) else: self.spin_boxes.set_values(cur_vals) if 2 < self.handler.bands_nr < 5: self.color_picker_connection(connect=False) self.color_btn.setColor(QColor(*self.spin_boxes.get_values()[:4])) self.color_picker_connection(connect=True) def set_values_from_picker(self, c): """Set bands spinboxes values after color change in the color picker""" values = None if self.handler.bands_nr > 2: values = [c.red(), c.green(), c.blue()] if self.handler.bands_nr == 4: values.append(c.alpha()) if values: self.spin_boxes.set_values(values) def set_nodata(self): """Set NoData value(s) for each band of current raster.""" if not self.raster: self.uc.bar_warn('Select a raster layer to define/change NoData value!') return if self.handler.provider.userNoDataValues(1): note = '\nNote: there is a user defined NODATA value.\nCheck the raster properties (Transparency).' else: note = '' dt = self.handler.provider.dataType(1) # current NODATA value if self.handler.provider.sourceHasNoDataValue(1): cur_nodata = self.handler.provider.sourceNoDataValue(1) if dt < 6: cur_nodata = '{0:d}'.format(int(float(cur_nodata))) else: cur_nodata = '' label = 'Define/change raster NODATA value.\n\n' label += 'Raster src_data type: {}.{}'.format(dtypes[dt]['name'], note) nd, ok = QInputDialog.getText(None, "Define NODATA Value", label, QLineEdit.Normal, str(cur_nodata)) if not ok: return if not is_number(nd): self.uc.show_warn('Wrong NODATA value!') return new_nodata = int(nd) if dt < 6 else float(nd) # set the NODATA value for each band res = [] for nr in self.handler.bands_range: res.append(self.handler.provider.setNoDataValue(nr, new_nodata)) self.handler.provider.sourceHasNoDataValue(nr) if False in res: self.uc.show_warn('Setting new NODATA value failed!') else: self.uc.bar_info('Successful setting new NODATA values!', dur=2) self.set_active_raster() self.raster.triggerRepaint() def check_undo_redo_btns(self): """Enable/Disable undo and redo buttons based on availability of undo/redo for current raster.""" self.undo_btn.setDisabled(True) self.redo_btn.setDisabled(True) if self.raster is None or self.raster.id() not in self.changes: return changes = self.changes[self.raster.id()] if changes.nr_undos() > 0: self.undo_btn.setEnabled(True) if changes.nr_redos() > 0: self.redo_btn.setEnabled(True) def enable_toolbar_actions(self, enable=True): """Enable / disable all toolbar actions but Help (for vectors and unsupported rasters)""" for widget in self.actions + [self.width_unit_cbo, self.line_width_sbox]: widget.setEnabled(enable) if widget in self.actions_always_on: widget.setEnabled(True) self.spin_boxes.enable(enable) @staticmethod def check_layer(layer): """Check if we can work with the raster""" if layer is None: return False if layer.type() != QgsMapLayerType.RasterLayer: return False if layer.providerType() != 'gdal': return False if all([ layer.isValid(), layer.crs() is not None, check_gdal_driver_create_option(layer), # GDAL driver has CREATE option os.path.isfile(layer.dataProvider().dataSourceUri()), # is it a local file? ]): return True else: return False def set_bands_cbo(self): self.bands_cbo.currentIndexChanged.disconnect(self.update_active_bands) self.bands_cbo.clear() for band in self.handler.bands_range: self.bands_cbo.addItem(f"{band}", [band]) if self.handler.bands_nr > 1: self.bands_cbo.addItem(self.RGB, [1, 2, 3]) self.bands_cbo.setCurrentIndex(0) self.bands_cbo.currentIndexChanged.connect(self.update_active_bands) def update_active_bands(self, idx): bands = self.bands_cbo.currentData() self.handler.active_bands = bands self.spin_boxes.create_spinboxes(bands, self.handler.data_types, self.handler.nodata_values) self.color_btn.setEnabled(len(bands) > 1) self.exp_dlg_btn.setEnabled(len(bands) == 1) def set_active_raster(self): """Active layer has changed - check if it is a raster layer and prepare it for the plugin""" old_spin_boxes_values = self.spin_boxes.get_values() self.crs_transform = None layer = self.iface.activeLayer() if self.check_layer(layer): self.raster = layer self.crs_transform = None if self.project.crs() == self.raster.crs() else \ QgsCoordinateTransform(self.project.crs(), self.raster.crs(), self.project) self.handler = RasterHandler(self.raster, self.uc, self.debug) supported, unsupported_type = self.handler.write_supported() if supported: self.enable_toolbar_actions() self.set_bands_cbo() self.spin_boxes.create_spinboxes(self.handler.active_bands, self.handler.data_types, self.handler.nodata_values) if self.handler.bands_nr == len(old_spin_boxes_values): self.spin_boxes.set_values(old_spin_boxes_values) self.bands_cbo.setEnabled(self.handler.bands_nr > 1) self.color_btn.setEnabled(len(self.handler.active_bands) > 1) self.rbounds = self.raster.extent().toRectF().getCoords() self.handler.raster_changed.connect(self.add_to_undo) if self.raster.id() not in self.changes: self.changes[self.raster.id()] = RasterChanges(nr_to_keep=self.settings["undo_steps"]) else: msg = f"The raster has unsupported src_data type: {unsupported_type}" msg += "\nServal can't work with it, sorry..." self.uc.show_warn(msg) self.enable_toolbar_actions(enable=False) self.reset_raster() else: # unsupported raster self.enable_toolbar_actions(enable=False) self.reset_raster() self.check_undo_redo_btns() def add_to_undo(self, change): """Add the old and new blocks to undo stack.""" self.changes[self.raster.id()].add_change(change) self.check_undo_redo_btns() if self.logger: self.logger.debug(self.get_undo_redo_values()) def get_undo_redo_values(self): changes = self.changes[self.raster.id()] return f"nr undos: {changes.nr_undos()}, redos: {changes.nr_redos()}" def undo(self): undo_data = self.changes[self.raster.id()].undo() self.handler.write_block_undo(undo_data) self.raster.triggerRepaint() self.check_undo_redo_btns() def redo(self): redo_data = self.changes[self.raster.id()].redo() self.handler.write_block_undo(redo_data) self.raster.triggerRepaint() self.check_undo_redo_btns() def reset_raster(self): self.raster = None self.color_btn.setDisabled(True) def color_picker_connection(self, connect=True): if connect: self.color_btn.colorChanged.connect(self.set_values_from_picker) else: self.color_btn.colorChanged.disconnect(self.set_values_from_picker) @staticmethod def show_website(): QDesktopServices.openUrl(QUrl("https://github.com/lutraconsulting/serval/blob/master/Serval/docs/user_manual.md")) def recreate_spatial_index(self, layer): """Check if spatial index exists for the layer and if it is relatively old and eventually recreate it.""" ctime = self.spatial_index_time[layer.id()] if layer.id() in self.spatial_index_time else None if ctime is None or datetime.now() - ctime > timedelta(seconds=30): self.spatial_index = QgsSpatialIndex(layer.getFeatures(), None, QgsSpatialIndex.FlagStoreFeatureGeometries) self.spatial_index_time[layer.id()] = datetime.now() def get_nearest_feature(self, pt_feat, vlayer_id): """Given the point feature, return nearest feature from vlayer.""" vlayer = self.project.mapLayer(vlayer_id) self.recreate_spatial_index(vlayer) ptxy = pt_feat.geometry().asPoint() near_fid = self.spatial_index.nearestNeighbor(ptxy)[0] return vlayer.getFeature(near_fid) def nearest_feature_attr_value(self, pt_feat, vlayer_id, attr_name): """Find nearest feature to pt_feat and return its attr_name attribute value.""" near_feat = self.get_nearest_feature(pt_feat, vlayer_id) return near_feat[attr_name] def nearest_pt_on_line_interpolate_z(self, pt_feat, vlayer_id): """Find nearest line feature to pt_feat and interpolate z value from vertices.""" near_feat = self.get_nearest_feature(pt_feat, vlayer_id) near_geom = near_feat.geometry() closest_pt_dist = near_geom.lineLocatePoint(pt_feat.geometry()) closest_pt = near_geom.interpolate(closest_pt_dist) return closest_pt.get().z() def intersecting_features_attr_average(self, pt_feat, vlayer_id, attr_name, only_center): """ Find all features intersecting current feature (cell center, or raster cell polygon) and calculate average value of their attr_name attribute. """ vlayer = self.project.mapLayer(vlayer_id) self.recreate_spatial_index(vlayer) ptxy = pt_feat.geometry().asPoint() pt_x, pt_y = ptxy.x(), ptxy.y() dxy = 0.001 half_pix_x = self.handler.pixel_size_x / 2. half_pix_y = self.handler.pixel_size_y / 2. if only_center: cell = QgsRectangle(pt_x, pt_y, pt_x + dxy, pt_y + dxy) else: cell = QgsRectangle(pt_x - half_pix_x, pt_y - half_pix_y, pt_x + half_pix_x, pt_y + half_pix_y) inter_fids = self.spatial_index.intersects(cell) values = [] for fid in inter_fids: feat = vlayer.getFeature(fid) if not feat.geometry().intersects(cell): continue val = feat[attr_name] if not is_number(val): continue values.append(val) if len(values) == 0: return None return sum(values) / float(len(values)) def interpolate_from_mesh(self, pt_feat, mesh_layer_id, group, dataset, above_existing): """Interpolate from mesh.""" mesh_layer = self.project.mapLayer(mesh_layer_id) ptxy = pt_feat.geometry().asPoint() dataset_val = mesh_layer.datasetValue(QgsMeshDatasetIndex(group, dataset), ptxy) val = dataset_val.scalar() if math.isnan(val): return val if above_existing: ident_vals = self.handler.provider.identify(ptxy, QgsRaster.IdentifyFormatValue).results() org_val = list(ident_vals.values())[0] if org_val == self.handler.nodata_values[0]: return val return max(org_val, val) else: return val