use of org.spongepowered.common.inventory.lens.slots.SlotLens 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;
}
use of org.spongepowered.common.inventory.lens.slots.SlotLens in project SpongeCommon by SpongePowered.
the class AbstractLens method getSlotLens.
@Override
@Nullable
public SlotLens getSlotLens(Fabric fabric, int ordinal) {
if (ordinal < 0 || ordinal > this.maxOrdinal) {
return null;
}
int offset = 0;
for (final Lens child : this.spanningChildren) {
int count = child.slotCount();
if (ordinal < offset + count) {
return child.getSlotLens(fabric, ordinal - offset);
}
offset += count;
}
return null;
}
use of org.spongepowered.common.inventory.lens.slots.SlotLens in project SpongeCommon by SpongePowered.
the class AbstractContainerMenuMixin_Adapter_Inventory method inventoryAdapter$getSlot.
@Override
public Optional<org.spongepowered.api.item.inventory.Slot> inventoryAdapter$getSlot(int ordinal) {
org.spongepowered.api.item.inventory.Slot slot = this.impl$slots.get(ordinal);
if (slot == null) {
Lens rootLens = this.inventoryAdapter$getRootLens();
SlotLens slotLens = rootLens.getSlotLens(this.inventoryAdapter$getFabric(), ordinal);
if (slotLens == null) {
return Optional.empty();
}
slot = slotLens.getAdapter(this.inventoryAdapter$getFabric(), ((Inventory) this));
this.impl$slots.put(ordinal, slot);
}
return Optional.of(slot);
}
Aggregations