Changeset 2252


Ignore:
Timestamp:
Jan 19, 2006, 12:00:08 PM (18 years ago)
Author:
ole
Message:

Cosmetic

Location:
inundation
Files:
5 edited

Legend:

Unmodified
Added
Removed
  • inundation/pyvolution/combine_pts.py

    r2022 r2252  
    1 """Add two point files by the union of the fine and coarse points.
    2 Course points that are in the extent of the fine points are removed.
    3 
    4 The extent of the fine points file is assumed to be a rectangle, parallel
    5 to the x and y axis.
     1"""Functionality for aggregation of points files
    62
    73   Ole Nielsen, Stephen Roberts, Duncan Gray
    84   Geoscience Australia, 2005.   
    95"""
     6
    107from utilities.polygon import outside_polygon, inside_polygon
    118from Numeric import take, concatenate
     
    1613
    1714def combine_rectangular_points_files(fine_points_file,
    18                                     coarse_points_file,
    19                                     output_points_file,
     15                                     coarse_points_file,
     16                                     output_points_file,
    2017                                     verbose = False):
     18    """Add two point files by the union of the fine and coarse points.
    2119   
    22     from load_mesh.loadASCII import import_points_file, extent, point_atts2array, export_points_file, add_point_dictionaries,take_points
     20       Course points that are in the extent of the fine points are removed.
     21
     22       The extent of the fine points file is assumed to be a rectangle, parallel
     23       to the x and y axis.
     24    """
     25   
     26    from load_mesh.loadASCII import import_points_file, extent, point_atts2array, export_points_file, add_point_dictionaries, take_points
    2327
    2428   
    2529    # load fine points file
    26     if verbose:print "loading Point files"
     30    if verbose: print "loading Point files"
    2731    fine = import_points_file(fine_points_file)
    2832   
     
    3034    coarse = import_points_file(coarse_points_file)
    3135       
    32     if verbose:print "doing other stuff"
     36    if verbose: print "doing other stuff"
    3337    # convert to Numeric   
    3438    fine = point_atts2array(fine)
     
    4448        if coarse.has_key('geo_reference'):
    4549            coarse['pointlist'] = \
    46              fine['geo_reference'].change_points_geo_ref(coarse['pointlist'],
    47                                                          points_geo_ref=coarse['geo_reference']) 
     50                fine['geo_reference'].change_points_geo_ref(coarse['pointlist'],
     51                                                            points_geo_ref=coarse['geo_reference']) 
    4852    # find extent of course points
    4953    extent = extent(fine['pointlist'])   
     
    5559        t0 = time.time()
    5660    outside_coarse_indices = outside_polygon(coarse['pointlist'],
    57                                              extent,closed=True)
     61                                             extent, closed=True)
    5862    if verbose:
    5963        print "Points outside determined"
     
    6468       
    6569    # add fine points and out_points
    66     if verbose:print "Adding points"
     70    if verbose: print "Adding points"
    6771    combined = add_point_dictionaries(fine, coarse)
    6872
    6973    # save
    70     if verbose:print "writing points"
     74    if verbose: print "writing points"
    7175    export_points_file(output_points_file, combined)
    7276
    7377def add_points_files(fine_points_file,
    74                                     coarse_points_file,
    75                                     output_points_file,
    76                                      verbose = False):
     78                     coarse_points_file,
     79                     output_points_file,
     80                     verbose = False):
     81    """
     82    """
    7783   
    7884    from load_mesh.loadASCII import import_points_file, extent, point_atts2array, export_points_file, add_point_dictionaries,take_points
     
    113119       
    114120    # add fine points and out_points
    115     if verbose:print "Adding points"
     121    if verbose: print "Adding points"
    116122    combined = add_point_dictionaries(fine, coarse)
    117123
    118124    # save
    119     if verbose:print "writing points"
     125    if verbose: print "writing points"
    120126    export_points_file(output_points_file, combined)
    121127
    122128def reduce_points_to_mesh_extent(points_file,
    123                                     mesh_file,
    124                                     output_points_file,
    125                                      verbose = False):
     129                                 mesh_file,
     130                                 output_points_file,
     131                                 verbose = False):
     132    """
     133    """   
    126134   
    127135    from load_mesh.loadASCII import import_points_file, extent, point_atts2array, export_points_file, add_point_dictionaries,take_points, import_mesh_file
     
    145153            extent = \
    146154             points['geo_reference'].change_points_geo_ref(extent,
    147                                                          points_geo_ref=mesh['geo_reference']) 
     155                                                           points_geo_ref=mesh['geo_reference']) 
    148156    #print "extent",extent
    149157
  • inundation/pyvolution/test_all.py

    r2141 r2252  
    4949    files = get_test_files(path)
    5050
    51     #print 'Testing:', files
     51
    5252
    5353    #test = re.compile('^test_[\w]*.py$', re.IGNORECASE)
     
    6060        files.remove('test_all.py')
    6161
     62    print 'Testing:'
     63    for file in files:
     64        print '  ' + file
    6265
    6366    if globals().has_key('exclude'):
     
    7679
    7780    from os import sep
    78    
     81
    7982    #Attempt to compile all extensions
    8083    execfile('..' + sep + 'utilities' + sep + 'compile.py')
  • inundation/pyvolution/test_combine_pts.py

    r2075 r2252  
    3838        fine_dict['geo_reference'] = Geo_reference(56,2.0,1.0)
    3939       
    40         fine_file = tempfile.mktemp(".pts")
     40        fine_file = tempfile.mktemp('.pts')
    4141        export_points_file(fine_file, fine_dict)
    4242
     
    5151        coarse_dict['geo_reference'] = Geo_reference(56,4.0,3.0)
    5252       
    53         coarse_file = tempfile.mktemp(".pts")
     53        coarse_file = tempfile.mktemp('.pts')
    5454        export_points_file(coarse_file, coarse_dict)
    5555       
    56         out_file = tempfile.mktemp(".pts")
     56        out_file = tempfile.mktemp('.pts')
    5757
    5858        combine_rectangular_points_files(fine_file,coarse_file,out_file)
     
    6363
    6464        results = import_points_file(out_file,
    65                                   delimiter = ',')
     65                                     delimiter = ',')
    6666        answer = [[2.0, 0.0],
    6767                  [0.0, 1.0],
     
    7373       
    7474        self.failUnless(len(results['pointlist']) == len(answer),
    75                          'final number of points wrong. failed.')
     75                        'final number of points wrong. failed.')
    7676       
    7777        assert allclose(results['pointlist'], answer)
     
    8484       
    8585        self.failUnless(results['geo_reference'] == fine_dict['geo_reference'],
    86                          ' failed.')
     86                        ' failed.')
    8787        #clean up
    8888        os.remove(out_file)
     
    100100        fine_dict['geo_reference'] = Geo_reference(56,2.0,1.0)
    101101       
    102         fine_file = tempfile.mktemp(".pts")
     102        fine_file = tempfile.mktemp('.pts')
    103103        export_points_file(fine_file, fine_dict)
    104104
     
    113113        coarse_dict['geo_reference'] = Geo_reference(56,4.0,3.0)
    114114       
    115         coarse_file = tempfile.mktemp(".pts")
     115        coarse_file = tempfile.mktemp('.pts')
    116116        export_points_file(coarse_file, coarse_dict)
    117117       
    118         out_file = tempfile.mktemp(".pts")
     118        out_file = tempfile.mktemp('.pts')
    119119
    120120        combine_rectangular_points_files(fine_file,coarse_file,out_file)
     
    149149       
    150150        self.failUnless(results['geo_reference'] == fine_dict['geo_reference'],
    151                          ' failed.')
     151                        ' failed.')
    152152        #clean up
    153153        os.remove(out_file)
     
    167167        fine_dict['geo_reference'] = Geo_reference(56,2.0,1.0)
    168168       
    169         fine_file = tempfile.mktemp(".pts")
     169        fine_file = tempfile.mktemp('.pts')
    170170        export_points_file(fine_file, fine_dict)
    171171
     
    180180        coarse_dict['geo_reference'] = Geo_reference(56,4.0,3.0)
    181181       
    182         coarse_file = tempfile.mktemp(".pts")
     182        coarse_file = tempfile.mktemp('.pts')
    183183        export_points_file(coarse_file, coarse_dict)
    184184       
    185         out_file = tempfile.mktemp(".pts")
     185        out_file = tempfile.mktemp('.pts')
    186186        try:
    187187            combine_rectangular_points_files(fine_file,coarse_file,out_file)
     
    189189            pass
    190190        else:
    191             self.failUnless(0 ==1,
    192                         'bad pts files did not raise error!')
     191            self.failUnless(0 == 1,
     192                            'bad pts files did not raise error!')
    193193        #clean up
    194194        os.remove(fine_file)
     
    216216        fine_dict['geo_reference'] = Geo_reference(56,x_origin,y_origin)
    217217       
    218         points_file = tempfile.mktemp(".pts")
     218        points_file = tempfile.mktemp('.pts')
    219219        export_points_file(points_file, fine_dict)
    220220
     
    236236        mesh['geo_reference'] = Geo_reference(56,x_origin+7.,y_origin+7.)
    237237       
    238         mesh_file = tempfile.mktemp(".tsh")
     238        mesh_file = tempfile.mktemp('.tsh')
    239239        export_mesh_file(mesh_file, mesh)
    240240       
    241         out_file = tempfile.mktemp(".pts")
     241        out_file = tempfile.mktemp('.pts')
    242242
    243243        reduce_points_to_mesh_extent(points_file,mesh_file,out_file)
Note: See TracChangeset for help on using the changeset viewer.