use of org.spongepowered.common.inventory.lens.impl.slot.DelegatingSlotLens in project SpongeCommon by SpongePowered.
the class SpongeQuery method reduce.
protected Map<Lens, Integer> reduce(Fabric fabric, Lens lens, Map<Lens, Integer> matches) {
if (matches.isEmpty()) {
return Collections.emptyMap();
}
// Check if all matches are the direct children of this lens
List<Lens> lensSlots = lens.getChildren();
if (lensSlots.size() == matches.size() && matches.keySet().containsAll(lensSlots)) {
// return parent lens instead of constructing a new for the query result
matches.clear();
matches.put(lens, 0);
return matches;
}
// Remove duplicate slot-lenses
Map<SlotLens, Map<Key<?>, Object>> lenses = new LinkedHashMap<>();
Map<Lens, Integer> toRemove = new HashMap<>();
for (Map.Entry<Lens, Integer> entry : matches.entrySet()) {
final Lens slotLens = entry.getKey();
if (slotLens.slotCount() == 1) {
// Remove Lens with one slot
toRemove.put(slotLens, matches.get(slotLens));
// Find SlotLens for that Lens
final SlotLens sl = slotLens.getSlotLens(fabric, 0);
final Lens parent = slotLens.getParent();
final Map<Key<?>, Object> dataAt = parent == null ? Collections.emptyMap() : parent.getDataFor(slotLens);
// Collect all data for the SlotLens
lenses.computeIfAbsent(sl, k -> new HashMap<>()).putAll(dataAt);
}
}
// remove all single-slot lenses
matches.keySet().removeAll(toRemove.keySet());
for (Map.Entry<SlotLens, Map<Key<?>, Object>> entry : lenses.entrySet()) {
final Map<Key<?>, Object> data = entry.getValue();
if (data.isEmpty()) {
// add back slot-lenses
matches.put(entry.getKey(), toRemove.getOrDefault(entry.getKey(), 0));
} else {
// with data if found
final QueriedSlotLens delegatingSlotLens = new QueriedSlotLens(entry.getKey(), data);
matches.put(delegatingSlotLens, toRemove.getOrDefault(entry.getKey(), 0));
}
}
return matches;
}
Aggregations