From efd533331d6a7f0c51ef857af448a6c84c3084ed Mon Sep 17 00:00:00 2001
From: Camil Staps
Date: Fri, 12 Feb 2016 15:01:00 +0100
Subject: Removed spaces in path

---
 Assignment1/CamilStaps-assignment1-shift.hs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 Assignment1/CamilStaps-assignment1-shift.hs

(limited to 'Assignment1/CamilStaps-assignment1-shift.hs')

diff --git a/Assignment1/CamilStaps-assignment1-shift.hs b/Assignment1/CamilStaps-assignment1-shift.hs
new file mode 100644
index 0000000..35a34e7
--- /dev/null
+++ b/Assignment1/CamilStaps-assignment1-shift.hs
@@ -0,0 +1,14 @@
+import System.Environment
+import Data.String.Utils
+
+main = do
+	args <- getArgs
+	let input = replace " " "" (head args)
+	print $ map (`shiftN` (read $ args!!1)) input
+
+-- shiftN c i: shift c i times forward (i.e. shiftN 'A' 1 == 'B')
+-- Only for uppercase letters
+shiftN :: Char -> Int -> Char
+shiftN c 0 = c
+shiftN 'Z' i = shiftN 'A' (i-1)
+shiftN c i = shiftN (toEnum ((fromEnum c) + 1)) (i-1)
\ No newline at end of file
-- 
cgit v1.2.3