Дешифровка текста из картинки

lab3/app/Main.hs

@@ -14,21 +14,18 @@ bitsPerByte = 1
 
 main :: IO ()
 main = do
+    putStrLn "Чтение текста из файла \"resources/biography.txt\""    
     inputText <- readFile "resources/biography.txt"
-    putStrLn $ take 30 inputText
-    let alphabet = createAlphabetFromText inputText
-    putStrLn $ show (length alphabet)
-    let encryptedText = encryptCaesar alphabet caesarShift inputText
-    putStrLn $ take 30 encryptedText
-    let encryptedTextBits = textToBits encryptedText
-    putStrLn $ concat (take 30 (map show (VU.toList encryptedTextBits)))
-    let encryptedTextFromBits = bitsToText encryptedTextBits
-    putStrLn $ take 30 encryptedTextFromBits
-    let decryptedText = decryptCaesar alphabet caesarShift encryptedTextFromBits
-    putStrLn $ take 30 decryptedText
 
-    readImageResult <- readImage "resources/david.bmp"
-    case readImageResult of
+    putStrLn "\nШифрование текста"    
+    let alphabet = createAlphabetFromText inputText
+    putStrLn $ "Размер алфавита: " ++ show (length alphabet)
+    let encryptedText = encryptCaesar alphabet caesarShift inputText
+    let encryptedTextBits = textToBits encryptedText
+
+    putStrLn "\nКодирование текста в изображение"
+    readSourceImageResult <- readImage "resources/david.bmp"
+    case readSourceImageResult of
         Left err -> putStrLn $ "Ошибка при чтении изображения: " ++ err
         Right dynImg -> do
             let img = convertRGB8 dynImg
@@ -39,3 +36,16 @@ main = do
 
             let resultImage = generateImage (encodePixel bitsPerByte img bitsPadded) width height
             saveBmpImage "tmp/david.bmp" (ImageRGB8 resultImage)
+
+
+    putStrLn "\nДекодирование текста из изображения"
+    readEncodedImageResult <- readImage "tmp/david.bmp"
+    case readEncodedImageResult of
+        Left err -> putStrLn $ "Ошибка при чтении изображения: " ++ err
+        Right dynImg -> do
+            let img = convertRGB8 dynImg
+            let bits = VU.fromList $ extractBitsFromImage bitsPerByte img
+            let fullText = bitsToText bits
+            let encryptedTextFromImage = takeWhile (/= '\NUL') fullText
+            let decryptedText = decryptCaesar alphabet caesarShift encryptedTextFromImage
+            writeFile "tmp/biography.txt" decryptedText