forked from dtarb/TauDEM
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathArcGISStabilityIndex.py
77 lines (62 loc) · 2.98 KB
/
ArcGISStabilityIndex.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# Created by: Pabitra Dash
import os
import subprocess
import arcpy
# get the input parameters
slp_raster_file = arcpy.GetParameterAsText(0)
desc = arcpy.Describe(slp_raster_file)
slp_raster_file = str(desc.catalogPath)
sca_raster_file = arcpy.GetParameterAsText(1)
desc = arcpy.Describe(sca_raster_file)
sca_raster_file = str(desc.catalogPath)
cal_raster_file = arcpy.GetParameterAsText(2)
desc = arcpy.Describe(cal_raster_file)
cal_raster_file = str(desc.catalogPath)
capl_text_file = arcpy.GetParameterAsText(3)
min_terr_recharge = arcpy.GetParameterAsText(4)
max_terr_recharge = arcpy.GetParameterAsText(5)
si_raster_file = arcpy.GetParameterAsText(6)
sat_raster_file = arcpy.GetParameterAsText(7)
temp_output_files_directory = arcpy.GetParameterAsText(8)
is_delete_intermediate_output_files = arcpy.GetParameterAsText(9)
# create the cis_inputs.txt file from the provided above parameters
this_script_dir = os.path.dirname(os.path.realpath(__file__))
si_control_file = os.path.join(temp_output_files_directory, 'Si_Control.txt')
si_control_file = r'' + si_control_file
with open(si_control_file, 'w') as file_obj:
file_obj.write('# input parameters for combined stability index computation with road impact\n')
file_obj.write('# input files\n')
file_obj.write('slp=' + slp_raster_file + '\n')
file_obj.write('sca=' + sca_raster_file + '\n')
file_obj.write('cal=' + cal_raster_file + '\n')
file_obj.write('calpar=' + capl_text_file + '\n')
file_obj.write('# output files\n')
file_obj.write('si=' + si_raster_file + '\n')
file_obj.write('sat=' + sat_raster_file + '\n')
file_obj.write('# Additional parameters' + '\n')
file_obj.write('minimumterrainrecharge=' + min_terr_recharge + '\n')
file_obj.write('maximumterrainrecharge=' + max_terr_recharge + '\n')
file_obj.write('# temporary output file directory\n')
file_obj.write('temporary_output_files_directory=' + temp_output_files_directory + '\n')
if str(is_delete_intermediate_output_files) == 'true':
file_obj.write('is_delete_intermediate_output_files=True\n')
else:
file_obj.write('is_delete_intermediate_output_files=False\n')
# put quotes around file paths in case they have spaces
si_control_file = '"' + si_control_file + '"'
py_script_to_execute = os.path.join(this_script_dir, 'StabilityIndex.py')
py_script_to_execute = '"' + py_script_to_execute + '"'
cmd = py_script_to_execute + \
' --params ' + si_control_file
# show executing command
arcpy.AddMessage('\nEXECUTING COMMAND:\n' + cmd)
# Capture the contents of shell command and print it to the arcgis dialog box
process = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE)
arcpy.AddMessage('\nProcess started:\n')
start_message = "Please wait a few seconds. Computation is in progress ..."
arcpy.AddMessage('\n' + start_message + '\n')
for line in process.stdout.readlines():
if isinstance(line, bytes): # true in Python 3
line = line.decode()
if start_message not in line:
arcpy.AddMessage(line)