# HG changeset patch
# User Chris Cannam
# Date 1244552586 0
# Node ID 5a2845883dd6dce237f16bcbd2bfa54629a6ba0e
# Parent  f6a8fc35df93a08c0cef9061a237a54905d509d5
* update to putative segmenter program

diff -r f6a8fc35df93 -r 5a2845883dd6 framework/MainWindowBase.cpp
--- a/framework/MainWindowBase.cpp	Tue Jun 09 13:02:16 2009 +0000
+++ b/framework/MainWindowBase.cpp	Tue Jun 09 13:03:06 2009 +0000
@@ -2636,8 +2636,12 @@
     // Check whether we need to add or remove model from play source
     Model *model = layer->getModel();
     if (model) {
+        Model *source = model->getSourceModel();
         if (inAView) {
             m_playSource->addModel(model);
+            if (source && source != model) {
+                m_playSource->addModel(source);
+            }
         } else {
             bool found = false;
             for (int i = 0; i < m_paneStack->getPaneCount(); ++i) {
@@ -2645,14 +2649,20 @@
                 if (!pane) continue;
                 for (int j = 0; j < pane->getLayerCount(); ++j) {
                     Layer *pl = pane->getLayer(j);
-                    if (pl && pl->getModel() == model) {
+                    if (pl && (pl->getModel() == model ||
+                               pl->getModel() == source)) {
                         found = true;
                         break;
                     }
                 }
                 if (found) break;
             }
-            if (!found) m_playSource->removeModel(model);
+            if (!found) {
+                m_playSource->removeModel(model);
+                if (source && source != model) {
+                    m_playSource->removeModel(source);
+                }
+            }
         }
     }