Changeset 2073


Ignore:
Timestamp:
Nov 25, 2005, 10:24:53 AM (18 years ago)
Author:
duncan
Message:

making function not public

Location:
inundation/pyvolution
Files:
2 edited

Legend:

Unmodified
Added
Removed
  • inundation/pyvolution/data_manager.py

    r2069 r2073  
    25792579def sww2domain(filename,boundary=None,t=None,\
    25802580               fail_if_NaN=True,NaN_filler=0\
    2581                ,verbose = True,very_verbose = False):
     2581               ,verbose = False,very_verbose = False):
    25822582    """
    25832583    Usage: domain = sww2domain('file.sww',t=time (default = last time in file))
     
    33343334    fid.close()
    33353335
     3336#********************
     3337#*** END OF OBSOLETE FUNCTIONS
     3338#***************
     3339
    33363340def asc_csiro2sww(bath_dir,
    33373341                  elevation_dir,
     
    34133417    number_of_latitudes = bath_grid.shape[0]
    34143418    number_of_longitudes = bath_grid.shape[1]
    3415     number_of_times = len(os.listdir(elevation_dir))
    3416     number_of_points = number_of_latitudes*number_of_longitudes
     3419    #number_of_times = len(os.listdir(elevation_dir))
     3420    #number_of_points = number_of_latitudes*number_of_longitudes
    34173421    number_of_volumes = (number_of_latitudes-1)*(number_of_longitudes-1)*2
    34183422
     
    34253429    latitudes.reverse()
    34263430   
    3427     #print "latitudes - before get_min_max_indexes",latitudes
    3428     kmin, kmax, lmin, lmax = get_min_max_indexes(latitudes,longitudes,
     3431    #print "latitudes - before _get_min_max_indexes",latitudes
     3432    kmin, kmax, lmin, lmax = _get_min_max_indexes(latitudes,longitudes,
    34293433                                                 minlat=minlat, maxlat=maxlat,
    34303434                                                 minlon=minlon, maxlon=maxlon)
     
    34333437    bath_grid = bath_grid[kmin:kmax,lmin:lmax]
    34343438    #print "bath_grid",bath_grid
    3435     number_of_latitudes = bath_grid.shape[0]
    3436     number_of_longitudes = bath_grid.shape[1]
     3439    latitudes = latitudes[kmin:kmax]
     3440    longitudes = longitudes[lmin:lmax]
     3441    number_of_latitudes = len(latitudes)
     3442    number_of_longitudes = len(longitudes)
    34373443    number_of_times = len(os.listdir(elevation_dir))
    34383444    number_of_points = number_of_latitudes*number_of_longitudes
    34393445    number_of_volumes = (number_of_latitudes-1)*(number_of_longitudes-1)*2
    34403446    #print "number_of_points",number_of_points
    3441 
    3442     latitudes = latitudes[kmin:kmax]
    3443     longitudes = longitudes[lmin:lmax]
    3444     number_of_latitudes = len(latitudes)
    3445     number_of_longitudes = len(longitudes)
    3446 
    34473447
    34483448    #Work out the times
     
    35613561            v4 = vertices[l+1,k]
    35623562
    3563             volumes.append([v1,v2,v3]) #Upper element
    3564             volumes.append([v4,v3,v2]) #Lower element
     3563            #Note, this is different to the ferrit2sww code
     3564            #since the order of the lats is reversed.
     3565            volumes.append([v1,v3,v2]) #Upper element
     3566            volumes.append([v4,v2,v3]) #Lower element
    35653567
    35663568    volumes = array(volumes)
     
    35683570    geo_ref = Geo_reference(refzone,min(x),min(y))
    35693571    geo_ref.write_NetCDF(outfile)
     3572
     3573    # This will put the geo ref in the middle
     3574    #geo_ref = Geo_reference(refzone,(max(x)+min(x))/2.0,(max(x)+min(y))/2.)
     3575
     3576
     3577    if verbose:
     3578        print '------------------------------------------------'
     3579        print 'More Statistics:'
     3580        print '  Extent (/lon):'
     3581        print '    x in [%f, %f], len(lat) == %d'\
     3582              %(min(x), max(x),
     3583                len(x))
     3584        print '    y in [%f, %f], len(lon) == %d'\
     3585              %(min(y), max(y),
     3586                len(y))
     3587        print 'geo_ref: ',geo_ref
    35703588
    35713589    z = resize(bath_grid,outfile.variables['z'][:].shape)
     
    36283646    outfile.close()
    36293647   
    3630 def get_min_max_indexes(latitudes,longitudes,
     3648def _get_min_max_indexes(latitudes,longitudes,
    36313649                        minlat=None, maxlat=None,
    36323650                        minlon=None, maxlon=None):
  • inundation/pyvolution/test_data_manager.py

    r2064 r2073  
    1313from shallow_water import *
    1414from config import epsilon
     15import data_manager
    1516
    1617from coordinate_transforms.geo_reference import Geo_reference
     
    33343335        fid.close()
    33353336       
    3336         sww_file = 'a_test.sww'
     3337        sww_file = tempfile.mktemp(".sww")
     3338        #sww_file = 'a_test.sww'
    33373339        asc_csiro2sww(bath_dir,elevation_dir, ucur_dir, vcur_dir,
    33383340                      sww_file, fail_on_NaN = False, elevation_NaN_filler = 0,
     
    33873389       
    33883390        fid.close()
    3389    
     3391       
     3392        # is the sww file readable?
     3393        #Lets see if we can convert it to a dem!
     3394        #print "sww_file",sww_file
     3395        #dem_file = tempfile.mktemp(".dem")
     3396        domain = sww2domain(sww_file) ###, dem_file)
     3397        domain.check_integrity()
     3398
    33903399        #tidy up
    33913400        os.remove(bath_dir_filename)
     
    33963405        os.rmdir(elevation_dir)
    33973406
    3398 
     3407     
     3408       
    33993409        # remove sww file
    34003410        os.remove(sww_file)
     3411       
     3412        # remove dem file
     3413        #os.remove(dem_file)
    34013414
    34023415    def test_get_min_max_indexes(self):
     
    34063419        # k - lat
    34073420        # l - lon
    3408         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3409                                                       -10,4,-10,31)
     3421        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3422            latitudes,longitudes,
     3423            -10,4,-10,31)
    34103424
    34113425        #print "kmin",kmin;print "kmax",kmax
     
    34203434       
    34213435        ## 2nd test
    3422         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3423                                                       0.5,2.5,5,25)
     3436        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3437            latitudes,longitudes,
     3438            0.5,2.5,5,25)
    34243439        #print "kmin",kmin;print "kmax",kmax
    34253440        #print "lmin",lmin;print "lmax",lmax
     
    34343449       
    34353450        ## 3rd test
    3436         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
     3451        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(latitudes,
     3452                                                                    longitudes,
    34373453                                                      1.1,1.9,12,17)
    34383454        #print "kmin",kmin;print "kmax",kmax
     
    34493465       
    34503466        ## 4th test
    3451         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
     3467        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3468            latitudes,longitudes,
    34523469                                                      -0.1,1.9,-2,17)
    34533470        #print "kmin",kmin;print "kmax",kmax
     
    34623479                         'failed')
    34633480        ## 5th test
    3464         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3465                                                       0.1,1.9,2,17)
     3481        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3482            latitudes,longitudes,
     3483            0.1,1.9,2,17)
    34663484        #print "kmin",kmin;print "kmax",kmax
    34673485        #print "lmin",lmin;print "lmax",lmax
     
    34773495        ## 6th test
    34783496       
    3479         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3480                                                       1.5,4,18,32)
     3497        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3498            latitudes,longitudes,
     3499            1.5,4,18,32)
    34813500        #print "kmin",kmin;print "kmax",kmax
    34823501        #print "lmin",lmin;print "lmax",lmax
     
    34933512        ## 7th test
    34943513        m2d = array([[0,1,2,3],[4,5,6,7],[8,9,10,11],[12,13,14,15]])
    3495         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3496                                                       1.5,1.5,15,15)
     3514        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3515            latitudes,longitudes,
     3516            1.5,1.5,15,15)
    34973517        #print "kmin",kmin;print "kmax",kmax
    34983518        #print "lmin",lmin;print "lmax",lmax
     
    35193539        # k - lat
    35203540        # l - lon
    3521         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3522                                                       -37,-27,147,149.5)
     3541        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3542            latitudes,longitudes,
     3543            -37,-27,147,149.5)
    35233544
    35243545        #print "kmin",kmin;print "kmax",kmax
     
    35473568        # k - lat
    35483569        # l - lon
    3549         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes,
    3550                                                       -43,-37,148.5,149.5)
     3570        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3571            latitudes,longitudes,
     3572            -43,-37,148.5,149.5)
    35513573
    35523574       
     
    35703592        # k - lat
    35713593        # l - lon
    3572         kmin, kmax, lmin, lmax = get_min_max_indexes (latitudes,longitudes)
    3573 
     3594        kmin, kmax, lmin, lmax = data_manager._get_min_max_indexes(
     3595            latitudes,longitudes)
    35743596       
    35753597        #print "kmin",kmin;print "kmax",kmax
     
    35883610#-------------------------------------------------------------
    35893611if __name__ == "__main__":
    3590     #suite = unittest.makeSuite(Test_Data_Manager,'test_asc_csiro2sww4')
     3612    #suite = unittest.makeSuite(Test_Data_Manager,'test_get_min_max_indexes3')
    35913613    suite = unittest.makeSuite(Test_Data_Manager,'test')
    35923614    #suite = unittest.makeSuite(Test_Data_Manager,'test_sww2dem_asc_missing_points')
Note: See TracChangeset for help on using the changeset viewer.