Initial commit

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

@@ -0,0 +1,49 @@
+package emu.grasscutter.data;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import emu.grasscutter.Grasscutter;
+import emu.grasscutter.data.def.ReliquaryAffixData;
+import emu.grasscutter.data.def.ReliquaryMainPropData;
+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<>();
+	
+	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);
+	}
+}