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

public class Lenta2 extends JPanel {

    private Map<Integer, Integer> lenta2;
    private int dydis;
    
    
   
    public Lenta2(MouseListener handler, int dydis) {
        lenta2 = new HashMap<Integer, Integer>();
        this.addMouseListener(handler);
        this.dydis = dydis;
        this.setSize(dydis * 3, dydis * 3);
    }

    public void paint(Graphics g) {
        g.setColor(Color.BLACK);
        g.drawLine(dydis, 0, dydis, 3 * dydis);
        g.drawLine(2 * dydis, 0, 2 * dydis, 3 * dydis);
        g.drawLine(0, dydis, 3 * dydis, dydis);
        g.drawLine(0, 2 * dydis, 3 * dydis, 2 * dydis);

        Set<Integer> uzimtiLaukeliai = lenta2.keySet();
        for (int i : uzimtiLaukeliai) {
            g.setColor((lenta2.get(i) == 0) ? Color.YELLOW : Color.PINK);
            int xKoord = i / 3;
            int yKoord = i % 3;
            if (lenta2.get(i) == 0) {
                g.drawLine(xKoord * dydis, yKoord * dydis, (xKoord + 1) * dydis,
                        (yKoord + 1) * dydis);
                g.drawLine((xKoord + 1) * dydis, yKoord * dydis, xKoord * dydis,
                        (yKoord + 1) * dydis);
            } else {
                g.drawOval(xKoord * dydis, yKoord * dydis, dydis, dydis);
            }
        }
    }
   
    public Integer gautiFigura(int koord) {
        return lenta2.get(koord);
    }
   
    public void nustatytiFigura (int koord, int zaidejas) {
        lenta2.put(koord, zaidejas);
    }
   
    public void istrintiReiksmes() {
        lenta2.clear();
    }

}