1 files changed, 42 insertions, 0 deletions
diff --git a/effectivenessobserver.cpp b/effectivenessobserver.cpp
new file mode 100644
index 0000000..305b3f8
--- /dev/null
+++ b/effectivenessobserver.cpp
@@ -0,0 +1,42 @@
+#include "effectivenessobserver.h"
+
+EffectiveInteractionEnforcer::EffectiveInteractionEnforcer(const arma::uvec &sampleIndices)
+    : m_sampleIndices(sampleIndices)
+    , m_effectiveness(arma::zeros<arma::vec>(sampleIndices.n_elem))
+{
+}
+
+void EffectiveInteractionEnforcer::setSelection(const arma::uvec &selection)
+{
+    m_selection = selection;
+}
+
+void EffectiveInteractionEnforcer::setMeasureDifference(const arma::vec &measure)
+{
+    m_measure = measure;
+
+    if (m_selection.n_elem == 0) {
+        return;
+    }
+
+    arma::uvec selectionIndices(m_selection);
+    for (auto it = selectionIndices.begin(); it != selectionIndices.end(); it++) {
+        *it = m_sampleIndices[*it];
+    }
+
+    double diff = arma::mean(m_measure(selectionIndices));
+    int effectiveness;
+    if (diff > 0) {
+        effectiveness = 1;
+    } else if (diff < 0) {
+        effectiveness = -1;
+    } else {
+        effectiveness = 0;
+    }
+
+    for (auto it = m_selection.cbegin(); it != m_selection.cend(); it++) {
+        m_effectiveness[*it] = effectiveness;
+    }
+
+    emit effectivenessChanged(m_effectiveness);
+}