Fixes to make the ocean_level-variable work properly.

Author: tcld

worldengine/cli/main.py

@@ -5,6 +5,7 @@
 import pickle
 import random
 import worldengine.generation as geo
+from worldengine.simulations.basic import find_threshold_f
 from worldengine.common import array_to_matrix, set_verbose, print_verbose
 from worldengine.draw import draw_ancientmap_on_file, draw_biome_on_file, draw_ocean_on_file, \
     draw_precipitation_on_file, draw_grayscale_heightmap_on_file, draw_simple_elevation_on_file, \
@@ -80,7 +81,7 @@ def generate_rivers_map(world, filename):
 
 
 def generate_plates(seed, world_name, output_dir, width, height,
-                    num_plates=10):
+                    ocean_level, num_plates=10):
     """
     Eventually this method should be invoked when generation is called at
     asked to stop at step "plates", it should not be a different operation
@@ -93,6 +94,7 @@ def generate_plates(seed, world_name, output_dir, width, height,
     :return:
     """
     elevation, plates = generate_plates_simulation(seed, width, height,
+                                                   ocean_level=ocean_level,
                                                    num_plates=num_plates)
 
     world = World(world_name, width, height, seed, num_plates, -1.0, "plates")
@@ -101,8 +103,9 @@ def generate_plates(seed, world_name, output_dir, width, height,
 
     # Generate images
     filename = '%s/plates_%s.png' % (output_dir, world_name)
-    # TODO calculate appropriate sea_level
-    sea_level = 1.0
+    sea_level = find_threshold_f(world.elevation['data'], 1.0 - ocean_level,
+                                      ocean=None, max=1.0, mindist=0.000005)    
+    
     draw_simple_elevation_on_file(world.elevation['data'], filename, width,
                                   height, sea_level)
     print("+ plates image generated in '%s'" % filename)
@@ -300,7 +303,7 @@ def main():
     g_generate.add_argument('--ocean_level', dest='ocean_level', type=float,
                             help='amount of surface covered by ocean " +'
                                  '[default = %(default)s]',
-                            metavar="N", default=1.0)
+                            metavar="N", default=0.65)
 
     # -----------------------------------------------------
     g_ancient_map = parser.add_argument_group(
@@ -463,7 +466,8 @@ def main():
         print('starting (it could take a few minutes) ...')
 
         generate_plates(seed, world_name, args.output_dir, args.width,
-                        args.height, num_plates=args.number_of_plates)
+                        args.height, args.ocean_level,
+                        num_plates=args.number_of_plates)
 
     elif operation == 'ancient_map':
         print('')  # empty line

worldengine/generation.py

@@ -127,7 +127,7 @@ def initialize_ocean_and_thresholds(world, ocean_level=0.65):
     e = world.elevation['data']
     #Calculate the height of the ocean relative to the geometry
     ocean_maxlevel = find_threshold_f(e, 1.0 - ocean_level,
-                                      ocean=None, max=1.0, mindist=0.00001)
+                                      ocean=None, max=100.0, mindist=0.00001)
     ocean = fill_ocean(e, ocean_maxlevel)
 
     #TODO: The thresholds (and amounts of e.g. mountains) could change for every world, depending on the height above the ocean.

worldengine/plates.py

@@ -75,7 +75,7 @@ def world_gen(name, width, height, seed, num_plates=10, ocean_level=0.65,
     if verbose:
         start_time = time.time()
     place_oceans_at_map_borders(world)
-    initialize_ocean_and_thresholds(world)
+    initialize_ocean_and_thresholds(world, ocean_level)
     if verbose:
         elapsed_time = time.time() - start_time
         print("...plates.world_gen: oceans initialized. Elapsed time " +