def lm_master(argv=None): """Main function for linkage mapper. Called by ArcMap with parameters or run from command line with parameters entered in script below. Calls functions in dedicated scripts for each of 5 processing steps. """ # Setup global variables if not cfg.lm_configured: # Causing problems with iterative scripting if argv is None: argv = sys.argv cfg.configure(cfg.TOOL_LM, argv) gp = cfg.gp try: gprint = lu.gprint # Move results from earlier versions to new directory structure lu.move_old_results() gp.OverwriteOutput = True gp.pyramid = "NONE" gp.rasterstatistics = "NONE" # Create output directories if they don't exist if gp.Exists(cfg.OUTPUTDIR): gp.RefreshCatalog(cfg.OUTPUTDIR) lu.create_dir(cfg.OUTPUTDIR) lu.create_dir(cfg.LOGDIR) lu.create_dir(cfg.MESSAGEDIR) lu.create_dir(cfg.DATAPASSDIR) # Create fresh scratch directory if not restarting in midst of step 3 # if cfg.S2EUCDISTFILE != None: # if cfg.S2EUCDISTFILE.lower() == "restart": pass # else: lu.delete_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.ARCSCRATCHDIR) if cfg.TOOL == 'Linkage Mapper': cfg.logFilePath = lu.create_log_file(cfg.MESSAGEDIR, cfg.TOOL, cfg.PARAMS) lu.print_drive_warning() installD = gp.GetInstallInfo("desktop") gprint('\nLinkage Mapper Version ' + cfg.releaseNum) try: gprint('on ArcGIS ' + installD['ProductName'] + ' ' + installD['Version'] + ' Service Pack ' + installD['SPNumber']) except: pass if cfg.CONNECTFRAGS: gwarn = gp.AddWarning lu.dashline(1) gwarn( 'Custom mode: will run steps 1-2 ONLY to cluster core polygons within ' ) gwarn('the maximum Euclidean corridor distance from one another ') gwarn('into polygons with a single cluster_ID value.') gwarn( 'Make sure you have set a Maximum Euclidean corridor distance.' ) lu.dashline(2) cfg.STEP3 = False cfg.STEP4 = False cfg.STEP5 = False if cfg.MAXEUCDIST == None: raise RuntimeError('Maximum Euclidean distance required ' 'for custom cluster mode.') # Set data frame spatial reference to coordinate system of input data # Problems arise in this script (core raster creation) and in S2 # (generate near table) if they differ. lu.set_dataframe_sr() # Check core ID field and project directory name. lu.check_cores(cfg.COREFC, cfg.COREFN) lu.check_project_dir() # Identify first step cleanup link tables from that point lu.dashline(1) if cfg.STEP1: gprint('Starting at step 1.') firststep = 1 elif cfg.STEP2: gprint('Starting at step 2.') firststep = 2 elif cfg.STEP3: gprint('Starting at step 3.') firststep = 3 linkTableFile = lu.get_prev_step_link_table(step=3) # Check exists elif cfg.STEP4: gprint('Starting at step 4.') firststep = 4 linkTableFile = lu.get_prev_step_link_table(step=4) # Check exists elif cfg.STEP5: gprint('Starting at step 5.') firststep = 5 linkTableFile = lu.get_prev_step_link_table(step=5) # Check exists lu.clean_up_link_tables(firststep) # Make a local grid copy of resistance raster for cwd runs- # will run faster than gdb. # Don't know if raster is in a gdb if entered from TOC lu.delete_data(cfg.RESRAST) gprint('\nMaking temporary copy of resistance raster for this run.') gp.OutputCoordinateSystem = gp.describe(cfg.COREFC).SpatialReference gp.Extent = gp.Describe(cfg.RESRAST_IN).Extent gp.SnapRaster = cfg.RESRAST_IN gp.cellSize = gp.Describe(cfg.RESRAST_IN).MeanCellHeight # import pdb; pdb.set_trace() gp.CopyRaster_management(cfg.RESRAST_IN, cfg.RESRAST) if (cfg.STEP1) or (cfg.STEP3): # Make core raster file gprint('\nMaking temporary raster of core file for this run.') lu.delete_data(cfg.CORERAS) gp.FeatureToRaster_conversion( cfg.COREFC, cfg.COREFN, cfg.CORERAS, gp.Describe(cfg.RESRAST).MeanCellHeight) # # gp.RasterToPolygon_conversion(cfg.CORERAS, cfg.COREFC, # "NO_SIMPLIFY") def delete_final_gdb(finalgdb): """Deletes final geodatabase""" if gp.Exists(finalgdb) and cfg.STEP5: try: lu.clean_out_workspace(finalgdb) except: lu.dashline(1) msg = ('ERROR: Could not remove contents of geodatabase ' + finalgdb + '. \nIs it open in ArcMap? You may ' 'need to re-start ArcMap to release the file lock.') lu.raise_error(msg) lu.delete_dir(finalgdb) # Delete final output geodatabase delete_final_gdb(cfg.OUTPUTGDB_OLD) delete_final_gdb(cfg.OUTPUTGDB) delete_final_gdb(cfg.EXTRAGDB) delete_final_gdb(cfg.LINKMAPGDB) # Run linkage mapper processing steps if cfg.STEP1: s1.STEP1_get_adjacencies() if cfg.STEP2: s2.STEP2_build_network() if cfg.STEP3: s3.STEP3_calc_cwds() if cfg.STEP4: s4.STEP4_refine_network() if cfg.STEP5: s5.STEP5_calc_lccs() lu.dashline() gprint('Results from this run can be found in your output ' 'directory:') gprint(cfg.OUTPUTDIR) # Clean up lu.delete_dir(cfg.SCRATCHDIR) # lu.delete_data(cfg.FCORES) gp.addmessage('\nDone with linkage mapping.\n') # Return GEOPROCESSING specific errors except arcgisscripting.ExecuteError: lu.exit_with_geoproc_error(_SCRIPT_NAME) # Return any PYTHON or system specific errors except: lu.exit_with_python_error(_SCRIPT_NAME) finally: lu.dashline() gprint('A record of run settings and messages can be found in your ' 'log directory:') gprint(cfg.MESSAGEDIR) lu.dashline(2) lu.close_log_file()
def circuitscape_master(argv=None): """ """ gprint = lu.gprint gwarn = arcpy.AddWarning if argv is None: argv = sys.argv cfg.configure(cfg.TOOL_CS, argv) gp = cfg.gp try: lu.create_dir(cfg.LOGDIR) lu.create_dir(cfg.MESSAGEDIR) cfg.logFilePath=lu.create_log_file(cfg.MESSAGEDIR, cfg.TOOL, cfg.PARAMS) CSPATH = lu.get_cs_path() if CSPATH == None: msg = ('Cannot find an installation of Circuitscape 3.5.5' '\nor greater in your Program Files directory.') arcpy.AddError(msg) lu.write_log(msg) exit(1) try: csDir, fn = path.split(CSPATH) if 'flush' not in open(path.join(csDir,'cs_compute.py')).read(): gwarn('\n---------------------------------------------') gwarn('Your version of Circuitscape is out of date. ') gwarn('---------------------------------------------\n') gwarn('Please get the latest from www.circuitscape.org.') gwarn('The new version interacts more smoothly with ArcMap.') gprint('Proceeding...\n') except: pass lu.print_drive_warning() # Check core ID field. lu.check_cores(cfg.COREFC, cfg.COREFN) gp.OutputCoordinateSystem = gp.describe(cfg.COREFC).SpatialReference # Set data frame spatial reference to coordinate system of input data lu.set_dataframe_sr() gp.pyramid = "NONE" gp.rasterstatistics = "NONE" # Move adj and cwd results from earlier versions to datapass directory lu.move_old_results() if cfg.CWDCUTOFF > 0: lu.delete_dir(cfg.SCRATCHDIR) # restart code- in progress if cfg.CWDCUTOFF < 0: cfg.CWDCUTOFF = cfg.CWDCUTOFF * -1 if not cfg.DOPINCH and not cfg.DOCENTRALITY: msg = ('ERROR: Please choose at least one option: pinch point or\n' 'network centrality analysis.') lu.raise_error(msg) lu.create_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.ARCSCRATCHDIR) if cfg.DO_ALLPAIRS: # Fixme: move raster path to config S5CORRIDORRAS = path.join(cfg.OUTPUTGDB,cfg.PREFIX + "_corridors") if not gp.Exists(S5CORRIDORRAS): S5CORRIDORRAS = path.join(cfg.OUTPUTGDB, cfg.PREFIX + "_lcc_mosaic_int") if not gp.Exists(S5CORRIDORRAS): msg = ('ERROR: Corridor raster created in step 5 is required' '\nfor all-pair analyses, but was not found.') lu.raise_error(msg) if cfg.DOPINCH: if cfg.CWDCUTOFF == '#' or cfg.CWDCUTOFF == 0: msg = ('ERROR: CWD cutoff distance is required for pinch point' ' analyses.') lu.raise_error(msg) # Make a local grid copy of resistance raster- # will run faster than gdb. lu.delete_data(cfg.RESRAST) if not gp.Exists(cfg.RESRAST_IN): msg = ('ERROR: Resistance raster is required for pinch point' ' analyses, but was not found.') lu.raise_error(msg) desc = arcpy.Describe(cfg.RESRAST_IN) if hasattr(desc, "catalogPath"): cfg.RESRAST_IN = arcpy.Describe(cfg.RESRAST_IN).catalogPath arcpy.env.extent = cfg.RESRAST_IN arcpy.env.snapRaster = cfg.RESRAST_IN gprint('\nMaking local copy of resistance raster.') gp.CopyRaster_management(cfg.RESRAST_IN, cfg.RESRAST) if cfg.DOCENTRALITY: gprint("Creating output folder: " + cfg.CENTRALITYBASEDIR) if path.exists(cfg.CENTRALITYBASEDIR): shutil.rmtree(cfg.CENTRALITYBASEDIR) lu.create_dir(cfg.CENTRALITYBASEDIR) gp.CreateFolder_management(cfg.CENTRALITYBASEDIR, cfg.CIRCUITOUTPUTDIR_NM) gp.CreateFolder_management(cfg.CENTRALITYBASEDIR, cfg.CIRCUITCONFIGDIR_NM) lu.clean_out_workspace(cfg.CORECENTRALITYGDB) s7.STEP7_calc_centrality() if not cfg.SAVECENTRALITYDIR: lu.delete_dir(cfg.CENTRALITYBASEDIR) if cfg.DOPINCH: if cfg.CWDCUTOFF > 0: # Negative values mean we're restarting gprint("Creating output folder: " + cfg.CIRCUITBASEDIR) lu.delete_dir(cfg.CIRCUITBASEDIR) lu.create_dir(cfg.CIRCUITBASEDIR) gp.CreateFolder_management(cfg.CIRCUITBASEDIR, cfg.CIRCUITOUTPUTDIR_NM) gp.CreateFolder_management(cfg.CIRCUITBASEDIR, cfg.CIRCUITCONFIGDIR_NM) s8.STEP8_calc_pinchpoints() if not cfg.SAVE_TEMP_CIRCUIT_FILES: lu.delete_dir(cfg.SCRATCHDIR) if not cfg.SAVECIRCUITDIR: lu.delete_dir(cfg.CIRCUITBASEDIR) gprint('\nDONE!\n') # Return GEOPROCESSING specific errors except arcgisscripting.ExecuteError: lu.exit_with_geoproc_error(_SCRIPT_NAME) # Return any PYTHON or system specific errors except: lu.exit_with_python_error(_SCRIPT_NAME)
def circuitscape_master(argv=None): """ """ gprint = lu.gprint if argv is None: argv = sys.argv argv.append(get_cs_path()) # Add Circuitscape path cfg.configure(cfg.TOOL_CS, argv) try: lu.create_dir(cfg.LOGDIR) lu.create_dir(cfg.MESSAGEDIR) cfg.logFilePath = lu.create_log_file(cfg.PARAM_NAMES, argv) if cfg.DOPINCH : lu.log_metadata(cfg.COREFC, [cfg.RESRAST_IN]) else: lu.log_metadata(cfg.COREFC) if cfg.CSPATH is None: lu.raise_error("Cannot find an installation of Circuitscape" "\nin your Program Files directory.") lu.print_drive_warning() # Check core ID field. lu.check_cores(cfg.COREFC, cfg.COREFN) arcpy.env.outputCoordinateSystem = arcpy.Describe(cfg.COREFC).SpatialReference # Set data frame spatial reference to coordinate system of input data lu.set_dataframe_sr() arcpy.env.pyramid = "NONE" arcpy.env.rasterStatistics = "NONE" # Move adj and cwd results from earlier versions to datapass directory lu.move_old_results() if cfg.CWDCUTOFF > 0: lu.delete_dir(cfg.SCRATCHDIR) # restart code- in progress if cfg.CWDCUTOFF < 0: cfg.CWDCUTOFF = cfg.CWDCUTOFF * -1 if not cfg.DOPINCH and not cfg.DOCENTRALITY: msg = ('ERROR: Please choose at least one option: pinch point or\n' 'network centrality analysis.') lu.raise_error(msg) lu.create_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.ARCSCRATCHDIR) if cfg.DO_ALLPAIRS: # Fixme: move raster path to config S5CORRIDORRAS = path.join(cfg.OUTPUTGDB,cfg.PREFIX + "_corridors") if not arcpy.Exists(S5CORRIDORRAS): S5CORRIDORRAS = path.join(cfg.OUTPUTGDB, cfg.PREFIX + "_lcc_mosaic_int") if not arcpy.Exists(S5CORRIDORRAS): msg = ('ERROR: Corridor raster created in step 5 is required' '\nfor all-pair analyses, but was not found.') lu.raise_error(msg) if cfg.DOPINCH: if cfg.CWDCUTOFF == '#' or cfg.CWDCUTOFF == 0: msg = ('ERROR: CWD cutoff distance is required for pinch point' ' analyses.') lu.raise_error(msg) # Make a local grid copy of resistance raster- # will run faster than gdb. lu.delete_data(cfg.RESRAST) if not arcpy.Exists(cfg.RESRAST_IN): msg = ('ERROR: Resistance raster is required for pinch point' ' analyses, but was not found.') lu.raise_error(msg) arcpy.env.extent = cfg.RESRAST_IN desc = arcpy.Describe(cfg.RESRAST_IN) if hasattr(desc, "catalogPath"): cfg.RESRAST_IN = arcpy.Describe(cfg.RESRAST_IN).catalogPath gprint('\nMaking local copy of resistance raster.') try: arcpy.CopyRaster_management(cfg.RESRAST_IN, cfg.RESRAST) except Exception: msg = ('ERROR: Could not make a copy of your resistance raster. ' + 'Try re-starting ArcMap to release the file lock.') lu.raise_error(msg) arcpy.env.snapRaster = cfg.RESRAST if cfg.DOCENTRALITY: gprint("Creating output folder: " + cfg.CENTRALITYBASEDIR) if path.exists(cfg.CENTRALITYBASEDIR): shutil.rmtree(cfg.CENTRALITYBASEDIR) lu.create_dir(cfg.CENTRALITYBASEDIR) arcpy.CreateFolder_management(cfg.CENTRALITYBASEDIR, cfg.CIRCUITOUTPUTDIR_NM) arcpy.CreateFolder_management(cfg.CENTRALITYBASEDIR, cfg.CIRCUITCONFIGDIR_NM) lu.clean_out_workspace(cfg.CORECENTRALITYGDB) s7.STEP7_calc_centrality() if not cfg.SAVECENTRALITYDIR: lu.delete_dir(cfg.CENTRALITYBASEDIR) if cfg.DOPINCH: if cfg.CWDCUTOFF > 0: # Negative values mean we're restarting gprint("Creating output folder: " + cfg.CIRCUITBASEDIR) lu.delete_dir(cfg.CIRCUITBASEDIR) lu.create_dir(cfg.CIRCUITBASEDIR) arcpy.CreateFolder_management(cfg.CIRCUITBASEDIR, cfg.CIRCUITOUTPUTDIR_NM) arcpy.CreateFolder_management(cfg.CIRCUITBASEDIR, cfg.CIRCUITCONFIGDIR_NM) s8.STEP8_calc_pinchpoints() if not cfg.SAVE_TEMP_CIRCUIT_FILES: lu.delete_dir(cfg.SCRATCHDIR) if not cfg.SAVECIRCUITDIR: lu.delete_dir(cfg.CIRCUITBASEDIR) gprint('\nDONE!\n') # Return GEOPROCESSING specific errors except arcpy.ExecuteError: lu.exit_with_geoproc_error(_SCRIPT_NAME) # Return any PYTHON or system specific errors except Exception: lu.exit_with_python_error(_SCRIPT_NAME)
def lm_master(argv=None): """Main function for linkage mapper. Called by ArcMap with parameters or run from command line with parameters entered in script below. Calls functions in dedicated scripts for each of 5 processing steps. """ # Setup global variables if not cfg.lm_configured: # Causing problems with iterative scripting if argv is None: argv = sys.argv cfg.configure(cfg.TOOL_LM, argv) gp = cfg.gp try: gprint = lu.gprint # Move results from earlier versions to new directory structure lu.move_old_results() gp.OverwriteOutput = True gp.pyramid = "NONE" gp.rasterstatistics = "NONE" # Create output directories if they don't exist if gp.Exists(cfg.OUTPUTDIR): gp.RefreshCatalog(cfg.OUTPUTDIR) lu.create_dir(cfg.OUTPUTDIR) lu.create_dir(cfg.LOGDIR) lu.create_dir(cfg.MESSAGEDIR) lu.create_dir(cfg.DATAPASSDIR) # Create fresh scratch directory if not restarting in midst of step 3 # if cfg.S2EUCDISTFILE != None: # if cfg.S2EUCDISTFILE.lower() == "restart": pass # else: lu.delete_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.ARCSCRATCHDIR) if cfg.TOOL == 'Linkage Mapper': cfg.logFilePath = lu.create_log_file(cfg.MESSAGEDIR, cfg.TOOL, cfg.PARAMS) lu.print_drive_warning() installD = gp.GetInstallInfo("desktop") gprint('\nLinkage Mapper Version ' + cfg.releaseNum) try: gprint('on ArcGIS ' + installD['ProductName'] + ' ' + installD['Version'] + ' Service Pack ' + installD['SPNumber']) except: pass if cfg.CONNECTFRAGS: gwarn = gp.AddWarning lu.dashline(1) gwarn('Custom mode: will run steps 1-2 ONLY to cluster core polygons within ') gwarn('the maximum Euclidean corridor distance from one another ') gwarn('into polygons with a single cluster_ID value.') gwarn('Make sure you have set a Maximum Euclidean corridor distance.') lu.dashline(2) cfg.STEP3 = False cfg.STEP4 = False cfg.STEP5 = False if cfg.MAXEUCDIST == None: raise RuntimeError('Maximum Euclidean distance required ' 'for custom cluster mode.') # Set data frame spatial reference to coordinate system of input data # Problems arise in this script (core raster creation) and in S2 # (generate near table) if they differ. lu.set_dataframe_sr() # Check core ID field and project directory name. lu.check_cores(cfg.COREFC, cfg.COREFN) lu.check_project_dir() # Identify first step cleanup link tables from that point lu.dashline(1) if cfg.STEP1: gprint('Starting at step 1.') firststep = 1 elif cfg.STEP2: gprint('Starting at step 2.') firststep = 2 elif cfg.STEP3: gprint('Starting at step 3.') firststep = 3 linkTableFile = lu.get_prev_step_link_table(step=3) # Check exists elif cfg.STEP4: gprint('Starting at step 4.') firststep = 4 linkTableFile = lu.get_prev_step_link_table(step=4) # Check exists elif cfg.STEP5: gprint('Starting at step 5.') firststep = 5 linkTableFile = lu.get_prev_step_link_table(step=5) # Check exists lu.clean_up_link_tables(firststep) # Make a local grid copy of resistance raster for cwd runs- # will run faster than gdb. # Don't know if raster is in a gdb if entered from TOC lu.delete_data(cfg.RESRAST) gprint('\nMaking temporary copy of resistance raster for this run.') gp.OutputCoordinateSystem = gp.describe(cfg.COREFC).SpatialReference gp.Extent = gp.Describe(cfg.RESRAST_IN).Extent gp.SnapRaster = cfg.RESRAST_IN gp.cellSize = gp.Describe(cfg.RESRAST_IN).MeanCellHeight # import pdb; pdb.set_trace() gp.CopyRaster_management(cfg.RESRAST_IN, cfg.RESRAST) if (cfg.STEP1) or (cfg.STEP3): # Make core raster file gprint('\nMaking temporary raster of core file for this run.') lu.delete_data(cfg.CORERAS) gp.FeatureToRaster_conversion(cfg.COREFC, cfg.COREFN, cfg.CORERAS, gp.Describe(cfg.RESRAST).MeanCellHeight) # # gp.RasterToPolygon_conversion(cfg.CORERAS, cfg.COREFC, # "NO_SIMPLIFY") def delete_final_gdb(finalgdb): """Deletes final geodatabase""" if gp.Exists(finalgdb) and cfg.STEP5: try: lu.clean_out_workspace(finalgdb) except: lu.dashline(1) msg = ('ERROR: Could not remove contents of geodatabase ' + finalgdb + '. \nIs it open in ArcMap? You may ' 'need to re-start ArcMap to release the file lock.') lu.raise_error(msg) lu.delete_dir(finalgdb) # Delete final output geodatabase delete_final_gdb(cfg.OUTPUTGDB_OLD) delete_final_gdb(cfg.OUTPUTGDB) delete_final_gdb(cfg.EXTRAGDB) delete_final_gdb(cfg.LINKMAPGDB) # Run linkage mapper processing steps if cfg.STEP1: s1.STEP1_get_adjacencies() if cfg.STEP2: s2.STEP2_build_network() if cfg.STEP3: s3.STEP3_calc_cwds() if cfg.STEP4: s4.STEP4_refine_network() if cfg.STEP5: s5.STEP5_calc_lccs() lu.dashline() gprint('Results from this run can be found in your output ' 'directory:') gprint(cfg.OUTPUTDIR) # Clean up lu.delete_dir(cfg.SCRATCHDIR) # lu.delete_data(cfg.FCORES) gp.addmessage('\nDone with linkage mapping.\n') # Return GEOPROCESSING specific errors except arcgisscripting.ExecuteError: lu.exit_with_geoproc_error(_SCRIPT_NAME) # Return any PYTHON or system specific errors except: lu.exit_with_python_error(_SCRIPT_NAME) finally: lu.dashline() gprint('A record of run settings and messages can be found in your ' 'log directory:') gprint(cfg.MESSAGEDIR) lu.dashline(2) lu.close_log_file()
def lm_master(argv=None): """Main function for linkage mapper. Called by ArcMap with parameters or run from command line with parameters entered in script below. Calls functions in dedicated scripts for each of 5 processing steps. """ # Setup global variables if cfg.TOOL != cfg.TOOL_CC: if argv is None: argv = sys.argv cfg.configure(cfg.TOOL_LM, argv) gprint = lu.gprint try: # Move results from earlier versions to new directory structure lu.move_old_results() arcpy.env.pyramid = "NONE" arcpy.env.rasterStatistics = "NONE" # Create output directories if they don't exist lu.create_dir(cfg.OUTPUTDIR) lu.create_dir(cfg.LOGDIR) lu.create_dir(cfg.MESSAGEDIR) lu.create_dir(cfg.DATAPASSDIR) if cfg.TOOL != cfg.TOOL_CC: lu.delete_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.SCRATCHDIR) lu.create_dir(cfg.ARCSCRATCHDIR) if cfg.TOOL == cfg.TOOL_LM: cfg.logFilePath = lu.create_log_file(cfg.PARAM_NAMES, argv) lu.write_custom_to_log(cfg.LMCUSTSETTINGS) lu.log_metadata(cfg.COREFC, [cfg.RESRAST_IN]) lu.print_drive_warning() if cfg.CONNECTFRAGS: lu.dashline(1) lu.warn('Custom mode: will run steps 1-2 ONLY to cluster core polygons within ') lu.warn('the maximum Euclidean corridor distance from one another ') lu.warn('into polygons with a single cluster_ID value.') lu.warn('Make sure you have set a Maximum Euclidean corridor distance.') lu.dashline(2) cfg.STEP3 = False cfg.STEP4 = False cfg.STEP5 = False if cfg.MAXEUCDIST == None: raise RuntimeError('Maximum Euclidean distance required ' 'for custom cluster mode.') # Set data frame spatial reference to coordinate system of input data # Problems arise in this script (core raster creation) and in S2 # (generate near table) if they differ. lu.set_dataframe_sr() # Check core ID field and project directory name. lu.check_cores(cfg.COREFC, cfg.COREFN) lu.check_project_dir() # Identify first step cleanup link tables from that point lu.dashline(1) if cfg.STEP1: gprint('Starting at step 1.') firststep = 1 elif cfg.STEP2: gprint('Starting at step 2.') firststep = 2 elif cfg.STEP3: gprint('Starting at step 3.') firststep = 3 linkTableFile = lu.get_prev_step_link_table(step=3) # Check exists elif cfg.STEP4: gprint('Starting at step 4.') firststep = 4 linkTableFile = lu.get_prev_step_link_table(step=4) # Check exists elif cfg.STEP5: gprint('Starting at step 5.') firststep = 5 linkTableFile = lu.get_prev_step_link_table(step=5) # Check exists lu.clean_up_link_tables(firststep) # Make a local grid copy of resistance raster for cwd runs- # will run faster than gdb. # Don't know if raster is in a gdb if entered from TOC lu.delete_data(cfg.RESRAST) gprint('\nMaking temporary copy of resistance raster for this run.') arcpy.env.outputCoordinateSystem = arcpy.Describe(cfg.COREFC).SpatialReference arcpy.env.extent = arcpy.Describe(cfg.RESRAST_IN).Extent arcpy.env.snapRaster = cfg.RESRAST_IN arcpy.env.cellSize = arcpy.Describe(cfg.RESRAST_IN).MeanCellHeight try: arcpy.CopyRaster_management(cfg.RESRAST_IN, cfg.RESRAST) except Exception: msg = ('ERROR: Could not make a copy of your resistance raster. ' + 'Try re-starting ArcMap to release the file lock.') lu.raise_error(msg) if (cfg.STEP1) or (cfg.STEP3): # Make core raster file gprint('\nMaking temporary raster of core file for this run.') lu.delete_data(cfg.CORERAS) arcpy.FeatureToRaster_conversion(cfg.COREFC, cfg.COREFN, cfg.CORERAS, arcpy.Describe(cfg.RESRAST).MeanCellHeight) def delete_final_gdb(finalgdb): """Deletes final geodatabase""" if arcpy.Exists(finalgdb) and cfg.STEP5: try: lu.clean_out_workspace(finalgdb) except Exception: lu.dashline(1) msg = ('ERROR: Could not remove contents of geodatabase ' + finalgdb + '. \nIs it open in ArcMap? You may ' 'need to re-start ArcMap to release the file lock.') lu.raise_error(msg) lu.delete_dir(finalgdb) # Delete final output geodatabase delete_final_gdb(cfg.OUTPUTGDB_OLD) delete_final_gdb(cfg.OUTPUTGDB) delete_final_gdb(cfg.EXTRAGDB) delete_final_gdb(cfg.LINKMAPGDB) # Run linkage mapper processing steps if cfg.STEP1: s1.STEP1_get_adjacencies() if cfg.STEP2: s2.STEP2_build_network() if cfg.STEP3: s3.STEP3_calc_cwds() if cfg.STEP4: s4.STEP4_refine_network() if cfg.STEP5: s5.STEP5_calc_lccs() lu.dashline() gprint('Results from this run can be found in your output ' 'directory:') gprint(cfg.OUTPUTDIR) # Clean up lu.delete_dir(cfg.SCRATCHDIR) arcpy.AddMessage('\nDone with linkage mapping.\n') # Return GEOPROCESSING specific errors except arcpy.ExecuteError: lu.exit_with_geoproc_error(_SCRIPT_NAME) # Return any PYTHON or system specific errors except Exception: lu.exit_with_python_error(_SCRIPT_NAME) finally: lu.dashline() gprint('A record of run settings and messages can be found in your ' 'log directory:') gprint(cfg.MESSAGEDIR) lu.dashline(2) lu.close_log_file()