Changeset 2622 for production/merimbula_2005/prepare.py

Timestamp:

Mar 28, 2006, 2:36:03 PM (18 years ago)

Author:

steve

Message:

Updated merimbula viewer

File:

• production/merimbula_2005/prepare.py (modified) (8 diffs)

production/merimbula_2005/prepare.py

@@ -1 +1 @@
 
 
-def prepare_timeboundary(filename):
-    """Converting tide time series to NetCDF tms file.
+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 time boundary from %s' %filename
+    print 'Preparing wind timeseries %s' %filename
     from Numeric import array, zeros, Float, asarray
 
@@ -13 +13 @@
 
     #Skip first line
-    line = fid.readline()
+    #line = fid.readline()
 
     #Read remaining lines
@@ -22 +22 @@
     N = len(lines)
     T = zeros(N, Float)  #Time
-    Q = zeros(N, Float)  #Values
+    S = zeros(N, Float)  #Speed
+    B = zeros(N, Float)  #Bearing
 
     Told = 0.0
     Sold = ' '
     Lold = ' '
+
     for i, line in enumerate(lines):
         fields = line.split()
@@ -46 +48 @@
         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 i>3270:
+#            T[i] = T[i]+3600
+#
         if T[i]<Told :
             print Lold
@@ -58 +60 @@
             print i, T[i]-Told
 
-        Q[i] = float(fields[2])
+        S[i] = float(fields[2])
+        B[i] = float(fields[3])
 
         Told = T[i]
@@ -65 +68 @@
 
 
+    #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
@@ -90 +183 @@
     fid.close()
 
-
 #-------------------------------------------------------------
 if __name__ == "__main__":
@@ -96 +188 @@
     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