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

class Tegnekontroll
  extends JFrame
  implements ActionListener {

  Tegner tegner;
  JButton firkant, sirkel;

  Tegnekontroll() {
    tegner  = new Tegner();
    firkant = new JButton("Firkant");
    firkant.addActionListener(this);
    sirkel  = new JButton("Sirkel");
    sirkel.addActionListener(this);

    // Plasser komponenter
    JPanel knapperad = new JPanel();
    knapperad.add(firkant);
    knapperad.add(sirkel);
    add(knapperad, BorderLayout.SOUTH);
    add(tegner, BorderLayout.CENTER);

    setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    pack();
    setVisible(true);
  }

  public void actionPerformed(ActionEvent e) {
    // Bestem om tegneren skal tegne firkant eller sirkel.
    tegner.oppdater(e.getSource() == firkant);
    repaint();
  }

  public static void main(String[] args) {
    SwingUtilities.invokeLater
      (new Runnable() {
      public void run() {
        new Tegnekontroll();
      }
    });
  }
}