/**
   @version 1.00 2001-09-24
   @author Cay Horstmann
*/
package DOMTreeTest;
import java.awt.*;
import java.awt.geom.*;
import java.io.*;
import java.util.*;
import java.awt.event.*;
import javax.swing.*;
import javax.xml.parsers.*;
import javax.xml.transform.*;
import javax.xml.transform.dom.*;
import javax.xml.transform.stream.*;
import org.w3c.dom.*;


/**
   This program shows how to write an XML file. It saves
   a file describing a modern drawing in SVG format.
*/
public class XMLWriteTest
{  
   public static void main(String[] args)
   {  
      XMLWriteFrame frame = new XMLWriteFrame();
      frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
      frame.show();
   }
}

/**
   A frame with a panel for showing a modern drawing.
*/
class XMLWriteFrame extends JFrame
{
   public XMLWriteFrame()
   {
      setTitle("XMLWriteTest");
      setSize(WIDTH, HEIGHT);

      // add panel to frame

      panel = new RectanglePanel();
      Container contentPane = getContentPane();
      contentPane.add(panel);

      // set up menu bar

      JMenuBar menuBar = new JMenuBar();
      setJMenuBar(menuBar);

      JMenu menu = new JMenu("File");
      menuBar.add(menu);

      JMenuItem newItem = new JMenuItem("New");
      menu.add(newItem);
      newItem.addActionListener(new 
         ActionListener()
         {
            public void actionPerformed(ActionEvent event)
            {
               panel.newDrawing();
            }
         });

      JMenuItem saveItem = new JMenuItem("Save");
      menu.add(saveItem);
      saveItem.addActionListener(new 
         ActionListener()
         {
            public void actionPerformed(ActionEvent event)
            {
               try
               {
                  saveDocument();
               }
               catch (TransformerException exception)
               {
                  JOptionPane.showMessageDialog(
                     XMLWriteFrame.this, exception.toString());
               }
               catch (IOException exception)
               {
                  JOptionPane.showMessageDialog(
                     XMLWriteFrame.this, exception.toString());
               }
            }
         });

      JMenuItem exitItem = new JMenuItem("Exit");
      menu.add(exitItem);
      exitItem.addActionListener(new
         ActionListener()
         {
            public void actionPerformed(ActionEvent event)
            {
               System.exit(0);
            }
         });

   }

   /**
      Saves the drawing in SVG format.
   */
   public void saveDocument() 
      throws TransformerException, IOException
   {
      JFileChooser chooser = new JFileChooser();
      if (chooser.showSaveDialog(this) != JFileChooser.APPROVE_OPTION)
         return;
      File f = chooser.getSelectedFile();
      Document doc = panel.buildDocument();         
      Transformer t = TransformerFactory
         .newInstance().newTransformer();

      t.setOutputProperty("doctype-system", 
"http://www.w3.org/TR/2000/CR-SVG-20000802/DTD/svg-20000802.dtd"
         );
      t.setOutputProperty("doctype-public", 
         "-//W3C//DTD SVG 20000802//EN");

      t.transform(new DOMSource(doc), 
         new StreamResult(new FileOutputStream(f)));      
   }

   public static final int WIDTH = 300;
   public static final int HEIGHT = 200;  

   private RectanglePanel panel;
}

/**
   A panel that shows a set of colored rectangles
*/
class RectanglePanel extends JPanel
{ 
   public RectanglePanel()
   {  
      rects = new ArrayList();
      colors = new ArrayList();
      generator = new Random();

      DocumentBuilderFactory factory 
         = DocumentBuilderFactory.newInstance();
      try
      {
         builder = factory.newDocumentBuilder();
      }
      catch (ParserConfigurationException exception)
      {
         exception.printStackTrace();
      }
   }

   /**
      Create a new random drawing.
   */
   public void newDrawing()
   {
      int n = 10 + generator.nextInt(20);
      rects.clear();
      for (int i = 1; i <= n; i++)
      {
         int x = generator.nextInt(getWidth());
         int y = generator.nextInt(getHeight());
         int width = generator.nextInt(getWidth() - x);
         int height = generator.nextInt(getHeight() - y);
         rects.add(new Rectangle(x, y, width, height));
         int r = generator.nextInt(256);
         int g = generator.nextInt(256);
         int b = generator.nextInt(256);
         colors.add(new Color(r, g, b));
      }
      repaint();
   }

   public void paintComponent(Graphics g)
   {  
      if (rects.size() == 0) newDrawing();
      super.paintComponent(g);
      Graphics2D g2 = (Graphics2D)g;      

      // draw all rectangles
      for (int i = 0; i < rects.size(); i++)
      {
         g2.setColor((Color)colors.get(i));
         g2.fill((Rectangle2D)rects.get(i));      
      }
   }

   /**
      Creates an SVG document of the current drawing.
      @return the DOM tree of the SVG document
   */
   public Document buildDocument()
   {

      Document doc = builder.newDocument();
      Element svgElement = doc.createElement("svg");
      doc.appendChild(svgElement);
      svgElement.setAttribute("width", "" + getWidth());
      svgElement.setAttribute("height", "" + getHeight());
      
      for (int i = 0; i < rects.size(); i++)
      {
         Color c = (Color)colors.get(i);
         Rectangle2D r = (Rectangle2D)rects.get(i);
         Element rectElement = doc.createElement("rect");
         rectElement.setAttribute("x", "" + r.getX());
         rectElement.setAttribute("y", "" + r.getY());
         rectElement.setAttribute("width", "" + r.getWidth());
         rectElement.setAttribute("height", "" + r.getHeight());
         rectElement.setAttribute("fill", colorToString(c));
         svgElement.appendChild(rectElement);
      }

      return doc;
   }

   /**
      Converts a color to a hex value.
      @param c a color
      @return a string of the form #rrggbb
   */
   private static String colorToString(Color c)
   {
      StringBuffer buffer = new StringBuffer();
      buffer.append(Integer.toHexString(
         c.getRGB() & 0xFFFFFF));
      while(buffer.length() < 6) buffer.insert(0, '0');
      buffer.insert(0, '#');
      return buffer.toString();
   }

   private ArrayList rects;
   private ArrayList colors;
   private Random generator;
   private DocumentBuilder builder;
}