comparison src/EM.cpp @ 54:a54df67e607e preshift

Merge from default branch

     for (int i = 0; i < (int)column.size(); ++i) {
         sum += column[i];
     }
     for (int i = 0; i < (int)column.size(); ++i) {
         column[i] /= sum;
+    }
+}
+
+void
+EM::normaliseSources(Grid &sources)
+{
+    V denominators(sources[0].size());
+
+    for (int i = 0; i < (int)sources.size(); ++i) {
+        for (int j = 0; j < (int)sources[i].size(); ++j) {
+            denominators[j] += sources[i][j];
+        }
+    }
+
+    for (int i = 0; i < (int)sources.size(); ++i) {
+        for (int j = 0; j < (int)sources[i].size(); ++j) {
+            sources[i][j] /= denominators[j];
+        }
     }
 }
 
 void
 EM::iterate(V column)

             }
             if (m_sourceSparsity != 1.0) {
                 newSources[i][n] = pow(newSources[i][n], m_sourceSparsity);
             }
         }
-        normalise(newSources[i]);
-    }
+    }
+    normaliseSources(newSources);
 
     m_pitches = newPitches;
     m_sources = newSources;
 }