X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pom.git;a=blobdiff_plain;f=pom_solver.h;fp=pom_solver.h;h=b0624b40063d55eb4d218a8402efe15a439fc160;hp=0000000000000000000000000000000000000000;hb=97a7e68f234cc09807d2d55f550e2516be0e9093;hpb=48c9926a2ed03737a3b024a85cda348caebf4cfe
diff --git a/pom_solver.h b/pom_solver.h
new file mode 100644
index 0000000..b0624b4
--- /dev/null
+++ b/pom_solver.h
@@ -0,0 +1,75 @@
+
+//////////////////////////////////////////////////////////////////////////////////
+// This program is free software: you can redistribute it and/or modify //
+// it under the terms of the version 3 of the GNU General Public License //
+// as published by the Free Software Foundation. //
+// //
+// This program is distributed in the hope that it will be useful, but //
+// WITHOUT ANY WARRANTY; without even the implied warranty of //
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU //
+// General Public License for more details. //
+// //
+// You should have received a copy of the GNU General Public License //
+// along with this program. If not, see . //
+// //
+// Written by Francois Fleuret //
+// (C) Ecole Polytechnique Federale de Lausanne //
+// Contact for comments & bug reports //
+//////////////////////////////////////////////////////////////////////////////////
+
+#ifndef POM_SOLVER_H
+#define POM_SOLVER_H
+
+#include "misc.h"
+#include "integral_proba_view.h"
+#include "normal_law.h"
+#include "room.h"
+
+class POMSolver {
+
+ // At each pixel the proba for the pixel to be off
+
+ ProbaView neg;
+
+ // At each pixel, 0 if the view is 0, and the proba for the pixel to
+ // be off if the view is 1 (or, more mathematically: neg * view)
+
+ ProbaView neg_view;
+
+ // Integral images to speed-up computation
+
+ IntegralProbaView ii_neg, ii_neg_view;
+
+ // Distribution of surface_difference / surface_synthetic
+
+ NormalLaw global_difference;
+
+ void compute_average_images(int camera,
+ Room *room,
+ Vector *proba_absence);
+
+ // Adds to every sum[i] the value log(P(X_i = 1 | V_camera) / P(X_i
+ // = 0 | V_camera)), given the other P(X_j = 1 | V)
+
+ void add_log_ratio(int camera,
+ Room *room,
+ ProbaView *view,
+ Vector *proba_absence,
+ Vector *sum);
+
+public:
+
+ POMSolver(Room *room);
+
+ // Uses the computation above for the various cameras and the prior
+ // to refresh proba_absence. Iterates as many times as specified.
+
+ void solve(Room *room,
+ Vector *prior,
+ Vector *views,
+ Vector *proba_presence,
+ int nb_frame,
+ char *convergence_file_format);
+};
+
+#endif