aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/logic/src/main/java/tools/refinery/logic/term/cardinalityinterval/NonEmptyCardinalityInterval.java
blob: efe1464ee9ea32acf3e1569ef75d931ee8c31e5f (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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
/*
 * SPDX-FileCopyrightText: 2021-2024 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.logic.term.cardinalityinterval;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import tools.refinery.logic.term.uppercardinality.FiniteUpperCardinality;
import tools.refinery.logic.term.uppercardinality.UpperCardinality;

import java.util.Objects;
import java.util.function.BinaryOperator;
import java.util.function.IntBinaryOperator;

public record NonEmptyCardinalityInterval(int lowerBound, UpperCardinality upperBound) implements CardinalityInterval {
	public NonEmptyCardinalityInterval {
		if (lowerBound < 0) {
			throw new IllegalArgumentException("lowerBound must not be negative");
		}
		if (upperBound.compareToInt(lowerBound) < 0) {
			throw new IllegalArgumentException("lowerBound must not be larger than upperBound");
		}
	}

	@Override
	@Nullable
	public Integer getConcrete() {
        return isConcrete() ? lowerBound : null;
    }

	@Override
	public boolean isConcrete() {
		return upperBound instanceof FiniteUpperCardinality finiteUpperCardinality &&
				finiteUpperCardinality.finiteUpperBound() == lowerBound;
	}

	@Override
	@NotNull
	public Integer getArbitrary() {
		return lowerBound;
	}

	@Override
	public boolean isError() {
		return false;
	}

	@Override
	public boolean isRefinementOf(CardinalityInterval other) {
		if (!(other instanceof NonEmptyCardinalityInterval nonEmptyOther)) {
			return false;
		}
		return lowerBound >= nonEmptyOther.lowerBound() && upperBound.compareTo(nonEmptyOther.upperBound()) <= 0;
	}

	@Override
	public CardinalityInterval min(CardinalityInterval other) {
		return lift(other, Math::min, UpperCardinality::min);
	}

	@Override
	public CardinalityInterval max(CardinalityInterval other) {
		return lift(other, Math::max, UpperCardinality::max);
	}

	@Override
	public CardinalityInterval add(CardinalityInterval other) {
		return lift(other, Integer::sum, UpperCardinality::add);
	}

	@Override
	public CardinalityInterval multiply(CardinalityInterval other) {
		return lift(other, (a, b) -> a * b, UpperCardinality::multiply);
	}

	@Override
	public CardinalityInterval meet(CardinalityInterval other) {
		return lift(other, Math::max, UpperCardinality::min);
	}

	@Override
	public CardinalityInterval join(CardinalityInterval other) {
		return lift(other, Math::min, UpperCardinality::max, this);
	}

	@Override
	public CardinalityInterval take(int count) {
		int newLowerBound = Math.max(lowerBound - count, 0);
		var newUpperBound = upperBound.take(count);
		if (newUpperBound == null) {
			return CardinalityIntervals.ERROR;
		}
		return CardinalityIntervals.between(newLowerBound, newUpperBound);
	}

	private CardinalityInterval lift(CardinalityInterval other, IntBinaryOperator lowerOperator,
									 BinaryOperator<UpperCardinality> upperOperator,
									 CardinalityInterval whenEmpty) {
		if (other instanceof NonEmptyCardinalityInterval nonEmptyOther) {
			return CardinalityIntervals.between(lowerOperator.applyAsInt(lowerBound, nonEmptyOther.lowerBound),
					upperOperator.apply(upperBound, nonEmptyOther.upperBound));
		}
		if (other instanceof EmptyCardinalityInterval) {
			return whenEmpty;
		}
		throw new IllegalArgumentException("Unknown CardinalityInterval: " + other);
	}

	private CardinalityInterval lift(CardinalityInterval other, IntBinaryOperator lowerOperator,
									 BinaryOperator<UpperCardinality> upperOperator) {
		return lift(other, lowerOperator, upperOperator, CardinalityIntervals.ERROR);
	}

	@Override
	public String toString() {
		if (upperBound instanceof FiniteUpperCardinality finiteUpperCardinality &&
				finiteUpperCardinality.finiteUpperBound() == lowerBound) {
			return "[%d]".formatted(lowerBound);
		}
		return "[%d..%s]".formatted(lowerBound, upperBound);
	}

	@Override
	public boolean equals(Object o) {
		if (this == o) return true;
		if (o == null || getClass() != o.getClass()) return false;
		NonEmptyCardinalityInterval that = (NonEmptyCardinalityInterval) o;
		return lowerBound == that.lowerBound && Objects.equals(upperBound, that.upperBound);
	}

	@Override
	public int hashCode() {
		return lowerBound * 31 + upperBound.hashCode();
	}
}