Квадрат, который движется внутри окна по диагонали и "отражается" от стенок под углом, "зеркальным" до угла движения к стенке

import java.awt.Graphics;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.Timer;
public class Main {
    public static void main(String[] args) {
        new JFrame() {
            {
                add(new JPanel() {
                    int[] pos = new int[] {100,100};
                    int[] offset = new int[] {-1,1};
                    int[] size = new int[] {50,50}; 
                    public void paint(Graphics g) {
                        super.paintComponent(g);
                        g.fillRect(pos[0],pos[1],size[0],size[1]);
                        pos[0]+=offset[0];
                        if (pos[0]<=0)
                            offset[0]=Math.abs(offset[0]);
                        if (pos[0]+size[0]>=getWidth())
                            offset[0]=Math.abs(offset[0])*-1;
                        pos[1]+=offset[1];
                        if(pos[1]<=0)
                            offset[1]=Math.abs(offset[1]);
                        if(pos[1]+size[1]>=getHeight())
                            offset[1]=Math.abs(offset[1])*-1;
                    }
                });
                setVisible(true);setSize(300,300);setDefaultCloseOperation(3);
                new Timer(1000/240, new ActionListener() {public void actionPerformed(ActionEvent e) {
                    repaint();
                }}) {{start();}};
            }
        };
    }
}