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