Peg.java

import java.util.*;

public class Peg 
{
    //class constants
    private static final char VIOLET = 'V';
    private static final char INDIGO = 'I';
    private static final char BLUE = 'B';
    private static final char GREEN = 'G';
    private static final char YELLOW = 'Y';
    private static final char ORANGE = 'O';
    private static final char RED = 'R';
    private static final char PINK = 'P';
    private static final char CYAN = 'C';
    private static final char MAROON = 'M';
    
    //instance variables
    private char color;
    private int numColors;
    private Character[] colors = {VIOLET, INDIGO, BLUE, GREEN, YELLOW, ORANGE, RED, PINK, CYAN, MAROON};
    
    //default constructor generates a peg
    //with a random color within the range
    //of colors for that difficulty level
    public Peg(int nc)
    {
        numColors = nc;
        
        Random rand = new Random();
        
        int randomNum = rand.nextInt(numColors);
        
        color = colors[randomNum];
    }
    
    //constructor that assigns color to the peg
    public Peg(int nc, Character c)
    {
        numColors = nc;
        
        for (int i = 0; i < numColors; i++)
        {
            if (colors[i] == c)
            {
                color = c;
                return;
            }
        }
        
        throw new IllegalArgumentException(c + " is not a correct peg color.");
    }
    
    public char getColor()
    {
        return color;
    }

    public boolean equals(Peg other)
    {
        if (this.color == other.color)
            return true;
        else
            return false;
    }
}