def main(): ''' main function ''' try: # Set the whitebox-tools executable directory # (change this to point to where you have the whitebox-tools.exe file) wb_dir = os.path.dirname( os.path.abspath(__file__)) + "/target/release/" wbt = WhiteboxTools() wbt.set_whitebox_dir(wb_dir) # Prints the whitebox-tools help...a listing of available commands print(wbt.help()) # Prints the whitebox-tools license print(wbt.license()) # Prints the whitebox-tools version print("Version information: {}".format(wbt.version())) # List all available tools in whitebox-tools print(wbt.list_tools()) # Lists tools with 'lidar' or 'LAS' in tool name or description. print(wbt.list_tools(['lidar', 'LAS'])) # print(wbt.tool_help("dev_from_mean_elev")) print(wbt.tool_help("elev_percentile")) # Sets verbose mode (True or False). Most tools will suppress output (e.g. updating # progress) when verbose mode is False. The default is True # wbt.set_verbose_mode(False) # needed to specify complete file names (with paths) to tools that you run. wbt.set_working_dir( os.path.dirname(os.path.abspath(__file__)) + "/testdata/") name = "elev_percentile" args = [ "--input=\"DEM.dep\"", "--output=\"DEV_101.dep\"", "--filter=101" ] # Run the tool and check the return value if wbt.run_tool(name, args, callback) != 0: print("ERROR running {}".format(name)) except: print("Unexpected error:", sys.exc_info()[0]) raise
def main(): """ main function """ try: # name = input("Tool name: ") or 'DevFromMeanElev' # input_file = input("Input file: ") or 'DEM no OTOs.dep' # output_file = input("Output file: ") or 'tmp.dep' # filter_size = int(input("Filter size: ") or "101") # setup WhiteboxTools wbt = WhiteboxTools() wb_dir = os.path.dirname( os.path.abspath(__file__)) + "/target/release/" wbt.set_whitebox_dir(wb_dir) # wkdir = "/Users/johnlindsay/Documents/data/GarveyGlenWatershed/" # wkdir = "/Users/johnlindsay/Documents/data/DanData/" # wkdir = "/Users/johnlindsay/Desktop/WhiteboxGAT-mac/resources/samples/Vermont DEM/" wkdir = "Users/johnlindsay/Documents/Data/NewBrunswick/" wbt.set_working_dir(wkdir) # Tool name and arguments # name = "DevFromMeanElev" # name = "ElevsPercentile" # name = "RelativeTopographicPosition" # filter_size = 101 # args = ["--input=\"DEM no OTOs.dep\"", # "--output=\"tmp33.dep\"", # "--filterx={0}".format(filter_size), # "--filtery={0}".format(filter_size)] # args = ["--input=\"{0}\"".format(input_file), # "--output=\"{0}\"".format(output_file), # "--filterx={0}".format(filter_size), # "--filtery={0}".format(filter_size)] # args = ["--input=\"DEM.dep\"", # "--output=\"tmp33.dep\"", # "--filterx=101", # "--filtery=101"] # name = "RemoveOffTerrainObjects" # args = ["--input=\"DEM no OTOs.dep\"", # "--output=\"tmp34.dep\"", # "--filter=10", # "--slope=10.0"] # name = "LidarInfo" # args = ["--input=\"points.las\"", # "--vlr", # "--geokeys"] # name = "LidarHillshade" # args = ["--input=\"RGB_5_529_150502_1754__0_270112_2848.las\"", # "--output=\"hillshade2.las\"", # "--azimuth=315.0", # "--altitude=30.0", # "--radius=2.5"] # name = "LidarTophatTransform" # args = ["--input=\"RGB_5_529_150502_1754__0_270112_2848.las\"", # "--output=\"filtered12m_RGB_5_529_150502_1754__0_270112_2848.las\"", # "--radius=12.0"] def custom_callback(value): ''' A custom callback for dealing with tool output. ''' if not hasattr(custom_callback, 'prev_line_progress'): custom_callback.prev_line_progress = False if not hasattr(custom_callback, 'prev_line_len'): custom_callback.prev_line_len = -1 if "%" in value: # wbt.cancel_op = True if custom_callback.prev_line_progress: if len(value) < custom_callback.prev_line_len: print(' ', end="\r") print('{0}'.format(value), end="\r") else: custom_callback.prev_line_progress = True print(value) else: if custom_callback.prev_line_progress: print('\n{0}'.format(value)) custom_callback.prev_line_progress = False else: print(value) custom_callback.prev_line_len = len(value) # file_names = ["RGB_3_529_150514_1353__0_294688_2784.las", # "RGB_3_529_150514_1353__0_276256_2784.las"] # filter_size = 18.0 # for file_name in file_names: # name = "LidarGroundPointFilter" # args = ["--input=\"{0}\"".format(file_name), # "--output=\"filtered{0}m_{1}\"".format( # str(filter_size).replace(".", "_"), file_name), # "--radius={0}".format(filter_size), # "--otoheight=0.75"] # # Run the tool and check the return value # ret = wbt.run_tool(name, args, custom_callback) # if ret == 1: # print("ERROR running {}".format(name)) # elif ret == 2: # print("Operation cancelled while running {}".format(name)) # name = "FillMissingData" # args = ["--input=\"mosaic_FR.dep\"", # "--output=\"tmp1.dep\"", # "--filter=41"] # name = "RemoveOffTerrainObjects" # args = ["--input=\"tmp1_clipped.dep\"", # "--output=\"mosaic_FR no OTOs4.dep\"", # "--filter=251", # "--slope=10.0"] # name = "Hillshade" # args = ["--input=\"tmp10.dep\"", # "--output=\"tmp11.dep\"", # "--azimuth=315.0", # "--altitude=20.0"] # name = "D8FlowAccumulation" # args = ["--input=\"DEM final.dep\"", # "--output=\"tmp13.dep\"", # "--out_type=\"cells\"", # "--log"] # name = "BufferRaster" # args = ["--input=\"tmp11.dep\"", # "--output=\"tmp12.dep\"", # "--size=450", # "--gridcells"] # name = "Quantiles" # args = ["--input=\"DEM final.dep\"", # "--output=\"tmp10.dep\"", # "--num_quantiles=15"] # name = "Clump" # args = ["--input=\"tmp10.dep\"", # "--output=\"tmp15.dep\""] # name = "Clump" # args = ["--input=\"tmp12.dep\"", # "--output=\"tmp16.dep\"", # "--diag", # "--zero_back"] # name = "DevFromMeanElev" # args = ["--input=\"Rel_Low_Denoised.dep\"", # "--output=\"tmp6.dep\"", # "--filter=411"] name = "MaxElevationDeviation" args = [ "--dem='alosDEM1.dep'", "--out_mag='mag1.dep'", "--out_scale='scale1.dep'", "--min_scale=15", "--max_scale=60", "--step=5" ] # Run the tool and check the return value ret = wbt.run_tool(name, args, custom_callback) if ret == 1: print("ERROR running {}".format(name)) elif ret == 2: print("Operation cancelled while running {}".format(name)) # app = SampleApp() # app.mainloop() except: print("Unexpected error:", sys.exc_info()[0]) raise