From b79c0330c0e3d65a4b6bd56c0931a4d6245e9d55 Mon Sep 17 00:00:00 2001
From: natrixaeria <upezu@student.kit.edu>
Date: Fri, 20 Dec 2019 06:39:40 +0100
Subject: Add color output to `output`

---
 src/solvers.rs | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

diff --git a/src/solvers.rs b/src/solvers.rs
index 9bf0ad6..ce5568c 100644
--- a/src/solvers.rs
+++ b/src/solvers.rs
@@ -32,20 +32,18 @@ impl Wall {
 
     pub fn output(&self, n: u32, h: u32) {
         let mut stones = vec![0; n as usize];
+        let mut toggle = 0;
+        let colors = ["\x1b[31m", "\x1b[32m", "\x1b[33m", "\x1b[34m", "\x1b[35m", "\x1b[36m"];
         for row in 0..h {
             self.calculate_row(row, &mut stones);
             for &len in stones.iter() {
-                if len <= 1 {
-                    print!("◼");
-                } else if len > 1 {
-                    print!("◢");
-                    for _ in 0..(len - 2) {
-                        print!("◼");
-                    }
-                    print!("◣");
+                print!("{}", colors[toggle]);
+                toggle = (toggle + 1) % colors.len();
+                for _ in 0..len {
+                    print!("█");
                 }
             }
-            println!("");
+            println!("\x1b[m");
         }
     }
 }
-- 
cgit v1.2.3-54-g00ecf