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

feat: add findPeaks Function

Browse Source
This commit is contained in:
kradchen
2025-10-23 15:49:32 +08:00
parent 52da7bcd35
commit 6b16f6e01a
6 changed files with 240 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
test/Function2D_Cuda_Test.cpp
View File

@@ -5,6 +5,7 @@
#include "Function.h"
#include "Matrix.h"
#include "TestUtility.h"
#include "MatlabReader.h"
#include "Function2D.h"
#include "Function2D.cuh"
@@ -18,11 +19,15 @@ protected:
static void TearDownTestCase(){
}
public:
Aurora::Matrix mSignal;
Aurora::CudaMatrix dmSignal;
Aurora::Matrix B;
Aurora::CudaMatrix dB;
void SetUp(){
MatlabReader m("/home/krad/TestData/peaks.mat");
mSignal = m.read("AScan_env_norm");
dmSignal = mSignal.toDeviceMatrix();
}
void TearDown(){
}
@@ -997,3 +1002,17 @@ TEST_F(Function2D_Cuda_Test, hilbert) {
EXPECT_NEAR(ret1[i], ret2.getValue(i), 0.01);
}
}
TEST_F(Function2D_Cuda_Test, findPeaks) {
int* indexs = new int[mSignal.getDimSize(1)*10];
auto ret1 = Aurora::findPeaks(dmSignal,10, 0.2, 0.05,indexs);
auto reH = ret1.toHostMatrix();
for(unsigned int i=0; i<10; ++i)
{
printf("%d,",indexs[i]);
}
delete [] indexs;
}

18
test/Function2D_Test.cpp
View File

@@ -2,6 +2,8 @@
#include <vector>
#include "TestUtility.h"
#include "MatlabReader.h"
#include "Matrix.h"
#include "Function.h"
#include "Function1D.h"
@@ -16,7 +18,11 @@ protected:
}
static void TearDownTestCase(){
}
public:
Aurora::Matrix mSignal;
void SetUp(){
MatlabReader m("/home/krad/TestData/peaks.mat");
mSignal = m.read("AScan_env_norm");
}
void TearDown(){
}
@@ -573,3 +579,15 @@ TEST_F(Function2D_Test, sub2ind) {
}
TEST_F(Function2D_Test, findPeaks) {
int* indexs = new int[mSignal.getDimSize(1)*10];
Aurora::findPeaksHost(mSignal,10, 0.2, 0.05,indexs);
for(unsigned int i=0; i<10; ++i)
{
printf("%d,",indexs[i]);
}
delete [] indexs;
}