LMU/DKS
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added a couple new theory functions (dglktfzf, dglktflf, fmuf).

Browse Source
This commit is contained in:
suter_a
2025-06-10 14:20:45 +02:00
parent 21b4c591b9
commit 9d21fc5400
3 changed files with 67 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
CMakeLists.txt
View File

@ -1,8 +1,8 @@
CMAKE_MINIMUM_REQUIRED (VERSION 3.2)
PROJECT (DKS)
SET (DKS_VERSION_MAJOR 1)
SET (DKS_VERSION_MINOR 1)
SET (DKS_VERSION_PATCH 4)
SET (DKS_VERSION_MINOR 2)
SET (DKS_VERSION_PATCH 0)
set (DKS_VERSION ${DKS_VERSION_MAJOR}.${DKS_VERSION_MINOR}.${DKS_VERSION_PATCH})
SET (PACKAGE \"dks\")
SET (PACKAGE_BUGREPORT \"locans.uldis@psi.ch\")

38
src/CUDA/NVRTCKernels/CudaChiSquareKernel.cu
View File

@ -36,6 +36,25 @@ __device__ double sekt(double t, double lambda) {
return (1.0/3.0) + (2.0/3.0)*(1.0 - lambdat) * exp(-lambdat);
}
__device__ double dglktfzf(double t, double sigma, double hopp) {
double nut = hopp*t;
return exp(-sqrt(4.0*pow(sigma/hopp,2.0)*(exp(-nut)-1.0+nut)));
}
__device__ double dglktflf(double t, double nu0, double sigma, double hopp) {
double w0 = TWO_PI*nu0;
double w0_2 = w0*w0;
double w0_t = w0*t;
double nu_2 = hopp*hopp;
double nu_t = hopp*t;
double Gamma_t = ((w0_2+nu_2)*nu_t+(w0_2-nu_2)*(1.0-exp(-nu_t)*cos(w0_t))-2.0*hopp*w0*exp(-nu_t)*sin(w0_t))/pow(w0_2+nu_2,2.0);
if (Gamma_t < 0.0)
Gamma_t = 0.0;
return exp(-sqrt(4.0*sigma*hopp*Gamma_t));
}
__device__ double lgkt(double t, double lambda, double sigma) {
double lambdat = lambda*t;
double sigmatsq = pow(sigma*t, 2.0);
@ -69,6 +88,12 @@ __device__ double rahf(double t, double nu, double lambda) {
return (1.0/6.0)*(1.0-nuth)*exp(-nuth) + (1.0/3.0)*(1.0-nut/4.0)*exp(-0.25*(nut+2.44949*lamt));
}
__device__ double ab(double t, double sigma, double gamma) {
double gt = gamma*t;
return exp(-pow(sigma/gamma,2.0)*(exp(-gt) - 1.0 + gt));
}
__device__ double tf(double t, double phi, double nu) {
double tmp_nu = TWO_PI*nu*t;
double tmp_phi = DEG_TO_RAD*phi;
@ -144,12 +169,6 @@ __device__ double ib(double t, double alpha, double phi, double nu, double lambd
return alpha*j0(wt+ph)*exp(-lambdaT*t) + (1.0-alpha)*exp(-lambdaL*t);
}
__device__ double ab(double t, double sigma, double gamma) {
double gt = gamma*t;
return exp(-pow(sigma/gamma,2.0)*(exp(-gt) - 1.0 + gt));
}
__device__ double snkzf(double t, double Delta0, double Rb) {
double D0t2 = pow(Delta0*t, 2.0);
double aa = 1.0/(1.0+pow(Rb,2.0)*D0t2);
@ -184,6 +203,13 @@ __device__ double dnktf(double t, double phi, double nu, double Delta0, double R
return sqrt(aa)*exp(-Delta0*Delta0*theta*aa)*cos(wt+ph);
}
__device__ double fmuf(double t, double wd) {
double sqrt3 = sqrt(3.0);
double wd_t = wd*t;
return (3.0+cos(sqrt3*wd_t)+(1.0-1.0/sqrt3)*cos(((3.0-sqrt3)/2.0)*wd_t)+(1.0+1.0/sqrt3)*cos(((3.0 + sqrt3)/2.0)*wd_t))/6.0;
}
/** Theory and chisquare functions.
* Based on the compiler flags set theory is calculated either in single hist mode or asymetric.
* Based on the compiler flags calculate either chisq or MLE

39
src/OpenCL/OpenCLKernels/OpenCLChiSquareRuntime.cl
View File

@ -59,6 +59,25 @@ double sekt(double t, double lambda) {
return (1.0/3.0) + (2.0/3.0)*(1.0 - lambdat) * exp(-lambdat);
}
double dglktfzf(double t, double sigma, double hopp) {
double nut = hopp*t;
return exp(-sqrt(4.0*pow(sigma/hopp,2.0)*(exp(-nut)-1.0+nut)));
}
double dglktflf(double t, double nu0, double sigma, double hopp) {
double w0 = TWO_PI*nu0;
double w0_2 = w0*w0;
double w0_t = w0*t;
double nu_2 = hopp*hopp;
double nu_t = hopp*t;
double Gamma_t = ((w0_2+nu_2)*nu_t+(w0_2-nu_2)*(1.0-exp(-nu_t)*cos(w0_t))-2.0*hopp*w0*exp(-nu_t)*sin(w0_t))/pow(w0_2+nu_2,2.0);
if (Gamma_t < 0.0)
Gamma_t = 0.0;
return exp(-sqrt(4.0*sigma*hopp*Gamma_t));
}
double lgkt(double t, double lambda, double sigma) {
double lambdat = lambda*t;
double sigmatsq = pow(sigma*t, 2.0);
@ -92,6 +111,12 @@ double rahf(double t, double nu, double lambda) {
return (1.0/6.0)*(1.0-nuth)*exp(-nuth) + (1.0/3.0)*(1.0-nut/4.0)*exp(-0.25*(nut+2.44949*lamt));
}
double ab(double t, double sigma, double gamma) {
double gt = gamma*t;
return exp(-pow(sigma/gamma,2.0)*(exp(-gt) - 1.0 + gt));
}
double tf(double t, double phi, double nu) {
double tmp_nu = TWO_PI*nu*t;
double tmp_phi = DEG_TO_RAD * phi;
@ -167,12 +192,6 @@ double ib(double t, double alpha, double phi, double nu, double lambdaT, double
return alpha*bessj0(wt+ph)*exp(-lambdaT*t) + (1.0-alpha)*exp(-lambdaL*t);
}
double ab(double t, double sigma, double gamma) {
double gt = gamma*t;
return exp(-pow(sigma/gamma,2.0)*(exp(-gt) - 1.0 + gt));
}
double snkzf(double t, double Delta0, double Rb) {
double D0t2 = pow(Delta0*t, 2.0);
double aa = 1.0/(1.0+pow(Rb,2.0)*D0t2);
@ -207,6 +226,14 @@ double dnktf(double t, double phi, double nu, double Delta0, double Rb, double n
return sqrt(aa)*exp(-Delta0*Delta0*theta*aa)*cos(wt+ph);
}
double fmuf(double t, double wd) {
double sqrt3 = sqrt(3.0);
double wd_t = wd*t;
return (3.0+cos(sqrt3*wd_t)+(1.0-1.0/sqrt3)*cos(((3.0-sqrt3)/2.0)*wd_t)+(1.0+1.0/sqrt3)*cos(((3.0 + sqrt3)/2.0)*wd_t))/6.0;
}
__kernel void kernelChiSquareSingleHisto(__global double *data, __global double *err,
__global double *par, __global double *chisq, __global int *map, __global double *funcv,
int length, int numpar, int numfunc, int nummap,