source: production/merimbula_2005/prepare.py @ 3233

def prepare_wind_stress(filename):
    """Converting wind timeseries to NetCDF tms file.
    This is a 'throw-away' code taylor made for files like
    'Benchmark_2_input.txt' from the LWRU2 benchmark
    """

    print 'Preparing wind timeseries %s' %filename
    from Numeric import array, zeros, Float, asarray

    fid = open(filename)

    #Skip first line
    #line = fid.readline()

    #Read remaining lines
    lines = fid.readlines()
    fid.close()


    N = len(lines)
    T = zeros(N, Float)  #Time
    S = zeros(N, Float)  #Speed
    B = zeros(N, Float)  #Bearing

    Told = 0.0
    Sold = ' '
    Lold = ' '

    for i, line in enumerate(lines):
        fields = line.split()

        #print fields

        l_time = (fields[0]+' '+fields[1])[0:-1]
        from time import strptime, mktime

        s_time = strptime(l_time,'%d/%m/%y  %H:%M:%S')

        #print s_time

        T[i] = float(mktime(s_time))

        if i==0:
            Tstart = T[0]

        T[i] = T[i] - Tstart
        #this is specific to this data set. deals with daylight saving
#        if i>3270:
#            T[i] = T[i]+3600
#
        if T[i]<Told :
            print Lold
            print l_time
            print Sold
            print s_time
            print Told
            print T[i]
            print i, T[i]-Told

        S[i] = float(fields[2])
        B[i] = float(fields[3])

        Told = T[i]
        Sold = s_time
        Lold = l_time


    #print T
    #Create tms file
    from Scientific.IO.NetCDF import NetCDFFile

    outfile = filename[:-4] + '.tms'
    print 'Writing to', outfile
    fid = NetCDFFile(outfile, 'w')

    fid.institution = 'Australian National University'
    fid.description = 'Input wind for Merimbula'
    fid.starttime = 0.0
    fid.createDimension('number_of_timesteps', len(T))
    fid.createVariable('time', Float, ('number_of_timesteps',))
    fid.variables['time'][:] = T

    fid.createVariable('speed', Float, ('number_of_timesteps',))
    fid.variables['speed'][:] = S[:]

    fid.createVariable('bearing', Float, ('number_of_timesteps',))
    fid.variables['bearing'][:] = B[:]


    fid.close()


def prepare_timeboundary(filename):
    """Converting tide time series to NetCDF tms file.
    This is a 'throw-away' code taylor made for files like
    'Benchmark_2_input.txt' from the LWRU2 benchmark
    """

    print 'Preparing time boundary from %s' %filename
    from Numeric import array, zeros, Float, asarray

    fid = open(filename)

    #Skip first line
    line = fid.readline()

    #Read remaining lines
    lines = fid.readlines()
    fid.close()


    N = len(lines)
    T = zeros(N, Float)  #Time
    Q = zeros(N, Float)  #Values

    Told = 0.0
    Sold = ' '
    Lold = ' '
    for i, line in enumerate(lines):
        fields = line.split()

        #print fields

        l_time = (fields[0]+' '+fields[1])[0:-1]
        from time import strptime, mktime

        s_time = strptime(l_time,'%d/%m/%y  %H:%M:%S')

        #print s_time

        T[i] = float(mktime(s_time))

        if i==0:
            Tstart = T[0]

        T[i] = T[i] - Tstart
        #this is specific to this data set. deals with daylight saving
        if i>3270:
            T[i] = T[i]+3600

        if T[i]<Told :
            print Lold
            print l_time
            print Sold
            print s_time
            print Told
            print T[i]
            print i, T[i]-Told

        Q[i] = float(fields[2])

        Told = T[i]
        Sold = s_time
        Lold = l_time


    #print T
    #Create tms file
    from Scientific.IO.NetCDF import NetCDFFile

    outfile = filename[:-4] + '.tms'
    print 'Writing to', outfile
    fid = NetCDFFile(outfile, 'w')

    fid.institution = 'Australian National University'
    fid.description = 'Input wave for Merimbula'
    fid.starttime = 0.0
    fid.createDimension('number_of_timesteps', len(T))
    fid.createVariable('time', Float, ('number_of_timesteps',))
    fid.variables['time'][:] = T

    fid.createVariable('stage', Float, ('number_of_timesteps',))
    fid.variables['stage'][:] = Q[:]

    fid.createVariable('xmomentum', Float, ('number_of_timesteps',))
    fid.variables['xmomentum'][:] = 0.0

    fid.createVariable('ymomentum', Float, ('number_of_timesteps',))
    fid.variables['ymomentum'][:] = 0.0

    fid.close()

#-------------------------------------------------------------
if __name__ == "__main__":
    import project
    print 'Prepare Open sea boundary condition from ',project.original_boundary_filename
    prepare_timeboundary(project.original_boundary_filename )


    print 'Prepare wind from ',project.original_wind_filename
    prepare_wind_stress(project.original_wind_filename )

    #Preparing points
    print 'Prepare bathymetry from xya file ',project.bathymetry_filename
    from pyvolution.data_manager import xya2pts
    xya2pts(project.bathymetry_filename, verbose = True)


#    fit_to_mesh_file(mesh_file, point_file, mesh_output_file,
#                     alpha=DEFAULT_ALPHA, verbose= False,
#                     expand_search = False,
#                     data_origin = None,
#                     mesh_origin = None,
#                     precrop = False,
#                     display_errors = True):