public static Map<KLabel, KLabel> collectionFor(Module m) {
return stream(m.productions())
.filter(p -> p.att().contains(Attribute.ASSOCIATIVE_KEY))
.flatMap(
p -> {
Set<Tuple2<KLabel, KLabel>> set = new HashSet<>();
set.add(Tuple2.apply(p.klabel().get(), p.klabel().get()));
if (p.att().contains(Attribute.UNIT_KEY)) {
set.add(
Tuple2.apply(
KLabel(p.att().<String>get(Attribute.UNIT_KEY).get()), p.klabel().get()));
}
if (p.att().contains("element")) {
set.add(
Tuple2.apply(KLabel(p.att().<String>get("element").get()), p.klabel().get()));
}
if (p.att().contains("wrapElement")) {
set.add(
Tuple2.apply(
KLabel(p.att().<String>get("wrapElement").get()), p.klabel().get()));
}
return set.stream();
})
.distinct()
.collect(Collectors.toMap(Tuple2::_1, Tuple2::_2));
}
public Module gen(Module mod) {
Set<Sentence> res = new HashSet<>();
for (Sort sort : iterable(mod.definedSorts())) {
Production prod =
Production(
"is" + sort.name(),
Sorts.Bool(),
Seq(
Terminal("is" + sort.name()),
Terminal("("),
NonTerminal(Sorts.K()),
Terminal(")")),
Att().add(Attribute.FUNCTION_KEY).add(Attribute.PREDICATE_KEY, sort.name()));
if (!mod.productions().contains(prod)) res.add(prod);
}
return Module(
mod.name(),
mod.imports(),
(scala.collection.immutable.Set<Sentence>) mod.localSentences().$bar(immutable(res)),
mod.att());
}
