1 files changed, 12 insertions, 0 deletions
diff --git a/height-map-display/src/map/square.c b/height-map-display/src/map/square.c
index 54add55..9109123 100644
--- a/height-map-display/src/map/square.c
+++ b/height-map-display/src/map/square.c
@@ -40,3 +40,15 @@ Square* find_selected_square(Square*** squares, int x, int y) {
 
     return NULL;
 }
+
+float* get_terrain_color(Terrain_Type terrain) {
+    static float colors[6][3] = {
+        {0.0f, 0.0f, 0.0f}, // init
+        {0.0f, 0.5f, 0.0f}, // rough
+        {0.2f, 0.7f, 0.2f}, // fairway
+        {0.3f, 0.8f, 0.3f}, // green
+        {0.1f, 0.2f, 0.8f}, // water
+        {0.9f, 0.7f, 0.7f}, // sand
+    };
+    return colors[terrain];
+}