1
0
cuberite-2a/source/LinearInterpolation.cpp

252 lines
7.2 KiB
C++
Raw Normal View History

// LinearInterpolation.cpp
// Implements methods for linear interpolation over 1D, 2D and 3D arrays
#include "Globals.h"
#include "LinearInterpolation.h"
/*
// Perform an automatic test upon program start (use breakpoints to debug):
extern void Debug3DNoise(float * a_Noise, int a_SizeX, int a_SizeY, int a_SizeZ, const AString & a_FileNameBase);
class Test
{
public:
Test(void)
{
// DoTest1();
DoTest2();
}
void DoTest1(void)
{
float In[8] = {0, 1, 2, 3, 1, 2, 2, 2};
float Out[3 * 3 * 3];
LinearInterpolate1DArray(In, 4, Out, 9);
LinearInterpolate2DArray(In, 2, 2, Out, 3, 3);
LinearInterpolate3DArray(In, 2, 2, 2, Out, 3, 3, 3);
LOGD("Out[0]: %f", Out[0]);
}
void DoTest2(void)
{
float In[3 * 3 * 3];
for (int i = 0; i < ARRAYCOUNT(In); i++)
{
In[i] = (float)(i % 5);
}
float Out[15 * 16 * 17];
LinearInterpolate3DArray(In, 3, 3, 3, Out, 15, 16, 17);
Debug3DNoise(Out, 15, 16, 17, "LERP test");
}
} gTest;
//*/
// Puts linearly interpolated values from one array into another array. 1D version
void LinearInterpolate1DArray(
float * a_Src,
int a_SrcSizeX,
float * a_Dst,
int a_DstSizeX
)
{
a_Dst[0] = a_Src[0];
int DstSizeXm1 = a_DstSizeX - 1;
int SrcSizeXm1 = a_SrcSizeX - 1;
float fDstSizeXm1 = (float)DstSizeXm1;
float fSrcSizeXm1 = (float)SrcSizeXm1;
for (int x = 1; x < DstSizeXm1; x++)
{
int SrcIdx = x * SrcSizeXm1 / DstSizeXm1;
float ValLo = a_Src[SrcIdx];
float ValHi = a_Src[SrcIdx + 1];
float Ratio = (float)x * fSrcSizeXm1 / fDstSizeXm1 - SrcIdx;
a_Dst[x] = ValLo + (ValHi - ValLo) * Ratio;
}
a_Dst[a_DstSizeX - 1] = a_Src[a_SrcSizeX - 1];
}
// Puts linearly interpolated values from one array into another array. 2D version
void LinearInterpolate2DArray(
float * a_Src,
int a_SrcSizeX, int a_SrcSizeY,
float * a_Dst,
int a_DstSizeX, int a_DstSizeY
)
{
ASSERT(a_DstSizeX > 0);
ASSERT(a_DstSizeX < MAX_INTERPOL_SIZEX);
ASSERT(a_DstSizeY > 0);
ASSERT(a_DstSizeY < MAX_INTERPOL_SIZEY);
// Calculate interpolation ratios and src indices along each axis:
float RatioX[MAX_INTERPOL_SIZEX];
float RatioY[MAX_INTERPOL_SIZEY];
int SrcIdxX[MAX_INTERPOL_SIZEX];
int SrcIdxY[MAX_INTERPOL_SIZEY];
for (int x = 1; x < a_DstSizeX; x++)
{
SrcIdxX[x] = x * (a_SrcSizeX - 1) / (a_DstSizeX - 1);
RatioX[x] = ((float)(x * (a_SrcSizeX - 1)) / (a_DstSizeX - 1)) - SrcIdxX[x];
}
for (int y = 1; y < a_DstSizeY; y++)
{
SrcIdxY[y] = y * (a_SrcSizeY - 1) / (a_DstSizeY - 1);
RatioY[y] = ((float)(y * (a_SrcSizeY - 1)) / (a_DstSizeY - 1)) - SrcIdxY[y];
}
// Special values at the ends. Notice especially the last indices being (size - 2) with ratio set to 1, to avoid index overflow:
SrcIdxX[0] = 0;
RatioX[0] = 0;
SrcIdxY[0] = 0;
RatioY[0] = 0;
SrcIdxX[a_DstSizeX - 1] = a_SrcSizeX - 2;
RatioX[a_DstSizeX - 1] = 1;
SrcIdxY[a_DstSizeY - 1] = a_SrcSizeY - 2;
RatioY[a_DstSizeY - 1] = 1;
// Output all the dst array values using the indices and ratios:
int idx = 0;
for (int y = 0; y < a_DstSizeY; y++)
{
int idxLoY = a_SrcSizeX * SrcIdxY[y];
int idxHiY = a_SrcSizeX * (SrcIdxY[y] + 1);
float ry = RatioY[y];
for (int x = 0; x < a_DstSizeX; x++)
{
// The four src corners of the current "cell":
float LoXLoY = a_Src[SrcIdxX[x] + idxLoY];
float HiXLoY = a_Src[SrcIdxX[x] + 1 + idxLoY];
float LoXHiY = a_Src[SrcIdxX[x] + idxHiY];
float HiXHiY = a_Src[SrcIdxX[x] + 1 + idxHiY];
// Linear interpolation along the X axis:
float InterpXLoY = LoXLoY + (HiXLoY - LoXLoY) * RatioX[x];
float InterpXHiY = LoXHiY + (HiXHiY - LoXHiY) * RatioX[x];
// Linear interpolation along the Y axis:
a_Dst[idx] = InterpXLoY + (InterpXHiY - InterpXLoY) * ry;
idx += 1;
}
}
}
/// Puts linearly interpolated values from one array into another array. 3D version
void LinearInterpolate3DArray(
float * a_Src,
int a_SrcSizeX, int a_SrcSizeY, int a_SrcSizeZ,
float * a_Dst,
int a_DstSizeX, int a_DstSizeY, int a_DstSizeZ
)
{
ASSERT(a_DstSizeX > 0);
ASSERT(a_DstSizeX < MAX_INTERPOL_SIZEX);
ASSERT(a_DstSizeY > 0);
ASSERT(a_DstSizeY < MAX_INTERPOL_SIZEY);
ASSERT(a_DstSizeZ > 0);
ASSERT(a_DstSizeZ < MAX_INTERPOL_SIZEZ);
// Calculate interpolation ratios and src indices along each axis:
float RatioX[MAX_INTERPOL_SIZEX];
float RatioY[MAX_INTERPOL_SIZEY];
float RatioZ[MAX_INTERPOL_SIZEZ];
int SrcIdxX[MAX_INTERPOL_SIZEX];
int SrcIdxY[MAX_INTERPOL_SIZEY];
int SrcIdxZ[MAX_INTERPOL_SIZEZ];
for (int x = 1; x < a_DstSizeX; x++)
{
SrcIdxX[x] = x * (a_SrcSizeX - 1) / (a_DstSizeX - 1);
RatioX[x] = ((float)(x * (a_SrcSizeX - 1)) / (a_DstSizeX - 1)) - SrcIdxX[x];
}
for (int y = 1; y < a_DstSizeY; y++)
{
SrcIdxY[y] = y * (a_SrcSizeY - 1) / (a_DstSizeY - 1);
RatioY[y] = ((float)(y * (a_SrcSizeY - 1)) / (a_DstSizeY - 1)) - SrcIdxY[y];
}
for (int z = 1; z < a_DstSizeZ; z++)
{
SrcIdxZ[z] = z * (a_SrcSizeZ - 1) / (a_DstSizeZ - 1);
RatioZ[z] = ((float)(z * (a_SrcSizeZ - 1)) / (a_DstSizeZ - 1)) - SrcIdxZ[z];
}
// Special values at the ends. Notice especially the last indices being (size - 2) with ratio set to 1, to avoid index overflow:
SrcIdxX[0] = 0;
RatioX[0] = 0;
SrcIdxY[0] = 0;
RatioY[0] = 0;
SrcIdxZ[0] = 0;
RatioZ[0] = 0;
SrcIdxX[a_DstSizeX - 1] = a_SrcSizeX - 2;
RatioX[a_DstSizeX - 1] = 1;
SrcIdxY[a_DstSizeY - 1] = a_SrcSizeY - 2;
RatioY[a_DstSizeY - 1] = 1;
SrcIdxZ[a_DstSizeZ - 1] = a_SrcSizeZ - 2;
RatioZ[a_DstSizeZ - 1] = 1;
// Output all the dst array values using the indices and ratios:
int idx = 0;
for (int z = 0; z < a_DstSizeZ; z++)
{
int idxLoZ = a_SrcSizeX * a_SrcSizeY * SrcIdxZ[z];
int idxHiZ = a_SrcSizeX * a_SrcSizeY * (SrcIdxZ[z] + 1);
float rz = RatioZ[z];
for (int y = 0; y < a_DstSizeY; y++)
{
int idxLoY = a_SrcSizeX * SrcIdxY[y];
int idxHiY = a_SrcSizeX * (SrcIdxY[y] + 1);
float ry = RatioY[y];
for (int x = 0; x < a_DstSizeX; x++)
{
// The eight src corners of the current "cell":
float LoXLoYLoZ = a_Src[SrcIdxX[x] + idxLoY + idxLoZ];
float HiXLoYLoZ = a_Src[SrcIdxX[x] + 1 + idxLoY + idxLoZ];
float LoXHiYLoZ = a_Src[SrcIdxX[x] + idxHiY + idxLoZ];
float HiXHiYLoZ = a_Src[SrcIdxX[x] + 1 + idxHiY + idxLoZ];
float LoXLoYHiZ = a_Src[SrcIdxX[x] + idxLoY + idxHiZ];
float HiXLoYHiZ = a_Src[SrcIdxX[x] + 1 + idxLoY + idxHiZ];
float LoXHiYHiZ = a_Src[SrcIdxX[x] + idxHiY + idxHiZ];
float HiXHiYHiZ = a_Src[SrcIdxX[x] + 1 + idxHiY + idxHiZ];
// Linear interpolation along the Z axis:
float LoXLoYInZ = LoXLoYLoZ + (LoXLoYHiZ - LoXLoYLoZ) * rz;
float HiXLoYInZ = HiXLoYLoZ + (HiXLoYHiZ - HiXLoYLoZ) * rz;
float LoXHiYInZ = LoXHiYLoZ + (LoXHiYHiZ - LoXHiYLoZ) * rz;
float HiXHiYInZ = HiXHiYLoZ + (HiXHiYHiZ - HiXHiYLoZ) * rz;
// Linear interpolation along the Y axis:
float LoXInYInZ = LoXLoYInZ + (LoXHiYInZ - LoXLoYInZ) * ry;
float HiXInYInZ = HiXLoYInZ + (HiXHiYInZ - HiXLoYInZ) * ry;
// Linear interpolation along the X axis:
a_Dst[idx] = LoXInYInZ + (HiXInYInZ - LoXInYInZ) * RatioX[x];
idx += 1;
} // for x
} // for y
} // for z
}