RockyFactory.java

package model.flammable;

import model.ElementFactory;
import model.PositionGenerator;
import util.Position;

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

public class RockyFactory implements ElementFactory<Rocky>, PositionGenerator {
    private final Random random;
    private int count;


    public RockyFactory(Random random, int count) {
        this.random = random;
        this.count = count;
    }

    @Override
    public List<Rocky> createElements(int rowCount, int columnCount) {
        List<Rocky> rocky = new ArrayList<>();
        for (int i = 0; i < count; i++) {
            Position randomPosition = generateRandomPosition(rowCount, columnCount);
            rocky.add(new Rocky(randomPosition));
        }
        return rocky;
    }

    @Override
    public int getCount() {
        return count;
    }

    @Override
    public Position generateRandomPosition(int rowCount, int columnCount) {
        int row = random.nextInt(rowCount);
        int column = random.nextInt(columnCount);
        return new Position(row, column);
    }
}