Allow deleting plots in exported plot figures

Author: onnodb

UI/TDVFigureWindow.m

@@ -2,7 +2,8 @@
 
     properties (Constant)
 
-        figureTag = 'twomdv-plot';
+        figureTag          = 'twomdv-plot';
+        plotContextMenuTag = 'twomdv-plot-cm';
 
     end
 
@@ -11,9 +12,7 @@
     methods (Static)
 
         function addData(h, fd)
-            if ~TDVFigureWindow.isValidFigureWindow(h)
-                error('Invalid TWOMDV Figure Window handle');
-            end
+            TDVFigureWindow.checkValidFigureWindow(h);
 
             figure(h);
 
@@ -24,21 +23,42 @@ function addData(h, fd)
             if isa(fd, 'FdDataCollection')
                 for i = 1:fd.length
                     figFdc.add(fd.items{i});
-                    plot(fd.items{i}.d, fd.items{i}.f);
+                    hPlot = plot(fd.items{i}.d, fd.items{i}.f);
+                    set(hPlot ...
+                        , 'UserData',           fd.items{i} ...
+                        , 'UIContextMenu',      TDVFigureWindow.getContextMenu(h) ...
+                        );
                 end
             else
                 error('Invalid argument "fd".');
             end
         end
 
-        function [h] = create()
-            h = figure(...
+        function [hFig] = create()
+            hFig = figure(...
                       'Tag',            TDVFigureWindow.figureTag ...
                     , 'UserData',       FdDataCollection() ...
                     );
             xlabel('Distance (um)');
             ylabel('Force (pN)');
             hold('on');
+
+            % Context menu for plots
+            hMenu = uicontextmenu(hFig, 'Tag', TDVFigureWindow.plotContextMenuTag);
+            uimenu(hMenu ...
+                , 'Label',          'Delete This Plot' ...
+                , 'Callback',       @(h,e) TDVFigureWindow.deleteSelectedData(hFig) ...
+                );
+        end
+
+        function deleteSelectedData(h)
+            TDVFigureWindow.checkValidFigureWindow(h);
+            fd = get(gco(h), 'UserData');
+            if strcmp(get(gco(h), 'Type'), 'line') && isa(fd, 'FdData')
+                fdc = get(h, 'UserData');
+                fdc.remove(fd);
+                delete(gco(h));
+            end
         end
 
         function [h] = findAll()
@@ -58,5 +78,23 @@ function addData(h, fd)
 
     end
 
+    % ------------------------------------------------------------------------
+
+    methods (Static, Access=private)
+
+        function checkValidFigureWindow(h)
+            if ~TDVFigureWindow.isValidFigureWindow(h)
+                error('Invalid TWOMDV Figure Window handle');
+            end
+        end
+
+        function [hMenu] = getContextMenu(h)
+            TDVFigureWindow.checkValidFigureWindow(h);
+            hMenu = findobj(h, 'Type', 'uicontextmenu', ...
+                            'Tag', TDVFigureWindow.plotContextMenuTag);
+        end
+
+    end
+
 end