diff --git a/scripts/world_create.py b/scripts/world_create.py index 7595e9e..cbfa399 100755 --- a/scripts/world_create.py +++ b/scripts/world_create.py @@ -225,13 +225,19 @@ def read_template(temp_file_name): if __name__ == '__main__': - arg_list = sys.argv + # Check for required command-line argument + if len(sys.argv) < 2: + print("Usage: world_create.py [invert]") + print(" : name of the world file to create (e.g., my_world.world)") + print(" [invert] : optional flag to invert image (default: no inversion)") + sys.exit(1) + arg_list = sys.argv world_name = arg_list[1] invert = False - if (len(arg_list) > 2): + if len(arg_list) > 2: flag = arg_list[2] - if (flag == "invert"): + if flag == "invert": invert = True rospack = rospkg.RosPack() @@ -251,4 +257,4 @@ def read_template(temp_file_name): # Double inversion for image preservation # rescale_and_resize(heightmap_path, img_size) - print('Done') + print('Done') \ No newline at end of file