From 4efea17ca60f4f1b70cfa7440188195e9e528fb9 Mon Sep 17 00:00:00 2001 From: James Harle <jdha@noc.ac.uk> Date: Wed, 24 Apr 2019 12:27:54 +0100 Subject: [PATCH] Solving bug with vector quantities (Issue 3, jdha/PyNEMO) --- pynemo/utils/nemo_bdy_lib.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/pynemo/utils/nemo_bdy_lib.py b/pynemo/utils/nemo_bdy_lib.py index 4ee96f4..2ba1215 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 + -- GitLab