133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
|
# -*- coding: UTF-8 -*-
|
|||
|
|
"""
|
|||
|
|
@Project:__init__.py
|
|||
|
|
@File:pspHAAlphaDecomposition.py Cloude-Pottier分解
|
|||
|
|
@Function: Cloude-Pottier eigenvector/eigenvalue based decomposition of a 3x3 coherency matrix [T3]
|
|||
|
|
(Averaging using a sliding window)
|
|||
|
|
V1.0.1:（1）可选分解特征；（2）bin转tif格式
|
|||
|
|
@Contact:
|
|||
|
|
@Author:SHJ
|
|||
|
|
@Date:2021/9/24 9:06
|
|||
|
|
@Version:1.0.1
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import subprocess
|
|||
|
|
import logging
|
|||
|
|
|
|||
|
|
logger = logging.getLogger("mylog")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PspCloudePottierDecomposition:
|
|||
|
|
"""
|
|||
|
|
调用polsarpro4.2.0的Cloude-Pottier极化分解 h_a_alpha_decomposition_T3.exe
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def __init__(
|
|||
|
|
self,
|
|||
|
|
exeDir,
|
|||
|
|
inT3Dir,
|
|||
|
|
outDir,
|
|||
|
|
exeDecomposeName='h_a_alpha_decomposition_T3.exe'):
|
|||
|
|
"""
|
|||
|
|
:param exeDir:exe所在目录
|
|||
|
|
:param inT3Dir:T3矩阵目录
|
|||
|
|
:param outDir:输出目录
|
|||
|
|
"""
|
|||
|
|
self.__exeName = exeDecomposeName
|
|||
|
|
self.__exeDir = exeDir
|
|||
|
|
self.__inT3Dir = inT3Dir
|
|||
|
|
self.__outDir = outDir
|
|||
|
|
self.__DecompostFlag = False
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def api_h_a_alpha_decomposition_T3(
|
|||
|
|
self,
|
|||
|
|
rectX,
|
|||
|
|
rectY,
|
|||
|
|
rectWidth,
|
|||
|
|
rectHeight,
|
|||
|
|
Nwin=1):
|
|||
|
|
"""
|
|||
|
|
:param rectX:有效区域x
|
|||
|
|
:param rectY:有效区域y
|
|||
|
|
:param rectWidth:有效区域宽
|
|||
|
|
:param rectHeight:有效区域高
|
|||
|
|
:param Nwin :Size of the (Nwin, Nwin) sliding window used to compute local estimates. (int)
|
|||
|
|
"""
|
|||
|
|
if self.__DecompostFlag:
|
|||
|
|
return True
|
|||
|
|
if len(self.__exeDir) == 0:
|
|||
|
|
if not os.path.exists(self.__exeName):
|
|||
|
|
logger.error(self.__exeName + ' not exists.')
|
|||
|
|
return False
|
|||
|
|
exePath = self.__exeName
|
|||
|
|
else:
|
|||
|
|
if not os.path.exists(self.__exeDir + '\\' + self.__exeName):
|
|||
|
|
logger.error(self.__exeName + ' not exists.')
|
|||
|
|
return False
|
|||
|
|
exePath = self.__exeDir + '\\' + self.__exeName
|
|||
|
|
|
|||
|
|
if not self._checkT3Matrix(self.__inT3Dir):
|
|||
|
|
logger.error('T3 Matrix check failed.')
|
|||
|
|
return False
|
|||
|
|
if not os.path.exists(self.__outDir):
|
|||
|
|
os.makedirs(self.__outDir)
|
|||
|
|
|
|||
|
|
alpbetdelgam = 1
|
|||
|
|
Lambda = 1
|
|||
|
|
alpha = 1
|
|||
|
|
entropy = 1
|
|||
|
|
anisotropy = 1
|
|||
|
|
|
|||
|
|
CombHA = 1
|
|||
|
|
CombH1mA = 1
|
|||
|
|
Comb1mHA = 1
|
|||
|
|
Comb1mH1mA = 1
|
|||
|
|
|
|||
|
|
Off_lig = rectX
|
|||
|
|
Off_col = rectY
|
|||
|
|
Sub_Nlig = rectWidth
|
|||
|
|
Sub_Ncol = rectHeight
|
|||
|
|
|
|||
|
|
para_list = [
|
|||
|
|
exePath,
|
|||
|
|
self.__inT3Dir,
|
|||
|
|
self.__outDir,
|
|||
|
|
Nwin,
|
|||
|
|
Off_lig,
|
|||
|
|
Off_col,
|
|||
|
|
Sub_Nlig,
|
|||
|
|
Sub_Ncol,
|
|||
|
|
alpbetdelgam,
|
|||
|
|
Lambda,
|
|||
|
|
alpha,
|
|||
|
|
entropy,
|
|||
|
|
anisotropy,
|
|||
|
|
CombHA,
|
|||
|
|
CombH1mA,
|
|||
|
|
Comb1mHA,
|
|||
|
|
Comb1mH1mA]
|
|||
|
|
cmd = " ".join(str(i) for i in para_list)
|
|||
|
|
config_path = os.path.join(self.__inT3Dir, 'config.txt')
|
|||
|
|
shutil.copyfile(config_path, os.path.join(self.__outDir, 'config.txt'))
|
|||
|
|
result_tuple = subprocess.getstatusoutput(cmd)
|
|||
|
|
|
|||
|
|
if result_tuple[0] != 1 or result_tuple[1].find('error') != -1:
|
|||
|
|
raise Exception(result_tuple[1])
|
|||
|
|
self.__DecompostFlag = True
|
|||
|
|
return True
|
|||
|
|
@staticmethod
|
|||
|
|
def _checkT3Matrix(T3Dir):
|
|||
|
|
# 检测T3矩阵
|
|||
|
|
if not os.path.exists(T3Dir):
|
|||
|
|
return False
|
|||
|
|
file_name_in_out = ['T11.bin', 'T12_real.bin', 'T12_imag.bin',
|
|||
|
|
'T13_real.bin', 'T13_imag.bin', 'T22.bin',
|
|||
|
|
'T23_real.bin', 'T23_imag.bin', 'T33.bin']
|
|||
|
|
for item in file_name_in_out:
|
|||
|
|
if not os.path.exists(T3Dir + "\\" + item):
|
|||
|
|
return False
|
|||
|
|
return True
|