import javax.swing.*;
import java.awt.*;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;

public class Main {

    public static void main(String[] args) {
        SwingUtilities.invokeLater(() -> {
            Okno okno = new Okno();
        });
    }
}

class Okno extends JFrame {

    private JRadioButton linia, prostokat, elipsa, zielony, czerwony, niebieski;
    private ButtonGroup ksztalty, kolory;
    private JLabel l, p, e, z, c, n;
    private int x, y;
    private JPanel panel;

    public Okno() {
        this.setDefaultCloseOperation(EXIT_ON_CLOSE);
        this.setVisible(true);
        this.setSize(500, 500);
        this.setLayout(null);

        JPanel ks = ksztalty();
        ks.setBounds(0, 0, 400, 44);
        this.add(ks);

        JPanel ko = kolory();
        ko.setBounds(0, 50, 400, 44);
        this.add(ko);

        this.addMouseListener(new MouseAdapter() {
            @Override
            public void mousePressed(MouseEvent e) {
                pobierzWsp(e);
            }

            @Override
            public void mouseReleased(MouseEvent e) {
                rysuj(e);
            }
        });
    }

    private JPanel ksztalty() {
        JPanel temp = new JPanel(new FlowLayout());
        this.l = new JLabel("Linia");
        this.linia = new JRadioButton();
        this.p = new JLabel("Prostokat");
        this.prostokat = new JRadioButton();
        this.e = new JLabel("Elipsa");
        this.elipsa = new JRadioButton();
        this.ksztalty = new ButtonGroup();
        this.ksztalty.add(linia);
        this.ksztalty.add(prostokat);
        this.ksztalty.add(elipsa);

        temp.add(l);
        temp.add(linia);
        temp.add(p);
        temp.add(prostokat);
        temp.add(e);
        temp.add(elipsa);

        return temp;
    }

    private JPanel kolory() {
        JPanel temp = new JPanel(new FlowLayout());
        this.z = new JLabel("Zielony");
        this.zielony = new JRadioButton();
        this.c = new JLabel("Czerwony");
        this.czerwony = new JRadioButton();
        this.n = new JLabel("Niebieski");
        this.niebieski = new JRadioButton();
        this.kolory = new ButtonGroup();

        this.kolory.add(zielony);
        this.kolory.add(czerwony);
        this.kolory.add(niebieski);

        temp.add(z);
        temp.add(zielony);
        temp.add(c);
        temp.add(czerwony);
        temp.add(n);
        temp.add(niebieski);

        return temp;
    }

    private void pobierzWsp(MouseEvent e) {
        this.x = e.getX();
        this.y = e.getY();
    }

    private void rysuj(MouseEvent e) {
        int xKonc = e.getX();
        int yKonc = e.getY();

        String typ = null;
        if (linia.isSelected())
            typ = "1";
        else if (prostokat.isSelected())
            typ = "2";
        else if (elipsa.isSelected())
            typ = "3";

        Color color = null;
        if (zielony.isSelected())
            color = Color.green;
        else if (niebieski.isSelected())
            color = Color.blue;
        else if (czerwony.isSelected())
            color = Color.red;

        System.out.println(x + " " + y + " " + xKonc + " " + yKonc);
        panel = new Rysowanie(color, typ, x, y, xKonc, yKonc);
        panel.setBounds(0, 0, 500, 500);
        panel.revalidate();
        panel.repaint();

        this.add(panel);
        this.revalidate();
        this.repaint();
    }
}
class Rysowanie extends JPanel {

    private Color color;
    private String type;
    private int x, y, xKonc, yKonc;

    public Rysowanie(Color color, String type, int x, int y, int xKonc, int yKonc) {
        this.setPreferredSize(new Dimension(500, 500));
        this.color = color;
        this.setLayout(null);
        this.type = type;
        this.x = x;
        this.y = y;
        this.xKonc = xKonc;
        this.yKonc = yKonc;
    }

    @Override
    public void paintComponent(Graphics g) {
        super.paintComponent(g);
        g.setColor(this.color);
        switch (type) {
            case "1":
                g.drawLine(x, y, xKonc - x, yKonc - y);
                break;
            case "2":
                g.drawRect(x, y, xKonc - x, yKonc - y);
                break;
            case "3":
                g.drawOval(x, y, xKonc - x, yKonc - y);
                break;
        }
    }
}