/*
 * AwtLoad.java
 *
 * Created on March 12, 2003, 12:08 PM
 */
import java.awt.*;
import java.util.Hashtable;
import java.util.StringTokenizer;

public class AwtLoad extends Component
{
  private Polygon p1, p2;
  private int prevWidth=0, prevHeight=0;
  private boolean repetitive = false;
  private boolean singlePaint = false;
  private boolean cloneGraphics = false;
  private int paintCount = 0;
  private int repeatCount = 0;
  public boolean running = true;
  private boolean doTriangles = false;
  private boolean doText = false;
  private boolean doLines = false;
  private boolean doBackground = true;
  private boolean multiColored = false;
  private boolean doGc = false;
  static private Color[] colors = {
    Color.black,
    Color.blue,
    Color.cyan,
    Color.darkGray,
    Color.gray,
    Color.green,
    Color.lightGray,
    Color.magenta,
    Color.orange,
    Color.pink,
    Color.red,
    Color.white,
    Color.yellow
  };
  static private int colorIndex = 0;
  static private Color getNextColor ()
  {
    if (++colorIndex >= colors.length)
      colorIndex = 0;
    return colors[colorIndex];
  }
  
  /** Hash table to look up command-line arguments */
  protected Hashtable arguments = new Hashtable ();
  
  public AwtLoad (String[] args)
  {
    System.out.println("AwtLoad constructor starting");
    for( int i=0; i<args.length; i++ )
    {
      StringTokenizer tokens = new StringTokenizer (args[i],"=");
      String argname = tokens.nextToken ().toUpperCase ();
      String argvalue = "";
      if( tokens.hasMoreTokens () )
        argvalue = tokens.nextToken ();
      arguments.put (argname,argvalue);
    }
    if (arguments.containsKey ("CLONE"))
    {
      cloneGraphics = true;
      singlePaint = true;
      repetitive = true;
    }
    else if (arguments.containsKey ("SINGLEPAINT"))
    {
      singlePaint = true;
      repetitive = true;
    }
    else
      repetitive = arguments.containsKey ("REPEAT");  // load-test with lots of repaints
    if (arguments.containsKey ("BLANK") )
      doBackground = false;
    else 
    {
      if (arguments.containsKey ("NOBG"))
        doBackground = false;
      if (arguments.containsKey ("TRIANGLES"))
        doTriangles = true;
      if (arguments.containsKey ("LINES"))
        doLines = true;
      if (arguments.containsKey ("TEXT"))
        doText = true;
    }
    multiColored = arguments.containsKey ("COLOR") || arguments.containsKey ("COLOUR");
    repeatCount = getArgumentInteger ("REPEAT");
    doGc = arguments.containsKey ("GC");
    System.out.println ("Running with:");
    System.out.println ("  repetitive = " + repetitive);
    System.out.println ("  repeatCount = " + repeatCount);
    System.out.println ("  singlePaint = " + singlePaint);
    System.out.println ("  multiColored = " + multiColored);
    System.out.println ("  cloneGraphics = " + cloneGraphics);
    System.out.println ("  doBackground = " + doBackground);
    System.out.println ("  doTriangles = " + doTriangles);
    System.out.println ("  doLines = " + doLines);
    System.out.println ("  doText = " + doText);
    System.out.println("AwtLoad constructor finished");
  }
  
  static public void help()
  {
    System.out.println("AWT Load testing program");
    System.out.println("Valid args:");
    System.out.println(" REPEAT        repeat indefinitely");
    System.out.println(" REPEAT=nnn    paint nnn times (default 1)");
    System.out.println(" SINGLEPAINT   do all the drawing in a single paint, with no repaint (implies REPEAT)");
    System.out.println(" BLANK         empty repaints - no graphical objects");
    System.out.println(" NOBG          no background fill");
    System.out.println(" LINES         draw some lines");
    System.out.println(" TRIANGLES     draw some filled 3-sided polygons");
    System.out.println(" TEXT          show some text");
    System.out.println(" COLOR         use crazy colours (COLOUR also works)");
    System.out.println(" GC            do a System.gc every paint");
    System.out.println(" CLONE         clone the Graphics object for each repaint (implies SINGLEPAINT)");
  }
  
  /** Get the value of a command-line argument which was given with an equals sign and a value.
   * @param arg The upper-case text of the command-line argument to check
   * @return The value of the argument or null
   */
  public String getArgumentValue (String arg)
  {
    String returnValue = (String)arguments.get (arg);
    if( returnValue != null && returnValue.equals ("") )
      returnValue = null;
    return returnValue;
  }
  
  /** Get the value of a command-line argument which was given with an equals sign and a value.
   * @param arg The upper-case text of the command-line argument to check
   * @return The value of the argument or null
   */
  public int getArgumentInteger (String arg)
  {
    int returnInt = 0;
    String returnValue = (String)arguments.get (arg);
    if( returnValue != null && ! returnValue.equals ("") )
    {
      try
      {
        returnInt = Integer.parseInt (returnValue);
      }
      catch (Throwable t)
      {
      }
    }
    return returnInt;
  }
  
  public void paint ( Graphics g )
  {
    do
    {
      if (repetitive)
      {
        if ((++paintCount%100)==0)
          System.out.println ("Paint count = " + paintCount);
        if (repeatCount != 0 && paintCount >= repeatCount)
          repetitive = false;
      }
      else
        System.out.println("paint " + getWidth() + " by " + getHeight() );
      if( getWidth () != prevWidth || getHeight () != prevHeight )
      {
        p1 = null;
        prevWidth = getWidth();
        prevHeight = getHeight();
        System.out.println("paint size now " + getWidth() + " by " + getHeight() );
      }
      if( p1 == null )
      {
        int h = getHeight();
        int w = getWidth();
        p1 = new Polygon ();

        p1.addPoint ( w/4, h/5 );
        p1.addPoint ( w/3+30, h/2 );
        p1.addPoint ( w/6, h*2/3 );

        p2 = new Polygon ();

        p2.addPoint ( w/2, h/8 );
        p2.addPoint ( w-w/6, h/2 );
        p2.addPoint ( w/3-30, h/2 );

      }
      if (cloneGraphics)
      {
        g = getGraphics();  // basically clones g
      }
      if (doBackground)
      {
        g.setColor (Color.white);
        g.fillRect (0,0,getWidth (),getHeight ());
      }
      if (doTriangles)
      {
        g.setColor (multiColored?getNextColor():Color.blue);
        g.fillPolygon( p1 );
        g.setColor (multiColored?getNextColor():Color.orange);
        g.fillPolygon ( p2 );
      }
      if (doLines)
      {
        int offset = paintCount % getHeight ();
        g.setColor (multiColored?getNextColor():Color.red);
        g.drawLine (offset,0,getWidth (),getHeight ()-offset);
        g.setColor (multiColored?getNextColor():Color.red);
        g.drawLine (getWidth (),offset,0,getHeight ()-offset);
      }
      if (doText)
      {
        g.setColor (multiColored?getNextColor():Color.black);
        String s = "Paint number " + paintCount;
        g.drawString (s,5,25);
      }
      if (!repetitive)
        running = false;
      else
      {
        if (cloneGraphics)
          g.dispose ();
        if (doGc)
          System.gc();
        if (singlePaint)
          Toolkit.getDefaultToolkit ().sync ();
        else
          repaint();
      }
    } while (singlePaint && running);
  }
  
  public void finalize () throws Throwable
  {
    System.out.println("AwtLoad finalize called");
    repetitive = false;
    super.finalize ();
  }
  
  public static void main ( String[] args )
  {
    if (args.length == 0)
    {
      help();
    }
    else
    {
      Natives.configureGarbageCollector (1000000);
      try
      {
        System.out.println("AwtLoad - gcj/libjava/xlib AWT load test");
        java.awt.Frame f = new java.awt.Frame ();
        System.out.println("AwtLoad created frame");
        AwtLoad comp = new AwtLoad (args);
        f.add ( comp );
        System.out.println("AwtLoad created and added component");
        f.pack ();
        System.out.println("AwtLoad packed");
        f.setVisible (true);
        System.out.println("AwtLoad setVisible(true)");
        f.repaint();
        System.out.println("AwtLoad repaint called");
        while (comp.running)
          Thread.currentThread ().sleep (100);  // wait for final paint to finish
        Toolkit.getDefaultToolkit ().sync ();
        f.remove (comp);
        f.dispose ();
        f = null;
        comp = null;
        Toolkit.getDefaultToolkit ().sync ();
        for (int i=0; i<5; i++)
        {
          System.out.println("AwtLoad collecting stuff - pass " + (i+1) );
          System.gc();
          Natives.flushGC ();
        }
        System.out.println("AwtLoad Stopping now");
        System.exit (0);
      }
      catch (Throwable t)
      {
        t.printStackTrace ();
      }
    }
  }
  
  public Dimension getPreferredSize ()
  {
    final Dimension dim = new Dimension (640,480);
    return dim;
  }
  public Dimension getMinimumSize ()
  {
    final Dimension dim = new Dimension (20,20);
    return dim;
  }
  
}