From 58c286a6f2d9d593d46888681bdf33066a21c70a Mon Sep 17 00:00:00 2001
From: tlatorre <tlatorre@uchicago.edu>
Date: Fri, 21 Sep 2018 13:03:14 -0500
Subject: update likelihood function to include the probability of the path
 coefficients

---
 src/likelihood.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/likelihood.c b/src/likelihood.c
index e045512..5f11f95 100644
--- a/src/likelihood.c
+++ b/src/likelihood.c
@@ -334,6 +334,7 @@ double nll_muon(event *ev, double T0, double *pos, double *dir, double t0, doubl
     double tmean = 0.0;
     muon_energy *m;
     double pmt_dir[3], R, cos_theta, theta, wavelength0, n_d2o, theta_cerenkov, s;
+    double logp_path;
 
     double mu_direct[MAX_PMTS];
     double ts[MAX_PMTS];
@@ -502,5 +503,9 @@ double nll_muon(event *ev, double T0, double *pos, double *dir, double t0, doubl
         }
     }
 
-    return kahan_sum(nll,nhit);
+    logp_path = 0.0;
+    for (i = 0; i < n; i++)
+        logp_path += pow(z1[i],2) + pow(z2[i],2);
+
+    return kahan_sum(nll,nhit) + logp_path;
 }
-- 
cgit