Free cookie consent management tool by TermsFeed Policy Generator

source: branches/HeuristicLab.Problems.BioBoost/HeuristicLab.Problems.BioBoost.Views/3.3/MenuItems/ExportSolutionMenuItem.cs @ 13072

Last change on this file since 13072 was 13072, checked in by gkronber, 7 years ago

#2499: added code from HeuristicLab.BioBoost.Views (from private repository) nothing much has been changed

File size: 7.6 KB
Line 
1using HeuristicLab.BioBoost.Representation;
2using HeuristicLab.Common;
3using HeuristicLab.Core;
4using HeuristicLab.Data;
5using HeuristicLab.MainForm;
6using HeuristicLab.Optimization;
7using HeuristicLab.Optimizer;
8using System;
9using System.Collections.Generic;
10using System.Globalization;
11using System.IO;
12using System.Linq;
13using System.Text;
14using System.Windows.Forms;
15using MenuItem = HeuristicLab.MainForm.WindowsForms.MenuItem;
16
17
18namespace HeuristicLab.BioBoost.Views {
19  internal sealed class ExportSolutionMenuItem : MenuItem, IOptimizerUserInterfaceItemProvider {
20    public override string Name {
21      get { return "Export CSV"; }
22    }
23
24    public override IEnumerable<string> Structure {
25      get { return new string[] { "&File" }; }
26    }
27
28    public override int Position {
29      get { return 1450; }
30    }
31
32    protected override void OnToolStripItemSet(EventArgs e) {
33      ToolStripItem.Enabled = false;
34    }
35
36    protected override void OnActiveViewChanged(object sender, EventArgs e) {
37      IContentView activeView = MainFormManager.MainForm.ActiveView as IContentView;
38      if (activeView == null) {
39        ToolStripItem.Enabled = false;
40        return;
41      }
42
43      var content = activeView.Content;
44      RunCollection runCollection = content as RunCollection;
45      var optimizer = content as IOptimizer;
46      if (runCollection == null && optimizer != null) {
47        runCollection = ((IOptimizer)content).Runs;
48      }
49
50      var alg = content as IAlgorithm;
51      var solution = content as BioBoostCompoundSolution;
52
53      ToolStripItem.Enabled =
54        // at least on run with a result || or the alg is paused and results contains a bioboost solution
55        (runCollection != null && runCollection.Any(run => run.Results.Any(p => p.Value is BioBoostCompoundSolution))) ||
56        (alg != null && alg.ExecutionState == ExecutionState.Paused && alg.Results.Any(r => r.Value is BioBoostCompoundSolution)) ||
57        (solution != null)
58        ;
59    }
60
61    public override void Execute() {
62      var folderBrowserDialog = new FolderBrowserDialog();
63      folderBrowserDialog.Description = "Select the export directory.";
64      folderBrowserDialog.RootFolder = Environment.SpecialFolder.Desktop;
65      if (folderBrowserDialog.ShowDialog() == DialogResult.OK) {
66        var folderName = folderBrowserDialog.SelectedPath;
67        IContentView activeView = (IContentView)MainFormManager.MainForm.ActiveView;
68        var mainForm = (MainForm.WindowsForms.MainForm)MainFormManager.MainForm;
69        mainForm.AddOperationProgressToContent(activeView.Content, "Export solutions.");
70
71        Action<IContentView> action = (view) => {
72          var i = 0;
73          // export the solutions of all runs
74          foreach (var run in view.Content.GetObjectGraphObjects(excludeStaticMembers: true).OfType<Run>()) {
75            var solution =
76              run.GetObjectGraphObjects(excludeStaticMembers: true).OfType<BioBoostCompoundSolution>().FirstOrDefault();
77            if (solution == null) continue;
78            ExportSolution(folderName, i++, run.Name, solution);
79          }
80          // export the solutions in the results of the paused algorithm
81          var alg = view.Content as IAlgorithm;
82          if (alg != null && alg.ExecutionState == ExecutionState.Paused) {
83            foreach (var solution in alg.Results.Select(r => r.Value).OfType<BioBoostCompoundSolution>()) {
84              ExportSolution(folderName, i++, alg.Name, solution);
85            }
86          }
87
88          // export the solution itself
89          var bioBoostSolution = view.Content as BioBoostCompoundSolution;
90          if (bioBoostSolution != null)
91            ExportSolution(folderName, i++, bioBoostSolution.Name, bioBoostSolution);
92        };
93
94        action.BeginInvoke(activeView, delegate(IAsyncResult result) {
95          try {
96            action.EndInvoke(result);
97          } catch (SystemException e) {
98            var dialog = new HeuristicLab.PluginInfrastructure.ErrorDialog(e);
99            dialog.ShowDialog();
100          }
101          mainForm.RemoveOperationProgressFromContent(activeView.Content);
102        }, null);
103      }
104    }
105
106    private void ExportSolution(string folderName, int i, string name, BioBoostCompoundSolution solution) {
107      var csv = CreateCsvSolution(solution);
108      var filename = string.Format("{0}-{1}.csv", i, name);
109      // remove invalid characters
110      foreach (var ch in Path.GetInvalidFileNameChars()) {
111        filename = filename.Replace(ch, '_');
112
113      }
114      var path = Path.Combine(folderName, filename);
115      if (path.Length >= 248) throw new PathTooLongException("The path is longer than 248 characters. Try to save your files in a different folder");
116      if (filename.Length >= 260) throw new PathTooLongException("The filename is longer than 260 characters. Please change the name of runs to something shorter.");
117      File.WriteAllText(path, csv);
118    }
119
120    private string CreateCsvSolution(BioBoostCompoundSolution content) {
121      content.EvaluateAndUpdateAllResults();
122      var sb = new StringBuilder();
123      // header
124      sb.Append("NUTS_ID").Append(",");
125      sb.Append(string.Join(",", content.IntValues.Select(x => string.Format("\"{0}\"", x.Key)))).Append(",");
126      sb.Append(string.Join(",", content.IntValues.Select(x => string.Format("\"{0}-Region\"", x.Key)))).Append(",");
127      sb.Append(string.Join(",", content.DoubleValues.Select(x => string.Format("\"{0}\"", x.Key)))).Append(",");
128      sb.Append(string.Join(",", content.StringValues.Select(x => string.Format("\"{0}\"", x.Key))));
129      sb.Append(Environment.NewLine);
130      // rows
131      for (int i = 0; i < content.LocationNames.Length; i++) {
132        sb.Append(content.LocationNames[i]);
133        foreach (var kvp in content.IntValues) { sb.Append(",").Append(kvp.Value[i].ToString("D", CultureInfo.InvariantCulture)); }
134        foreach (var kvp in content.IntValues) { sb.Append(",").Append(kvp.Value[i] < 0 ? "" : content.LocationNames[kvp.Value[i]]); }
135        foreach (var kvp in content.DoubleValues) { sb.Append(",").Append(kvp.Value[i].ToString("G", CultureInfo.InvariantCulture)); }
136        foreach (var kvp in content.StringValues) { sb.Append(",").Append(kvp.Value[i]); }
137        sb.Append(Environment.NewLine);
138      }
139
140      return sb.ToString();
141    }
142
143    /*
144    private string CreateJsonSolution(BioBoostCompoundSolution content) {
145      var builder = new StringBuilder();
146      builder.Append("{ \"type\": \"FeatureCollection\", \n\"features\": [");
147      for (int i = 0; i < content.LocationNames.Length; i++) {
148        if (i > 0) builder.Append(",");
149        builder.Append("\n{ \"type\": \"Feature\", \"properties\": { \"NUTS_ID\": \"");
150        builder.Append(content.LocationNames[i]);
151        builder.Append("\"");
152        foreach (var kvp in content.IntValues) {
153          var target = content.LocationNames[i];
154          if (kvp.Value[i] >= 0) {
155            target = content.LocationNames[kvp.Value[i]];
156          }
157          builder.Append(",\n                                     \"");
158          builder.Append(kvp.Key);
159          builder.Append("\": \"");
160          builder.Append(target);
161          builder.Append("\"");
162        }
163        foreach (var kvp in content.DoubleValues) {
164          builder.Append(",\n                                     \"");
165          builder.Append(kvp.Key);
166          builder.Append("\": ");
167          builder.Append(kvp.Value[i].ToString("R", CultureInfo.InvariantCulture));
168        }
169        builder.Append("} }");
170      }
171      builder.Append("\n]\n}");
172
173      return builder.ToString();
174    }
175    */
176  }
177}
Note: See TracBrowser for help on using the repository browser.