source: trunk/HeuristicLab.Clients.Hive.JobManager/3.3/Views/HiveResourceSelector.cs @ 16202

Last change on this file since 16202 was 16202, checked in by jzenisek, 12 months ago

#2839: fixed several bugs:

  • renamed tab "Slaves" to "Resources"
  • fixed bugs in job manager > resource selector reported by fholzinger
  • adapted client-side handling of disabled resource ancestors
  • adapted parentship check in HiveAdminClient
  • adapted seting of enabled state of controls for project details, project resources
  • introduced "inactive" tag in ProjectsView
File size: 32.4 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2017 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
4 *
5 * This file is part of HeuristicLab.
6 *
7 * HeuristicLab is free software: you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation, either version 3 of the License, or
10 * (at your option) any later version.
11 *
12 * HeuristicLab is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
19 */
20#endregion
21
22using System;
23using System.Collections.Generic;
24using System.Drawing;
25using System.Linq;
26using System.Windows.Forms;
27using HeuristicLab.Collections;
28using HeuristicLab.Common.Resources;
29using HeuristicLab.Core;
30using HeuristicLab.Core.Views;
31using HeuristicLab.MainForm;
32using HeuristicLab.MainForm.WindowsForms;
33
34namespace HeuristicLab.Clients.Hive.JobManager.Views {
35  [View("Hive Project Selector View")]
36  [Content(typeof(IItemList<Project>), true)]
37  public partial class HiveProjectSelector : ItemView, IDisposable {
38    private const int greenFlagImageIndex = 0;
39    private const int redFlagImageIndex = 1;
40    private const int slaveImageIndex = 0;
41    private const int slaveGroupImageIndex = 1;
42    public const string additionalSlavesGroupName = "Additional Slaves";
43    public const string additionalSlavesGroupDescription = "Contains additional slaves which are either ungrouped or the parenting slave group is not assigned to the selected project.";
44
45    private const string CURRENT_SELECTION_TAG = " [current assignment]";
46    private const string NEW_SELECTION_TAG = " [new assignment]";
47    private const string CHANGED_SELECTION_TAG = " [changed assignment]";
48    private const string ADDED_SELECTION_TAG = " [added assignment]";
49    private const string REMOVED_SELECTION_TAG = " [removed assignment]";
50    private const string SELECTED_TAG = " [assigned]";
51    private const string INCLUDED_TAG = " [included]";
52    private const string ADDED_INCLUDE_TAG = " [added include]";
53    private const string REMOVED_INCLUDE_TAG = " [removed include]";
54
55    private TreeNode additionalNode;
56
57    private readonly HashSet<TreeNode> mainTreeNodes = new HashSet<TreeNode>();
58    private readonly HashSet<TreeNode> filteredTreeNodes = new HashSet<TreeNode>();
59
60    private readonly HashSet<Resource> availableResources = new HashSet<Resource>();
61    private readonly HashSet<Resource> assignedResources = new HashSet<Resource>();
62    private readonly HashSet<Resource> includedResources = new HashSet<Resource>();
63    private readonly HashSet<Resource> newAssignedResources = new HashSet<Resource>();
64    private readonly HashSet<Resource> newIncludedResources = new HashSet<Resource>();
65
66    private readonly Color addedAssignmentColor = Color.FromArgb(255, 87, 191, 193); // #57bfc1
67    private readonly Color removedAssignmentColor = Color.FromArgb(255, 236, 159, 72); // #ec9f48
68    private readonly Color addedIncludeColor = Color.FromArgb(25, 169, 221, 221); // #a9dddd
69    private readonly Color removedIncludeColor = Color.FromArgb(25, 249, 210, 145); // #f9d291
70    private readonly Color selectedBackColor = Color.DodgerBlue;
71    private readonly Color selectedForeColor = Color.White;
72    private readonly Color controlTextColor = SystemColors.ControlText;
73    private readonly Color grayTextColor = SystemColors.GrayText;
74
75    private string currentSearchString;
76
77    private void resetHiveResourceSelector() {
78      lastSelectedProject = null;
79      selectedProject = null;
80      projectId = null;
81    }
82
83    private Guid jobId;
84    public Guid JobId {
85      get { return jobId; }
86      set {
87        if (jobId == value) return;
88        jobId = value;
89        resetHiveResourceSelector();
90      }
91    }
92
93    private Guid? projectId;
94    public Guid? ProjectId {
95      get { return projectId; }
96      set {
97        if (projectId == value) return;
98        projectId = value;
99      }
100    }
101
102    private Guid? selectedProjectId;
103    public Guid? SelectedProjectId {
104      get { return selectedProjectId; }
105      set {
106        if (selectedProjectId == value) return;
107        selectedProjectId = value;
108      }
109    }
110
111    private IEnumerable<Guid> selectedResourceIds;
112    public IEnumerable<Guid> SelectedResourceIds {
113      get { return selectedResourceIds; }
114      set {
115        if (selectedResourceIds == value) return;
116        selectedResourceIds = value;
117      }
118    }
119
120    public bool ChangedProjectSelection {
121      get {
122        if ((lastSelectedProject == null && selectedProject != null) 
123          || (lastSelectedProject != null && selectedProject == null)
124          || (lastSelectedProject != null && selectedProject != null && lastSelectedProject.Id != selectedProject.Id))
125          return true;
126        else return false;
127      }
128    }
129
130    public bool ChangedResources {
131      get { return !assignedResources.SetEquals(newAssignedResources); }
132    }
133
134    private Project lastSelectedProject;
135    private Project selectedProject;
136    public Project SelectedProject {
137      get { return selectedProject; }
138      set {
139        if (selectedProject == value) return;
140
141        if ((JobId == Guid.Empty || JobId == null) 
142          && (value == null || SelectedProject == null || value.Id != SelectedProject.Id)) selectedResourceIds = null;
143        lastSelectedProject = selectedProject;
144        selectedProject = value;
145
146        UpdateResourceTree();
147        ExtractStatistics();
148        OnSelectedProjectChanged();
149      }
150    }
151
152    public int AssignedCores {
153      get {
154        HashSet<Slave> newAssignedSlaves = new HashSet<Slave>(newAssignedResources.OfType<Slave>());
155        foreach(var slaveGroup in newAssignedResources.OfType<SlaveGroup>()) {
156          foreach(var slave in HiveClient.Instance.GetAvailableResourceDescendants(slaveGroup.Id).OfType<Slave>()) {
157            newAssignedSlaves.Add(slave);
158          }
159        }
160        return newAssignedSlaves.Sum(x => x.Cores.GetValueOrDefault());
161      }
162    }
163
164    public IEnumerable<Resource> AssignedResources {
165      get { return newAssignedResources; }
166      set {
167        if (newAssignedResources == value) return;
168        newAssignedResources.Clear();
169        foreach(var resource in value) {
170          newAssignedResources.Add(resource);
171        }
172      }
173    }
174
175
176    public new IItemList<Project> Content {
177      get { return (IItemList<Project>)base.Content; }
178      set { base.Content = value; }
179    }
180
181    public HiveProjectSelector() {
182      InitializeComponent();
183
184      projectsImageList.Images.Add(VSImageLibrary.FlagGreen);
185      projectsImageList.Images.Add(VSImageLibrary.FlagRed);
186      resourcesImageList.Images.Add(VSImageLibrary.MonitorLarge);
187      resourcesImageList.Images.Add(VSImageLibrary.NetworkCenterLarge);
188    }
189
190    #region Overrides
191    protected override void OnContentChanged() {
192      base.OnContentChanged();
193
194      if (Content != null) {               
195        if (SelectedProjectId.HasValue && SelectedProjectId.Value != Guid.Empty) {
196          SelectedProject = GetSelectedProjectById(SelectedProjectId.Value);
197        } else {
198          SelectedProject = null;
199        }
200        //ExtractStatistics();
201        UpdateProjectTree();
202
203      } else {
204        lastSelectedProject = null;
205        selectedProject = null;
206        selectedProjectId = null;
207        projectsTreeView.Nodes.Clear();
208        resourcesTreeView.Nodes.Clear();
209      }
210    }
211
212    #endregion
213
214    #region Event Handlers
215    private void HiveProjectSelector_Load(object sender, EventArgs e) {
216      projectsTreeView.Nodes.Clear();
217      resourcesTreeView.Nodes.Clear();
218    }
219
220    private void searchTextBox_TextChanged(object sender, EventArgs e) {
221      currentSearchString = searchTextBox.Text.ToLower();
222      //UpdateFilteredTree();
223      UpdateProjectTree();
224    }
225
226    private void projectsTreeView_MouseDoubleClick(object sender, MouseEventArgs e) {
227      OnProjectsTreeViewDoubleClicked();
228    }
229
230    private void projectsTreeView_BeforeSelect(object sender, TreeViewCancelEventArgs e) {
231      var node = (Project)e.Node.Tag;
232      if (HiveClient.Instance.DisabledParentProjects.Contains(node)) {
233        e.Cancel = true;
234      }
235    }
236
237    private void projectsTreeView_AfterSelect(object sender, TreeViewEventArgs e) {
238      var node = (Project)e.Node.Tag;
239
240      if (node == null) {
241        projectsTreeView.SelectedNode = null;
242      } else if (HiveClient.Instance.DisabledParentProjects.Contains(node)) {
243        return;
244      } else {
245        ResetTreeNodes(projectsTreeView.Nodes);
246        e.Node.BackColor = selectedBackColor;
247        e.Node.ForeColor = selectedForeColor;
248
249        if (node.Id == projectId) {
250          e.Node.Text += CURRENT_SELECTION_TAG;
251        } else if (projectId == null || projectId == Guid.Empty) {
252          e.Node.Text += NEW_SELECTION_TAG;
253        } else {
254          e.Node.Text += CHANGED_SELECTION_TAG;
255        }
256
257
258      }
259      SelectedProject = node;
260    }
261
262    private void resourcesTreeView_MouseDown(object sender, MouseEventArgs e) {
263      var node = resourcesTreeView.GetNodeAt(new Point(e.X, e.Y));
264
265      if (node == null && e.Button == MouseButtons.Left) {
266        resourcesTreeView.SelectedNode = null;
267        ExtractStatistics();
268      }
269    }
270
271    private void resourcesTreeView_BeforeSelect(object sender, TreeViewCancelEventArgs e) {
272      if(e.Node == null) {
273        e.Cancel = true;
274        resourcesTreeView.SelectedNode = null;
275        ExtractStatistics();
276      } else if (e.Node == additionalNode) {
277        e.Cancel = true;
278      } else {
279        var r = (Resource)e.Node.Tag;
280        if(r != null && !HiveClient.Instance.DisabledParentResources.Contains(r)) {
281          ExtractStatistics(r);
282        } else {
283          e.Cancel = true;
284        }
285      }
286
287    }
288
289    private void resourcesTreeView_BeforeCheck(object sender, TreeViewCancelEventArgs e) {
290      if(e.Node == null || e.Node == additionalNode) {
291        e.Cancel = true;
292      } else {
293        var checkedResource = (Resource)e.Node.Tag;
294        if (checkedResource == null
295          || checkedResource.Id == Guid.Empty
296          || HiveClient.Instance.DisabledParentResources.Contains(checkedResource)
297          || newIncludedResources.Contains(checkedResource)) {
298          e.Cancel = true;
299        }
300      }
301    }
302
303    private void resourcesTreeView_AfterCheck(object sender, TreeViewEventArgs e) {
304      var checkedResource = (Resource)e.Node.Tag;
305      if (e.Node.Checked) {
306        newAssignedResources.Add(checkedResource);
307      } else {
308        newAssignedResources.Remove(checkedResource);
309      }
310
311      UpdateResourceTreeAfterCheck();
312      if(resourcesTreeView.SelectedNode == null)
313        ExtractStatistics();
314      OnAssignedResourcesChanged();
315    }
316
317    private void resourcesTreeView_NodeMouseDoubleClick(object sender, TreeNodeMouseClickEventArgs e) {
318      return;
319    }
320    #endregion
321
322    #region Helpers
323
324    private Project GetSelectedProjectById(Guid projectId) {
325      return Content.Where(x => x.Id == projectId).SingleOrDefault();
326    }
327
328    private void UpdateProjectTree() {
329
330      if (string.IsNullOrEmpty(currentSearchString)) {
331        BuildProjectTree(Content);
332      } else {
333        HashSet<Project> filteredProjects = new HashSet<Project>();
334        foreach(var project in Content) {
335          if(project.Name.ToLower().Contains(currentSearchString.ToLower())) {
336            filteredProjects.Add(project);
337            filteredProjects.UnionWith(Content.Where(p => HiveClient.Instance.GetAvailableProjectAncestors(project.Id).Select(x => x.Id).Contains(p.Id)));
338          }
339        }
340        BuildProjectTree(filteredProjects);
341      }
342    }
343
344    private void BuildProjectTree(IEnumerable<Project> projects) {
345      projectsTreeView.Nodes.Clear();
346      if (!projects.Any()) return;
347
348      var disabledParentProjects = HiveClient.Instance.DisabledParentProjects;
349      // select all top level projects (withouth parent, or without any ancestor within current project collection)
350      var mainProjects = new HashSet<Project>(projects.Where(x => x.ParentProjectId == null));
351      //var parentedMainProjects = new HashSet<Project>(projects
352      //  .Where(x => x.ParentProjectId.HasValue
353      //  && !projects.Select(y => y.Id).Contains(x.ParentProjectId.Value)
354      //  && !projects.SelectMany(y => HiveClient.Instance.ProjectAncestors[y.Id]).Contains(x.ParentProjectId.Value)));
355      //mainProjects.UnionWith(parentedMainProjects);
356      var mainDisabledParentProjects = new HashSet<Project>(disabledParentProjects.Where(x => x.ParentProjectId == null || x.ParentProjectId == Guid.Empty));
357      mainProjects.UnionWith(mainDisabledParentProjects);
358      var subProbjects = new HashSet<Project>(projects.Union(disabledParentProjects).Except(mainProjects));
359      //foreach (var p in subProbjects) {
360      //  p.ParentProjectId = HiveClient.Instance.ProjectAncestors[p.Id].Where(x => projects.Select(y => y.Id).Contains(x)).FirstOrDefault();
361      //}
362
363      var stack = new Stack<Project>(mainProjects.OrderByDescending(x => x.Name));
364
365      TreeNode currentNode = null;
366      Project currentProject = null;
367
368      while(stack.Any()) {
369        var newProject = stack.Pop();
370        var newNode = new TreeNode(newProject.Name) { Tag = newProject };
371
372        while (currentNode != null && newProject.ParentProjectId != currentProject.Id) {
373          currentNode = currentNode.Parent;
374          currentProject = currentNode == null ? null : (Project)currentNode.Tag;
375        }
376
377        if (currentNode == null) {
378          projectsTreeView.Nodes.Add(newNode);
379          newNode.ImageIndex = greenFlagImageIndex;
380        } else {
381          currentNode.Nodes.Add(newNode);
382          newNode.ImageIndex = redFlagImageIndex;
383        }
384
385        newNode.SelectedImageIndex = newNode.ImageIndex;
386
387        if(disabledParentProjects.Contains(newProject)) {
388          newNode.Checked = false;
389          newNode.ForeColor = grayTextColor;
390        }
391        else if (SelectedProject != null && SelectedProject.Id.Equals(newProject.Id)) {
392          newNode.BackColor = selectedBackColor;
393          newNode.ForeColor = selectedForeColor;
394          if(SelectedProject.Id == projectId) {
395            newNode.Text += CURRENT_SELECTION_TAG;
396          } else if (projectId == null || projectId == Guid.Empty) {
397            newNode.Text += NEW_SELECTION_TAG;
398          } else {
399            newNode.Text += CHANGED_SELECTION_TAG;
400          }
401        }
402
403        if (!string.IsNullOrEmpty(currentSearchString) && newProject.Name.ToLower().Contains(currentSearchString.ToLower())) {
404          newNode.BackColor = Color.LightBlue;
405        }
406
407        var childProjects = subProbjects.Where(x => x.ParentProjectId == newProject.Id);
408        if (childProjects.Any()) {
409          foreach (var project in childProjects.OrderByDescending(x => x.Name)) {
410            subProbjects.Remove(project);
411            stack.Push(project);
412          }
413          currentNode = newNode;
414          currentProject = newProject;
415        }
416      }
417
418      projectsTreeView.ExpandAll();
419    }
420
421    private static IEnumerable<Resource> GetAssignedResourcesForProject(Guid projectId) {
422      var assignedProjectResources = HiveServiceLocator.Instance.CallHiveService(s => s.GetAssignedResourcesForProject(projectId));
423      return HiveClient.Instance.Resources.Where(x => assignedProjectResources.Select(y => y.ResourceId).Contains(x.Id));
424    }
425
426    private static IEnumerable<Resource> GetAssignedResourcesForJob(Guid jobId) {
427      var assignedJobResources = HiveServiceLocator.Instance.CallHiveService(s => s.GetAssignedResourcesForJob(jobId));
428      return HiveClient.Instance.Resources.Where(x => assignedJobResources.Select(y => y.ResourceId).Contains(x.Id));
429    }
430
431    private void UpdateResourceTree() {
432      UpdateAvailableResources();
433      UpdateAssignedResources();
434      UpdateIncludedResources();
435      BuildResourceTree(availableResources);
436    }
437
438    private void UpdateResourceTreeAfterCheck() {
439      resourcesTreeView.BeforeCheck -= resourcesTreeView_BeforeCheck;
440      resourcesTreeView.AfterCheck -= resourcesTreeView_AfterCheck;
441      UpdateNewAssignedResources();
442      UpdateNewIncludedResources();
443      SetTreeNodes(resourcesTreeView.Nodes);
444      resourcesTreeView.BeforeCheck += resourcesTreeView_BeforeCheck;
445      resourcesTreeView.AfterCheck += resourcesTreeView_AfterCheck;
446    }
447
448    private void UpdateAvailableResources() {
449      availableResources.Clear();
450      if (selectedProject != null) {
451        var assignedProjectResources = GetAssignedResourcesForProject(selectedProject.Id);
452        foreach (var resource in assignedProjectResources) {
453          availableResources.Add(resource);
454          foreach(var descendant in HiveClient.Instance.Resources.Where(x => HiveClient.Instance.ResourceDescendants[resource.Id].Contains(x.Id))) {
455            availableResources.Add(descendant);
456          }
457        }
458      }
459      //ExtractStatistics();
460      //OnAssignedResourcesChanged();
461    }
462
463    private void UpdateAssignedResources() {
464      assignedResources.Clear();
465      newAssignedResources.Clear();
466
467      if (JobId == Guid.Empty || JobId == null) { // new, unchanged jobs get all avaialable resources
468        // update new assigned resources
469        if(selectedResourceIds == null) {
470          foreach (var resource in availableResources
471            .Where(x => !x.ParentResourceId.HasValue
472            || !availableResources.Select(y => y.Id).Contains(x.ParentResourceId.Value))) {
473            newAssignedResources.Add(resource);
474          }
475        } else {
476          foreach(var resource in availableResources.Where(x => selectedResourceIds.Contains(x.Id))) {
477            newAssignedResources.Add(resource);
478          }
479        }
480      } else if(selectedProject.Id == projectId) { // existent, unchanged jobs get all assigned resources
481        // update assigned resources
482        var assignedJobResources = GetAssignedResourcesForJob(JobId);
483        foreach (var resource in assignedJobResources) {
484          assignedResources.Add(resource);
485          if (selectedResourceIds == null) {
486            newAssignedResources.Add(resource);
487          }
488        }
489
490        if(selectedResourceIds != null) {
491          foreach (var resource in availableResources.Where(x => selectedResourceIds.Contains(x.Id))) {
492            newAssignedResources.Add(resource);
493          }
494        }
495      }
496
497      //ExtractStatistics();
498      OnAssignedResourcesChanged();
499    }
500
501    private void UpdateNewAssignedResources() {
502      for(int i = newAssignedResources.Count-1; i>=0; i--) {
503        if(newAssignedResources.Intersect(HiveClient.Instance.GetAvailableResourceAncestors(newAssignedResources.ElementAt(i).Id)).Any()) {
504          newAssignedResources.Remove(newAssignedResources.ElementAt(i));
505        }
506      }
507    }
508
509    private void UpdateIncludedResources() {
510      includedResources.Clear();
511      newIncludedResources.Clear();
512
513      if (JobId != Guid.Empty) {
514        foreach (var item in assignedResources) {
515          foreach (var descendant in HiveClient.Instance.GetAvailableResourceDescendants(item.Id)) {
516            includedResources.Add(descendant);
517          }
518        }
519      }
520
521      foreach (var item in newAssignedResources) {
522        foreach (var descendant in HiveClient.Instance.GetAvailableResourceDescendants(item.Id)) {
523          newIncludedResources.Add(descendant);
524        }
525      }
526    }
527
528    private void UpdateNewIncludedResources() {
529      newIncludedResources.Clear();
530      foreach (var item in newAssignedResources) {
531        foreach (var descendant in HiveClient.Instance.GetAvailableResourceDescendants(item.Id)) {
532          newIncludedResources.Add(descendant);
533        }
534      }
535    }
536
537    private void BuildResourceTree(IEnumerable<Resource> resources) {
538      resourcesTreeView.Nodes.Clear();
539      if (!resources.Any()) return;
540
541      resourcesTreeView.BeforeCheck -= resourcesTreeView_BeforeCheck;
542      resourcesTreeView.AfterCheck -= resourcesTreeView_AfterCheck;
543
544      //var disabledParentResources = HiveClient.Instance.DisabledParentResources;
545      var disabledParentResources = HiveClient.Instance.GetDisabledResourceAncestors(resources);
546      var mainResources = new HashSet<Resource>(resources.OfType<SlaveGroup>().Where(x => x.ParentResourceId == null));
547      //var parentedMainResources = new HashSet<Resource>(resources.OfType<SlaveGroup>()
548      //  .Where(x => x.ParentResourceId.HasValue && !resources.Select(y => y.Id).Contains(x.ParentResourceId.Value)));
549      //mainResources.UnionWith(parentedMainResources);
550      //var subResources = new HashSet<Resource>(resources.Except(mainResources));
551      var mainDisabledParentResources = new HashSet<Resource>(disabledParentResources.Where(x => x.ParentResourceId == null || x.ParentResourceId == Guid.Empty));
552      mainResources.UnionWith(mainDisabledParentResources);
553      var subResources = new HashSet<Resource>(resources.Union(disabledParentResources).Except(mainResources).OrderByDescending(x => x.Name));
554
555      var addedAssignments = newAssignedResources.Except(assignedResources);
556      var removedAssignments = assignedResources.Except(newAssignedResources);
557      var addedIncludes = newIncludedResources.Except(includedResources);
558      var removedIncludes = includedResources.Except(newIncludedResources);
559
560      TreeNode currentNode = null;
561      Resource currentResource = null;
562
563      var stack = new Stack<Resource>(mainResources.OrderByDescending(x => x.Name));
564
565      while (stack.Any()) {
566        var newResource = stack.Pop();
567        var newNode = new TreeNode(newResource.Name) { Tag = newResource };
568
569        // search for parent node of newNode and save in currentNode
570        // necessary since newNodes (stack top items) might be siblings
571        // or grand..grandparents of previous node (currentNode)
572        while (currentNode != null && newResource.ParentResourceId != currentResource.Id) {
573          currentNode = currentNode.Parent;
574          currentResource = currentNode == null ? null : (Resource)currentNode.Tag;
575        }
576
577        if (currentNode == null) {
578          resourcesTreeView.Nodes.Add(newNode);
579        } else {
580          currentNode.Nodes.Add(newNode);
581        }
582
583        if (disabledParentResources.Contains(newResource)) {
584          newNode.Checked = false;
585          newNode.ForeColor = grayTextColor;
586        } else if (newAssignedResources.Select(x => x.Id).Contains(newResource.Id)) {
587          newNode.Checked = true;
588          if(!addedAssignments.Select(x => x.Id).Contains(newResource.Id) && !removedAssignments.Select(x => x.Id).Contains(newResource.Id)) {
589            newNode.Text += SELECTED_TAG;
590          }
591        } else if (newIncludedResources.Select(x => x.Id).Contains(newResource.Id)) {
592          newNode.Checked = true;
593          newNode.ForeColor = grayTextColor;
594        }
595
596        if (includedResources.Select(x => x.Id).Contains(newResource.Id) && newIncludedResources.Select(x => x.Id).Contains(newResource.Id)) {
597          newNode.Text += INCLUDED_TAG;
598        } else if (addedIncludes.Select(x => x.Id).Contains(newResource.Id)) {
599          newNode.BackColor = addedIncludeColor;
600          newNode.ForeColor = grayTextColor;
601          newNode.Text += ADDED_INCLUDE_TAG;
602        } else if (removedIncludes.Select(x => x.Id).Contains(newResource.Id)) {
603          newNode.BackColor = removedIncludeColor;
604          newNode.Text += REMOVED_INCLUDE_TAG;
605        }
606
607        if (addedAssignments.Select(x => x.Id).Contains(newResource.Id)) {
608          newNode.BackColor = addedAssignmentColor;
609          newNode.ForeColor = controlTextColor;
610          newNode.Text += ADDED_SELECTION_TAG;
611        } else if (removedAssignments.Select(x => x.Id).Contains(newResource.Id)) {
612          newNode.BackColor = removedAssignmentColor;
613          newNode.ForeColor = controlTextColor;
614          newNode.Text += REMOVED_SELECTION_TAG;
615        }
616
617        if (newResource is Slave) {
618          newNode.ImageIndex = slaveImageIndex;
619        } else {
620          newNode.ImageIndex = slaveGroupImageIndex;
621
622          var childResources = subResources.Where(x => x.ParentResourceId == newResource.Id);
623          if (childResources.Any()) {
624            foreach (var resource in childResources.OrderByDescending(x => x.Name)) {
625              subResources.Remove(resource);
626              stack.Push(resource);
627            }
628            currentNode = newNode;
629            currentResource = newResource;
630          }
631        }
632        newNode.SelectedImageIndex = newNode.ImageIndex;
633      }
634
635      var singleSlaves = subResources.OfType<Slave>();
636      if (singleSlaves.Any()) {
637
638        additionalNode = new TreeNode(additionalSlavesGroupName) {
639          ForeColor = SystemColors.GrayText,
640          Tag = new SlaveGroup() {
641            Name = additionalSlavesGroupName,
642            Description = additionalSlavesGroupDescription
643          }
644        };
645
646        foreach (var slave in singleSlaves.OrderBy(x => x.Name)) {
647          var slaveNode = new TreeNode(slave.Name) { Tag = slave };
648          additionalNode.Nodes.Add(slaveNode);
649        }
650
651        resourcesTreeView.Nodes.Add(additionalNode);
652      }
653
654      ExpandResourceNodesOfInterest(resourcesTreeView.Nodes);
655
656      resourcesTreeView.BeforeCheck += resourcesTreeView_BeforeCheck;
657      resourcesTreeView.AfterCheck += resourcesTreeView_AfterCheck;
658    }
659
660    private void ExpandResourceNodesOfInterest(TreeNodeCollection nodes) {
661      foreach(TreeNode n in nodes) {
662        Resource r = (Resource)n.Tag;
663        if(n.Nodes.Count > 0) {
664          if(HiveClient.Instance.GetAvailableResourceDescendants(r.Id).OfType<SlaveGroup>().Any()
665            || HiveClient.Instance.GetAvailableResourceDescendants(r.Id).OfType<Slave>().Intersect(assignedResources.Union(newAssignedResources)).Any()) {
666            n.Expand();
667            ExpandResourceNodesOfInterest(n.Nodes);
668          } else {
669            n.Collapse();
670          }   
671        } else {
672          n.Collapse();
673        }
674      }
675    }
676
677    private void CollapseSlaveOnlyNodes(TreeNode tn) {
678      if (!(tn.Tag is Resource)) return;
679
680      Resource r = (Resource)tn.Tag;
681      if (HiveClient.Instance.GetAvailableResourceDescendants(r.Id).OfType<SlaveGroup>().Any()) {
682        tn.Expand();
683        foreach (TreeNode n in tn.Nodes) CollapseSlaveOnlyNodes(n);
684      } else {
685        tn.Collapse();
686      }
687    }
688
689    private void ExtractStatistics(Resource resource = null) { 
690      HashSet<Slave> newAssignedSlaves = new HashSet<Slave>(newAssignedResources.OfType<Slave>());
691      foreach (var slaveGroup in newAssignedResources.OfType<SlaveGroup>()) {
692        foreach (var slave in HiveClient.Instance.GetAvailableResourceDescendants(slaveGroup.Id).OfType<Slave>()) {
693          newAssignedSlaves.Add(slave);
694        }
695      }
696
697      HashSet<Slave> selectedSlaves = null;
698
699      if (resource != null) {
700        var slaveGroup = resource as SlaveGroup;
701        if (slaveGroup != null) {
702          selectedSlaves = new HashSet<Slave>(HiveClient.Instance.GetAvailableResourceDescendants(slaveGroup.Id).OfType<Slave>());
703        } else {
704          selectedSlaves = new HashSet<Slave>(new[] { resource as Slave });
705        }
706      } else {
707        selectedSlaves = newAssignedSlaves;
708      }
709
710      int sumCores = selectedSlaves.Sum(x => x.Cores.GetValueOrDefault());
711      int sumFreeCores = selectedSlaves.Sum(x => x.FreeCores.GetValueOrDefault());
712      double sumMemory = selectedSlaves.Sum(x => x.Memory.GetValueOrDefault()) / 1024.0;
713      double sumFreeMemory = selectedSlaves.Sum(x => x.FreeMemory.GetValueOrDefault()) / 1024.0;
714
715      coresSummaryLabel.Text = string.Format("{0} Total ({1} Free / {2} Used)", sumCores, sumFreeCores, sumCores - sumFreeCores);
716      memorySummaryLabel.Text = string.Format("{0:0.00} GB Total ({1:0.00} GB Free / {2:0.00} GB Used)", sumMemory, sumFreeMemory, sumMemory - sumFreeMemory);
717    }
718
719    private void StyleTreeNode(TreeNode n, string name) {
720      n.Text = name;
721      n.BackColor = Color.Transparent;
722      n.ForeColor = Color.Black;
723
724      if(n.Tag is Project) {
725        var p = (Project)n.Tag;
726        if(HiveClient.Instance.DisabledParentProjects.Select(x => x.Id).Contains(p.Id)) {
727          n.Checked = false;
728          n.ForeColor = grayTextColor;
729        }
730      } else if(n.Tag is Resource) {
731        var r = (Resource)n.Tag;
732        if(HiveClient.Instance.DisabledParentResources.Select(x => x.Id).Contains(r.Id) || n == additionalNode) {
733          n.Checked = false;
734          n.ForeColor = grayTextColor;
735        }
736      }
737    }
738
739    private void ResetTreeNodes(TreeNodeCollection nodes) {
740      foreach (TreeNode n in nodes) {
741        string name = "";
742        if (n.Tag is Project) name = ((Project)n.Tag).Name;
743        else if (n.Tag is Resource) name = ((Resource)n.Tag).Name;
744        StyleTreeNode(n, name);
745        if (n.Nodes.Count > 0) {
746          ResetTreeNodes(n.Nodes);
747        }
748      }
749    }
750
751    private void SetTreeNodes(TreeNodeCollection nodes) {
752      var addedAssignments = newAssignedResources.Except(assignedResources);
753      var removedAssignments = assignedResources.Except(newAssignedResources);
754      var addedIncludes = newIncludedResources.Except(includedResources);
755      var removedIncludes = includedResources.Except(newIncludedResources);
756
757      foreach (TreeNode n in nodes) {
758
759        if(n.Tag is Resource) {
760          // reset
761          var resource = (Resource)n.Tag;
762          n.Text = resource.Name;
763          n.BackColor = Color.Transparent;
764          n.ForeColor = Color.Black;
765          n.Checked = false;
766
767          if (HiveClient.Instance.DisabledParentResources.Select(x => x.Id).Contains(resource.Id) || n == additionalNode) {
768            n.ForeColor = grayTextColor;
769          }
770
771          // add additional info
772          if (newAssignedResources.Select(x => x.Id).Contains(resource.Id)) {
773            n.Checked = true;
774            if (!addedAssignments.Select(x => x.Id).Contains(resource.Id) && !removedAssignments.Select(x => x.Id).Contains(resource.Id)) {
775              n.Text += SELECTED_TAG;
776            }
777          } else if (newIncludedResources.Select(x => x.Id).Contains(resource.Id)) {
778            n.Checked = true;
779            n.ForeColor = grayTextColor;
780          }
781
782          if (includedResources.Select(x => x.Id).Contains(resource.Id) && newIncludedResources.Select(x => x.Id).Contains(resource.Id)) {
783            n.Text += INCLUDED_TAG;
784          } else if (addedIncludes.Select(x => x.Id).Contains(resource.Id)) {
785            n.BackColor = addedIncludeColor;
786            n.ForeColor = grayTextColor;
787            n.Text += ADDED_INCLUDE_TAG;
788          } else if (removedIncludes.Select(x => x.Id).Contains(resource.Id)) {
789            n.BackColor = removedIncludeColor;
790            n.Text += REMOVED_INCLUDE_TAG;
791          }
792
793          if (addedAssignments.Select(x => x.Id).Contains(resource.Id)) {
794            n.BackColor = addedAssignmentColor;
795            n.ForeColor = controlTextColor;
796            n.Text += ADDED_SELECTION_TAG;
797          } else if (removedAssignments.Select(x => x.Id).Contains(resource.Id)) {
798            n.BackColor = removedAssignmentColor;
799            n.ForeColor = controlTextColor;
800            n.Text += REMOVED_SELECTION_TAG;
801          }
802        }
803
804        if(n.Nodes.Count > 0) {
805          SetTreeNodes(n.Nodes);
806        }
807      }
808    }
809
810    #endregion
811
812    #region Events
813    public event EventHandler SelectedProjectChanged;
814    private void OnSelectedProjectChanged() {
815      var handler = SelectedProjectChanged;
816      if (handler != null) handler(this, EventArgs.Empty);
817    }
818
819    public event EventHandler AssignedResourcesChanged;
820    private void OnAssignedResourcesChanged() {
821      var handler = AssignedResourcesChanged;
822      if (handler != null) handler(this, EventArgs.Empty);
823    }
824
825    public event EventHandler ProjectsTreeViewDoubleClicked;
826    private void OnProjectsTreeViewDoubleClicked() {
827      var handler = ProjectsTreeViewDoubleClicked;
828      if (handler != null) handler(this, EventArgs.Empty);
829    }
830    #endregion
831  }
832}
Note: See TracBrowser for help on using the repository browser.