Free cookie consent management tool by TermsFeed Policy Generator

source: trunk/HeuristicLab.Problems.ExternalEvaluation/3.4/ExternalEvaluationProblem.cs @ 16565

Last change on this file since 16565 was 16565, checked in by gkronber, 5 years ago

#2520: merged changes from PersistenceOverhaul branch (r16451:16564) into trunk

File size: 8.4 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2019 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.Threading;
27using Google.ProtocolBuffers;
28using HeuristicLab.Analysis;
29using HeuristicLab.Common;
30using HeuristicLab.Core;
31using HeuristicLab.Data;
32using HeuristicLab.Optimization;
33using HeuristicLab.Parameters;
34using HEAL.Attic;
35
36namespace HeuristicLab.Problems.ExternalEvaluation {
37  [Item("External Evaluation Problem (single-objective)", "A problem that is evaluated in a different process.")]
38  [Creatable(CreatableAttribute.Categories.ExternalEvaluationProblems, Priority = 100)]
39  [StorableType("115EB3A5-A8A8-4A2E-9799-9485FE896DEC")]
40  // BackwardsCompatibility3.3
41  // Rename class to SingleObjectiveExternalEvaluationProblem
42  public class ExternalEvaluationProblem : SingleObjectiveBasicProblem<IEncoding>, IExternalEvaluationProblem {
43
44    public static new Image StaticItemImage {
45      get { return HeuristicLab.Common.Resources.VSImageLibrary.Type; }
46    }
47
48    #region Parameters
49    public OptionalValueParameter<EvaluationCache> CacheParameter {
50      get { return (OptionalValueParameter<EvaluationCache>)Parameters["Cache"]; }
51    }
52    public IValueParameter<CheckedItemCollection<IEvaluationServiceClient>> ClientsParameter {
53      get { return (IValueParameter<CheckedItemCollection<IEvaluationServiceClient>>)Parameters["Clients"]; }
54    }
55    public IValueParameter<SolutionMessageBuilder> MessageBuilderParameter {
56      get { return (IValueParameter<SolutionMessageBuilder>)Parameters["MessageBuilder"]; }
57    }
58    public IFixedValueParameter<SingleObjectiveOptimizationSupportScript> SupportScriptParameter {
59      get { return (IFixedValueParameter<SingleObjectiveOptimizationSupportScript>)Parameters["SupportScript"]; }
60    }
61
62    private IFixedValueParameter<BoolValue> MaximizationParameter {
63      get { return (IFixedValueParameter<BoolValue>)Parameters["Maximization"]; }
64    }
65    #endregion
66
67    #region Properties
68    public new IEncoding Encoding {
69      get { return base.Encoding; }
70      set { base.Encoding = value; }
71    }
72    public EvaluationCache Cache {
73      get { return CacheParameter.Value; }
74    }
75    public CheckedItemCollection<IEvaluationServiceClient> Clients {
76      get { return ClientsParameter.Value; }
77    }
78    public SolutionMessageBuilder MessageBuilder {
79      get { return MessageBuilderParameter.Value; }
80    }
81    public SingleObjectiveOptimizationSupportScript OptimizationSupportScript {
82      get { return SupportScriptParameter.Value; }
83    }
84    private ISingleObjectiveOptimizationSupport OptimizationSupport {
85      get { return SupportScriptParameter.Value; }
86    }
87    #endregion
88
89    [StorableConstructor]
90    protected ExternalEvaluationProblem(StorableConstructorFlag _) : base(_) { }
91    protected ExternalEvaluationProblem(ExternalEvaluationProblem original, Cloner cloner) : base(original, cloner) { }
92    public override IDeepCloneable Clone(Cloner cloner) {
93      return new ExternalEvaluationProblem(this, cloner);
94    }
95    public ExternalEvaluationProblem()
96      : base() {
97      Parameters.Remove("Maximization"); // readonly in base class
98      Parameters.Add(new FixedValueParameter<BoolValue>("Maximization", "Set to false if the problem should be minimized.", new BoolValue()));
99      Parameters.Add(new OptionalValueParameter<EvaluationCache>("Cache", "Cache of previously evaluated solutions."));
100      Parameters.Add(new ValueParameter<CheckedItemCollection<IEvaluationServiceClient>>("Clients", "The clients that are used to communicate with the external application.", new CheckedItemCollection<IEvaluationServiceClient>() { new EvaluationServiceClient() }));
101      Parameters.Add(new ValueParameter<SolutionMessageBuilder>("MessageBuilder", "The message builder that converts from HeuristicLab objects to SolutionMessage representation.", new SolutionMessageBuilder()) { Hidden = true });
102      Parameters.Add(new FixedValueParameter<SingleObjectiveOptimizationSupportScript>("SupportScript", "A script that can provide neighborhood and analyze the results of the optimization.", new SingleObjectiveOptimizationSupportScript()));
103
104      Operators.Add(new BestScopeSolutionAnalyzer());
105    }
106
107    #region Single Objective Problem Overrides
108    public override bool Maximization {
109      get { return Parameters.ContainsKey("Maximization") && ((IValueParameter<BoolValue>)Parameters["Maximization"]).Value.Value; }
110    }
111
112    public virtual void SetMaximization(bool maximization) {
113      MaximizationParameter.Value.Value = maximization;
114    }
115
116    public override double Evaluate(Individual individual, IRandom random) {
117      var qualityMessage = Evaluate(BuildSolutionMessage(individual));
118      if (!qualityMessage.HasExtension(SingleObjectiveQualityMessage.QualityMessage_))
119        throw new InvalidOperationException("The received message is not a SingleObjectiveQualityMessage.");
120      return qualityMessage.GetExtension(SingleObjectiveQualityMessage.QualityMessage_).Quality;
121    }
122    public virtual QualityMessage Evaluate(SolutionMessage solutionMessage) {
123      return Cache == null
124        ? EvaluateOnNextAvailableClient(solutionMessage)
125        : Cache.GetValue(solutionMessage, EvaluateOnNextAvailableClient, GetQualityMessageExtensions());
126    }
127
128    public override void Analyze(Individual[] individuals, double[] qualities, ResultCollection results, IRandom random) {
129      OptimizationSupport.Analyze(individuals, qualities, results, random);
130    }
131
132    public override IEnumerable<Individual> GetNeighbors(Individual individual, IRandom random) {
133      return OptimizationSupport.GetNeighbors(individual, random);
134    }
135    #endregion
136
137    public virtual ExtensionRegistry GetQualityMessageExtensions() {
138      var extensions = ExtensionRegistry.CreateInstance();
139      extensions.Add(SingleObjectiveQualityMessage.QualityMessage_);
140      return extensions;
141    }
142
143    #region Evaluation
144    private HashSet<IEvaluationServiceClient> activeClients = new HashSet<IEvaluationServiceClient>();
145    private object clientLock = new object();
146
147    private QualityMessage EvaluateOnNextAvailableClient(SolutionMessage message) {
148      IEvaluationServiceClient client = null;
149      lock (clientLock) {
150        client = Clients.CheckedItems.FirstOrDefault(c => !activeClients.Contains(c));
151        while (client == null && Clients.CheckedItems.Any()) {
152          Monitor.Wait(clientLock);
153          client = Clients.CheckedItems.FirstOrDefault(c => !activeClients.Contains(c));
154        }
155        if (client != null)
156          activeClients.Add(client);
157      }
158      try {
159        return client.Evaluate(message, GetQualityMessageExtensions());
160      }
161      finally {
162        lock (clientLock) {
163          activeClients.Remove(client);
164          Monitor.PulseAll(clientLock);
165        }
166      }
167    }
168
169    private SolutionMessage BuildSolutionMessage(Individual individual, int solutionId = 0) {
170      lock (clientLock) {
171        SolutionMessage.Builder protobufBuilder = SolutionMessage.CreateBuilder();
172        protobufBuilder.SolutionId = solutionId;
173        foreach (var variable in individual.Values) {
174          try {
175            MessageBuilder.AddToMessage(variable.Value, variable.Key, protobufBuilder);
176          }
177          catch (ArgumentException ex) {
178            throw new InvalidOperationException(string.Format("ERROR while building solution message: Parameter {0} cannot be added to the message", Name), ex);
179          }
180        }
181        return protobufBuilder.Build();
182      }
183    }
184    #endregion
185  }
186}
Note: See TracBrowser for help on using the repository browser.