Examples with Query org.spongepowered.api.item.inventory.query.Query used on opensource projects

Example 1 with Query

use of org.spongepowered.api.item.inventory.query.Query 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;
}

Example 2 with Query

use of org.spongepowered.api.item.inventory.query.Query in project SpongeCommon by SpongePowered.

the class DefaultSingleBlockCarrier method inventory.

static Inventory inventory(final Direction from, final BlockCarrier thisThing) {
    if (thisThing instanceof WorldlyContainer) {
        final net.minecraft.core.Direction facing = DirectionFacingProvider.INSTANCE.get(from).get();
        final int[] slots = ((WorldlyContainer) thisThing).getSlotsForFace(facing);
        if (slots.length == 0) {
            return new EmptyInventoryImpl(thisThing.inventory());
        }
        // build query for each slot
        final Query.Builder builder = Query.builder();
        Arrays.stream(slots).mapToObj(slot -> QueryTypes.KEY_VALUE.get().of(SlotMatchers.index(slot))).forEach(builder::and);
        final Query query = builder.build();
        return thisThing.inventory().query(query);
    }
    return thisThing.inventory();
}