aboutsummaryrefslogtreecommitdiffstats
path: root/store/src/main/java/org/eclipse/viatra/solver/data/map/MapAsIterable.java
blob: 22b5e6c166bd2d235bb4e2378d658461852ca590 (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
package org.eclipse.viatra.solver.data.map;

import java.util.Iterator;
import java.util.function.BiFunction;
import java.util.function.BiPredicate;

public class MapAsIterable<K,V,D> implements Iterable<D> {
	private final VersionedMap<K, V> internal;
	private final BiFunction<K, V, D> entryTransformation;
	private final BiPredicate<K,V> filtering;
	
	public MapAsIterable(VersionedMap<K, V> internal, BiFunction<K, V, D> entryTransformation, BiPredicate<K,V> filtering) {
		this.internal = internal;
		this.entryTransformation = entryTransformation;
		this.filtering = filtering;
	}
	public MapAsIterable(VersionedMap<K, V> internal, BiFunction<K, V, D> entryTransformation) {
		this.internal = internal;
		this.entryTransformation = entryTransformation;
		this.filtering = ((k,v)->true);
	}
	@Override
	public Iterator<D> iterator() {
		return new CursorAsIterator<>(internal.getAll(), entryTransformation, filtering);
	}
}