Changeset 15631 for branches/2877_HiveImprovements/HeuristicLab.Clients.Hive.Administrator/3.3/Views/ResourcesView.cs

Timestamp:

01/19/18 02:12:20 (6 years ago)

Author:

swagner

Message:

#2877: Improved groups/slaves treeview of Hive Administrator

• enabled selecting and moving multiple slaves at once

File:

• branches/2877_HiveImprovements/HeuristicLab.Clients.Hive.Administrator/3.3/Views/ResourcesView.cs (modified) (8 diffs)

branches/2877_HiveImprovements/HeuristicLab.Clients.Hive.Administrator/3.3/Views/ResourcesView.cs

@@ -44 +44 @@
     }
 
-    public const string UngroupedGroupName = "UNGROUPED";
+    public const string UNGROUPED_GROUP_NAME = "UNGROUPED";
+    private TreeNode ungroupedGroupNode;
     private const int slaveImageIndex = 0;
     private const int slaveGroupImageIndex = 1;
@@ -140 +141 @@
         var ungroupedSlaves = new SlaveGroup {
           Id = Guid.NewGuid(),
-          Name = UngroupedGroupName,
+          Name = UNGROUPED_GROUP_NAME,
           Description = "Slaves without a group"
         };
-        var ungroupedSlavesCount = 0;
-        var ungroupedNode = CreateTreeNodeFromResource(ungroupedSlaves);
+        ungroupedGroupNode = CreateTreeNode(ungroupedSlaves);
 
         var parentlessResources = from r in Content
@@ -152 +152 @@
         foreach (Resource r in parentlessResources) {
           if (r.GetType() == typeof(SlaveGroup)) {  // root node
-            var root = CreateTreeNodeFromResource(r);
+            var root = CreateTreeNode(r);
             treeSlaveGroup.Nodes.Add(root);
             BuildSlaveGroupTree((SlaveGroup)r, root);
           } else if (r.GetType() == typeof(Slave)) {  // ungrouped slaves
-            ungroupedNode.Nodes.Add(CreateTreeNodeFromResource(r));
-            ungroupedSlavesCount++;
+            ungroupedGroupNode.Nodes.Add(CreateTreeNode(r));
           }
         }
-        if (ungroupedSlavesCount > 0)
-          ungroupedNode.Text += "   [" + ungroupedSlavesCount.ToString() + "]";
-        treeSlaveGroup.Nodes.Add(ungroupedNode);
+        if (ungroupedGroupNode.Nodes.Count > 0)
+          ungroupedGroupNode.Text += "   [" + ungroupedGroupNode.Nodes.Count.ToString() + "]";
+        treeSlaveGroup.Nodes.Add(ungroupedGroupNode);
       }
     }
@@ -172 +171 @@
                         select r;
       foreach (var g in childGroups) {
-        var childNode = CreateTreeNodeFromResource(g);
+        var childNode = CreateTreeNode(g);
         node.Nodes.Add(childNode);
         BuildSlaveGroupTree(g, childNode);
@@ -183 +182 @@
       var childSlavesCount = 0;
       foreach (var s in childSlaves) {
-        node.Nodes.Add(CreateTreeNodeFromResource(s));
+        node.Nodes.Add(CreateTreeNode(s));
         childSlavesCount++;
       }
@@ -190 +189 @@
     }
 
-    private TreeNode CreateTreeNodeFromResource(Resource r) {
+    private TreeNode CreateTreeNode(Resource r) {
       var node = new TreeNode();
       node.Text = r.Name;
@@ -239 +238 @@
     private bool IsAuthorized(Resource resource) {
       return resource != null
-          && resource.Name != UngroupedGroupName
+          && resource != ungroupedGroupNode.Tag
           && resource.Id != Guid.Empty
           && UserInformation.Instance.UserExists
@@ -347 +346 @@
     }
 
+    private void treeSlaveGroup_ItemDrag(object sender, ItemDragEventArgs e) {
+      var nodes = GetCheckedNodes(treeSlaveGroup.Nodes).ToList();
+      var draggedNode = (TreeNode)e.Item;
+      if (!draggedNode.Checked) nodes.Add(draggedNode);
+      nodes.Remove(ungroupedGroupNode);
+      var resources = nodes.Select(x => x.Tag).OfType<Resource>().ToList();
+
+      if ((resources.Count > 0) && resources.TrueForAll((r) => IsAuthorized(r))) {
+        DataObject data = new DataObject();
+        data.SetData(HeuristicLab.Common.Constants.DragDropDataFormat, resources);
+        var action = DoDragDrop(data, DragDropEffects.Copy | DragDropEffects.Link | DragDropEffects.Move);
+        if (action.HasFlag(DragDropEffects.Move)) {
+          foreach (var node in nodes) node.Remove();
+          ungroupedGroupNode.Text = ((Resource)ungroupedGroupNode.Tag).Name;
+          if (ungroupedGroupNode.Nodes.Count > 0)
+            ungroupedGroupNode.Text += "   [" + ungroupedGroupNode.Nodes.Count.ToString() + "]";
+
+        }
+      }
+    }
+
+    private IEnumerable<TreeNode> GetCheckedNodes(TreeNodeCollection nodes) {
+      if (nodes != null) {
+        foreach (var node in nodes.OfType<TreeNode>()) {
+          if (node.Checked) yield return node;
+          foreach (var child in GetCheckedNodes(node.Nodes))
+            yield return child;
+        }
+      }
+    }
+
+    private void treeSlaveGroup_DragEnterOver(object sender, DragEventArgs e) {
+      e.Effect = DragDropEffects.None;
+      var resources = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) as IEnumerable<Resource>;
+      var node = treeSlaveGroup.GetNodeAt(treeSlaveGroup.PointToClient(new Point(e.X, e.Y)));
+      var group = node?.Tag as SlaveGroup;
+
+      if ((resources != null) && (node != null) && (group != null) && IsAuthorized(group) && (!resources.Contains(group))) {
+        if (e.AllowedEffect.HasFlag(DragDropEffects.Move)) e.Effect = DragDropEffects.Move;
+      }
+    }
+
     private void treeSlaveGroup_DragDrop(object sender, DragEventArgs e) {
-      if (e.Data.GetDataPresent("System.Windows.Forms.TreeNode", false)) {
-        Point pt = ((TreeView)sender).PointToClient(new Point(e.X, e.Y));
-        TreeNode destNode = ((TreeView)sender).GetNodeAt(pt);
-        TreeNode newNode = (TreeNode)e.Data.GetData("System.Windows.Forms.TreeNode");
-
-        if (destNode.TreeView == newNode.TreeView) {
-          if (destNode.Text == UngroupedGroupName || (destNode.Parent != null && destNode.Parent.Text == UngroupedGroupName)) {
-            MessageBox.Show(string.Format("You can't drag items to the group \"{0}\".{1}This group only contains slaves which haven't yet been assigned to a real group.",
-              UngroupedGroupName, Environment.NewLine), "HeuristicLab Hive Administrator", MessageBoxButtons.OK, MessageBoxIcon.Information);
-            return;
-          }
-
-          SlaveGroup sgrp = null;
-          TreeNode parentNode = null;
-          if (destNode.Tag != null && destNode.Tag is SlaveGroup) {
-            sgrp = (SlaveGroup)destNode.Tag;
-            parentNode = destNode;
-          } else if (destNode.Parent != null && destNode.Parent.Tag is SlaveGroup) {
-            sgrp = (SlaveGroup)destNode.Parent.Tag;
-            parentNode = destNode.Parent;
-          }
-
-          if (newNode.Tag is SlaveGroup && CheckParentsEqualsMovedNode(parentNode, newNode)) {
-            return;
-          }
-
-          SlaveGroup parent = (SlaveGroup)parentNode.Tag;
-
-          if (parent.OwnerUserId != null && !IsAuthorized(parent)) {
-            MessageBox.Show(string.Format("You don't have the permissions to drag items to the group \"{0}\".", ((Resource)parentNode.Tag).Name),
-              "HeuristicLab Hive Administrator", MessageBoxButtons.OK, MessageBoxIcon.Error);
-            return;
-          }
-
-          if (sgrp != null && newNode.Tag != null) {
-            //save parent group to get an id
-            if (sgrp.Id == Guid.Empty) {
-              sgrp.Store();
-            }
-
-            if (newNode.Tag is Slave) {
-              Slave slave = (Slave)newNode.Tag;
-              if (slave.ParentResourceId == null || (slave.ParentResourceId != null && slave.ParentResourceId != sgrp.Id)) {
-                slave.ParentResourceId = sgrp.Id;
-                newNode.Remove();
-                parentNode.Nodes.Clear();
-                BuildSlaveGroupTree(sgrp, parentNode);
-              }
-            } else if (newNode.Tag is SlaveGroup) {
-              SlaveGroup slaveGroup = (SlaveGroup)newNode.Tag;
-              if (slaveGroup.ParentResourceId == null || (slaveGroup.ParentResourceId != null && slaveGroup.ParentResourceId != sgrp.Id)) {
-                slaveGroup.ParentResourceId = sgrp.Id;
-                newNode.Remove();
-                parentNode.Nodes.Clear();
-                BuildSlaveGroupTree(sgrp, parentNode);
-              }
-            }
-          }
-        }
-      }
-    }
-
-    private bool CheckParentsEqualsMovedNode(TreeNode dest, TreeNode movedNode) {
-      TreeNode tmp = dest;
-
-      while (tmp != null) {
-        if (tmp == movedNode) {
-          return true;
-        }
-        tmp = tmp.Parent;
-      }
-      return false;
-    }
-
-    private void treeSlaveGroup_ItemDrag(object sender, ItemDragEventArgs e) {
-      TreeNode sourceNode = (TreeNode)e.Item;
-      if (IsAuthorized((Resource)sourceNode.Tag))
-        DoDragDrop(sourceNode, DragDropEffects.All);
-    }
-
-    private void treeSlaveGroup_DragEnter(object sender, DragEventArgs e) {
-      e.Effect = DragDropEffects.Move;
-    }
-
-    private void treeSlaveGroup_DragOver(object sender, DragEventArgs e) {
-      e.Effect = DragDropEffects.Move;
-    }
-
-    private void treeSlaveGroup_QueryContinueDrag(object sender, QueryContinueDragEventArgs e) {
-      e.Action = DragAction.Continue;
+      if (e.Effect != DragDropEffects.None) {
+        var node = treeSlaveGroup.GetNodeAt(treeSlaveGroup.PointToClient(new Point(e.X, e.Y)));
+        var group = (SlaveGroup)node.Tag;  // drop is only allowed on slave groups, validity of drop target is check in treeSlaveGroup_DragEnterOver
+        var resources = e.Data.GetData(HeuristicLab.Common.Constants.DragDropDataFormat) as IEnumerable<Resource>;
+
+        foreach (var r in resources) r.ParentResourceId = group.Id;
+        node.Nodes.Clear();
+        BuildSlaveGroupTree(group, node);
+      }
     }