Разработка простого графического редактора, добавить функцию заливки

package lr5;
 
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import javax.swing.event.*;
import java.awt.image.*;
 
public class LR5
{
    // Режим рисования
    int rezhim=0;
    int xPad;
    int xf;
    int yf;
    int yPad;
    int[] xCrds = new int[3];
    int[] yCrds = new int[3];
    boolean pressed=false;
 
    // текущий цвет
    Color maincolor;
    MyFrame f;
    MyPanel japan;
    JButton colorbutton;
    JColorChooser tcc;
 
    // поверхность рисования
    BufferedImage imag;
 
    public LR5()
    {
        f=new MyFrame("");
        f.setSize(800,800);
        f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        maincolor=Color.black;
        japan = new MyPanel();
        japan.setBounds(30,30,600,600);
        japan.setBackground(Color.white);
        japan.setOpaque(true);
        f.add(japan);
        JToolBar toolbar = new JToolBar("Toolbar", JToolBar.VERTICAL);
        JButton penbutton = new JButton(new ImageIcon("pen.png"));
        penbutton.addActionListener(new ActionListener() 
        {
            public void actionPerformed(ActionEvent event) 
            {
                rezhim=0; 
            }
        }
    );
        
        toolbar.add(penbutton);
        JButton lasticbutton = new JButton(new ImageIcon("lastic.png"));
        lasticbutton.addActionListener(new ActionListener()
        {
            public void actionPerformed(ActionEvent event)
            {
                rezhim=2; 
            } 
        }
        );
        
        toolbar.add(lasticbutton);
        JButton linebutton = new JButton(new ImageIcon("line.png"));
        
        linebutton.addActionListener(new ActionListener()
        {
            public void actionPerformed(ActionEvent event)
            {
                rezhim=4; 
            } 
        });
    toolbar.add(linebutton);
    JButton elipsbutton = new JButton(new ImageIcon("elips.png"));
    elipsbutton.addActionListener(new ActionListener()
    {
 
        public void actionPerformed(ActionEvent event) {
 
        rezhim=5;
        }
    });
 
    toolbar.add(elipsbutton);
    JButton rectbutton = new JButton(new ImageIcon("rect.png"));
    rectbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
            rezhim=6;
        }
    });
 
    toolbar.add(rectbutton);
    JButton treug = new JButton(new ImageIcon("triangle.png"));
    treug.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent e)
        {
            rezhim=7; 
        }
    });
 
    toolbar.add(treug);
    toolbar.setBounds(0, 0, 30, 300);
    f.add(toolbar);
 
    // Панель инструментов
    JToolBar colorbar = new JToolBar("Colorbar", JToolBar.HORIZONTAL);
    colorbar.setBounds(30, 0, 260, 30);
    colorbutton = new JButton();
    colorbutton.setBackground(maincolor);
    colorbutton.setBounds(15, 5, 20, 20);
    colorbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
        ColorDialog coldi = new ColorDialog(f,"Выбор цвета");
 
        coldi.setVisible(true); 
        } 
    });
 
    colorbar.add(colorbutton);
    JButton redbutton = new JButton();
    redbutton.setBackground(Color.red);
    redbutton.setBounds(40, 5, 15, 15);
    redbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
        maincolor = Color.red;
        colorbutton.setBackground(maincolor); 
        } 
    });
    colorbar.add(redbutton);
    
    JButton orangebutton = new JButton();
    orangebutton.setBackground(Color.orange);
    orangebutton.setBounds(60, 5, 15, 15);
    orangebutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
            maincolor = Color.orange;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(orangebutton);
    
    JButton yellowbutton = new JButton();
    yellowbutton.setBackground(Color.yellow);
    yellowbutton.setBounds(80, 5, 15, 15);
    yellowbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
        maincolor = Color.yellow;
        colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(yellowbutton);
    
    JButton greenbutton = new JButton();
    greenbutton.setBackground(Color.green);
    greenbutton.setBounds(100, 5, 15, 15);
    greenbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event) {
            maincolor = Color.green;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(greenbutton);
 
    JButton bluebutton = new JButton();
    bluebutton.setBackground(Color.blue);
    bluebutton.setBounds(120, 5, 15, 15);
    bluebutton.addActionListener(new ActionListener(){
        public void actionPerformed(ActionEvent event)
        {
            maincolor = Color.blue;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(bluebutton);
 
    JButton cyanbutton = new JButton();
    cyanbutton.setBackground(Color.cyan);
    cyanbutton.setBounds(140, 5, 15, 15);
    cyanbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
            maincolor = Color.cyan;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(cyanbutton);
 
    JButton magentabutton = new JButton();
    magentabutton.setBackground(Color.magenta);
    magentabutton.setBounds(160, 5, 15, 15);
    magentabutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
            maincolor = Color.magenta;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(magentabutton);
 
    JButton whitebutton = new JButton();
    whitebutton.setBackground(Color.white);
    whitebutton.setBounds(180, 5, 15, 15);
    whitebutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event){
            maincolor = Color.white;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(whitebutton);
 
    JButton blackbutton = new JButton();
    blackbutton.setBackground(Color.black);
    blackbutton.setBounds(200, 5, 15, 15);
    blackbutton.addActionListener(new ActionListener()
    {
        public void actionPerformed(ActionEvent event)
        {
            maincolor = Color.black;
            colorbutton.setBackground(maincolor);
        }
    });
    colorbar.add(blackbutton);
 
    colorbar.setLayout(null);
    f.add(colorbar);
    tcc = new JColorChooser(maincolor);
    tcc.getSelectionModel().addChangeListener(new ChangeListener()
    {
        public void stateChanged(ChangeEvent e){
            maincolor = tcc.getColor();
            colorbutton.setBackground(maincolor);
        }
    });
 
    japan.addMouseMotionListener(new MouseMotionAdapter()
    {
        public void mouseDragged(MouseEvent e)
        {
 
            if (pressed==true)
            {
                Graphics g = imag.getGraphics();
                Graphics2D g2 = (Graphics2D)g;
                
                // установка цвета
                g2.setColor(maincolor);
                switch (rezhim)
                {
                // карандаш
                    case 0:
                        g2.drawLine(xPad, yPad, e.getX(), e.getY());
                        break;
 
                    // ластик
                    case 2:
                        g2.setStroke(new BasicStroke(3.0f));
                        g2.setColor(Color.WHITE);
                        g2.drawLine(xPad, yPad, e.getX(), e.getY());
                        break;
                }
                xPad=e.getX();
                yPad=e.getY();
            }
            japan.repaint();
        }
    });
 
    japan.addMouseListener(new MouseAdapter()
    {
        public void mouseClicked(MouseEvent e)
        {
            Graphics g = imag.getGraphics();
            Graphics2D g2 = (Graphics2D)g;
            
            // установка цвета
            g2.setColor(maincolor);
            switch (rezhim)
            {
                // карандаш
                case 0:
                    g2.drawLine(xPad, yPad, xPad+1, yPad+1);
                    break;
 
                // кисть
                case 1:
                    g2.setStroke(new BasicStroke(3.0f));
                    g2.drawLine(xPad, yPad, xPad+1, yPad+1);
                    break;
 
                // ластик
                case 2:
                    g2.setStroke(new BasicStroke(3.0f));
                    g2.setColor(Color.WHITE);
                    g2.drawLine(xPad, yPad, xPad+1, yPad+1);
                    break;
            }
            xPad=e.getX();
            yPad=e.getY();
            pressed=true;
            japan.repaint();
        }
        public void mousePressed(MouseEvent e) 
        {
            xPad=e.getX();
            yPad=e.getY();
            xf=e.getX();
            yf=e.getY();
            pressed=true;
        }
        
        public void mouseReleased(MouseEvent e) 
        {
            Graphics g = imag.getGraphics();
            Graphics2D g2 = (Graphics2D)g;
            
            // установка цвета
            g2.setColor(maincolor);
            xCrds[0] = xf;
            yCrds[0] = yf;
            xCrds[2] = xPad;
            yCrds[2] = yf;
            xCrds[1] = xCrds[0]+(xCrds[2]-xCrds[0])/2;
            yCrds[1] = yPad;
            
            // Общие рассчеты для овала и прямоугольника
            int x1=xf, x2=xPad, y1=yf, y2=yPad;
            if(xf>xPad)
            {
                x2=xf; x1=xPad;
            }
            if(yf>yPad)
            {
                y2=yf; y1=yPad;
            }
            switch(rezhim)
            {
            // линия
            case 4:
                g.drawLine(xf, yf, e.getX(), e.getY());
                break;
            
            // круг
            case 5:
                g.drawOval(x1, y1, (x2-x1), (y2-y1));
                break;
 
            // прямоугольник
            case 6:
                g.drawRect(x1, y1, (x2-x1), (y2-y1));
                break;
 
            case 7:
                g.drawPolygon(xCrds, yCrds, 3);
                break;
            }
            xf=0; yf=0;
            pressed=false;
            japan.repaint();
        }
    }
    );
 
    japan.addKeyListener(new KeyAdapter()
    {
        public void keyTyped(KeyEvent e) 
        {
            if(rezhim==3)
            {
 
                Graphics g = imag.getGraphics();
 
                Graphics2D g2 = (Graphics2D)g;
 
                // установка цвета
 
                g2.setColor(maincolor);
 
                japan.repaint();
            }
        }
    });
    f.setLayout(null);
    f.setVisible(true);}
    
    public static void main(String[] args)
    {
        SwingUtilities.invokeLater(new Runnable()
        {
            public void run()
            {
            new LR5();
            }
        }
        ); 
    }
 
    class ColorDialog extends JDialog
    {
        public ColorDialog(JFrame owner, String title)
        {
            super(owner, title, true);
            add(tcc);
            setSize(200, 200);
        }
    }
 
    class MyFrame extends JFrame
    {
        public void paint(Graphics g){
        super.paint(g);}
        public MyFrame(String title)
        {
            super(title);
        }
    }
 
    class MyPanel extends JPanel{
        public void paintComponent (Graphics g)
        {
            if(imag==null)
            {
                imag = new BufferedImage(this.getWidth(), this.getHeight(), BufferedImage.TYPE_INT_RGB);
                Graphics2D d2 = (Graphics2D) imag.createGraphics();
                d2.setColor(Color.white);
                d2.fillRect(0, 0, this.getWidth(), this.getHeight());
            }
            super.paintComponent(g);
            g.drawImage(imag, 0, 0,this); 
        }
    }
}