Grasscutter/src/main/java/emu/grasscutter/data/GenshinDepot.java

package emu.grasscutter.data;

import java.util.ArrayList;
import java.util.List;

import org.danilopianini.util.FlexibleQuadTree;
import org.danilopianini.util.SpatialIndex;

import emu.grasscutter.Grasscutter;
import emu.grasscutter.data.def.ReliquaryAffixData;
import emu.grasscutter.data.def.ReliquaryMainPropData;
import emu.grasscutter.game.world.SpawnDataEntry;
import emu.grasscutter.game.world.SpawnDataEntry.SpawnGroupEntry;
import emu.grasscutter.utils.WeightedList;
import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;

public class GenshinDepot {
	private static Int2ObjectMap<WeightedList<ReliquaryMainPropData>> relicMainPropDepot = new Int2ObjectOpenHashMap<>();
	private static Int2ObjectMap<List<ReliquaryAffixData>> relicAffixDepot = new Int2ObjectOpenHashMap<>();
	
	private static Int2ObjectMap<SpatialIndex<SpawnGroupEntry>> spawnLists = new Int2ObjectOpenHashMap<>();
	
	public static void load() {
		for (ReliquaryMainPropData data : GenshinData.getReliquaryMainPropDataMap().values()) {
			if (data.getWeight() <= 0 || data.getPropDepotId() <= 0) {
				continue;
			}
			WeightedList<ReliquaryMainPropData> list = relicMainPropDepot.computeIfAbsent(data.getPropDepotId(), k -> new WeightedList<>());
			list.add(data.getWeight(), data);
		}
		for (ReliquaryAffixData data : GenshinData.getReliquaryAffixDataMap().values()) {
			if (data.getWeight() <= 0 || data.getDepotId() <= 0) {
				continue;
			}
			List<ReliquaryAffixData> list = relicAffixDepot.computeIfAbsent(data.getDepotId(), k -> new ArrayList<>());
			list.add(data);
		}
		// Let the server owner know if theyre missing weights
		if (relicMainPropDepot.size() == 0 || relicAffixDepot.size() == 0) {
			Grasscutter.getLogger().error("Relic properties are missing weights! Please check your ReliquaryMainPropExcelConfigData or ReliquaryAffixExcelConfigData files in your ExcelBinOutput folder.");
		}
	}
	
	public static ReliquaryMainPropData getRandomRelicMainProp(int depot) {
		WeightedList<ReliquaryMainPropData> depotList = relicMainPropDepot.get(depot);
		if (depotList == null) {
			return null;
		}
		return depotList.next();
	}
	
	public static List<ReliquaryAffixData> getRandomRelicAffixList(int depot) {
		return relicAffixDepot.get(depot);
	}
	
	public static Int2ObjectMap<SpatialIndex<SpawnGroupEntry>> getSpawnLists() {
		return spawnLists;
	}
	
	public static SpatialIndex<SpawnGroupEntry> getSpawnListById(int sceneId) {
		return getSpawnLists().computeIfAbsent(sceneId, id -> new FlexibleQuadTree<>());
	}
}