From 8373f12eaf63b6db27a1848202a86aeeb15c4d26 Mon Sep 17 00:00:00 2001
From: John Glover <john@john-glovers-macbook.local>
Date: Tue, 21 Dec 2010 11:25:11 +0000
Subject: Fixing bugs in SMSResidual.

---
 sms/sms.i | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'sms/sms.i')

diff --git a/sms/sms.i b/sms/sms.i
index a61d4b8..afde07a 100644
--- a/sms/sms.i
+++ b/sms/sms.i
@@ -500,6 +500,32 @@
     }
 }
 
+%extend SMS_ResidualParams
+{
+    void getResidual(int sizeArray, sfloat *pArray)
+    {
+        if(sizeArray < $self->residualSize)
+        {
+            sms_error("numpy array not big enough");
+            return;
+        }
+        int i;
+        for(i = 0; i < $self->residualSize; i++)
+            pArray[i] = $self->residual[i];
+    }
+    void getApprox(int sizeArray, sfloat *pArray)
+    {
+        if(sizeArray < $self->nCoeffs)
+        {
+            sms_error("numpy array not big enough");
+            return;
+        }
+        int i;
+        for(i = 0; i < $self->nCoeffs; i++)
+            pArray[i] = $self->approxEnvelope[i];
+    }
+}
+
 %extend SMS_ModifyParams
 {
     /* no need to return an error code, if sms_error is called, it will throw an exception in python */
-- 
cgit v1.2.3