Skip to content

Commit

Permalink
structure predicate stuff + fix sunflower plains trees
Browse files Browse the repository at this point in the history
  • Loading branch information
dags- committed May 14, 2020
1 parent 6cd959a commit a5fce36
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions src/main/java/com/terraforged/fm/structure/StructureManager.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
package com.terraforged.fm.structure;

import com.terraforged.fm.predicate.FeaturePredicate;
import net.minecraft.world.gen.feature.structure.Structure;

import java.util.HashMap;
import java.util.Map;

public class StructureManager {

private final Map<String, FeaturePredicate> predicates = new HashMap<>();

public void register(Structure<?> structure, FeaturePredicate predicate) {
register(structure.getStructureName(), predicate);
}

public void register(String structureName, FeaturePredicate predicate) {
predicates.put(structureName, predicate);
}

public FeaturePredicate getPredicate(Structure<?> structure) {
return getPredicate(structure.getStructureName());
}

public FeaturePredicate getPredicate(String structureName) {
return predicates.getOrDefault(structureName, FeaturePredicate.ALLOW);
}
}

0 comments on commit a5fce36

Please sign in to comment.