1 files changed, 59 insertions, 0 deletions
diff --git a/openEMS/python/openEMS/_nf2ff.pyx b/openEMS/python/openEMS/_nf2ff.pyx
new file mode 100644
index 0000000..8016694
--- /dev/null
+++ b/openEMS/python/openEMS/_nf2ff.pyx
@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2015,20016 Thorsten Liebig (Thorsten.Liebig@gmx.de)
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+cimport _nf2ff
+import numpy as np
+import os
+from CSXCAD.Utilities import CheckNyDir
+
+cdef class _nf2ff:
+    def __cinit__(self, freq, theta, phi, center, numThreads=0, **kw):
+        if type(freq) in [float, int]:
+            freq = list(float(freq))
+        if type(theta) in [float, int]:
+            theta = list(float(theta))
+        if type(phi) in [float, int]:
+            phi = list(float(phi))
+        self.thisptr = new cpp_nf2ff(freq, theta, phi, center, numThreads)
+
+        if 'verbose' in kw:
+            self.SetVerboseLevel(kw['verbose'])
+            del kw['verbose']
+
+        assert len(kw)==0, 'Unknown keyword(s): {}'.format(kw)
+
+    def AnalyseFile(self, e_file, h_file):
+        assert os.path.exists(e_file)
+        assert os.path.exists(h_file)
+        return self.thisptr.AnalyseFile(e_file.encode('UTF-8'), h_file.encode('UTF-8'))
+
+    def SetMirror(self, mirr_type, ny, pos):
+        if mirr_type<=0:
+            return
+        assert mirr_type<3
+        ny = CheckNyDir(ny)
+        self.thisptr.SetMirror(mirr_type, ny, pos)
+
+    def SetRadius(self, radius):
+        self.thisptr.SetRadius(radius)
+
+    def Write2HDF5(self, filename):
+        return self.thisptr.Write2HDF5(filename.encode('UTF-8'))
+
+    def SetVerboseLevel(self, level):
+        self.thisptr.SetVerboseLevel(level)