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

public class HiirJooned2 extends JApplet implements MouseListener{
   int abix, abiy;
   int loendur=0;
   java.util.List<Joon> jooned=new ArrayList<Joon>();
   JComboBox<String> tyybivalik=new JComboBox<String>();
   
   public HiirJooned2(){
      addMouseListener(this);
	  tyybivalik.addItem("Tavaline joon");
	  tyybivalik.addItem("Suunaga joon");
	  getContentPane().add(tyybivalik, BorderLayout.SOUTH);
   }
   public void paint(Graphics g){
     super.paint(g);
	 for(Joon j: jooned){
	    j.joonista(g);
	 }
   }
   public void mousePressed(MouseEvent e){
      if(loendur==0){
	    abix=e.getX();
		abiy=e.getY();
	  }
	  if(loendur==1){
	    if(tyybivalik.getSelectedIndex()==0){
	      jooned.add(new Joon(abix, abiy, e.getX(), e.getY()));
		}
	    if(tyybivalik.getSelectedIndex()==1){
	      jooned.add(new SuunagaJoon(abix, abiy, e.getX(), e.getY()));
		}
  	    repaint();
	  }
	  loendur++; //suurendab muutujat ühe võrra
	  if(loendur>1){
	     loendur=0;
	  }
   }
   public void mouseReleased(MouseEvent e){}
   public void mouseClicked(MouseEvent e){}
   public void mouseEntered(MouseEvent e){}
   public void mouseExited(MouseEvent e){}
   public static void main(String[] arg){
      JFrame f=new JFrame("Hiirevajutused");
	  f.add(new HiirJooned2());
	  f.setSize(400, 300);
	  f.setVisible(true);
   }
}