[LIB-9] Add functional for build image chess diagrams

[chesshog.git] / src / main / java / org / hedgecode / chess / position / Color.java

diff --git a/src/main/java/org/hedgecode/chess/position/Color.java b/src/main/java/org/hedgecode/chess/position/Color.java
index 742dc0d..79878d9 100644 (file)
--- a/src/main/java/org/hedgecode/chess/position/Color.java
+++ b/src/main/java/org/hedgecode/chess/position/Color.java
@@ -23,7 +23,35 @@ package org.hedgecode.chess.position;
  */
 public enum Color {
 
-    WHITE,
-    BLACK
+    WHITE ('w'),
+    BLACK ('b');
+
+    private char letter;
+
+    Color(char letter) {
+        this.letter = letter;
+    }
+
+    public char letter() {
+        return letter;
+    }
+
+    public static Color byLetter(char letter) {
+        for (Color color : Color.values()) {
+            if (color.letter() == letter)
+                return color;
+        }
+        return null;
+    }
+
+    public static Color byLetter(String letter) {
+        if (letter == null || letter.length() != 1)
+            return null;
+        for (Color color : Color.values()) {
+            if (color.letter() == letter.charAt(0))
+                return color;
+        }
+        return null;
+    }
 
 }