Archived Forums

"""RasterGenerator.py - generates rasters in an output directory, for the purpose of reproducing ArcGIS crashes."""

import os
import sys
import win32com.client

# Helper functions.

def log_returned_geoprocessor_messages(gp):
    i = 0
    while i < gp.MessageCount:
        sev = gp.GetSeverity(i)
        msg = gp.GetMessage(i)
        print msg
        if sev == 0:
            gp.AddMessage(msg)
        elif sev == 1:
            gp.AddWarning(msg)
        elif sev == 2:
            gp.AddError(msg)
        i = i + 1

def log_message(gp, msg):
    print msg
    gp.AddMessage(msg)

def log_warning(gp, msg):
    print msg
    gp.AddWarning(msg)

def log_error_and_exit(gp, msg):
    gp.AddError(msg)
    sys.exit(msg)

# Create the geoprocessor object.

gp = win32com.client.Dispatch("esriGeoprocessing.GPDispatch.1")

# Check out the spatial analyst extension.

try:
    gp.CheckOutExtension("spatial")
except Exception, e:
    log_error_and_exit(gp, "Failed to obtain a license for the spatial extension. gp.CheckOutExtension reported: " + str(e))

# Validate and parse the input parameters.

if len(sys.argv) < 4:
    log_error_and_exit(gp, "You must specify the input parameters: output directory, list of raster data types (FLOAT or INTEGER), and number of rasters to generate.")

output_dir = sys.argv[1]
raster_data_types = sys.argv[2]
num_rasters = sys.argv[3]

log_message(gp, "Output Directory = " + output_dir)
log_message(gp, "Raster Data Types = " + raster_data_types)
log_message(gp, "Number of Rasters = " + num_rasters)

if not os.path.isdir(output_dir) or not gp.Exists(output_dir):
    log_error_and_exit(gp, "The output directory \"" + output_dir + "\" does not appear to exist.")

raster_data_types_list = []
for data_type in raster_data_types.split(";"):
    if data_type.upper() != "FLOAT" and data_type.upper() != "INTEGER":
        log_error_and_exit(gp, "\"" + data_type + "\" is an invalid raster data type. The valid data types are \"FLOAT\" and \"INTEGER\".")
    raster_data_types_list.append(data_type.upper())

num_rasters_list = []
for num_str in num_rasters.split(";"):
    try:
        num = int(num_str)
    except:
        log_error_and_exit(gp, "\"" + num_str + "\" is an invalid number of rasters. It must be a positive integer.")
    if num < 1:
        log_error_and_exit(gp, "\"" + num_str + "\" is an invalid number of rasters. It must be a positive integer.")
    num_rasters_list.append(num)

if len(raster_data_types_list) != len(num_rasters_list):
    log_error_and_exit(gp, "The list of raster data types must be the same length as the list of numbers of rasters.")

if len(raster_data_types_list) == 0:
    log_error_and_exit(gp, "Please specify at least one item in the raster data types list and the number of rasters list.")

# Verify there are no rasters in the output directory.

try:
    gp.Workspace = output_dir
except Exception, e:
    log_error_and_exit(gp, "Could not check set the geoprocessor's workspace to \"" + output_dir + "\". The gp.Workspace property failed when set to that directory: " + str(e))

try:
    existing_rasters = gp.ListRasters("*")
except Exception, e:
    log_error_and_exit(gp, "Could not check for existing rasters in the directory \"" + output_dir + "\". The gp.ListRasters function failed: " + str(e))

existing_rasters.Reset()
r = existing_rasters.Next()
if r is not None and len(r.strip()) > 0:
    log_error_and_exit(gp, "Rasters alredy exist in the directory \"" + output_dir + "\". Please specify an empty directory.")

# Create the rasters.    

rasters_generated = 0
predicted_info_directory_entries = 0

for i in range(len(raster_data_types_list)):
    data_type = raster_data_types_list[i]
    for j in range(num_rasters_list[i]):

        # Print a warning if the next raster will cause Arc to crash.
        
        if data_type == "FLOAT":
            predicted_info_directory_entries = predicted_info_directory_entries + 2
        else:
            predicted_info_directory_entries = predicted_info_directory_entries + 3

        if divmod(predicted_info_directory_entries, 1024)[1] == 0:
            log_warning(gp, "***** The next raster will cause Arc to crash! *****")

        # Generate the raster

        raster = "%s\\test%04i" % (output_dir, rasters_generated + 1)
        
        if data_type == "FLOAT":
            gp.CreateConstantRaster_sa(raster, 1.1, "FLOAT", 1, "0 0 10 10")
        else:
            gp.CreateConstantRaster_sa(raster, 1, "INTEGER", 1, "0 0 10 10")

        log_returned_geoprocessor_messages(gp)

        rasters_generated = rasters_generated + 1            

        # Print a message if we were supposed to crash but we didn't.

        if divmod(predicted_info_directory_entries, 1024)[1] == 0:
            log_warning(gp, "***** Arc should have crashed. If you're seeing this message, it did not crash, and my algorithm for predicting its crashes is not correct. *****")

log_message(gp, str(rasters_generated) + " rasters generated successfully.")