magarena/src/magic/model/MagicColor.java

package magic.model;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.EnumSet;
import java.util.Iterator;
import java.util.List;

import magic.translate.MText;

public enum MagicColor {

    White(MagicColorStrings._S1, 'w'),
    Blue(MagicColorStrings._S2, 'u'),
    Black(MagicColorStrings._S3, 'b'),
    Red(MagicColorStrings._S4, 'r'),
    Green(MagicColorStrings._S5, 'g')
    ;

    public static final int NR_COLORS=values().length;

    private final String name;
    private final char symbol;
    private final int mask;
    private final String displayName;

    private MagicColor(final String name, final char symbol) {
        this.name = name;
        this.symbol = symbol;
        this.mask = 1 << ordinal();
        this.displayName = MText.get(name);
    }

    /**
     * Translatable color name.
     */
    public String getDisplayName() {
        return displayName;
    }

    public String getName() {
        return name;
    }

    @Override
    public String toString() {
        return name;
    }

    public char getSymbol() {
        return symbol;
    }

    public int getMask() {
        return mask;
    }

    public boolean hasColor(final int flags) {
        return (flags&mask)!=0;
    }

    public MagicAbility getProtectionAbility() {
        switch (this) {
            case White: return MagicAbility.ProtectionFromWhite;
            case Blue: return MagicAbility.ProtectionFromBlue;
            case Black: return MagicAbility.ProtectionFromBlack;
            case Red: return MagicAbility.ProtectionFromRed;
            case Green: return MagicAbility.ProtectionFromGreen;
        }
        throw new RuntimeException("No protection ability for MagicColor " + this);
    }

    public MagicSubType getLandSubType() {
        switch (this) {
            case White: return MagicSubType.Plains;
            case Blue: return MagicSubType.Island;
            case Black: return MagicSubType.Swamp;
            case Red: return MagicSubType.Mountain;
            case Green: return MagicSubType.Forest;
        }
        throw new RuntimeException("No land subtype for MagicColor " + this);
    }

    public MagicManaType getManaType() {
        switch (this) {
            case White: return MagicManaType.White;
            case Blue: return MagicManaType.Blue;
            case Black: return MagicManaType.Black;
            case Red: return MagicManaType.Red;
            case Green: return MagicManaType.Green;
        }
        return MagicManaType.Colorless;
    }

    public static int getFlags(final String colors) {
        int flags=0;
        for (int index=0;index<colors.length();index++) {
            final char symbol=colors.charAt(index);
            final MagicColor color=getColor(symbol);
            flags|=color.getMask();
        }
        return flags;
    }

    public static MagicColor getColor(final char symbol) {
        final char usymbol=Character.toLowerCase(symbol);
        for (final MagicColor color : values()) {
            if (color.symbol==usymbol) {
                return color;
            }
        }
        throw new RuntimeException("No corresponding MagicColor for " + symbol);
    }

    public static MagicColor getColor(final String colorName) {
        for (final MagicColor color : values()) {
            if (color.name.equalsIgnoreCase(colorName)) {
                return color;
            }
        }
        throw new RuntimeException("No corresponding MagicColor for " + colorName);
    }

    public static EnumSet<MagicColor> prefixColors(final List<String> tokens) {
        final EnumSet<MagicColor> colors = EnumSet.noneOf(MagicColor.class);
        boolean matched = true;
        for (Iterator<String> iterator = tokens.iterator(); iterator.hasNext() && matched;) {
            final String name = iterator.next();
            matched = false;
            for (final MagicColor color : values()) {
                if (color.name.equalsIgnoreCase(name)) {
                    matched = true;
                    colors.add(color);
                    iterator.remove();
                    break;
                }
            }
        }
        return colors;
    }

    public static String getRandomColors(final int count) {
        final List<MagicColor> colors = new ArrayList<>(Arrays.asList(values()));
        final StringBuilder colorText=new StringBuilder();
        for (int c=count;c>0;c--) {
            final int index=MagicRandom.nextRNGInt(colors.size());
            colorText.append(colors.remove(index).getSymbol());
        }
        return colorText.toString();
    }

    public static int numColors(final MagicSource source) {
        int numColors = 0;
        for (final MagicColor color : values()) {
            if (source.hasColor(color)) {
                numColors++;
            }
        }
        return numColors;
    }

    public static boolean isColorless(final MagicSource source) {
        return numColors(source) == 0;
    }

    public static boolean isMono(final MagicSource source) {
        return numColors(source) == 1;
    }

    public static boolean isMulti(final MagicSource source) {
        return numColors(source) > 1;
    }

    // returns clockwise distance from c1 to c2
    // eg. distance(Black, White) = 3
    public static int distance(final MagicColor c1, final MagicColor c2) {
        return (c2.ordinal() - c1.ordinal() + NR_COLORS) % NR_COLORS;
    }
}