import javax.swing.JFrame;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Color;
import java.util.ArrayList;
import javax.swing.JComponent;


/**
    This program draws Tartans, given a specification
      in terms of a Sett.
*/
public class A6 
{  

  // colors not available as constants from the Color class   
  private static Color DARK_BLUE = new Color(0,0,127); 
  private static Color DARK_GREEN = new Color(0,127,0); 
  private static Color BROWN = new Color(95,35,0);
  
  
   /**
       A method to help check how setts themselves are drawn
       @return an ArrayList of Setts to be tested
    */
   
   private static ArrayList<Sett> initializeSetts()                     {    
     ArrayList<Sett> setts = new ArrayList<Sett>();
     Sett royalStewart = new Sett( 
        new Color[] {Color.RED, Color.GRAY, 
                     Color.BLACK, Color.YELLOW, 
                     Color.BLACK, Color.WHITE, 
                      Color.BLACK, DARK_GREEN, 
                      Color.RED, Color.BLACK,  
                      Color.RED, Color.WHITE},
         new int[] {72, 8, 12, 2, 2, 2, 2, 16, 8, 2, 4, 2});
     setts.add(royalStewart);
     Sett blackWatch = new Sett(
         new Color[] {DARK_BLUE, Color.BLACK,
                      DARK_BLUE, Color.BLACK,
                      DARK_BLUE, Color.BLACK,
                      DARK_GREEN, Color.BLACK, 
                      DARK_GREEN, Color.BLACK,
                      DARK_BLUE, Color.BLACK,
                      DARK_BLUE, Color.BLACK},
         new int[] {22, 2, 2, 2, 2, 2, 16, 16, 2, 16, 16, 16, 2, 2});
     setts.add(blackWatch);                                       
     Sett macLeod = new Sett(
         new Color[] {Color.RED, Color.BLACK,
                      DARK_GREEN, Color.BLACK,
                      DARK_BLUE, Color.BLACK, Color.YELLOW},
         new int[] {6, 4, 30, 20, 40, 4, 4});
     setts.add(macLeod);                                       
     Sett macDonald = new Sett(
         new Color[] {DARK_GREEN, Color.RED, 
                      DARK_GREEN, Color.RED, 
                      DARK_GREEN, Color.BLACK, 
                      Color.RED, DARK_BLUE, 
                      Color.RED, DARK_BLUE, 
                      Color.RED, DARK_BLUE},
         new int[] {16, 2, 4, 6, 24, 24, 2, 24, 6, 4, 2, 16});
     setts.add(macDonald);                                       
     return setts;                                                     }
          
 
   /**
       A method to help check how tartans themselves are drawn
       @return an ArrayList of Setts whose tartans can be tested
    */
   
   private static ArrayList<Sett> initializeSettsForTartans()          {    
     ArrayList<Sett> setts = new ArrayList<Sett>();
     Sett robRoy = new Sett(  
        new Color[] {Color.RED, Color.BLACK}, 
        new int[] {8, 8});
     setts.add(robRoy);
     Sett macQuarrie = new Sett(
         new Color[] {Color.RED, DARK_GREEN,
                      Color.RED, DARK_GREEN,
                      Color.RED, DARK_GREEN},
         new int[] {32, 2, 2, 2, 8, 24});         
     setts.add(macQuarrie);
     Sett napier = new Sett(  
        new Color[] {Color.WHITE, DARK_BLUE, 
                      Color.BLACK, Color.WHITE, 
                      Color.BLACK, Color.WHITE, 
                      Color.BLACK, Color.WHITE, 
                      Color.BLACK, Color.WHITE, Color.BLACK}, 
        new int[] {2, 24, 8, 4, 4, 8, 4, 4, 4, 4, 8});
     setts.add(napier);
     Sett brodie = new Sett(
         new Color[] {Color.BLACK ,Color.RED,
                       Color.BLACK ,Color.YELLOW,
                       Color.BLACK, Color.RED},
         new int[] {4, 32, 16, 2, 16, 4});
     setts.add(brodie);             
     return setts;
    }
    
    
   /**
        A method to check the drawing of setts and tartans
        @param args is ignored
    */ 
   
   public static void main(String[] args)                              {
       
     // get and draw some setts  
     ArrayList<Sett> setts = initializeSetts();                                  
     
     for (int i=0; i<setts.size(); i++)                             {
         
       JFrame frame = new JFrame();
       int frameLength = 960;
       int frameHeight = 640;
       frame.setSize(frameLength, frameHeight);
       
       // in this application we don't want ot exit on close
       // frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
     
       SettComponent sc = new SettComponent(setts.get(i));
       frame.add(sc);
       frame.setVisible(true);                                           
       long finishTime = System.currentTimeMillis() + 4000;
       while (System.currentTimeMillis() < finishTime) {}            }
       
     // get and draw some tartans  
     ArrayList<Sett> settsForTartans = initializeSettsForTartans();      
     for (int i=0; i<settsForTartans.size(); i++)                     {
         
       JFrame frame2 = new JFrame();
       int frameLength = 960;
       int frameHeight = 640;
       frame2.setSize(frameLength, frameHeight);     
       TartanComponent tc = 
         new TartanComponent(new Tartan(0, 0, 
            frameLength, frameHeight, settsForTartans.get(i)));
       frame2.add(tc);
       frame2.setVisible(true);                                           
       long finishTime = System.currentTimeMillis() + 4000;
       while (System.currentTimeMillis() < finishTime) {}            } }
     
     
  /**
      A private class that wraps a sett for drawing purposes
   */

  private static class SettComponent extends JComponent
  {

     // the sett to be drawn
     
     private Sett sett;
     
     
     /**
          Constructs a component for a given sett
          @param t the sett
      */
     
     public SettComponent(Sett s)                         {
       sett = s;                                          }
       

          
     /**
       Paints the current tartan onto a the 2D analog
         of a given Graphics object
       @param g the Graphics object         
     */

   public void paintComponent(Graphics g)                    {
       
     // RecoverGraphics2D
     Graphics2D g2 = (Graphics2D) g;
     
     sett.draw(g2);                                          }

  } // end of SettComponent class
     
   
}