Changeset 8306 for trunk/anuga_work/development/gareth/balanced_basic/__init__.py

Timestamp:

Jan 18, 2012, 2:31:29 PM (13 years ago)

Author:

davies

Message:

Changes to swb2

File:

• trunk/anuga_work/development/gareth/balanced_basic/__init__.py (modified) (2 diffs)

trunk/anuga_work/development/gareth/balanced_basic/__init__.py

@@ -7 +7 @@
 
 # Make selected classes available directly
-from swb2_domain import Domain
+from balanced_basic.swb2_domain import Domain as Domain
+from balanced_basic.swb2_domain import Domain
 #, \
 #     Transmissive_boundary,\
@@ -20 +21 @@
 #
 
+#-----------------------------------------------------
+# Make selected classes available directly
+#-----------------------------------------------------
+from anuga.__metadata__ import __version__, __date__, __author__
+
+#from anuga.shallow_water.shallow_water_domain import Domain
+
+from anuga.abstract_2d_finite_volumes.quantity import Quantity
+
+from anuga.abstract_2d_finite_volumes.util import file_function, \
+                                        sww2timeseries, sww2csv_gauges
+
+from anuga.abstract_2d_finite_volumes.mesh_factory import rectangular_cross, \
+                                                    rectangular
+
+from anuga.file.csv_file import load_csv_as_building_polygons,  \
+                                load_csv_as_polygons
+
+from anuga.file.sts import create_sts_boundary
+
+from anuga.file.ungenerate import load_ungenerate
+
+from anuga.geometry.polygon import read_polygon, plot_polygons, polygon_area, \
+                                   inside_polygon, polygon_area
+from anuga.geometry.polygon_function import Polygon_function
+
+from anuga.abstract_2d_finite_volumes.pmesh2domain import \
+                                            pmesh_to_domain_instance
+
+from anuga.utilities.system_tools import file_length
+from anuga.utilities.sww_merge import sww_merge
+from anuga.utilities.file_utils import copy_code_files
+from anuga.utilities.numerical_tools import safe_acos as acos
+
+from anuga.geometry.polygon import read_polygon
+from anuga.caching import cache
+
+#-----------------------------
+# Standard Boundaries
+#-----------------------------
+from anuga.shallow_water.boundaries import File_boundary
+from anuga.shallow_water.boundaries import Reflective_boundary
+from anuga.shallow_water.boundaries import Field_boundary
+from anuga.shallow_water.boundaries import \
+                    Transmissive_stage_zero_momentum_boundary
+from anuga.shallow_water.boundaries import \
+                    Transmissive_momentum_set_stage_boundary
+from anuga.shallow_water.boundaries import \
+                    Transmissive_n_momentum_zero_t_momentum_set_stage_boundary
+
+
+#-----------------------------
+# SWW-specific Boundaries
+#-----------------------------
+from anuga.abstract_2d_finite_volumes.generic_boundary_conditions \
+                            import Dirichlet_boundary
+from anuga.abstract_2d_finite_volumes.generic_boundary_conditions \
+                            import Time_boundary
+from anuga.abstract_2d_finite_volumes.generic_boundary_conditions \
+                            import Time_space_boundary
+from anuga.abstract_2d_finite_volumes.generic_boundary_conditions \
+                            import Transmissive_boundary
+
+
+
+#-----------------------------
+# Shalow Water Tsunamis
+#-----------------------------
+
+from anuga.shallow_water.smf import slide_tsunami, slump_tsunami
+
+
+#-----------------------------
+# Forcing
+#-----------------------------
+from anuga.shallow_water.forcing import Inflow, Rainfall, Wind_stress
+
+#-----------------------------
+# File conversion utilities
+#-----------------------------
+from anuga.file_conversion.file_conversion import sww2obj, \
+                    timefile2netcdf, tsh2sww
+from anuga.file_conversion.urs2nc import urs2nc
+from anuga.file_conversion.urs2sww import urs2sww  
+from anuga.file_conversion.urs2sts import urs2sts
+from anuga.file_conversion.dem2pts import dem2pts                    
+from anuga.file_conversion.esri2sww import esri2sww   
+from anuga.file_conversion.sww2dem import sww2dem, sww2dem_batch 
+from anuga.file_conversion.asc2dem import asc2dem     
+from anuga.file_conversion.ferret2sww import ferret2sww     
+from anuga.file_conversion.dem2dem import dem2dem
+
+
+#-----------------------------
+# Mesh API
+#-----------------------------
+from anuga.pmesh.mesh_interface import create_mesh_from_regions
+
+#-----------------------------
+# SWW file access
+#-----------------------------
+from anuga.shallow_water.sww_interrogate import get_flow_through_cross_section
+    
+
+
+#---------------------------
+# Operators
+#---------------------------
+from anuga.operators.base_operator import Operator
+
+#---------------------------
+# Structure Operators
+#---------------------------
+from anuga.structures.structure_operator import Structure_operator
+from anuga.structures.boyd_box_operator import Boyd_box_operator
+from anuga.structures.boyd_pipe_operator import Boyd_pipe_operator
+
+#---------------------------
+# User Access Functions
+#---------------------------
+
+from anuga.utilities.system_tools import get_user_name, get_host_name, get_revision_number
+
+
+
+
+#-----------------------------
+# rectangular domains
+#-----------------------------
+def rectangular_cross_domain(*args, **kwargs):
+    """
+    Create a rectangular domain with triangulation made
+    up of m+1 by n+1 uniform rectangular cells divided
+    into 4 triangles in a cross pattern
+
+    Arguments
+    m:      number of cells in x direction
+    n:      number of cells in y direction
+    len1:   length of domain in x direction (left to right)
+            (default 1.0)
+    len2:   length of domain in y direction (bottom to top)
+            (default 1.0)
+    origin: tuple (x,y) specifying location of lower left corner
+            of domain (default (0,0))
+    """
+    points, vertices, boundary = rectangular_cross(*args, **kwargs)
+    return Domain(points, vertices, boundary)
+
+#----------------------------
+# Create domain from file
+#----------------------------
+def create_domain_from_file(file, DomainClass=Domain):
+    """
+    Create a domain from a file
+    """
+    return pmesh_to_domain_instance(file,DomainClass=DomainClass)
+
+#---------------------------
+# Create domain from regions
+#---------------------------
+
+def create_domain_from_regions(bounding_polygon,
+                               boundary_tags,
+                               maximum_triangle_area=None,
+                               mesh_filename=None,
+                               interior_regions=None,
+                               interior_holes=None,
+                               hole_tags=None,
+                               poly_geo_reference=None,
+                               mesh_geo_reference=None,
+                               minimum_triangle_angle=28.0,
+                               fail_if_polygons_outside=True,
+                               use_cache=False,
+                               verbose=True):
+    """Create domain from bounding polygons and resolutions.
+
+    bounding_polygon is a list of points in Eastings and Northings,
+    relative to the zone stated in poly_geo_reference if specified.
+    Otherwise points are just x, y coordinates with no particular 
+    association to any location.
+
+    boundary_tags is a dictionary of symbolic tags. For every tag there
+    is a list of indices referring to segments associated with that tag.
+    If a segment is omitted it will be assigned the default tag ''.
+
+    maximum_triangle_area is the maximal area per triangle
+    for the bounding polygon, excluding the  interior regions.
+
+    Interior_regions is a list of tuples consisting of (polygon,
+    resolution) for each region to be separately refined. Do not have
+    polygon lines cross or be on-top of each other.  Also do not have
+    polygon close to each other.
+    
+    NOTE: If a interior_region is outside the bounding_polygon it should 
+    throw an error
+    
+    interior_holes is a list of ploygons for each hole. These polygons do not
+    need to be closed, but their points must be specified in a counter-clockwise
+    order.
+
+    hole_tags  is a list of tag segment dictionaries.
+
+    This function does not allow segments to share points - use underlying
+    pmesh functionality for that
+
+    poly_geo_reference is the geo_reference of the bounding polygon and
+    the interior polygons.
+    If none, assume absolute.  Please pass one though, since absolute
+    references have a zone.
+    
+    mesh_geo_reference is the geo_reference of the mesh to be created.
+    If none is given one will be automatically generated.  It was use
+    the lower left hand corner of  bounding_polygon (absolute)
+    as the x and y values for the geo_ref.
+    
+    Returns the shallow water domain instance
+
+    Note, interior regions should be fully nested, as overlaps may cause
+    unintended resolutions. 
+
+    fail_if_polygons_outside: If True (the default) Exception in thrown
+    where interior polygons fall outside bounding polygon. If False, these
+    will be ignored and execution continued.
+        
+    
+    """
+
+
+    # Build arguments and keyword arguments for use with caching or apply.
+    args = (bounding_polygon,
+            boundary_tags)
+    
+    kwargs = {'maximum_triangle_area': maximum_triangle_area,
+              'mesh_filename': mesh_filename,
+              'interior_regions': interior_regions,
+              'interior_holes': interior_holes,
+              'hole_tags': hole_tags,
+              'poly_geo_reference': poly_geo_reference,
+              'mesh_geo_reference': mesh_geo_reference,
+              'minimum_triangle_angle': minimum_triangle_angle,
+              'fail_if_polygons_outside': fail_if_polygons_outside,
+              'verbose': verbose} #FIXME (Ole): See ticket:14
+
+    # Call underlying engine with or without caching
+    if use_cache is True:
+        try:
+            from anuga.caching import cache
+        except:
+            msg = 'Caching was requested, but caching module'+\
+                  'could not be imported'
+            raise (msg)
+
+
+        domain = cache(_create_domain_from_regions,
+                       args, kwargs,
+                       verbose=verbose,
+                       compression=False)
+    else:
+        domain = apply(_create_domain_from_regions,
+                       args, kwargs)
+
+    return domain
+
+        
+def _create_domain_from_regions(bounding_polygon,
+                                boundary_tags,
+                                maximum_triangle_area=None,
+                                mesh_filename=None,                           
+                                interior_regions=None,
+                                interior_holes=None,
+                                hole_tags=None,
+                                poly_geo_reference=None,
+                                mesh_geo_reference=None,
+                                minimum_triangle_angle=28.0,
+                                fail_if_polygons_outside=True,
+                                verbose=True):
+    """_create_domain_from_regions - internal function.
+
+    See create_domain_from_regions for documentation.
+    """
+
+    #from anuga.shallow_water.shallow_water_domain import Domain
+    from balanced_basic.swb2_domain import Domain
+    from anuga.pmesh.mesh_interface import create_mesh_from_regions
+    
+    create_mesh_from_regions(bounding_polygon,
+                             boundary_tags,
+                             maximum_triangle_area=maximum_triangle_area,
+                             interior_regions=interior_regions,
+                             filename=mesh_filename,
+                             interior_holes=interior_holes,
+                             hole_tags=hole_tags,
+                             poly_geo_reference=poly_geo_reference,
+                             mesh_geo_reference=mesh_geo_reference,
+                             minimum_triangle_angle=minimum_triangle_angle,
+                             fail_if_polygons_outside=fail_if_polygons_outside,
+                             use_cache=False,
+                             verbose=verbose)
+
+    domain = Domain(mesh_filename, use_cache=False, verbose=verbose)
+
+
+    return domain
+    
+import logging as log
+
+from anuga.config import use_psyco, g, velocity_protection
+if use_psyco:
+    # try using psyco if available
+    try:
+        import psyco
+    except:
+        import os
+        import sys
+        if os.name == 'posix' and os.uname()[4] in ['x86_64', 'ia64']:
+            pass
+            # Psyco isn't supported on 64 bit systems, but it doesn't matter
+        elif sys.version[:3] == '2.7' :
+            pass
+            # Psyco isn't available for python 2.7 (16/05/2011)
+        else:
+            log.critical('WARNING: psyco (speedup) could not be imported, '
+                         'you may want to consider installing it')
+    else:
+        psyco.full() # aggressively compile everything
+        #psyco.background() # attempt to profile code - only compile most used
+