計算機科学のブログ

HaskellのI/O バイナリデータの操作 JPEGのグリッチング foldMを使ってI/Oアクションを連結する リスト、Control.Monadモジュール

入門Haskellプログラミング (Will Kurt(著)、株式会社クイープ(監修、翻訳)、翔泳社)のUNIT4(HaskellのI/O)、LESSON25(バイナリデータの操作)、25.2(JPEGのグリッチング)、foldMを使ってI/Oアクションを連結する、クイックチェック 25-4の解答を求めてみる。

コード

import qualified BinIface as BC
import Control.Monad (foldM)
import qualified Data.ByteString.Char8 as BC
import System.Environment (getArgs)
import System.Random (randomRIO)

intToChar :: Int -> Char
intToChar n = toEnum $ n `mod` 256

intToBC :: Int -> BC.ByteString
intToBC n = BC.pack [intToChar n]

replaceByte :: Int -> Int -> BC.ByteString -> BC.ByteString
replaceByte loc charVal bytes =
  let (before, rest) = BC.splitAt loc bytes
      after = BC.drop 1 rest
      newChar = intToBC charVal
   in mconcat [before, newChar, after]

randomReplaceByte :: BC.ByteString -> IO BC.ByteString
randomReplaceByte bytes = do
  let bytesLength = BC.length bytes
  loc <- randomRIO (1, bytesLength)
  charVal <- randomRIO (0, 255)
  return $ replaceByte loc charVal bytes

-- クイックチェック 25-4のリスト
glitchActions :: [BC.ByteString -> IO BC.ByteString]
glitchActions =
  [ randomReplaceByte,
    -- randomSortSection,
    randomReplaceByte,
    -- randomSortSection,
    randomReplaceByte
  ]

main :: IO ()
main = do
  args <- getArgs
  let fname = head args
  imageFile <- BC.readFile fname
  glitched <-
    foldM
      (\b f -> f b)
      imageFile
      glitchActions
  let glitchedFileName = mconcat ["glitched_", fname]
  BC.writeFile glitchedFileName glitched
  print "all done"

入出力結果(Terminal, Zsh)

% runghc sample04.hs lovecraft.jpg 
"all done"
% diff lovecraft.jpg glitched_lovecraft.jpg 
Binary files lovecraft.jpg and glitched_lovecraft.jpg differ
%