1 | using System;
|
---|
2 | using System.Collections.Generic;
|
---|
3 | using System.Linq;
|
---|
4 | using HeuristicLab.Algorithms.GeneticAlgorithm;
|
---|
5 | using HeuristicLab.Core;
|
---|
6 | using HeuristicLab.Data;
|
---|
7 | using HeuristicLab.Encodings.PermutationEncoding;
|
---|
8 | using HeuristicLab.Optimization;
|
---|
9 | using HeuristicLab.Problems.TravelingSalesman;
|
---|
10 | using Newtonsoft.Json;
|
---|
11 | using Newtonsoft.Json.Linq;
|
---|
12 | using Newtonsoft.Json.Serialization;
|
---|
13 |
|
---|
14 | namespace HeuristicLab.JsonInterface {
|
---|
15 | public class JCGenerator {
|
---|
16 |
|
---|
17 | private JObject template = JObject.Parse(@"{
|
---|
18 | 'Metadata': {
|
---|
19 | 'Algorithm':'',
|
---|
20 | 'Problem':''
|
---|
21 | },
|
---|
22 | 'Objects': [],
|
---|
23 | 'Types': {}
|
---|
24 | }");
|
---|
25 |
|
---|
26 | private Dictionary<string, string> TypeList = new Dictionary<string, string>();
|
---|
27 |
|
---|
28 | JArray jsonItems = new JArray();
|
---|
29 | private void PopulateJsonItems(JsonItem item) {
|
---|
30 | if (item.Parameters != null) {
|
---|
31 | if(item.Range == null)
|
---|
32 | jsonItems.Add(Serialize(item));
|
---|
33 | foreach (var p in item.Parameters)
|
---|
34 | if(p.Parameters != null)
|
---|
35 | PopulateJsonItems(p);
|
---|
36 | }
|
---|
37 | }
|
---|
38 |
|
---|
39 | private JObject Serialize(JsonItem item) {
|
---|
40 | JObject obj = JObject.FromObject(item, Settings());
|
---|
41 | obj["StaticParameters"] = obj["Parameters"];
|
---|
42 | obj["FreeParameters"] = obj["Parameters"];
|
---|
43 |
|
---|
44 | obj.Property("Parameters")?.Remove();
|
---|
45 | RefactorFreeParameters(obj, null);
|
---|
46 | RefactorStaticParameters(obj);
|
---|
47 |
|
---|
48 | obj.Property("Default")?.Remove();
|
---|
49 | obj.Property("Type")?.Remove();
|
---|
50 |
|
---|
51 | TypeList.Add(item.Path, item.Type);
|
---|
52 | return obj;
|
---|
53 | }
|
---|
54 |
|
---|
55 | public string GenerateTemplate(IAlgorithm algorithm, IProblem problem, params string[] freeParameters) {
|
---|
56 | algorithm.Problem = problem;
|
---|
57 | JsonItem algorithmData = JsonItemConverter.Extract(algorithm);
|
---|
58 | JsonItem problemData = JsonItemConverter.Extract(problem);
|
---|
59 | PopulateJsonItems(algorithmData);
|
---|
60 | PopulateJsonItems(problemData);
|
---|
61 |
|
---|
62 | template["Metadata"]["Algorithm"] = algorithm.Name;
|
---|
63 | template["Metadata"]["Problem"] = problem.Name;
|
---|
64 | template["Objects"] = jsonItems;
|
---|
65 | template["Types"] = JObject.FromObject(TypeList);
|
---|
66 |
|
---|
67 | return CustomJsonWriter.Serialize(template);
|
---|
68 | }
|
---|
69 |
|
---|
70 | #region Helper
|
---|
71 | private void RefactorFreeParameters(JToken token, string[] freeParameters) {
|
---|
72 |
|
---|
73 |
|
---|
74 | IList<JObject> objToRemove = new List<JObject>();
|
---|
75 | TransformNodes(x => {
|
---|
76 | var p = x.ToObject<JsonItem>();
|
---|
77 |
|
---|
78 | /*bool isSelected = false;
|
---|
79 | string name = x["Name"].ToObject<string>();
|
---|
80 | foreach (var selected in freeParameters)
|
---|
81 | isSelected = (name == selected || isSelected);
|
---|
82 | */
|
---|
83 | if (/*!isSelected ||*/ p.Default == null || (p.Default != null && p.Default.GetType() == typeof(string) && p.Range == null)) {
|
---|
84 | objToRemove.Add(x);
|
---|
85 | } else {
|
---|
86 | x.Property("Path")?.Remove();
|
---|
87 | x.Property("Type")?.Remove();
|
---|
88 | x.Property("Parameters")?.Remove();
|
---|
89 | }
|
---|
90 | }, token["FreeParameters"]);
|
---|
91 | foreach (var x in objToRemove) x.Remove();
|
---|
92 |
|
---|
93 | }
|
---|
94 |
|
---|
95 | private void RefactorStaticParameters(JToken token) {
|
---|
96 | IList<JObject> objToRemove = new List<JObject>();
|
---|
97 | TransformNodes(x => {
|
---|
98 | var p = x.ToObject<JsonItem>();
|
---|
99 | x.Property("Range")?.Remove();
|
---|
100 | x.Property("Operators")?.Remove();
|
---|
101 | x.Property("Parameters")?.Remove();
|
---|
102 | x.Property("Type")?.Remove();
|
---|
103 | if (p.Default == null) objToRemove.Add(x);
|
---|
104 | }, token["StaticParameters"]);
|
---|
105 | foreach (var x in objToRemove) x.Remove();
|
---|
106 | }
|
---|
107 |
|
---|
108 | private JsonSerializer Settings() => new JsonSerializer() {
|
---|
109 | TypeNameHandling = TypeNameHandling.None,
|
---|
110 | NullValueHandling = NullValueHandling.Ignore,
|
---|
111 | ReferenceLoopHandling = ReferenceLoopHandling.Serialize
|
---|
112 | };
|
---|
113 |
|
---|
114 | private void TransformNodes(Action<JObject> action, params JToken[] tokens) {
|
---|
115 | foreach(JObject obj in tokens.SelectMany(x => x.Children<JObject>()))
|
---|
116 | action(obj);
|
---|
117 | }
|
---|
118 | #endregion
|
---|
119 | }
|
---|
120 | }
|
---|