sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add interp and repmat function.

Browse Source
This commit is contained in:
sunwen
2023-04-20 15:34:38 +08:00
parent 9c0667a65a
commit d73dc8be23
8 changed files with 245 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/Function3D.cpp
View File

@@ -1 +1,46 @@
#include "Function3D.h"
#include "Function2D.h"
#include "Function.h"
using namespace Aurora;
Matrix Aurora::interp3(const Matrix& aX, const Matrix& aY, const Matrix& aZ, const Matrix& aV, const Matrix& aX1, const Matrix& aY1, const Matrix& aZ1,InterpnMethod aMethod)
{
if (aV.getDims() != 3)
{
return Matrix();
}
if ( aX1.getDimSize(0) != aY1.getDimSize(0) ||
aX1.getDimSize(0) != aZ1.getDimSize(0) ||
aZ1.getDimSize(0) != aY1.getDimSize(0))
{
return Matrix();
}
int nx1 = aX1.getDimSize(0);
int zAxisNum = aV.getDimSize(2);
int columnNum = aV.getDimSize(1);
int rowNum = aV.getDimSize(0);
if(aX.getDimSize(0) != columnNum || aY.getDimSize(0) != rowNum || aZ.getDimSize(0) != zAxisNum)
{
return Matrix();
}
std::shared_ptr<double> resultData = std::shared_ptr<double>(new double[nx1], std::default_delete<double[]>());
for (int i = 0; i < nx1; ++i)
{
std::shared_ptr<double> zResultData = std::shared_ptr<double>(new double[zAxisNum], std::default_delete<double[]>());
for(int j =0; j < zAxisNum; ++j)
{
zResultData.get()[j] = interp2(aX,aY,aV($,$,j).toMatrix(),aX1(i).toMatrix(),aY1(i).toMatrix(),aMethod).getData()[0];
}
Matrix xResult(zResultData,std::vector<int>{columnNum});
resultData.get()[i] = interp1(aZ,xResult,aZ1(i).toMatrix(),aMethod).getData()[0];
}
return Matrix(resultData,std::vector<int>{nx1});
}
Matrix Aurora::interpn(const Matrix& aX, const Matrix& aY, const Matrix& aZ, const Matrix& aV, const Matrix& aX1, const Matrix& aY1, const Matrix& aZ1,InterpnMethod aMethod)
{
return Aurora::interp3(aY,aX,aZ,aV,aY1,aX1,aZ1,aMethod);
}