Methods in TheEndBiomeData that query the intended type.

fabric-biome-api-v1/src/main/java/net/fabricmc/fabric/api/biome/v1/TheEndBiomes.java

@@ -103,4 +103,49 @@ public static void addMidlandsBiome(RegistryKey<Biome> highlands, RegistryKey<Bi
 	public static void addBarrensBiome(RegistryKey<Biome> highlands, RegistryKey<Biome> barrens, double weight) {
 		TheEndBiomeData.addEndBarrensReplacement(highlands, barrens, weight);
 	}
+
+	/**
+	 * Returns true if the given biome was added as a main end Biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End by mods.
+	 * @param biome The biome to search for
+	 */
+	public static boolean canGenerateAsMainIslandBiome(RegistryKey<Biome> biome){
+		return TheEndBiomeData.canGenerateAsEndBiomeIn(BiomeKeys.THE_END, biome);
+	}
+
+	/**
+	 * Returns true if the given biome was added as a small end islands Biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End by mods.
+	 * @param biome The biome to search for
+	 */
+	public static boolean canGenerateAsSmallIslandsBiome(RegistryKey<Biome> biome){
+		return TheEndBiomeData.canGenerateAsEndBiomeIn(BiomeKeys.SMALL_END_ISLANDS, biome);
+	}
+
+	/**
+	 * Returns true if the given biome was added as a Highland Biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End by mods.
+	 * @param biome The biome to search for
+	 */
+	public static boolean canGenerateAsHighlandsBiome(RegistryKey<Biome> biome){
+		return TheEndBiomeData.canGenerateAsEndBiomeIn(BiomeKeys.END_HIGHLANDS, biome);
+	}
+
+	/**
+	 * Returns true if the given biome was added as midland biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End as midland biome by mods.
+	 * @param biome The biome to search for
+	 */
+	public static boolean canGenerateAsEndMidlands(RegistryKey<Biome> biome){
+		return TheEndBiomeData.canGenerateAsEndMidlands(biome);
+	}
+
+	/**
+	 * Returns true if the given biome was added as barrens biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End as barrens biome by mods.
+	 * @param biome The biome to search for
+	 */
+	public static boolean canGenerateAsEndBarrens(RegistryKey<Biome> biome){
+		return TheEndBiomeData.canGenerateAsEndBarrens(biome);
+	}
 }

fabric-biome-api-v1/src/main/java/net/fabricmc/fabric/impl/biome/TheEndBiomeData.java

@@ -90,6 +90,38 @@ public static void addEndBarrensReplacement(RegistryKey<Biome> highlands, Regist
 		ADDED_BIOMES.add(barrens);
 	}
 
+	private static boolean containsValue(Map<RegistryKey<Biome>, WeightedPicker<RegistryKey<Biome>>> map, RegistryKey<Biome> biome){
+		return map.values().stream().anyMatch(w->w.getAny(biome).isPresent());
+	}
+
+	/**
+	 * Returns true if the given biome was added for the specified keyBiome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End by mods.
+	 *
+	 * @param keyBiome The parent biome type
+	 * @param biome The biome you are looking for
+	 */
+	public static boolean canGenerateAsEndBiomeIn(RegistryKey<Biome> keyBiome, RegistryKey<Biome> biome){
+		WeightedPicker<RegistryKey<Biome>> highlandBiomes = END_BIOMES_MAP.get(keyBiome);
+		return highlandBiomes!=null && highlandBiomes.getAny(biome).isPresent();
+	}
+
+	/**
+	 * Returns true if the given biome was added as midland biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End as midland biome by mods.
+	 */
+	public static boolean canGenerateAsEndMidlands(RegistryKey<Biome> biome){
+		return containsValue(END_MIDLANDS_MAP, biome);
+	}
+
+	/**
+	 * Returns true if the given biome was added as barrens biome in the end, considering the Vanilla end biomes,
+	 * and any biomes added to the End as barrens biome by mods.
+	 */
+	public static boolean canGenerateAsEndBarrens(RegistryKey<Biome> biome){
+		return containsValue(END_BARRENS_MAP, biome);
+	}
+
 	public static Overrides createOverrides(Registry<Biome> biomeRegistry) {
 		return new Overrides(biomeRegistry);
 	}

fabric-biome-api-v1/src/main/java/net/fabricmc/fabric/impl/biome/WeightedPicker.java

@@ -18,6 +18,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Optional;
 import java.util.function.Function;
 
 import com.google.common.base.Preconditions;
@@ -100,6 +101,16 @@ WeightedEntry<T> search(final double target) {
 		return entries.get(low);
 	}
 
+	/**
+	 * Returns any {@link WeightedEntry} that contains the passed element.
+	 *
+	 * @param element The Element you are looking for
+	 * @return The result of the search
+	 */
+	Optional<WeightedEntry<T>> getAny(T element){
+		return entries.stream().filter(w -> w.entry==element).findAny();
+	}
+
 	/**
 	 * Represents a modded entry in a list, and its corresponding weight.
 	 *