aboutsummaryrefslogtreecommitdiff
path: root/src/likelihood.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/likelihood.h')
-rw-r--r--src/likelihood.h16
1 files changed, 15 insertions, 1 deletions
diff --git a/src/likelihood.h b/src/likelihood.h
index be003d3..921a97d 100644
--- a/src/likelihood.h
+++ b/src/likelihood.h
@@ -52,6 +52,20 @@
* quantity from simulation. */
#define PHOTONS_PER_MEV 400.0
+/* Maximum number of vertices to fit. */
+#define MAX_VERTICES 2
+
+typedef struct vertex {
+ int id;
+ double T0;
+ double pos[3];
+ double dir[3];
+ double t0;
+ double z1[10];
+ double z2[10];
+ size_t n;
+} vertex;
+
typedef struct particle {
int id;
double mass;
@@ -69,6 +83,6 @@ double particle_get_energy(double x, particle *p);
void particle_free(particle *p);
double time_pdf(double t, double mu_noise, double mu_indirect, double *mu_direct, double *mu_shower, size_t n, double *ts, double *ts_shower, double tmean, double sigma, double *ts_sigma);
double time_cdf(double t, double mu_noise, double mu_indirect, double *mu_direct, double *mu_shower, size_t n, double *ts, double *ts_shower, double tmean, double sigma, double *ts_sigma);
-double nll_muon(event *ev, int id, double T0, double *pos, double *dir, double t0, double *z1, double *z2, size_t n, double dx, double dx_shower, int fast);
+double nll_muon(event *ev, vertex *v, size_t n, double dx, double dx_shower, int fast);
#endif