221 lines
8.1 KiB
Python
221 lines
8.1 KiB
Python
import shutil
|
|
from rscder.gui.actions import ActionManager
|
|
from rscder.utils.icons import IconInstance
|
|
from rscder.utils.project import Project, PairLayer, ResultPointLayer
|
|
from rscder.plugins.basic import BasicPlugin
|
|
from PyQt5.QtWidgets import QDialog, QHBoxLayout, QFileDialog, QComboBox, QVBoxLayout, QPushButton, QLabel, QLineEdit, QAction
|
|
from PyQt5.QtGui import QIcon
|
|
from PyQt5.QtCore import Qt
|
|
from osgeo import gdal, gdal_array
|
|
import numpy as np
|
|
import threading
|
|
class ExportDialog(QDialog):
|
|
|
|
TXT='*.txt'
|
|
TIF='*.tif'
|
|
|
|
def __init__(self, parent=None, out_type = TXT):
|
|
super().__init__(parent)
|
|
|
|
self.setWindowTitle('Export')
|
|
self.setWindowIcon(IconInstance().LOGO)
|
|
|
|
self.out_path = None
|
|
self.result_layer = None
|
|
|
|
result_layer_select_label = QLabel('选择结果:')
|
|
|
|
result_layer_select = QComboBox(self)
|
|
|
|
result_layer_select.addItem('---', None)
|
|
for layer in Project().layers.values():
|
|
for result_layer in layer.layers:
|
|
if isinstance(result_layer, ResultPointLayer):
|
|
result_layer_select.addItem(IconInstance().VECTOR, result_layer.name, result_layer)
|
|
|
|
def on_result_layer_select(index):
|
|
self.result_layer = result_layer_select.currentData()
|
|
|
|
result_layer_select.currentIndexChanged.connect(on_result_layer_select)
|
|
|
|
out_path_label = QLabel('输出路径:')
|
|
out_path_text = QLineEdit(self)
|
|
out_path_text.setReadOnly(True)
|
|
out_path_text.setPlaceholderText('选择输出路径')
|
|
|
|
def on_out_path_btn():
|
|
select_file = QFileDialog.getSaveFileName(self, '选择输出路径', '', out_type)
|
|
if select_file[0]:
|
|
out_path_text.setText(select_file[0])
|
|
self.out_path = select_file[0]
|
|
out_path_btn = QPushButton('...', self)
|
|
out_path_btn.clicked.connect(on_out_path_btn)
|
|
|
|
ok_btn = QPushButton('确定', self)
|
|
ok_btn.clicked.connect(self.accept)
|
|
cancel_btn = QPushButton('取消', self)
|
|
cancel_btn.clicked.connect(self.reject)
|
|
|
|
|
|
hbox1 = QHBoxLayout()
|
|
hbox1.addWidget(result_layer_select_label)
|
|
hbox1.addWidget(result_layer_select)
|
|
|
|
hbox2 = QHBoxLayout()
|
|
hbox2.addWidget(out_path_label)
|
|
hbox2.addWidget(out_path_text)
|
|
hbox2.addWidget(out_path_btn)
|
|
|
|
hbox3 = QHBoxLayout()
|
|
hbox3.addWidget(ok_btn,0,alignment=Qt.AlignHCenter)
|
|
hbox3.addWidget(cancel_btn,0,alignment=Qt.AlignHCenter)
|
|
|
|
vbox = QVBoxLayout()
|
|
vbox.addLayout(hbox1)
|
|
vbox.addLayout(hbox2)
|
|
vbox.addLayout(hbox3)
|
|
|
|
self.setLayout(vbox)
|
|
|
|
class ExportPlugin(BasicPlugin):
|
|
|
|
@staticmethod
|
|
def info():
|
|
return {
|
|
'name': 'Export',
|
|
'description': 'Export to other format',
|
|
'author': 'RSCDER',
|
|
}
|
|
|
|
def set_action(self):
|
|
self.export_txt = QAction(IconInstance().DOCUMENT, '导出为 Arcgis 兼容的TXT', self.mainwindow)
|
|
self.export_txt.triggered.connect(self.export_txt_action)
|
|
|
|
self.export_bin = QAction(IconInstance().DOCUMENT, '导出栅格二值变化检测结果')
|
|
self.export_bin.triggered.connect(self.export_bin_action)
|
|
|
|
self.export_rgb = QAction(IconInstance().DOCUMENT, '导出变化幅度渲染结果')
|
|
self.export_rgb.triggered.connect(self.export_rgb_action)
|
|
ActionManager().export_menu.addAction(self.export_txt)
|
|
ActionManager().export_menu.addAction(self.export_bin)
|
|
ActionManager().export_menu.addAction(self.export_rgb)
|
|
|
|
# self.ctx['toolbar'].addAction(self.export_txt)
|
|
|
|
def export_bin_action(self):
|
|
dialog = ExportDialog(self.mainwindow, ExportDialog.TIF)
|
|
if dialog.exec_():
|
|
result:ResultPointLayer = dialog.result_layer
|
|
out = dialog.out_path
|
|
if result:
|
|
shutil.copy(result.result_path['path'], out)
|
|
self.message_box.info('导出成功')
|
|
|
|
def export_txt_action(self):
|
|
dialog = ExportDialog(self.mainwindow, ExportDialog.TXT)
|
|
if dialog.exec_():
|
|
result = dialog.result_layer
|
|
out = dialog.out_path
|
|
if result:
|
|
shutil.copy(result.path, out)
|
|
self.message_box.info('导出成功')
|
|
|
|
def export_rgb_action(self):
|
|
dialog = ExportDialog(self.mainwindow, ExportDialog.TIF)
|
|
btn = QPushButton('选择样式文件')
|
|
self.style_path = ''
|
|
def select_style():
|
|
select_file = QFileDialog.getOpenFileName(self.mainwindow, '选择样式文件', '*.*')
|
|
if select_file[0]:
|
|
# style_path.setText(select_file[0])
|
|
self.style_path = select_file[0]
|
|
btn.setText(select_file[0])
|
|
btn.clicked.connect(select_style)
|
|
dialog.layout().addWidget(btn)
|
|
|
|
default_style = [
|
|
[0, 0, 255, 0],
|
|
[1, 255, 0, 0]
|
|
]
|
|
|
|
if self.style_path == '':
|
|
style = default_style
|
|
else:
|
|
style = np.loadtxt(self.style_path, comments='#', delimiter=',')
|
|
style = style.tolist()
|
|
|
|
if dialog.exec_():
|
|
result = dialog.result_layer.result_path['cmi']
|
|
out = dialog.out_path
|
|
t = threading.Thread(target=self.render, args=(style, result, out)) # 创建线程并执行RENDER函数
|
|
# self.render(style, result, out)
|
|
t.start()
|
|
|
|
def render(self, style:list, path, out):
|
|
self.send_message.emit('正在导出' )
|
|
data = gdal_array.LoadFile(path)
|
|
if len(data.shape) == 3:
|
|
data = data[0]
|
|
data = data / 255.0 # scale to 0-1 range for gdal_array.SaveArray()
|
|
# TODO: check data range is 0-1
|
|
|
|
def get_color(v):
|
|
# first_color = []
|
|
# second_color = []
|
|
# first_value = 0
|
|
# second_value = 1
|
|
# for s in style:
|
|
# if s[0] <= v:
|
|
# first_value = s[0]
|
|
# first_color = s[1:]
|
|
# else:
|
|
# second_value = s[0]
|
|
# second_color = s[1:]
|
|
# break
|
|
# if second_value == 1:
|
|
# return np.array(style[-1][1:])
|
|
|
|
# first_dis = (v - first_value) / (second_value - first_value)
|
|
# second_dis = (second_value - v) / (second_value - first_value)
|
|
# first_color = np.array(first_color)
|
|
# second_color = np.array(second_color)
|
|
# color = first_color* first_dis + second_color * second_dis
|
|
# return np.floor(color)
|
|
|
|
if v==0:
|
|
return np.array(style[-1][1:])
|
|
elif v==1:
|
|
return np.array(style[-2][1:])
|
|
else:
|
|
first_color =style[-1][1:]
|
|
second_color=style[-2][1:]
|
|
|
|
first_color = np.array(first_color)
|
|
second_color=np.array(second_color)
|
|
color_inter=first_color*(1-v)+second_color*v
|
|
return color_inter
|
|
|
|
|
|
|
|
|
|
get_color = np.frompyfunc(get_color, nin=1, nout=1)
|
|
|
|
Y, X = data.shape
|
|
rgbs = get_color(data.reshape(-1,))
|
|
rgbs = np.stack(rgbs).reshape((Y, X, 3)).astype(np.uint8)
|
|
|
|
driver = gdal.GetDriverByName('GTiff')
|
|
# out_tif = os.path.join(Project().other_path, 'temp.tif')
|
|
out_ds = driver.Create(out, X, Y, 3, gdal.GDT_Byte)
|
|
ds = gdal.Open(path)
|
|
geo = ds.GetGeoTransform()
|
|
proj = ds.GetProjectionRef()
|
|
out_ds.SetGeoTransform(geo)
|
|
out_ds.SetProjection(proj)
|
|
|
|
for i in range(3):
|
|
out_ds.GetRasterBand(i+1).WriteArray(rgbs[..., i])
|
|
|
|
del out_ds
|
|
|
|
self.send_message.emit('导出成功，结果保存至:'+out ) # 发送消息到控制台中显示。 创建完成消息后，关闭窗口。) |