// 
// Decompiled by Procyon v0.6.0
// 

package com.hypixel.hytale.server.core.modules.entityui.asset;

import com.hypixel.hytale.assetstore.AssetMap;
import java.util.Set;
import com.hypixel.hytale.assetstore.AssetUpdateQuery;
import java.util.Iterator;
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import com.hypixel.hytale.protocol.packets.assets.UpdateEntityUIComponents;
import com.hypixel.hytale.protocol.UpdateType;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
import com.hypixel.hytale.protocol.Packet;
import java.util.Map;
import javax.annotation.Nonnull;
import com.hypixel.hytale.assetstore.map.IndexedLookupTableAssetMap;
import com.hypixel.hytale.server.core.asset.packet.AssetPacketGenerator;

public class EntityUIComponentPacketGenerator extends AssetPacketGenerator<String, EntityUIComponent, IndexedLookupTableAssetMap<String, EntityUIComponent>>
{
    @Nonnull
    @Override
    public Packet generateInitPacket(@Nonnull final IndexedLookupTableAssetMap<String, EntityUIComponent> assetMap, @Nonnull final Map<String, EntityUIComponent> assets) {
        final Int2ObjectMap<com.hypixel.hytale.protocol.EntityUIComponent> configs = new Int2ObjectOpenHashMap<com.hypixel.hytale.protocol.EntityUIComponent>();
        for (final Map.Entry<String, EntityUIComponent> entry : assets.entrySet()) {
            configs.put(assetMap.getIndex(entry.getKey()), entry.getValue().toPacket());
        }
        return new UpdateEntityUIComponents(UpdateType.Init, assetMap.getNextIndex(), configs);
    }
    
    @Nonnull
    @Override
    public Packet generateUpdatePacket(@Nonnull final IndexedLookupTableAssetMap<String, EntityUIComponent> assetMap, @Nonnull final Map<String, EntityUIComponent> loadedAssets, @Nonnull final AssetUpdateQuery query) {
        final Int2ObjectMap<com.hypixel.hytale.protocol.EntityUIComponent> components = new Int2ObjectOpenHashMap<com.hypixel.hytale.protocol.EntityUIComponent>();
        for (final Map.Entry<String, EntityUIComponent> entry : loadedAssets.entrySet()) {
            components.put(assetMap.getIndex(entry.getKey()), entry.getValue().toPacket());
        }
        return new UpdateEntityUIComponents(UpdateType.AddOrUpdate, assetMap.getNextIndex(), components);
    }
    
    @Nonnull
    @Override
    public Packet generateRemovePacket(@Nonnull final IndexedLookupTableAssetMap<String, EntityUIComponent> assetMap, @Nonnull final Set<String> removed, @Nonnull final AssetUpdateQuery query) {
        final Int2ObjectMap<com.hypixel.hytale.protocol.EntityUIComponent> configs = new Int2ObjectOpenHashMap<com.hypixel.hytale.protocol.EntityUIComponent>();
        for (final String entry : removed) {
            configs.put(assetMap.getIndex(entry), new com.hypixel.hytale.protocol.EntityUIComponent());
        }
        return new UpdateEntityUIComponents(UpdateType.Remove, assetMap.getNextIndex(), configs);
    }
}