r4228: Interpolation fixes

[ctsim.git] / include / interpolator.h

diff --git a/include/interpolator.h b/include/interpolator.h
index 6b3469ba01a4e4674352473beff7024880dfd5c1..a3dd43a403ed64a82668f86307790864f8bdb2b5 100644 (file)
--- a/include/interpolator.h
+++ b/include/interpolator.h
@@ -2,7 +2,7 @@
 **  This is part of the CTSim program
 **  Copyright (c) 1983-2001 Kevin Rosenberg
 **
-**  $Id: interpolator.h,v 1.8 2003/01/30 21:53:16 kevin Exp $
+**  $Id: interpolator.h,v 1.9 2003/03/23 18:37:42 kevin Exp $
 **
 **  This program is free software; you can redistribute it and/or modify
 **  it under the terms of the GNU General Public License (version 2) as
@@ -62,30 +62,98 @@ public:
   {}
   
   T interpolate (double dXPos, double dYPos)
-  {
-    int iFloorX = static_cast<int>(floor(dXPos));
-    int iFloorY = static_cast<int>(floor (dYPos));
-    double dXFrac = dXPos - iFloorX;
-    double dYFrac = dYPos - iFloorY;
-
-    T result = 0;
-
-    if (iFloorX < 0 || iFloorY < 0 || iFloorX > m_nx-1 || iFloorY > m_ny-1)
-      result = 0;
-    else if (iFloorX == m_nx - 1 && iFloorY == m_ny - 1)
+{
+  int iFloorX = static_cast<int>(floor(dXPos));
+  int iFloorY = static_cast<int>(floor (dYPos));
+  double dXFrac = dXPos - iFloorX;
+  double dYFrac = dYPos - iFloorY;
+  
+  T result = 0;
+  
+  if (iFloorX < 0 || iFloorY < 0 || iFloorX > m_nx-1 || iFloorY > m_ny-1)
+    result = 0;
+  else if (iFloorX == m_nx - 1 && iFloorY == m_ny - 1)
       result = static_cast<T>(m_ppMatrix[m_nx-1][m_ny-1]);
-    else if (iFloorX == m_nx - 1)
-      result = static_cast<T>(m_ppMatrix[iFloorX][iFloorY] + dYFrac * (m_ppMatrix[iFloorX][iFloorY+1] - m_ppMatrix[iFloorX][iFloorY]));
+  else if (iFloorX == m_nx - 1)
+    result = static_cast<T>(m_ppMatrix[iFloorX][iFloorY] + dYFrac * (m_ppMatrix[iFloorX][iFloorY+1] - m_ppMatrix[iFloorX][iFloorY]));
     else if (iFloorY == m_ny - 1)
       result = static_cast<T>(m_ppMatrix[iFloorX][iFloorY] + dXFrac * (m_ppMatrix[iFloorX+1][iFloorY] - m_ppMatrix[iFloorX][iFloorY]));
+  else
+    result = static_cast<T>
+      ((1 - dXFrac) * (1 - dYFrac) * m_ppMatrix[iFloorX][iFloorY] + 
+       dXFrac * (1 - dYFrac) * m_ppMatrix[iFloorX+1][iFloorY] + 
+       dYFrac * (1 - dXFrac) * m_ppMatrix[iFloorX][iFloorY+1] +
+       dXFrac * dYFrac * m_ppMatrix[iFloorX+1][iFloorY+1]);
+  
+  return result;
+}
+    };
+
+
+template<class T>
+class BilinearPolarInterpolator {
+private:
+  T** const m_ppMatrix;
+  const int m_nAngle;
+  const int m_nPos;
+  int m_nCenterPos;
+  
+public:
+  BilinearPolarInterpolator (T** ppMatrix, unsigned int nAngle,
+                            unsigned int nPos)
+  : m_ppMatrix(ppMatrix), m_nAngle(nAngle), m_nPos(nPos)
+  {
+    if (m_nPos %2)
+      m_nCenterPos = (m_nPos - 1) / 2;
     else
-      result = static_cast<T>((1 - dXFrac) * (1 - dYFrac) * m_ppMatrix[iFloorX][iFloorY] + 
-        dXFrac * (1 - dYFrac) * m_ppMatrix[iFloorX+1][iFloorY] + 
-        dYFrac * (1 - dXFrac) * m_ppMatrix[iFloorX][iFloorY+1] +
-        dXFrac * dYFrac * m_ppMatrix[iFloorX+1][iFloorY+1]);
+      m_nCenterPos = m_nPos / 2;
+  }
 
-    return result;
+  T interpolate (double dAngle, double dPos)
+{
+  int iFloorAngle = static_cast<int>(floor(dAngle));
+  int iFloorPos = static_cast<int>(floor (dPos));
+  double dAngleFrac = dAngle - iFloorAngle;
+  double dPosFrac = dPos - iFloorPos;
+  
+  T result = 0;
+  
+  if (iFloorAngle < -1 || iFloorPos < 0 || iFloorAngle > m_nAngle-1 || iFloorPos > m_nPos-1)
+    result = 0;
+  else if (iFloorAngle == -1 && iFloorPos == m_nPos-1)
+    result = static_cast<T>(m_ppMatrix[0][m_nPos-1] + dAngleFrac * (m_ppMatrix[m_nAngle-1][iFloorPos] - m_ppMatrix[0][iFloorPos]));
+  else if (iFloorAngle == m_nAngle - 1 && iFloorPos == m_nPos-1)
+    result = static_cast<T>(m_ppMatrix[m_nAngle-1][m_nPos-1] + dAngleFrac * (m_ppMatrix[0][iFloorPos] - m_ppMatrix[m_nAngle-1][iFloorPos]));
+  else if (iFloorPos == m_nPos - 1)
+    result = static_cast<T>(m_ppMatrix[iFloorAngle][iFloorPos] + dAngleFrac * (m_ppMatrix[iFloorAngle+1][iFloorPos] - m_ppMatrix[iFloorAngle][iFloorPos]));
+  else {
+    if (iFloorAngle == m_nAngle-1) {
+      int iUpperAngle = 0;
+      int iLowerPos = (m_nPos-1) - iFloorPos;
+      int iUpperPos = (m_nPos-1) - (iFloorPos+1);
+      result = static_cast<T>
+       ((1-dAngleFrac) * (1-dPosFrac) * m_ppMatrix[iFloorAngle][iFloorPos] + 
+        dAngleFrac * (1-dPosFrac) * m_ppMatrix[iUpperAngle][iLowerPos] + 
+        dPosFrac * (1-dAngleFrac) * m_ppMatrix[iFloorAngle][iFloorPos+1] +
+        dAngleFrac * dPosFrac * m_ppMatrix[iUpperAngle][iUpperPos]);
+    } else if (iFloorAngle == -1) {
+      int iLowerAngle = m_nAngle - 1;
+      int iLowerPos = (m_nPos-1) - iFloorPos;
+      int iUpperPos = (m_nPos-1) - (iFloorPos+1);
+      result = static_cast<T>
+       ((1-dAngleFrac) * (1-dPosFrac) * m_ppMatrix[iLowerAngle][iLowerPos] + 
+        dAngleFrac * (1-dPosFrac) * m_ppMatrix[iFloorAngle+1][iFloorPos] + 
+        dPosFrac * (1-dAngleFrac) * m_ppMatrix[iLowerAngle][iUpperPos] +
+        dAngleFrac * dPosFrac * m_ppMatrix[iFloorAngle+1][iFloorPos+1]);
+    } else
+      result = static_cast<T>
+       ((1-dAngleFrac) * (1-dPosFrac) * m_ppMatrix[iFloorAngle][iFloorPos] + 
+        dAngleFrac * (1-dPosFrac) * m_ppMatrix[iFloorAngle+1][iFloorPos] + 
+        dPosFrac * (1-dAngleFrac) * m_ppMatrix[iFloorAngle][iFloorPos+1] +
+        dAngleFrac * dPosFrac * m_ppMatrix[iFloorAngle+1][iFloorPos+1]);
   }
+  return result;
+}
 };