diff options
Diffstat (limited to 'src/likelihood.h')
| -rw-r--r-- | src/likelihood.h | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/src/likelihood.h b/src/likelihood.h index bda46f0..d739802 100644 --- a/src/likelihood.h +++ b/src/likelihood.h @@ -19,6 +19,8 @@ #include "event.h" #include <stddef.h> /* for size_t */ +#include <gsl/gsl_interp.h> +#include <gsl/gsl_spline.h> #define PSUP_REFLECTION_TIME 80.0 @@ -80,6 +82,20 @@ typedef struct particle { double range; double *x; double *T; + double *cos_theta; + double *cdf_shower; + /* Spline for looking up cdf -> cos(theta). */ + gsl_spline *spline_shower; + gsl_interp_accel *acc_shower; + double *cdf_delta; + double *x_shower; + double *gamma_pdf; + double xlo_shower; + double xhi_shower; + double pos_a; + double pos_b; + gsl_spline *spline_delta; + gsl_interp_accel *acc_delta; size_t n; double a; double b; @@ -95,6 +111,6 @@ void particle_free(particle *p); double time_pdf(double t, double mu_noise, double mu_indirect, double *mu, size_t n, double *ts, double tmean, double *ts_sigma); double time_cdf(double t, double mu_noise, double mu_indirect, double *mu, size_t n, double *ts, double tmean, double *ts_sigma); double nll_best(event *ev); -double nll(event *ev, vertex *v, size_t n, double dx, double dx_shower, int fast, int charge_only, int hit_only); +double nll(event *ev, vertex *v, size_t n, double dx, int ns, const int fast, int charge_only, int hit_only); #endif |