add ability to fit for multiple vertices

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