Disabling begin epoch group on data manager nodes when you can't begin an epoch group

Author: cafarm

src/main/matlab/+symphonyui/+ui/+presenters/DataManagerPresenter.m

@@ -722,14 +722,14 @@ function onServiceChangedControllerState(obj, ~, ~)
         end
 
         function updateStateOfControls(obj)
-            hasOpenFile = obj.documentationService.hasOpenFile();
-            hasSource = hasOpenFile && ~isempty(obj.documentationService.getExperiment().sources);
-            hasEpochGroup = hasOpenFile && ~isempty(obj.documentationService.getCurrentEpochGroup());
+            hasSource = ~isempty(obj.documentationService.getExperiment().sources);
             controllerState = obj.acquisitionService.getControllerState();
             isStopped = controllerState.isStopped();
 
-            %obj.view.enableBeginEpochGroup(hasSource && isStopped);
-            %obj.view.enableEndEpochGroup(hasEpochGroup && isStopped);
+            enableBeginEpochGroup = hasSource && isStopped;
+            
+            obj.view.enableBeginEpochGroupMenu(obj.view.getExperimentNode(), enableBeginEpochGroup);
+            obj.view.enableBeginEpochGroupMenu(obj.view.getEpochGroupsFolderNode(), enableBeginEpochGroup);
         end
 
         function loadSettings(obj)

src/main/matlab/+symphonyui/+ui/+views/DataManagerView.m

@@ -619,6 +619,13 @@ function setEpochGroupNodeNormal(obj, node)
             menu = obj.addEntityContextMenus(menu);
             set(node, 'UIContextMenu', menu);
         end
+        
+        function enableBeginEpochGroupMenu(obj, node, tf) %#ok<INUSL>
+            menu = get(node, 'UIContextMenu');
+            children = get(menu, 'Children');
+            index = arrayfun(@(c)isequal(get(c, 'Label'), 'Begin Epoch Group...'), children);
+            set(children(index), 'Enable', appbox.onOff(tf));
+        end
 
         function n = addEpochBlockNode(obj, parent, name, entity)
             value.entity = entity;