aboutsummaryrefslogtreecommitdiffstats
path: root/Solvers/VIATRA-Solver/hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner/src/hu/bme/mit/inf/dslreasoner/viatrasolver/reasoner/dse/PunishSizeObjective.xtend
blob: 8505661c9ca8648c12e033245b70fb3c6c5db299 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.dse

import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PartialInterpretation
import hu.bme.mit.inf.dslreasoner.viatrasolver.partialinterpretationlanguage.partialinterpretation.PrimitiveElement
import hu.bme.mit.inf.dslreasoner.viatrasolver.reasoner.optimization.IThreeValuedObjective
import java.util.Comparator
import org.eclipse.viatra.dse.base.ThreadContext
import org.eclipse.viatra.dse.objectives.Comparators
import org.eclipse.xtend.lib.annotations.Accessors

class PunishSizeObjective implements IThreeValuedObjective {
	@Accessors int level = 3
	
	override createNew() {
		this
	}
	
	override init(ThreadContext context) {
		// Nothing to initialize.
	}
	
	override getComparator() {
		Comparators.LOWER_IS_BETTER
	}
	
	override getFitness(ThreadContext threadContext) {
		val model = threadContext.model
		if (model instanceof PartialInterpretation) {
			val size = model.newObjectCount
//			println('''size=«size»''')
			size as double
		} else {
			throw new IllegalArgumentException("notifier must be a PartialInterpretation")
		}
	}
	
	override getBestPossibleFitness(ThreadContext threadContext) {
		getFitness(threadContext)
	}
	
	override getWorstPossibleFitness(ThreadContext threadContext) {
		val model = threadContext.model
		if (model instanceof PartialInterpretation) {
			(model.newObjectCount + model.maxNewElements) as double
		} else {
			throw new IllegalArgumentException("notifier must be a PartialInterpretation")
		}
	}
	
	private def getNewObjectCount(PartialInterpretation interpretation) {
		interpretation.newElements.reject[it instanceof PrimitiveElement].size
	}
	
	override getName() {
		typeof(PunishSizeObjective).name
	}
	
	override isHardObjective() {
		false
	}
	
	override satisifiesHardObjective(Double fitness) {
		true
	}
	
	override setComparator(Comparator<Double> comparator) {
		throw new UnsupportedOperationException("Model generation objective comparator cannot be set.")
	}
	
}