364 lines
14 KiB
Python
364 lines
14 KiB
Python
from colorsys import hls_to_rgb
|
||
import os
|
||
from osgeo import gdal
|
||
from PyQt5.QtWidgets import QWidget, QApplication, QMainWindow, QToolBox
|
||
from PyQt5.QtWidgets import QDialog, QFileDialog, QLineEdit, QPushButton, QVBoxLayout, QHBoxLayout, QLabel, QMessageBox,QSpacerItem
|
||
from PyQt5.QtCore import Qt, QSize,QSettings,pyqtSignal,QThread
|
||
from PyQt5.QtGui import QIcon,QColor
|
||
from PyQt5 import QtGui
|
||
from threading import Thread
|
||
from rscder.utils.icons import IconInstance
|
||
from rscder.utils.setting import Settings
|
||
from qgis.gui import QgsMapCanvas
|
||
from rscder.utils.project import MultiBandRasterLayer, Project, RasterLayer
|
||
from rscder.gui.progress_bar import MetroCircleProgress
|
||
|
||
class progressDialog(QDialog):
|
||
def __init__(self, parent=None,name='default') -> None:
|
||
super(progressDialog,self).__init__(parent)
|
||
self.setWindowTitle(name)
|
||
self.setWindowIcon(IconInstance().RASTER)
|
||
self.setFixedSize(300,80)
|
||
# self.setWindowIcon(IconInstance().RASTER)
|
||
tlayout=QVBoxLayout()
|
||
tlayout.setContentsMargins(10,0,0,0)
|
||
hlayout=QHBoxLayout()
|
||
h2layout=QHBoxLayout()
|
||
self.label=QLabel('加载影像...')
|
||
hlayout.addWidget(self.label)
|
||
hlayout.addItem(QSpacerItem(30,0))
|
||
tlayout.addLayout(hlayout)
|
||
h2layout.addWidget(MetroCircleProgress(self, radius=7))
|
||
tlayout.addLayout(h2layout)
|
||
self.setLayout(tlayout)
|
||
def setlabel(self,content:str):
|
||
self.label.setText(content)
|
||
|
||
class loader(QDialog):
|
||
signal1=pyqtSignal(str)
|
||
def __init__(self, parent=None) -> None:
|
||
super().__init__(parent)
|
||
self.setWindowTitle('载入数据')
|
||
self.setWindowIcon(IconInstance().DATA_LOAD)
|
||
self.pyramid:bool=False
|
||
self.temp1=''
|
||
self.temp2=''
|
||
self.path1=''
|
||
self.path2=''
|
||
self.bands=['red:','green:','blue:','NIR:']
|
||
self.bandsorder=[3,2,1,4]
|
||
self.mapcanva1 = QgsMapCanvas(self)
|
||
self.mapcanva2 = QgsMapCanvas(self)
|
||
self.mapcanva1.setCanvasColor(QColor(0, 0, 0))
|
||
self.mapcanva2.setCanvasColor(QColor(0, 0, 0))
|
||
self.mapcanva1.setFixedWidth(200)
|
||
self.mapcanva1.setFixedHeight(200)
|
||
self.mapcanva2.setFixedWidth(200)
|
||
self.mapcanva2.setFixedHeight(200)
|
||
maplayout=QHBoxLayout()
|
||
maplayout.addWidget(self.mapcanva1)
|
||
maplayout.addWidget(self.mapcanva2)
|
||
|
||
path1_label = QLabel('时相1影像:')
|
||
path1_label.setFixedWidth(60)
|
||
path1_input = QLineEdit()
|
||
path1_input.setPlaceholderText('时相1影像')
|
||
path1_input.setToolTip('时相1影像')
|
||
path1_input.setReadOnly(True)
|
||
path1_input.setText(self.path1)
|
||
self.path1_input = path1_input
|
||
|
||
|
||
path1_open = QPushButton('...', self)
|
||
# path1_open.setEnabled(False)
|
||
path1_open.setFixedWidth(30)
|
||
path1_open.clicked.connect(self.open_file1)
|
||
|
||
|
||
path1_layout=QHBoxLayout()
|
||
path1_layout.addWidget(path1_label)
|
||
path1_layout.addWidget(path1_input)
|
||
path1_layout.addWidget(path1_open)
|
||
|
||
|
||
labels1=[QLabel() for i in range(4)]
|
||
style1_inputs=[QLineEdit() for i in range(4)]
|
||
for i in range(4):
|
||
labels1[i].setText(self.bands[i])
|
||
labels1[i].setFixedWidth(50)
|
||
style1_inputs[i].setText(str(self.bandsorder[i]))
|
||
style1_inputs[i].setFixedWidth(20)
|
||
self.style1_inputs=style1_inputs
|
||
style1_set = QPushButton(self.tr('确定'), self)
|
||
style1_set.setFixedWidth(40)
|
||
style1_set.clicked.connect(self.set_style1)
|
||
style1_set.setEnabled(False)
|
||
style1_layout=QHBoxLayout()
|
||
for i in range(4):
|
||
style1_layout.addWidget(labels1[i])
|
||
style1_layout.addWidget(style1_inputs[i])
|
||
style1_layout.addWidget(style1_set)
|
||
|
||
path2_label = QLabel('时相2影像:')
|
||
path2_label.setFixedWidth(60)
|
||
path2_input = QLineEdit()
|
||
path2_input.setPlaceholderText('时相2影像')
|
||
path2_input.setToolTip('时相2影像')
|
||
path2_input.setReadOnly(True)
|
||
path2_input.setText(self.path2)
|
||
self.path2_input = path2_input
|
||
|
||
path2_open = QPushButton('...', self)
|
||
path2_open.setFixedWidth(30)
|
||
# path2_open.setEnabled(False)
|
||
path2_open.clicked.connect(self.open_file2)
|
||
|
||
path2_layout=QHBoxLayout()
|
||
path2_layout.addWidget(path2_label)
|
||
path2_layout.addWidget(path2_input)
|
||
path2_layout.addWidget(path2_open)
|
||
|
||
|
||
labels2=[QLabel() for i in range(4)]
|
||
|
||
style2_inputs=[QLineEdit() for i in range(4)]
|
||
for i in range(4):
|
||
labels2[i].setText(self.bands[i])
|
||
labels2[i].setFixedWidth(50)
|
||
style2_inputs[i].setText(str(self.bandsorder[i]))
|
||
style2_inputs[i].setFixedWidth(20)
|
||
self.style2_inputs=style2_inputs
|
||
style2_set = QPushButton(self.tr('确定'), self)
|
||
style2_set.setFixedWidth(40)
|
||
style2_set.clicked.connect(self.set_style2)
|
||
style2_set.setEnabled(False)
|
||
self.open1=style1_set
|
||
self.open2=style2_set
|
||
style2_layout=QHBoxLayout()
|
||
for i in range(4):
|
||
style2_layout.addWidget(labels2[i])
|
||
style2_layout.addWidget(style2_inputs[i])
|
||
style2_layout.addWidget(style2_set)
|
||
|
||
ok_button = QPushButton(self.tr('确定'))
|
||
cancel_button = QPushButton(self.tr('取消'))
|
||
|
||
ok_button.clicked.connect(self.ok)
|
||
cancel_button.clicked.connect(self.cancel)
|
||
|
||
button_layout = QHBoxLayout()
|
||
button_layout.setDirection(QHBoxLayout.RightToLeft)
|
||
button_layout.addWidget(ok_button, 0, Qt.AlignRight)
|
||
button_layout.addWidget(cancel_button, 0, Qt.AlignRight)
|
||
|
||
main_layout = QVBoxLayout()
|
||
main_layout.addLayout(path1_layout)
|
||
main_layout.addLayout(style1_layout)
|
||
main_layout.addLayout(path2_layout)
|
||
main_layout.addLayout(style2_layout)
|
||
main_layout.addLayout(maplayout)
|
||
main_layout.addLayout(button_layout)
|
||
self.setLayout(main_layout)
|
||
|
||
def open_file1(self):
|
||
path1 = QFileDialog.getOpenFileNames(self, '打开数据1', Settings.General().last_path, '*.*')
|
||
if path1[0]!='':
|
||
try:
|
||
|
||
self.path1 = path1[0][0]
|
||
self.path1_input.setText(self.path1)
|
||
result=QMessageBox.question(self, '提示', '是否创建图像金字塔', QMessageBox.Yes | QMessageBox.No, QMessageBox.No) #默认关闭界面选择No
|
||
if result==QMessageBox.Yes:
|
||
progress1=progressDialog(self,'加载时相一')
|
||
progress1.setModal(False)
|
||
self.temp1=os.path.join(Project().other_path,'temp1.tif')
|
||
t1=GdalPreviewImage(self.path1,self.temp1,1024,self)
|
||
# t1.started.connect(progress1.show)
|
||
t1.finished.connect(self.loadfile1)
|
||
t1.finished.connect(lambda :progress1.setlabel('创建影像金字塔..'))
|
||
t2=build_pyramids_overviews(self.path1,self)
|
||
t2.finished.connect(progress1.hide)
|
||
t1.start()
|
||
t1.finished.connect(t2.start)
|
||
# t2.start()
|
||
progress1.show()
|
||
else:
|
||
progress1=progressDialog(self,'加载时相一')
|
||
progress1.setModal(False)
|
||
self.temp1=os.path.join(Project().other_path,'temp1.tif')
|
||
t1=GdalPreviewImage(self.path1,self.temp1,1024)
|
||
# t1.started.connect(progress1.show)
|
||
t1.finished.connect(self.loadfile1)
|
||
t1.finished.connect(progress1.hide)
|
||
t1.start()
|
||
progress1.show()
|
||
|
||
|
||
except:
|
||
return
|
||
def loadfile1(self):
|
||
try:
|
||
self.left_layer=MultiBandRasterLayer(path=self.temp1)
|
||
self.mapcanva1.setLayers([self.left_layer.layer])
|
||
self.mapcanva1.zoomToFeatureExtent(self.left_layer.layer.extent())
|
||
self.open1.setEnabled(True)
|
||
except:
|
||
return
|
||
def open_file2(self):
|
||
path2 = QFileDialog.getOpenFileNames(self, '打开数据2', Settings.General().last_path, '*.*')
|
||
if path2[0]!='':
|
||
|
||
self.path2 = path2[0][0]
|
||
self.path2_input.setText(self.path2)
|
||
result=QMessageBox.question(self, '提示', '是否创建图像金字塔', QMessageBox.Yes | QMessageBox.No, QMessageBox.No) #默认关闭界面选择No
|
||
|
||
if result==QMessageBox.Yes:
|
||
progress2=progressDialog(self,'加载时相二')
|
||
progress2.setModal(False)
|
||
# progress1.show
|
||
self.temp2=os.path.join(Project().other_path,'temp2.tif')
|
||
t1=GdalPreviewImage(self.path2,self.temp2,1024,self)
|
||
# t1.started.connect(progress1.show)
|
||
t1.finished.connect(self.loadfile2)
|
||
t1.finished.connect(lambda :progress2.setlabel('创建影像金字塔..'))
|
||
t2=build_pyramids_overviews(self.path2,self)
|
||
t2.finished.connect(progress2.hide)
|
||
t1.start()
|
||
t1.finished.connect(t2.start)
|
||
progress2.show()
|
||
else:
|
||
progress2=progressDialog(self,'加载时相二')
|
||
progress2.setModal(False)
|
||
self.temp2=os.path.join(Project().other_path,'temp2.tif')
|
||
t1=GdalPreviewImage(self.path2,self.temp2,1024)
|
||
# t1.started.connect(progress1.show)
|
||
t1.finished.connect(self.loadfile2)
|
||
t1.finished.connect(progress2.hide)
|
||
t1.start()
|
||
progress2.show()
|
||
|
||
|
||
|
||
def loadfile2(self):
|
||
try:
|
||
self.right_layer=MultiBandRasterLayer(path=self.temp2)
|
||
self.mapcanva2.setLayers([self.right_layer.layer])
|
||
self.mapcanva2.zoomToFeatureExtent(self.right_layer.layer.extent())
|
||
self.open2.setEnabled(True)
|
||
except:
|
||
return
|
||
def ok(self):
|
||
self.bandsorder1=[int(q.text()) for q in self.style1_inputs ]
|
||
self.style1={'r':self.bandsorder1[0],'g':self.bandsorder1[1],'b':self.bandsorder1[2],'NIR':self.bandsorder1[3]}
|
||
|
||
self.bandsorder2=[int(q.text()) for q in self.style2_inputs ]
|
||
self.style2={'r':self.bandsorder2[0],'g':self.bandsorder2[1],'b':self.bandsorder2[2],'NIR':self.bandsorder2[3]}
|
||
|
||
if self.path1 == '':
|
||
QMessageBox.warning(self, 'Warning', 'Please select pic 1!')
|
||
return
|
||
if self.path2 == '':
|
||
QMessageBox.warning(self, 'Warning', 'Please select pic 2!')
|
||
return
|
||
self.accept()
|
||
def cancel(self):
|
||
self.reject()
|
||
|
||
def open_alg(self,path):
|
||
pass
|
||
|
||
def set_style1(self):
|
||
self.bandsorder1=[int(q.text()) for q in self.style1_inputs ]
|
||
self.style1={'r':self.bandsorder1[0],'g':self.bandsorder1[1],'b':self.bandsorder1[2],'NIR':self.bandsorder1[3]}
|
||
self.left_layer.set_stlye(self.style1)
|
||
|
||
def set_style2(self):
|
||
self.bandsorder2=[int(q.text()) for q in self.style2_inputs ]
|
||
self.style2={'r':self.bandsorder2[0],'g':self.bandsorder2[1],'b':self.bandsorder2[2],'NIR':self.bandsorder2[3]}
|
||
self.right_layer.set_stlye(self.style2)
|
||
|
||
class GdalPreviewImage(QThread):
|
||
def __init__(self,srcFile,tarFilename,width=1024.0,parent=None) -> None:
|
||
super(GdalPreviewImage,self).__init__(parent)
|
||
self.srcFile=srcFile
|
||
self.tarFilename=tarFilename
|
||
self.width=width
|
||
def run(self):
|
||
try:
|
||
srcFile=self.srcFile
|
||
tarFilename=self.tarFilename
|
||
width=self.width
|
||
p=tarFilename
|
||
dataset = gdal.Open(srcFile, gdal.GA_ReadOnly)
|
||
srcProjection = dataset.GetProjection()
|
||
srcGeoTransform = dataset.GetGeoTransform()
|
||
srcWidth = dataset.RasterXSize
|
||
srcHeight = dataset.RasterYSize
|
||
srcBandCount = dataset.RasterCount
|
||
# srcNoDatas = [
|
||
# dataset.GetRasterBand(bandIndex).GetNoDataValue()
|
||
# for bandIndex in range(1, srcBandCount+1)
|
||
# ]
|
||
# print(srcNoDatas)
|
||
srcBandDataType = dataset.GetRasterBand(1).DataType
|
||
# 创建重采样后的栅格
|
||
outFilePath = p
|
||
resampleFactor=width/srcWidth
|
||
driver = gdal.GetDriverByName('GTiff')
|
||
outWidth = int(srcWidth * resampleFactor)
|
||
outHeight = int(srcHeight * resampleFactor)
|
||
outDataset = driver.Create(
|
||
outFilePath,
|
||
outWidth,
|
||
outHeight,
|
||
srcBandCount,
|
||
srcBandDataType
|
||
)
|
||
print(outDataset)
|
||
geoTransforms = list(srcGeoTransform)
|
||
geoTransforms[1] = geoTransforms[1]/resampleFactor
|
||
geoTransforms[5] = geoTransforms[5]/resampleFactor
|
||
outGeoTransform = tuple(geoTransforms)
|
||
outDataset.SetGeoTransform(outGeoTransform)
|
||
outDataset.SetProjection(srcProjection)
|
||
# for bandIndex in range(1, srcBandCount+1):
|
||
# band = outDataset.GetRasterBand(bandIndex)
|
||
# band.SetNoDataValue(srcNoDatas[bandIndex-1])
|
||
gdal.ReprojectImage(
|
||
dataset,
|
||
outDataset,
|
||
srcProjection,
|
||
srcProjection,
|
||
gdal.gdalconst.GRA_NearestNeighbour,
|
||
0.0, 0.0,
|
||
)
|
||
del outDataset
|
||
except:
|
||
pass
|
||
# return outFilePath
|
||
|
||
|
||
class build_pyramids_overviews(QThread):
|
||
def __init__(self,filename,parent=None) -> None:
|
||
super(build_pyramids_overviews,self).__init__(parent)
|
||
self.filename=filename
|
||
def run(self):
|
||
try:
|
||
filename=self.filename
|
||
image:gdal.Dataset = gdal.Open(filename, 0)
|
||
# 如果第二个参数设置为0，则金字塔文件建立在外面
|
||
# 如果第二个参数缺省或者为1，则金字塔文件建立在文件内
|
||
gdal.SetConfigOption('COMPRESS_OVERVIEW', 'DEFLATE')
|
||
ov_list = [2, 4,6, 8, 12,16,24, 32, 48,64,96,128]
|
||
image.BuildOverviews("NEAREST", len(ov_list),overviewlist=ov_list)
|
||
del image
|
||
except:
|
||
pass
|
||
|
||
if __name__ == '__main__':
|
||
import sys
|
||
|
||
app = QApplication(sys.argv)
|
||
w = progressDialog()
|
||
w.show()
|
||
sys.exit(app.exec_()) |