diff --git a/ColorFilter.java b/ColorFilter.java new file mode 100644 index 0000000..af588ec --- /dev/null +++ b/ColorFilter.java @@ -0,0 +1,30 @@ +import java.util.List; +import java.util.ArrayList; + +public class ColorFilter { + private String targetColor; + + public ColorFilter(String targetColor) { + this.targetColor = targetColor; + } + + public List filterByColor(List shapes) { + List filteredShapes = new ArrayList<>(); + + for (Shape shape : shapes) { + if (shape.getColor().equalsIgnoreCase(targetColor)) { + filteredShapes.add(shape); + } + } + + return filteredShapes; + } + + public String getTargetColor() { + return targetColor; + } + + public void setTargetColor(String targetColor) { + this.targetColor = targetColor; + } +} \ No newline at end of file