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

public class HiirKujundid extends JApplet implements MouseListener{
   int abix, abiy;
   int abi2x, abi2y;
   int loendur=0;
   java.util.List<Kujund> kujundid=new ArrayList<Kujund>();
   Kujund valitud=null;
   JComboBox<String> tyybivalik=new JComboBox<String>();
   
   public HiirKujundid(){
      addMouseListener(this);
	  tyybivalik.addItem("Tavaline joon");
	  tyybivalik.addItem("Suunaga joon");
	  tyybivalik.addItem("Kolmnurk");
	  //kujundid.add(new Kolmnurk(10, 10, 50, 10, 30, 60));
	  getContentPane().add(tyybivalik, BorderLayout.SOUTH);
   }
   public void paint(Graphics g){
     super.paint(g);
	 for(Kujund k: kujundid){
	    k.joonista(g);
	 }
   }
   public void mousePressed(MouseEvent e){
      boolean andmedKoos=false;	 	  
      if(loendur==0){
	    abix=e.getX();
		abiy=e.getY();
		for(Kujund k: kujundid){
		   if(k.kasPihtas(abix, abiy)){
		      valitud=k;
		   }
		}
	  }
	  if(loendur==1){
	    if(tyybivalik.getSelectedIndex()==0){
	      kujundid.add(new Joon(abix, abiy, e.getX(), e.getY())); 
		  andmedKoos=true;
		}
	    if(tyybivalik.getSelectedIndex()==1){
	      kujundid.add(new SuunagaJoon(abix, abiy, e.getX(), e.getY()));
		  andmedKoos=true;
		}
		if(tyybivalik.getSelectedIndex()==2){
		  abi2x=e.getX();
		  abi2y=e.getY();
		}
	  }
	  if(loendur==2){
	     if(tyybivalik.getSelectedIndex()==2){
		    kujundid.add(new Kolmnurk(abix, abiy, abi2x, abi2y, e.getX(), e.getY()));
			andmedKoos=true;
		 }
	  }
	  if(valitud!=null){
	     kujundid.remove(valitud);
		 valitud=null;
		 andmedKoos=true;
	  }
	  loendur++; //suurendab muutujat ühe võrra
	  if(andmedKoos){
	     loendur=0;
		 repaint();
	  }
   }
   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 HiirKujundid());
	  f.setSize(400, 300);
	  f.setVisible(true);
	  f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
   }
}