package balloonpeople;


import painter.SPainter;

import javax.swing.*;
import java.util.ArrayList;
import java.util.Scanner;

public class BalloonFamily {

    private ArrayList<BalloonPerson> balloonPeople;

    public BalloonFamily(int numberOfPeople) {
        this.balloonPeople = new ArrayList<>(numberOfPeople);
        for (int i = 0; i < numberOfPeople; i=i+1) {
            String name = getName("Name of family member # " + (i+1) + ": ");
            int age = getAge("Age of family member " + (i+1) + ": ");
            double height = getHeight("Height of family member # " + (i+1) + ": ");
            balloonPeople.add(new BalloonPerson(name, age, height));
        }
    }

    private String getName(String prompt) {
        String nss = JOptionPane.showInputDialog(null, prompt + "?");
        Scanner scanner = new Scanner(nss);
        return scanner.next();
    }

    private int getAge(String prompt) {
        String nss = JOptionPane.showInputDialog(null, prompt + "?");
        Scanner scanner = new Scanner(nss);
        return scanner.nextInt();
    }

    private double getHeight(String prompt) {
        String nss = JOptionPane.showInputDialog(null, prompt + "?");
        Scanner scanner = new Scanner(nss);
        return scanner.nextDouble();
    }

    public String toString() {
        int i = 0;
        while (i < balloonPeople.size()) {
            System.out.println(balloonPeople.get(i));
            i = i + 1;
        }
        return null;
    }

    public void painter(SPainter painter) {
        double y = painter.getWidth()/balloonPeople.size();
        if ( y < 100) {
            painter.draw("Too many family members selected, choose less " +
                    "Max: 4");
        } else {
            painter.mrt(painter.getWidth()/2);
            painter.mlt(y/2);
            balloonPeople.get(0).paintPerson(painter);
            for (int i = 1; i < balloonPeople.size(); i++) {
                painter.mlt(y);
                balloonPeople.get(i).paintPerson(painter);
                painter.center();


            }
        }
    }
}