import javax.swing.*;
import java.awt.*;
import java.util.Random;

public class FormEntityWithExtraGallery extends JComponent {
    private DrawningLocomotive _locomotiveFirst;
    private DrawningLocomotive _locomotiveSecond;
    private DrawningLocomotive _locomotiveThird;
    EntityWithExtraCreator<EntityLocomotive, IDrawningExtra> entityWithExtraCreator;

    public FormEntityWithExtraGallery() {
        JFrame formFrame = new JFrame("Gallery");
        formFrame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        formFrame.setSize(900, 500);
        formFrame.setVisible(true);
        formFrame.setLocationRelativeTo(null);

        Panel statusPanel = new Panel();
        statusPanel.setBackground(Color.WHITE);
        statusPanel.setLayout(new FlowLayout());
        setLayout(new BorderLayout());
        add(statusPanel, BorderLayout.SOUTH);

        Label speedLabel = new Label("Speed:         ");
        Label weightLabel = new Label("Weight:       ");
        Label colorLabel = new Label("Color:                    ");

        JButton randomlyFillArraysWithParts = new JButton("Randomly fill arrays with parts");
        randomlyFillArraysWithParts.addActionListener(e -> {
            Random random = new Random();
            entityWithExtraCreator = new EntityWithExtraCreator<>(100, 100);
            for (int i = 0; i < 100; i ++) {
                if (random.nextBoolean()) {
                    entityWithExtraCreator.Insert(new EntityLocomotive(random.nextInt(100), random.nextInt(100),
                            new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255))));
                }
                else {
                    entityWithExtraCreator.Insert(new EntityWarmlyLocomotive(random.nextInt(100), random.nextInt(100),
                            new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255)),
                            new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255)),
                            random.nextBoolean(), random.nextBoolean()));
                }
            }
            for (int i = 0; i < 100; i ++) {
                int extraRand = random.nextInt(3);
                switch (extraRand) {
                    case 0:
                        entityWithExtraCreator.Insert(new ExtraWheelsDraw(random.nextInt(3),
                                new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255))));
                        break;

                    case 1:
                        entityWithExtraCreator.Insert(new ExtraStarWheelDraw(random.nextInt(3),
                                new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255))));
                        break;

                    case 2:
                        entityWithExtraCreator.Insert(new ExtraRoundWheelDraw(random.nextInt(3),
                                new Color(random.nextInt(255),random.nextInt(255),random.nextInt(255))));
                        break;
                }
            }
            repaint();
        });
        statusPanel.add(randomlyFillArraysWithParts);

        JButton showRandomEntity = new JButton("Create entity from parts");
        showRandomEntity.addActionListener(e -> {
            if (entityWithExtraCreator == null) return;
            _locomotiveFirst = entityWithExtraCreator.getEntityWithExtra();
            _locomotiveFirst.SetPosition(200, 200, formFrame.getWidth(), formFrame.getHeight() - 75);

            _locomotiveSecond = entityWithExtraCreator.getEntityWithExtra();
            _locomotiveSecond.SetPosition(400, 200, formFrame.getWidth(), formFrame.getHeight() - 75);

            _locomotiveThird = entityWithExtraCreator.getEntityWithExtra();
            _locomotiveThird.SetPosition(600, 200, formFrame.getWidth(), formFrame.getHeight() - 75);
            repaint();
        });
        statusPanel.add(showRandomEntity);


        formFrame.getContentPane().add(this);
    }

    @Override
    protected void paintComponent(Graphics g) {
        super.paintComponent(g);
        Graphics2D g2 = (Graphics2D)g;
        if (_locomotiveFirst != null) _locomotiveFirst.DrawTransport(g2);
        if (_locomotiveSecond != null) _locomotiveSecond.DrawTransport(g2);
        if (_locomotiveThird != null) _locomotiveThird.DrawTransport(g2);
        super.repaint();
    }
}