Search in sources :

Example 1 with DelegatingSlotLens

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;
}
Also used : Inventory(org.spongepowered.api.item.inventory.Inventory) DelegatingLens(org.spongepowered.common.inventory.lens.impl.DelegatingLens) QueriedSlotLens(org.spongepowered.common.inventory.lens.impl.slot.QueriedSlotLens) SlotLens(org.spongepowered.common.inventory.lens.slots.SlotLens) HashMap(java.util.HashMap) InventoryBridge(org.spongepowered.common.bridge.world.inventory.InventoryBridge) Fabric(org.spongepowered.common.inventory.fabric.Fabric) QueryLens(org.spongepowered.common.inventory.lens.impl.QueryLens) Key(org.spongepowered.api.data.Key) EmptyInventoryImpl(org.spongepowered.common.inventory.EmptyInventoryImpl) LensRegistrar(org.spongepowered.common.inventory.lens.impl.LensRegistrar) LinkedHashMap(java.util.LinkedHashMap) Query(org.spongepowered.api.item.inventory.query.Query) List(java.util.List) Map(java.util.Map) Lens(org.spongepowered.common.inventory.lens.Lens) DelegatingSlotLens(org.spongepowered.common.inventory.lens.impl.slot.DelegatingSlotLens) Collections(java.util.Collections) InventoryAdapter(org.spongepowered.common.inventory.adapter.InventoryAdapter) HashMap(java.util.HashMap) LinkedHashMap(java.util.LinkedHashMap) QueriedSlotLens(org.spongepowered.common.inventory.lens.impl.slot.QueriedSlotLens) SlotLens(org.spongepowered.common.inventory.lens.slots.SlotLens) DelegatingSlotLens(org.spongepowered.common.inventory.lens.impl.slot.DelegatingSlotLens) LinkedHashMap(java.util.LinkedHashMap) DelegatingLens(org.spongepowered.common.inventory.lens.impl.DelegatingLens) QueriedSlotLens(org.spongepowered.common.inventory.lens.impl.slot.QueriedSlotLens) SlotLens(org.spongepowered.common.inventory.lens.slots.SlotLens) QueryLens(org.spongepowered.common.inventory.lens.impl.QueryLens) Lens(org.spongepowered.common.inventory.lens.Lens) DelegatingSlotLens(org.spongepowered.common.inventory.lens.impl.slot.DelegatingSlotLens) HashMap(java.util.HashMap) LinkedHashMap(java.util.LinkedHashMap) Map(java.util.Map) Key(org.spongepowered.api.data.Key) QueriedSlotLens(org.spongepowered.common.inventory.lens.impl.slot.QueriedSlotLens)

Aggregations

Collections (java.util.Collections)1 HashMap (java.util.HashMap)1 LinkedHashMap (java.util.LinkedHashMap)1 List (java.util.List)1 Map (java.util.Map)1 Key (org.spongepowered.api.data.Key)1 Inventory (org.spongepowered.api.item.inventory.Inventory)1 Query (org.spongepowered.api.item.inventory.query.Query)1 InventoryBridge (org.spongepowered.common.bridge.world.inventory.InventoryBridge)1 EmptyInventoryImpl (org.spongepowered.common.inventory.EmptyInventoryImpl)1 InventoryAdapter (org.spongepowered.common.inventory.adapter.InventoryAdapter)1 Fabric (org.spongepowered.common.inventory.fabric.Fabric)1 Lens (org.spongepowered.common.inventory.lens.Lens)1 DelegatingLens (org.spongepowered.common.inventory.lens.impl.DelegatingLens)1 LensRegistrar (org.spongepowered.common.inventory.lens.impl.LensRegistrar)1 QueryLens (org.spongepowered.common.inventory.lens.impl.QueryLens)1 DelegatingSlotLens (org.spongepowered.common.inventory.lens.impl.slot.DelegatingSlotLens)1 QueriedSlotLens (org.spongepowered.common.inventory.lens.impl.slot.QueriedSlotLens)1 SlotLens (org.spongepowered.common.inventory.lens.slots.SlotLens)1