From bb3e0630af885961d339e531197c614942907194 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Ne=C4=8Dada?= <marek.necada@aalto.fi>
Date: Wed, 19 Jul 2017 17:01:22 +0300
Subject: [PATCH] introduce TMatrix class

Former-commit-id: 1c6dea1510a91a498f4abc8fa47dd6f95f397da3
---
 qpms/tmatrices.py | 26 +++++++++++++++++++++++++-
 qpms/types.py     |  6 ------
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/qpms/tmatrices.py b/qpms/tmatrices.py
index 42c8f15..eb31a31 100644
--- a/qpms/tmatrices.py
+++ b/qpms/tmatrices.py
@@ -3,8 +3,9 @@ import quaternion, spherical_functions as sf # because of the Wigner matrices. T
 import re
 from scipy import interpolate
 from scipy.constants import hbar, e as eV, pi, c
-from qpms_c import get_mn_y#, get_nelem # TODO IMPORT get_nelem INTO THE FINAL MODULE
+from qpms_c import get_mn_y, get_nelem
 ň = np.newaxis
+from .types import NormalizationT
 
 # Transformations of spherical bases
 def WignerD_mm(l, quat):
@@ -335,3 +336,26 @@ def get_TMatrix_fromspec(tmatrix_spec):
         else:
             raise ValueError('not implemented: ', optype)
     return (TMatrices, freqs, lMax)
+
+class TMatrix(object):
+    def __init__(self, tmatrix_spec):
+        self.specification = tmatrix_spec
+        self.tmdata, self.freqs, self.lMax = get_TMatrix_fromspec(tmatrix_spec)
+        self.nelem = get_nelem(self.lMax) 
+        #self._interpolators = dict()
+        self.default_interpolator = interpolate.interp1d(self.freqs, 
+            self.tmdata, axis=0, kind='linear', fill_value='extrapolate')
+        self.normalization = NormalizationT.TAYLOR # TODO others are not supported by the loading functions
+    
+    def atfreq(self, freq):
+        freqarray = np.array(freq, copy=False)
+        if freqarray.shape: # not just a scalar
+            tm_interp = np.empty(freqarray.shape + (2, self.nelem, 2, self.nelem), dtype=np.complex_)
+            for i in np.ndindex(freqarray.shape):
+                tm_interp[i] = self.default_interpolator(freqarray[i])
+            return tm_interp
+        else: # scalar
+            return self.default_interpolator(freq)
+
+    __getitem__ = atfreq # might be changed later, use atfreq to be sure
+
diff --git a/qpms/types.py b/qpms/types.py
index 402808c..a335657 100644
--- a/qpms/types.py
+++ b/qpms/types.py
@@ -70,9 +70,3 @@ of (integer) coordinate tuples indicating which scatterers are to be included
 to the finite sublattice.
 """
 
-
-class TMatrix(object):
-    pass
-
-
-