/*
 * JDOMTester.java
 *
 * based on org.jdom.samples.SAXBuilderDemo in JDOM beta-5
 *
 */
import java.io.*;
import java.util.*;

import org.jdom.*;
import org.jdom.input.SAXBuilder;
import org.jdom.output.XMLOutputter;

/**
 * <p><code>JDOMTester</code> tests speed of parsing and building documents
 * with JDOM.
 */
public class JDOMTester {

    private static final String DEFAULT_SAX_DRIVER_CLASS = "org.apache.xerces.parsers.SAXParser";
    private String saxDriverClass;
    private SAXBuilder builder;

    /**
     * <p>
     * This will create an instance of <code>{@link SAXBuilder}</code>
     *   for use in the rest of this program.
     * </p>
     *
     * @param saxDriverClass <code>String</code> name of driver class to use.
     */
    public JDOMTester(String saxDriverClass) {
        this.saxDriverClass = saxDriverClass;
        this.builder = new SAXBuilder(saxDriverClass);
    }

  /**
   */
    private String readFile(String filename)
          throws IOException, JDOMException {

        BufferedReader in = new BufferedReader(new FileReader(filename));
        String line		= "";
        StringBuffer buffer = new StringBuffer();
        while  ( (line = in.readLine())!=null && !(line.trim().equals(""))) {
        		buffer.append(line);
        }
       return buffer.toString();

    }



  /**
   */
   private void parseElement(Element rootElement) {
       rootElement.getName();
       List attributeList = rootElement.getAttributes();
       for (int i = 0; i< attributeList.size(); i++ ) {
         	Attribute attr = (Attribute)attributeList.get(i);
            attr.getName();
            attr.getValue();
       }
       List childrenList = rootElement.getChildren();
       //System.out.println("Element " + rootElement.getName() + " has " + attributeList.size() + " attributes and "
       //						+ childrenList.size() + " children." );
        for (int j=0; j<childrenList.size(); j++ ) {
        	Element childElement = (Element) childrenList.get(j);
            List list = childElement.getChildren();
        	for (int k=0; k<list.size(); k++ ) {
				parseElement( (Element) list.get(k) );
            }
        }
    }

  /**
   */
	private long parseTest(String xmlDocument, int parse_count)
    	throws JDOMException {
        	long startTime = System.currentTimeMillis();
			for (int i=0; i < parse_count; i++) {
	        	Document doc = builder.build(new StringReader(xmlDocument));
				parseElement(doc.getRootElement());
			}
    	    long endTime = System.currentTimeMillis();
	        return (endTime-startTime);

    }

  /**
   */
    private void printName(Element element) {
       	System.out.println ("Element Name = " + element.getName() );
    }

  /**
   */
	private void addAttributes(Element element, int count) {
    	for (int i=0; i<count; i++) {
            element.addAttribute("attr"+i, "attr"+i+"-value");
		}
    }

  /**
   */
    private void addChildren(Element element, int attributeCount, int childrenCount, int levels) {
	     if (levels == 0)
    	 	return;

     	int currentLevel = 1;

     	List children = new ArrayList();
     	for (int i=0; i<childrenCount; i++ ) {
			Element elem = new Element("child" + i);
	 		addAttributes(elem, attributeCount);
    	    children.add(elem);
     	}

     	for (int j=0; j<levels; j++,currentLevel++ ) {
     		element.setChildren(children);
        	for (int k=0; k<children.size(); k++) {
        		addChildren((Element)children.get(k), attributeCount, (childrenCount), (levels-currentLevel) );
        	}
     	}
    }

  /**
   */
   	private long buildTest(String xmlDocument, int build_count)
   	throws JDOMException {
        long startTime = System.currentTimeMillis();
		for (int i=0; i < build_count; i++) {
	        Document doc = builder.build(new StringReader("<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?><master/>"));
			Element rootElement = doc.getRootElement();
            addAttributes(rootElement, 2);
            addChildren(rootElement, 2, 2, 2);    //attributes, children, levels
		}
        long endTime = System.currentTimeMillis();
	    return (endTime-startTime);
    }


  /**
   */
    public void test(String filename, int repeatCount, int parse_count, int build_count, OutputStream out)
        throws IOException, JDOMException {

		String xmlDocument = readFile(filename);
		long[]  parsetimes = new long[repeatCount];
        long[]  buildtimes = new long[repeatCount];
		for (int i = 0; i<repeatCount; i++ ) {
        	parsetimes[i] = parseTest(xmlDocument, parse_count);
        	buildtimes[i] = buildTest(xmlDocument, build_count);
		}

        System.out.println("===============================================================================");
		System.out.println("Performance Metric  " + "\t\t\tTimes for " + repeatCount + " trials");
        System.out.println("===============================================================================");
		System.out.println("Parse times with " + parse_count + " iterations \t" + printTimes(parsetimes) + " milliseconds");
		System.out.println("Build times with  " + build_count + " iterations\t" + printTimes(buildtimes) + " milliseconds");
        System.out.println("===============================================================================");

    }

  /**
   */
    private String printTimes( long[] times) {
     	String result="";
		long sum=0L;
        for (int i=0; i<times.length; i++) {
            sum += times[i];
         	result = result + times[i] + "  ";
        }
        long average = sum/times.length;
        return result + "with average of " + average + " ";
    }

  /**
   */
    public static void main(String[] args) {
        if ((args.length != 2) && (args.length != 3)) {
            System.out.println("Usage: java JDOMTester [XML document filename]  [repeatCount] <sax driver class>");
            return;
        }

        // Load command line parameters
        String filename = args[0];
        String saxDriverClass = DEFAULT_SAX_DRIVER_CLASS;
        int repeatCount = Integer.parseInt(args[1]);
        if (args.length == 3) {
            saxDriverClass = args[2];
        }

        // Create an instance of the tester and test
        try {
            JDOMTester tester = new JDOMTester(saxDriverClass);
			for (int i = 1000; i <= 1000; i=i*10) {
            	tester.test(filename, repeatCount, i, i, System.out);
        	}

        } catch (JDOMException e) {
            if (e.getRootCause() != null) {
                e.getRootCause().printStackTrace();
            } else {
                e.printStackTrace();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}