back-integrate DKS Fourier after DKS upgrade and modularization. Not yet tested other than compilation.

This commit is contained in:
2017-05-29 17:54:57 +02:00
parent c2e362d406
commit 2b7a391bbb
2 changed files with 3 additions and 54 deletions

View File

@@ -8,7 +8,7 @@
***************************************************************************/
/***************************************************************************
* Copyright (C) 2007-2016 by Andreas Suter *
* Copyright (C) 2007-2017 by Andreas Suter *
* andreas.suter@psi.ch *
* *
* This program is free software; you can redistribute it and/or modify *
@@ -298,12 +298,10 @@ PFourier::PFourier(TH1F *data, Int_t unitTag, Double_t startTime, Double_t endTi
fIn = 0;
fOut = 0;
/*//as
#ifdef HAVE_DKS
fInDKS = 0;
fOutDKS = 0;
#endif
*/
SetUseFFTW(useFFTW);
@@ -365,13 +363,11 @@ PFourier::PFourier(TH1F *data, Int_t unitTag, Double_t startTime, Double_t endTi
fIn = (fftw_complex *)fftw_malloc(sizeof(fftw_complex)*fNoOfBins);
fOut = (fftw_complex *)fftw_malloc(sizeof(fftw_complex)*fNoOfBins);
} else { // try DKS
/*//as
#ifdef HAVE_DKS
fInDKS = new double[fNoOfBins];
unsigned int size=fNoOfBins/2+1;
fOutDKS = new complex<double>[size];
#endif
*/
}
// check if memory allocation has been successful
@@ -381,17 +377,12 @@ PFourier::PFourier(TH1F *data, Int_t unitTag, Double_t startTime, Double_t endTi
return;
}
} else { // try DKS
/*//as
#ifdef HAVE_DKS
if ((fInDKS == 0) || (fOutDKS == 0)) {
fValid = false;
return;
}
#else
fValid = false;
return;
#endif
*/
fValid = false;
return;
}
@@ -405,7 +396,6 @@ PFourier::PFourier(TH1F *data, Int_t unitTag, Double_t startTime, Double_t endTi
fValid = false;
}
} else { // try DKS
/*//as
#ifdef HAVE_DKS
// init DKSBase
fDks.setAPI("Cuda", 4);
@@ -430,10 +420,7 @@ PFourier::PFourier(TH1F *data, Int_t unitTag, Double_t startTime, Double_t endTi
}
if ((fReal_ptr==0) || (fComp_ptr==0))
fValid = false;
#else
fValid = false;
#endif
*/
fValid = false;
}
}
@@ -454,7 +441,6 @@ PFourier::~PFourier()
if (fOut)
fftw_free(fOut);
} else {
/*//as
#ifdef HAVE_DKS
// free accelerator memory
fDks.freeMemory<double>(fReal_ptr, (int)fNoOfBins);
@@ -465,7 +451,6 @@ PFourier::~PFourier()
if (fOut)
delete [] fOutDKS;
#endif
*/
}
}
@@ -488,7 +473,6 @@ void PFourier::Transform(UInt_t apodizationTag)
if (fUseFFTW) {
fftw_execute(fFFTwPlan);
} else {
/*//as
#ifdef HAVE_DKS
int dimsize[3] = {(int)fNoOfBins, 1, 1};
int status=0, size=fNoOfBins/2+1;
@@ -498,11 +482,7 @@ void PFourier::Transform(UInt_t apodizationTag)
status = fDks.callR2CFFT(fReal_ptr, fComp_ptr, 1, dimsize);
// read data from accelerator
status = fDks.readData< complex<double> >(fComp_ptr, fOutDKS, size);
#else
fValid = false;
return;
#endif
*/
fValid = false;
return;
}
@@ -527,7 +507,6 @@ void PFourier::Transform(UInt_t apodizationTag)
fOut[i][1] = im;
}
} else { // try DKS
/*//as
UInt_t size=fNoOfBins/2+1;
for (UInt_t i=0; i<size; i++) {
phase = 2.0*PI/(fTimeResolution*fNoOfBins) * i * shiftTime;
@@ -537,7 +516,6 @@ void PFourier::Transform(UInt_t apodizationTag)
fOutDKS[i] = complex<double>(re, im);
#endif
}
*/
}
}
@@ -609,15 +587,11 @@ TH1F* PFourier::GetRealFourier(const Double_t scale)
}
} else {
for (UInt_t i=0; i<noOfFourierBins; i++) {
/*//as
#ifdef HAVE_DKS
realFourier->SetBinContent(i+1, scale*fOutDKS[i].real());
#else
realFourier->SetBinContent(i+1, PMUSR_UNDEFINED);
#endif
*/
realFourier->SetBinContent(i+1, PMUSR_UNDEFINED); //as needs to be removed
realFourier->SetBinError(i+1, 0.0);
}
}
return realFourier;
@@ -761,15 +735,11 @@ TH1F* PFourier::GetImaginaryFourier(const Double_t scale)
}
} else {
for (UInt_t i=0; i<noOfFourierBins; i++) {
/*//as
#ifdef HAVE_DKS
imaginaryFourier->SetBinContent(i+1, scale*fOutDKS[i].imag());
#else
imaginaryFourier->SetBinContent(i+1, PMUSR_UNDEFINED);
#endif
*/
imaginaryFourier->SetBinContent(i+1, PMUSR_UNDEFINED); //as needs to be removed
imaginaryFourier->SetBinError(i+1, 0.0);
}
}
return imaginaryFourier;
@@ -812,15 +782,11 @@ TH1F* PFourier::GetPowerFourier(const Double_t scale)
}
} else {
for (UInt_t i=0; i<noOfFourierBins; i++) {
/*//as
#ifdef HAVE_DKS
pwrFourier->SetBinContent(i+1, scale*sqrt(fOutDKS[i].real()*fOutDKS[i].real()+fOutDKS[i].imag()*fOutDKS[i].imag()));
#else
pwrFourier->SetBinContent(i+1, PMUSR_UNDEFINED);
#endif
*/
pwrFourier->SetBinContent(i+1, PMUSR_UNDEFINED); //as needs to be removed
pwrFourier->SetBinError(i+1, 0.0);
}
}
return pwrFourier;
@@ -863,7 +829,6 @@ TH1F* PFourier::GetPhaseFourier(const Double_t scale)
re = fOut[i][0];
im = fOut[i][1];
} else {
/*//as
#ifdef HAVE_DKS
re = fOutDKS[i].real();
im = fOutDKS[i].imag();
@@ -871,9 +836,6 @@ TH1F* PFourier::GetPhaseFourier(const Double_t scale)
re = 1.0;
im = 0.0;
#endif
*/
re = 1.0; //as needs to be removed
im = 0.0; //as needs to be removed
}
// calculate the phase
if (re == 0) {
@@ -946,18 +908,14 @@ void PFourier::PrepareFFTwInputData(UInt_t apodizationTag)
}
} else {
for (UInt_t i=0; i<fNoOfData-t0bin; i++) {
/*//as
#ifdef HAVE_DKS
fInDKS[i] = fData->GetBinContent(i+t0bin) - mean;
#endif
*/
}
for (UInt_t i=fNoOfData-t0bin; i<fNoOfBins; i++) {
/*//as
#ifdef HAVE_DKS
fInDKS[i] = 0.0;
#endif
*/
}
}
@@ -1021,11 +979,9 @@ void PFourier::ApodizeData(Int_t apodizationTag) {
if (fUseFFTW) {
fIn[i][0] *= q;
} else {
/*//as
#ifdef HAVE_DKS
fInDKS[i] *= q;
#endif
*/
}
}
}