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

Improve detectTofAndAttMex performance.

Browse Source
This commit is contained in:
kradchen
2023-05-30 13:09:09 +08:00
parent e3000a5989 77d23aaae4
commit 0c4a6e8c4a
1 changed files with 24 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/transmissionReconstruction/detection/detection.cpp
View File

@@ -306,33 +306,35 @@ namespace Recon {
size_t N = _AscanBlock.getDimSize(1);
size_t totalSize = N*M;
{
auto resDetect = new float[N]{0};
auto resEnvelope = new float[totalSize]{0};
auto resEnvelopeRef = new float[totalSize]{0};
auto aScans_r = new float[totalSize]{0};
auto aScansRef_r = new float[totalSize]{0};
{
// auto resDetect = new float[N]{0};
// auto resEnvelope = new float[totalSize]{0};
// auto resEnvelopeRef = new float[totalSize]{0};
// auto aScans_r = new float[totalSize]{0};
// auto aScansRef_r = new float[totalSize]{0};
// {
Matrix _AscanBlock_trim = _AscanBlock.getDimSize(0)!=M?_AscanBlock.block(0, 0, M-1):_AscanBlock;
Matrix _AscanRefBlock_trim = _AscanRefBlock.getDimSize(0)!=M?_AscanRefBlock.block(0, 0, M-1):_AscanRefBlock;
std::copy(_AscanBlock_trim.getData(), _AscanBlock_trim.getData() + totalSize, aScans_r);
std::copy(_AscanRefBlock_trim.getData(), _AscanRefBlock_trim.getData() + totalSize,
aScansRef_r);
}
calculateBankDetectAndHilbertTransformation(
aScans_r, aScansRef_r, N, M, resampleFactor, nthreads,
resDetect, resEnvelope, resEnvelopeRef);
delete [] aScans_r;
delete [] aScansRef_r;
// std::copy(_AscanBlock_trim.getData(), _AscanBlock_trim.getData() + totalSize, aScans_r);
// std::copy(_AscanRefBlock_trim.getData(), _AscanRefBlock_trim.getData() + totalSize,
// aScansRef_r);
// }
resDetectD = Aurora::malloc(N);
std::copy(resDetect, resDetect+ N, resDetectD);
delete [] resDetect;
resEnvelopeD = Aurora::malloc(totalSize);
std::copy(resEnvelope, resEnvelope+ totalSize, resEnvelopeD);
delete [] resEnvelope;
resEnvelopeRefD = Aurora::malloc(totalSize);
std::copy(resEnvelopeRef, resEnvelopeRef+ totalSize, resEnvelopeRefD);
delete [] resEnvelopeRef;
calculateBankDetectAndHilbertTransformation(
_AscanBlock_trim.getData(), _AscanBlock_trim.getData(), N, M, resampleFactor, nthreads,
resDetectD, resEnvelopeD, resEnvelopeRefD);
// delete [] aScans_r;
// delete [] aScansRef_r;
// resDetectD = Aurora::malloc(N);
// std::copy(resDetect, resDetect+ N, resDetectD);
// delete [] resDetect;
// std::copy(resEnvelope, resEnvelope+ totalSize, resEnvelopeD);
// delete [] resEnvelope;
// resEnvelopeRefD = Aurora::malloc(totalSize);
// std::copy(resEnvelopeRef, resEnvelopeRef+ totalSize, resEnvelopeRefD);
// delete [] resEnvelopeRef;
}
auto resDetect =Matrix::New(resDetectD,1,N);
auto resEnvelope =Matrix::New(resEnvelopeD,M,N);