MDEOptimiser implementation of CPS benchmark

1 files changed, 23 insertions, 0 deletions

diff --git a/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/src/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/TotalCostFitnessFunction.xtend b/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/src/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/TotalCostFitnessFunction.xtend
new file mode 100644
index 00000000..af65e442
--- /dev/null
+++ b/Domains/hu.bme.mit.inf.dslreasoner.domains.cps/src/hu/bme/mit/inf/dslreasoner/domains/cps/mdeo/TotalCostFitnessFunction.xtend
@@ -0,0 +1,23 @@
+package hu.bme.mit.inf.dslreasoner.domains.cps.mdeo
+
+import hu.bme.mit.inf.dslreasoner.domains.cps.CyberPhysicalSystem
+import uk.ac.kcl.inf.mdeoptimiser.libraries.core.optimisation.IGuidanceFunction
+import uk.ac.kcl.inf.mdeoptimiser.libraries.core.optimisation.interpreter.guidance.Solution
+
+class TotalCostFitnessFunction implements IGuidanceFunction {
+        override getName() {
+                "TotalCost"
+        }
+
+        override computeFitness(Solution solution) {
+                val cps = solution.model as CyberPhysicalSystem
+                var int cost = 0
+                for (appType : cps.applicationTypes) {
+                        cost += 5 * appType.instances.size
+                }
+                for (hostType : cps.hostTypes) {
+                        cost += hostType.cost * hostType.instances.size
+                }
+                cost
+        }
+}