Try to recover from failed quadrature.

Former-commit-id: 746709b224d825baf4a41ddfcd040a0c80fc45c8
This commit is contained in:
Marek Nečada 2019-11-12 07:18:20 +02:00
parent e4f368f47a
commit 1821a0d8f5
4 changed files with 43 additions and 9 deletions

View File

@ -47,11 +47,13 @@ try:
class DebugFlags(enum.IntFlag): # Should be IntFlag if python version >= 3.6 class DebugFlags(enum.IntFlag): # Should be IntFlag if python version >= 3.6
MISC = QPMS_DBGMSG_MISC MISC = QPMS_DBGMSG_MISC
THREADS = QPMS_DBGMSG_THREADS THREADS = QPMS_DBGMSG_THREADS
INTEGRATION = QPMS_DBGMSG_INTEGRATION
has_IntFlag = True has_IntFlag = True
except AttributeError: # For old versions of enum, use IntEnum instead except AttributeError: # For old versions of enum, use IntEnum instead
class DebugFlags(enum.IntEnum): class DebugFlags(enum.IntEnum):
MISC = QPMS_DBGMSG_MISC MISC = QPMS_DBGMSG_MISC
THREADS = QPMS_DBGMSG_THREADS THREADS = QPMS_DBGMSG_THREADS
INTEGRATION = QPMS_DBGMSG_INTEGRATION
has_IntFlag = False has_IntFlag = False
def dbgmsg_enable(qpms_dbgmsg_flags types): def dbgmsg_enable(qpms_dbgmsg_flags types):

View File

@ -146,6 +146,7 @@ cdef extern from "qpms_error.h":
ctypedef enum qpms_dbgmsg_flags: ctypedef enum qpms_dbgmsg_flags:
QPMS_DBGMSG_MISC QPMS_DBGMSG_MISC
QPMS_DBGMSG_THREADS QPMS_DBGMSG_THREADS
QPMS_DBGMSG_INTEGRATION
qpms_dbgmsg_flags qpms_dbgmsg_enable(qpms_dbgmsg_flags types) qpms_dbgmsg_flags qpms_dbgmsg_enable(qpms_dbgmsg_flags types)
qpms_dbgmsg_flags qpms_dbgmsg_disable(qpms_dbgmsg_flags types) qpms_dbgmsg_flags qpms_dbgmsg_disable(qpms_dbgmsg_flags types)

View File

@ -29,7 +29,8 @@ void qpms_pr_debug_at_flf(const char *filename, unsigned int linenum,
typedef enum { typedef enum {
QPMS_DBGMSG_MISC = 1, QPMS_DBGMSG_MISC = 1,
QPMS_DBGMSG_THREADS = 2 // Multithreading-related debug messages. QPMS_DBGMSG_THREADS = 2, // Multithreading-related debug messages.
QPMS_DBGMSG_INTEGRATION = 4 // Quadrature-related debug messages.
} qpms_dbgmsg_flags; } qpms_dbgmsg_flags;
void qpms_debug_at_flf(const char *filename, unsigned int linenum, void qpms_debug_at_flf(const char *filename, unsigned int linenum,

View File

@ -25,6 +25,8 @@
#include <errno.h> #include <errno.h>
#include <pthread.h> #include <pthread.h>
#include <gsl/gsl_integration.h> #include <gsl/gsl_integration.h>
#include <gsl/gsl_errno.h>
#include "optim.h"
// These are quite arbitrarily chosen constants for the quadrature in qpms_tmatrix_axialsym_fill() // These are quite arbitrarily chosen constants for the quadrature in qpms_tmatrix_axialsym_fill()
#define TMATRIX_AXIALSYM_INTEGRAL_EPSREL (1e-5) #define TMATRIX_AXIALSYM_INTEGRAL_EPSREL (1e-5)
@ -712,6 +714,38 @@ struct qpms_tmatrix_axialsym_fill_integration_thread_arg {
pthread_mutex_t *i2start_mutex; pthread_mutex_t *i2start_mutex;
}; };
// This wrapper retries to calculate the T-matrix integral and if it fails, reduces torelance.
static inline int axint_wrapper(const gsl_function *f, double epsabs, double epsrel,
size_t intlimit, gsl_integration_workspace *w, double *result, double *abserr) {
static int tol_exceeded = 0;
double errfac;
int retval;
for (errfac = 1; epsabs*errfac < 0.01 && epsrel*errfac < 0.01; errfac *= 8) {
retval = gsl_integration_qag(f, 0, M_PI, epsabs*errfac, epsrel*errfac,
intlimit, 2, w, result, abserr);
if(QPMS_LIKELY(!retval)) {
if(QPMS_LIKELY(errfac == 1))
return 0;
else {
QPMS_DEBUG(QPMS_DBGMSG_INTEGRATION,
"T-matrix quadrature succeeded only after %g-fold"
"tolerance reduction."); // TODO more details here?
return GSL_EROUND;
}
}
else if(retval == GSL_EROUND) {
if (!tol_exceeded) {
QPMS_WARN("T-matrix quadrature failed; retrying with worse tolerances."
" (this warning will be shown only once).");
}
tol_exceeded += 1;
}
else QPMS_ENSURE_SUCCESS(retval);
}
QPMS_PR_ERROR("T-matrix quadrature failed even after harsh tolerance reduction.");
}
static void * qpms_tmatrix_axialsym_fill_integration_thread(void *arg) { static void * qpms_tmatrix_axialsym_fill_integration_thread(void *arg) {
const struct qpms_tmatrix_axialsym_fill_integration_thread_arg *a = arg; const struct qpms_tmatrix_axialsym_fill_integration_thread_arg *a = arg;
struct tmatrix_axialsym_integral_param_t p = *a->p_template; struct tmatrix_axialsym_integral_param_t p = *a->p_template;
@ -749,27 +783,23 @@ static void * qpms_tmatrix_axialsym_fill_integration_thread(void *arg) {
// Re(R') // Re(R')
p.btype = QPMS_BESSEL_REGULAR; p.btype = QPMS_BESSEL_REGULAR;
p.realpart = true; p.realpart = true;
QPMS_ENSURE_SUCCESS(gsl_integration_qag(&f, 0, M_PI, a->epsabs, a->epsrel, axint_wrapper(&f, a->epsabs, a->epsrel, a->intlimit, w, &result, &abserr);
a->intlimit, 2, w, &result, &abserr));
a->R[iQR] = result; a->R[iQR] = result;
// Im(R') // Im(R')
p.realpart = false; p.realpart = false;
QPMS_ENSURE_SUCCESS(gsl_integration_qag(&f, 0, M_PI, a->epsabs, a->epsrel, axint_wrapper(&f, a->epsabs, a->epsrel, a->intlimit, w, &result, &abserr);
a->intlimit, 2, w, &result, &abserr));
a->R[iQR] += I*result; a->R[iQR] += I*result;
// Re(Q') // Re(Q')
p.btype = QPMS_HANKEL_PLUS; p.btype = QPMS_HANKEL_PLUS;
p.realpart = true; p.realpart = true;
QPMS_ENSURE_SUCCESS(gsl_integration_qag(&f, 0, M_PI, a->epsabs, a->epsrel, axint_wrapper(&f, a->epsabs, a->epsrel, a->intlimit, w, &result, &abserr);
a->intlimit, 2, w, &result, &abserr));
a->Q[iQR] = result; a->Q[iQR] = result;
// Im(Q') // Im(Q')
p.realpart = false; p.realpart = false;
QPMS_ENSURE_SUCCESS(gsl_integration_qag(&f, 0, M_PI, a->epsabs, a->epsrel, axint_wrapper(&f, a->epsabs, a->epsrel, a->intlimit, w, &result, &abserr);
a->intlimit, 2, w, &result, &abserr));
a->Q[iQR] += I*result; a->Q[iQR] += I*result;
} }
} }