aboutsummaryrefslogtreecommitdiffstats
path: root/subprojects/logic/src/main/java/tools/refinery/logic/term/cardinalityinterval/NonEmptyCardinalityInterval.java
blob: 0919fc36fc4db65358ae8eed00cbe0b4a42c566f (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
/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.logic.term.cardinalityinterval;

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
	public boolean isEmpty() {
		return false;
	}

	@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();
	}
}