update fit to fit for electrons and protons

1 files changed, 13 insertions, 1 deletions

diff --git a/src/likelihood.h b/src/likelihood.h
index d708f19..945fc46 100644
--- a/src/likelihood.h
+++ b/src/likelihood.h
@@ -33,6 +33,18 @@
 #define GTVALID 400.0
 #define BETA_MIN 0.8
 
-double nll_muon(event *ev, double T0, double *pos, double *dir, double t0, double *z1, double *z2, size_t n, double epsrel, int fast);
+typedef struct particle {
+    double mass;
+    double range;
+    double *x;
+    double *T;
+    size_t n;
+} particle;
+
+particle *particle_init(int id, double T0, double rho, size_t n);
+double particle_get_energy(double x, particle *p);
+void particle_free(particle *p);
+double get_expected_charge(double x, double T, double T0, double *pos, double *dir, double *pmt_pos, double *pmt_normal, double r, int reflected);
+double nll_muon(event *ev, int id, double T0, double *pos, double *dir, double t0, double *z1, double *z2, size_t n, double epsrel, int fast);
 
 #endif