CompositeGameElement.java

package work3;

import java.util.ArrayList;
import java.util.List;

/**
 * Композитний елемент, що містить інші елементи.
 * Шаблон: Composite (Composite).
 */
public class CompositeGameElement implements GameElement {
    private List<GameElement> children = new ArrayList<>();
    private String groupName;

    public CompositeGameElement(String groupName) {
        this.groupName = groupName;
    }

    /**
     * Додає елемент до групи.
     */
    public void add(GameElement element) {
        children.add(element);
    }

    /**
     * Видаляє елемент з групи.
     */
    public void remove(GameElement element) {
        children.remove(element);
    }

    @Override
    public double getArea() {
        double totalArea = 0;
        for (GameElement child : children) {
            totalArea += child.getArea();
        }
        System.out.println("Group '" + groupName + "' total area: " + totalArea);
        return totalArea;
    }
}