Free cookie consent management tool by TermsFeed Policy Generator

source: stable/HeuristicLab.Encodings.SymbolicExpressionTreeEncoding/3.4/SymbolicExpressionTreeNode.cs @ 12702

Last change on this file since 12702 was 12702, checked in by mkommend, 9 years ago

#2276: Merged changes to stable.

File size: 7.3 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2015 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 HeuristicLab.Common;
25using HeuristicLab.Core;
26using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;
27
28namespace HeuristicLab.Encodings.SymbolicExpressionTreeEncoding {
29  [StorableClass]
30  public class SymbolicExpressionTreeNode : DeepCloneable, ISymbolicExpressionTreeNode {
31    [Storable]
32    private IList<ISymbolicExpressionTreeNode> subtrees;
33    [Storable]
34    private ISymbol symbol;
35
36    // cached values to prevent unnecessary tree iterations
37    private ushort length;
38    private ushort depth;
39
40    public ISymbol Symbol {
41      get { return symbol; }
42      protected set { symbol = value; }
43    }
44
45    // parent relation is not persisted or cloned (will be set on AddSubtree or RemoveSubtree)
46    private ISymbolicExpressionTreeNode parent;
47    public ISymbolicExpressionTreeNode Parent {
48      get { return parent; }
49      set { parent = value; }
50    }
51
52    [StorableConstructor]
53    protected SymbolicExpressionTreeNode(bool deserializing) { }
54    protected SymbolicExpressionTreeNode(SymbolicExpressionTreeNode original, Cloner cloner)
55      : base(original, cloner) {
56      symbol = original.symbol; // symbols are reused
57      length = original.length;
58      depth = original.depth;
59      if (original.subtrees != null) {
60        subtrees = new List<ISymbolicExpressionTreeNode>(original.subtrees.Count);
61        foreach (var subtree in original.subtrees) {
62          var clonedSubtree = cloner.Clone(subtree);
63          subtrees.Add(clonedSubtree);
64          clonedSubtree.Parent = this;
65        }
66      }
67    }
68    public override IDeepCloneable Clone(Cloner cloner) {
69      return new SymbolicExpressionTreeNode(this, cloner);
70    }
71
72    internal SymbolicExpressionTreeNode()
73      : base() {
74      // don't allocate subtrees list here!
75      // because we don't want to allocate it in terminal nodes
76    }
77
78    public SymbolicExpressionTreeNode(ISymbol symbol)
79      : base() {
80      subtrees = new List<ISymbolicExpressionTreeNode>(3);
81      this.symbol = symbol;
82    }
83
84
85    [StorableHook(HookType.AfterDeserialization)]
86    private void AfterDeserialization() {
87      if (subtrees != null) {
88        foreach (var subtree in subtrees)
89          subtree.Parent = this;
90      }
91    }
92
93    public virtual bool HasLocalParameters {
94      get { return false; }
95    }
96
97    public virtual IEnumerable<ISymbolicExpressionTreeNode> Subtrees {
98      get { return subtrees; }
99    }
100
101    public virtual ISymbolicExpressionTreeGrammar Grammar {
102      get { return parent.Grammar; }
103    }
104
105    public int GetLength() {
106      if (length > 0) return length;
107      else {
108        length = 1;
109        if (subtrees != null) {
110          for (int i = 0; i < subtrees.Count; i++) {
111            checked { length += (ushort)subtrees[i].GetLength(); }
112          }
113        }
114        return length;
115      }
116    }
117
118    public int GetDepth() {
119      if (depth > 0) return depth;
120      else {
121        if (subtrees != null) {
122          for (int i = 0; i < subtrees.Count; i++) depth = Math.Max(depth, (ushort)subtrees[i].GetDepth());
123        }
124        depth++;
125        return depth;
126      }
127    }
128
129    public int GetBranchLevel(ISymbolicExpressionTreeNode child) {
130      return GetBranchLevel(this, child);
131    }
132
133    private static int GetBranchLevel(ISymbolicExpressionTreeNode root, ISymbolicExpressionTreeNode point) {
134      if (root == point)
135        return 0;
136      foreach (var subtree in root.Subtrees) {
137        int branchLevel = GetBranchLevel(subtree, point);
138        if (branchLevel < int.MaxValue)
139          return 1 + branchLevel;
140      }
141      return int.MaxValue;
142    }
143
144    public virtual void ResetLocalParameters(IRandom random) { }
145    public virtual void ShakeLocalParameters(IRandom random, double shakingFactor) { }
146
147    public int SubtreeCount {
148      get {
149        if (subtrees == null) return 0;
150        return subtrees.Count;
151      }
152    }
153
154    public virtual ISymbolicExpressionTreeNode GetSubtree(int index) {
155      return subtrees[index];
156    }
157    public virtual int IndexOfSubtree(ISymbolicExpressionTreeNode tree) {
158      return subtrees.IndexOf(tree);
159    }
160    public virtual void AddSubtree(ISymbolicExpressionTreeNode tree) {
161      subtrees.Add(tree);
162      tree.Parent = this;
163      ResetCachedValues();
164    }
165    public virtual void InsertSubtree(int index, ISymbolicExpressionTreeNode tree) {
166      subtrees.Insert(index, tree);
167      tree.Parent = this;
168      ResetCachedValues();
169    }
170    public virtual void RemoveSubtree(int index) {
171      subtrees[index].Parent = null;
172      subtrees.RemoveAt(index);
173      ResetCachedValues();
174    }
175
176    public IEnumerable<ISymbolicExpressionTreeNode> IterateNodesBreadth() {
177      var list = new List<ISymbolicExpressionTreeNode>(GetLength()) { this };
178      int i = 0;
179      while (i != list.Count) {
180        for (int j = 0; j != list[i].SubtreeCount; ++j)
181          list.Add(list[i].GetSubtree(j));
182        ++i;
183      }
184      return list;
185    }
186
187    public IEnumerable<ISymbolicExpressionTreeNode> IterateNodesPrefix() {
188      List<ISymbolicExpressionTreeNode> list = new List<ISymbolicExpressionTreeNode>();
189      ForEachNodePrefix((n) => list.Add(n));
190      return list;
191    }
192
193    public void ForEachNodePrefix(Action<ISymbolicExpressionTreeNode> a) {
194      a(this);
195      if (subtrees != null) {
196        //avoid linq to reduce memory pressure
197        for (int i = 0; i < subtrees.Count; i++) {
198          subtrees[i].ForEachNodePrefix(a);
199        }
200      }
201    }
202
203    public IEnumerable<ISymbolicExpressionTreeNode> IterateNodesPostfix() {
204      List<ISymbolicExpressionTreeNode> list = new List<ISymbolicExpressionTreeNode>();
205      ForEachNodePostfix((n) => list.Add(n));
206      return list;
207    }
208
209    public void ForEachNodePostfix(Action<ISymbolicExpressionTreeNode> a) {
210      if (subtrees != null) {
211        //avoid linq to reduce memory pressure
212        for (int i = 0; i < subtrees.Count; i++) {
213          subtrees[i].ForEachNodePostfix(a);
214        }
215      }
216      a(this);
217    }
218
219    public override string ToString() {
220      if (Symbol != null) return Symbol.Name;
221      return "SymbolicExpressionTreeNode";
222    }
223
224    private void ResetCachedValues() {
225      length = 0; depth = 0;
226      SymbolicExpressionTreeNode parentNode = parent as SymbolicExpressionTreeNode;
227      if (parentNode != null) parentNode.ResetCachedValues();
228    }
229  }
230}
Note: See TracBrowser for help on using the repository browser.