Merge pull request #22 from csyhuang/develop

Release 0.3.7

.travis.yml

@@ -1,7 +1,10 @@
 language: python
 python:
   - "2.7"
-  - "3.6"
+  - "3.4"
+  - "3.5"
+  - "3.6"      # current default Python on Travis CI
+  - "3.7"
 
 install:
   - pip install --upgrade pip setuptools wheel
@@ -12,6 +15,6 @@ install:
 
 # command to run tests
 script:
-  - coverage run --source=hn2016_falwa setup.py test
+  - python setup.py pytest
 
 after_success: pip install codecov && codecov

examples/nh2018_science/demo_script_for_nh2018.ipynb

@@ -18,7 +18,7 @@
     "This notebook demonstrate how to compute local wave activity and all the flux terms in equations (2) and (3) in NH2018 with the updated functionality in the python package `hn2016_falwa`. To run the script, please install the \n",
     "package `hn2016_falwa` using\n",
     "```\n",
-    "python setup.py install\n",
+    "python setup.py develop\n",
     "```\n",
     "after cloning the [GitHub repo](http://github.com/csyhuang/hn2016_falwa).\n",
     "\n",
@@ -503,7 +503,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.5"
+   "version": "3.5.6"
   }
  },
  "nbformat": 4,

hn2016_falwa/basis.py

@@ -35,7 +35,6 @@ def eqvlat(ylat, vort, area, n_points, planet_radius=6.378e+6, vgrad=None):
     vort_min = np.min([vort.min(), vort.min()])
     vort_max = np.max([vort.max(), vort.max()])
     q_part_u = np.linspace(vort_min, vort_max, n_points, endpoint=True)
-    #dq = q_part_u[2] - q_part_u[1]
     aa = np.zeros(q_part_u.size)  # to sum up area
     vort_flat = vort.flatten()  # Flatten the 2D arrays to 1D
     area_flat = area.flatten()

hn2016_falwa/beta_version.py

@@ -1,6 +1,7 @@
 def input_jk_output_index(j,k,kmax):
     return j*(kmax) + k
 
+
 def extrap1d(interpolator):
 
     xs = interpolator.x

hn2016_falwa/constant.py

@@ -0,0 +1,7 @@
+# Define physical constants
+P0 = 1000.  # Ground pressure level. Unit: hPa
+SCALE_HEIGHT = 7000.  # Unit: m
+CP = 1004. # specific heat at constant pressure for air (cp) = 1004 J/kg-K
+DRY_GAS_CONSTANT = 287.
+EARTH_RADIUS = 6.378e+6  # Unit: m
+EARTH_OMEGA = 7.29e-5

hn2016_falwa/download_data.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python
+from enum import Enum
+from ecmwfapi import ECMWFDataServer
+server = ECMWFDataServer()
+
+
+class ERAICode(Enum):
+    u = '131.128' # zonal wind
+    v = '132.128' # meridional wind
+    t = '130.128' # temperature
+
+def retrieve_erai(start_date, end_date, file_suffix, var_list):
+    """
+    Args:
+        start_date(datetime.date)
+        start_date(datetime.date)
+        file_suffix(str)
+        var_list(list)
+
+    Return:
+        A boolean indicating whether the download is successful
+    """
+
+    start_date_str = start_date.strftime(format='%Y-%m-%d')
+    end_date_str = end_date.strftime(format='%Y-%m-%d')
+    fname = "{}_to_{}_{}.nc".format(start_date_str, end_date_str, file_suffix)
+    param_str = "/".join([x.value for x in var_list])
+
+    try:
+        server.retrieve({
+            "class": "ei",
+            "dataset": "interim",
+            "date": "{}/to/{}".format(start_date_str, end_date_str),
+            "expver": "1",
+            "grid": "1.5/1.5",
+            "levelist": "1/2/3/5/7/10/20/30/50/70/100/125/150/175/200/225/250/300/350/400/450/500/550/600/650/700/750/775/800/825/850/875/900/925/950/975/1000",
+            "levtype": "pl",
+            "param": param_str,
+            "step": "0",
+            "stream": "oper",
+            "format": "netcdf",
+            "time": "00:00:00/06:00:00/12:00:00/18:00:00",
+            "type": "an",
+            "target": fname,
+        })
+        print('Finished downloading {}'.format(fname))
+        return True
+    except:
+        print('Failed downloading {}'.format(fname))
+        return False
+
+