diff --git a/src/main/java/model/elements/FireHandler.java b/src/main/java/model/elements/FireHandler.java new file mode 100644 index 0000000000000000000000000000000000000000..a06c7731fb791203a8f9381e7bcdc58147507052 --- /dev/null +++ b/src/main/java/model/elements/FireHandler.java @@ -0,0 +1,33 @@ +package model.elements; + +import model.Fire; +import model.ModelElement; +import util.Position; + +public class FireHandler implements ElementHandler { + private final Fire fire; + + public FireHandler(Fire fire) { + this.fire = fire; + } + + @Override + public void addElement(Position position) { + fire.getFirePositions().add(position); + } + + @Override + public void removeElement(Position position) { + fire.getFirePositions().remove(position); + } + + @Override + public boolean hasElement(Position position) { + return fire.getFirePositions().contains(position); + } + + @Override + public ModelElement getModelElement() { + return ModelElement.FIRE; + } +}