diff --git a/pynemo/utils/nemo_bdy_lib.py b/pynemo/utils/nemo_bdy_lib.py index 4ee96f47f0afe5cb011bd96ebcac8426c1ae9fd3..2ba12156d6db360ded1227ff6d4d17b78b573bbb 100644 --- a/pynemo/utils/nemo_bdy_lib.py +++ b/pynemo/utils/nemo_bdy_lib.py @@ -10,13 +10,16 @@ def sub2ind(shap, subx, suby): ind = (subx * shap[0]) + suby return ind - # THIS FUNCTION MAY BE BROKEN def rot_rep(pxin, pyin, dummy, cd_todo, gcos, gsin): """rotate function""" - if cd_todo.lower() in ['en to i', 'ij to e']: + if cd_todo.lower() == 'en to i': x_var, y_var = pxin, pyin - elif cd_todo.lower() in ['en to j', 'ij to n']: + elif cd_todo.lower() == 'en to j': x_var, y_var = pyin, pxin*-1 + elif cd_todo.lower() == 'ij to e': + x_var, y_var = pxin, pyin*-1 + elif cd_todo.lower() == 'ij to n': + x_var, y_var = pyin, pxin else: raise SyntaxError('rot_rep cd_todo %s is invalid' %cd_todo) return x_var * gcos + y_var * gsin @@ -124,4 +127,4 @@ def bdy_transport(): Keyword arguments: """ raise NotImplementedError - \ No newline at end of file +