import javax.xml.parsers.*;
import javax.xml.transform.*;
import javax.xml.transform.dom.*;
import javax.xml.transform.stream.*;
import java.io.*;
import java.awt.*;
import java.awt.event.*;

import org.w3c.dom.*;

public class XMLJoonis3 extends Frame implements ActionListener, 
   MouseListener, MouseMotionListener{
  
  Button nupp = new Button("Lae");
  Button salvesta=new Button("Salvesta");
  Button tryki=new Button("Trüki");
  Button kustuta=new Button("Kustuta");
  String failinimi="joonistusandmed.xml";
  Choice kujundivalik=new Choice();
	Choice tegevusevalik = new Choice();
  Color[] varvid={Color.pink, Color.black, Color.yellow};
  String[] varvinimed={"Lilla", "Must", "Kollane"};
  Choice varvivalik=new Choice();
  Document d;
  Node juur, kujund, nihutatavKujund;
  MouseEvent vajutuseKoht;
 
  public XMLJoonis3(){
      setLayout(new FlowLayout());
      add(nupp);    add(salvesta);
      add(tryki);   add(kustuta);
      add(kujundivalik); add(varvivalik);
	  add(tegevusevalik);
      nupp.addActionListener(this);
      salvesta.addActionListener(this);
      tryki.addActionListener(this);
      kustuta.addActionListener(this);
      addMouseListener(this);
      addMouseMotionListener(this);
      addWindowListener(new WindowAdapter(){
        public void windowClosing(WindowEvent evt){          
         System.exit(0);
        }
      });
      kujundivalik.add("Joon");
      kujundivalik.add("Ristkülik");
      for(int i=0; i<varvid.length; i++){
         varvivalik.add(varvinimed[i]);
      }
	  tegevusevalik.add("Lisa");
	  tegevusevalik.add("Eemalda");
	  tegevusevalik.add("Nihuta");
      alusta();
      setSize(400,300);
      setVisible(true);  
  }
  
  public void actionPerformed(ActionEvent e){
    if(e.getSource()==nupp){
     try{
         d=DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(failinimi);
         juur=(Element)d.getFirstChild();
         repaint();
     } catch(Exception viga){System.out.println("Probleem lugemisel: "+viga);   }
    }
    if(e.getSource()==salvesta){
      try{
             Transformer t=TransformerFactory.newInstance().newTransformer();
             FileOutputStream valja=new FileOutputStream(failinimi);
             t.transform(new DOMSource(d), new StreamResult(valja));
             valja.close();
      } catch(Exception viga){ viga.printStackTrace();   }
    } 
    if(e.getSource()==tryki){System.out.println(d.getFirstChild());}
    if(e.getSource()==kustuta){alusta();}
  }
  
  void muudaJoonistusVarv(Graphics g, Node s6lm){
     //Muutke joonistusvärv vastavalt sõlmest tulnud andmetele
     int punane=Integer.parseInt(s6lm.getFirstChild().getTextContent());
     int roheline=Integer.parseInt(s6lm.getFirstChild().getNextSibling().getTextContent());
     int sinine=Integer.parseInt(s6lm.getFirstChild().getNextSibling().getNextSibling().getTextContent());
     g.setColor(new Color(punane, roheline, sinine));
  }
  
  public void paint(Graphics g){
    NodeList jooned=((Element)d.getFirstChild()).getElementsByTagName("joon");
    for(int i=0; i<jooned.getLength(); i++){
      Node esimene=jooned.item(i).getFirstChild();
      if(esimene.getNodeName().equals("varv")){
         muudaJoonistusVarv(g, esimene);
      }
      NodeList punktid=((Element)jooned.item(i)).getElementsByTagName("punkt");      
      for(int j=1; j<punktid.getLength(); j++){
        Node p1=punktid.item(j-1);
        Node p2=punktid.item(j);
        int x1=Integer.parseInt(((Element)p1).getElementsByTagName("x").item(0).getFirstChild().getNodeValue());
        int y1=Integer.parseInt(p1.getFirstChild().getNextSibling().getFirstChild().getNodeValue());
        int x2=Integer.parseInt(p2.getFirstChild().getFirstChild().getNodeValue());
        int y2=Integer.parseInt(p2.getFirstChild().getNextSibling().getFirstChild().getNodeValue());
        g.drawLine(x1, y1, x2, y2);
      }
    }
    NodeList ristkylikud=((Element)d.getFirstChild()).getElementsByTagName("ristkylik");
    for(int i=0; i<ristkylikud.getLength(); i++){
       Node esimene=ristkylikud.item(i).getFirstChild();
       if(esimene.getNodeName().equals("varv")){
         muudaJoonistusVarv(g, esimene);
       }
       NodeList punktid=ristkylikud.item(i).getChildNodes();
       int x1=Integer.parseInt(punktid.item(1).getFirstChild().getTextContent());
       int y1=Integer.parseInt(punktid.item(1).getFirstChild().getNextSibling().getTextContent());
       int x2=Integer.parseInt(punktid.item(2).getFirstChild().getTextContent());
       int y2=Integer.parseInt(punktid.item(2).getFirstChild().getNextSibling().getTextContent());
       if(x1>x2){
         int abi=x1; x1=x2; x2=abi;
       }
       if(y1>y2){
         int abi=y1; y1=y2; y2=abi;
       }
       g.drawRect(x1, y1, x2-x1, y2-y1);
    }
  }
  
  
  
  public void mousePressed(MouseEvent e) {
	  if (tegevusevalik.getSelectedIndex() == 0)
	  {
		  if (kujundivalik.getSelectedIndex() == 0)
		  {
			  kujund = d.createElement("joon");
		  }
		  if (kujundivalik.getSelectedIndex() == 1)
		  {
			  kujund = d.createElement("ristkylik");
		  }
		  juur.appendChild(kujund);
		  lisaVarv();
		  lisaPunkt(e.getX(), e.getY());
	  }
	  if (tegevusevalik.getSelectedIndex() >= 1) {
		  NodeList ristkylikud = ((Element)d.getFirstChild()).getElementsByTagName("ristkylik");
		  for (int i = 0; i < ristkylikud.getLength(); i++)
		  {
			  NodeList punktid = ristkylikud.item(i).getChildNodes();
			  int x1 = Integer.parseInt(punktid.item(1).getFirstChild().getTextContent());
			  int y1 = Integer.parseInt(punktid.item(1).getFirstChild().getNextSibling().getTextContent());
			  int x2 = Integer.parseInt(punktid.item(2).getFirstChild().getTextContent());
			  int y2 = Integer.parseInt(punktid.item(2).getFirstChild().getNextSibling().getTextContent());
			  if (x1 > x2)
			  {
				  int abi = x1; x1 = x2; x2 = abi;
			  }
			  if (y1 > y2)
			  {
				  int abi = y1; y1 = y2; y2 = abi;
			  }
//			  g.drawRect(x1, y1, x2 - x1, y2 - y1);
			  //Kui vajutati ristküliku sisse
			  if (e.getX() > x1 && e.getX() < x2 &&
				 e.getY() > y1 && e.getY() < y2) {
			   Node r = ristkylikud.item(i);
			   //kustutamine
			   if(tegevusevalik.getSelectedIndex()==1){
			     r.getParentNode().removeChild(r);
			     i--;
			     repaint();
			   }
			   if(tegevusevalik.getSelectedIndex()==2){
			     nihutatavKujund=r;
			     vajutuseKoht=e;
			   }
			  }
		  }
	     
	  }
  }
  
  public void mouseDragged(MouseEvent e){
	  if (tegevusevalik.getSelectedIndex() == 0)
	  {

		  if (kujundivalik.getSelectedIndex() == 0)
		  {
			  lisaPunkt(e.getX(), e.getY());
		  }
	  }
  }

  public void mouseReleased(MouseEvent e) {
	  if (tegevusevalik.getSelectedIndex() == 0)
	  {
		  if (kujundivalik.getSelectedIndex() == 1)
		  {
			  lisaPunkt(e.getX(), e.getY());
		  }
	  } 
	  if(tegevusevalik.getSelectedIndex()==2){
	     int dx=e.getX()-vajutuseKoht.getX();
	     int dy=e.getY()-vajutuseKoht.getY();
	     NodeList xnoded=((Element)nihutatavKujund).
	            getElementsByTagName("x");
	     NodeList ynoded=((Element)nihutatavKujund).
	            getElementsByTagName("y");
	     for(int i=0; i<xnoded.getLength(); i++){
 	      xnoded.item(i).setTextContent(String.valueOf(Integer.parseInt(
	       xnoded.item(i).getTextContent())+dx));
	      ynoded.item(i).setTextContent(String.valueOf(Integer.parseInt(
	       ynoded.item(i).getTextContent())+dy));
	     }
	  }
	  repaint(); 
  }

  void lisaVarv(){
     Color c=varvid[varvivalik.getSelectedIndex()];
     Element varv=d.createElement("varv");
     Element punane=d.createElement("punane");
     Element roheline=d.createElement("roheline");
     Element sinine=d.createElement("sinine");
     varv.appendChild(punane);
     varv.appendChild(roheline);
     varv.appendChild(sinine);
     punane.appendChild(d.createTextNode(String.valueOf(c.getRed())));
     roheline.appendChild(d.createTextNode(String.valueOf(c.getGreen())));
     sinine.appendChild(d.createTextNode(String.valueOf(c.getBlue())));
     kujund.appendChild(varv);
  }

  void lisaPunkt(int x, int y){
    Element punkt=d.createElement("punkt");
    Element px=d.createElement("x");
    px.appendChild(d.createTextNode(x+""));
    punkt.appendChild(px);
    Element py=d.createElement("y");
    py.appendChild(d.createTextNode(y+""));
    punkt.appendChild(py);
    kujund.appendChild(punkt);  
  }

  public void alusta(){
      try{
         d=DocumentBuilderFactory.newInstance().newDocumentBuilder().newDocument();
         juur=d.createElement("koordinaadid");
         d.appendChild(juur);
         repaint();
      } catch(Exception viga){System.out.println("Viga dokumendi loomisel: "+viga);}           
  }

  public void mouseClicked(MouseEvent e) { }
  public void mouseEntered(MouseEvent e) {}
  public void mouseExited(MouseEvent e) {}
  public void mouseMoved(MouseEvent e){}
  
  
  public static void main(String argumendid[]) throws Exception{
    new XMLJoonis3();
  }
}