Haskell - File処理(PDB形式のファイルからauthorを抽出(Data.Listモジュール, isPrefixOf関数))

すごいHaskellたのしく学ぼう!
(オーム社)
Miran Lipovača (著)
田中 英行 村主 崇行 (翻訳)
[Kindle版]

開発環境

• OS X Mavericks - Apple(OS)
• BBEdit - Bare Bones Software, Inc., Emacs (Text Editor)
• Haskell (純粋関数型プログラミング言語)
• GHC (The Glasgow Haskell Compiler) (処理系)
• The Haskell Platform (インストール方法、モジュール等)

初めてのコンピュータサイエンス(Jennifer Campbell、Paul Gries、Jason Montojo、Greg Wilson(著)長尾 高弘(翻訳))の9章(集合と辞書)、9.5(練習問題)、3.をHaskellで解いてみる。

その他参考書籍

• プログラミングHaskell (オーム社) Graham Hutton(著) 山本 和彦(翻訳)
• Real World Haskell―実戦で学ぶ関数型言語プログラミング (オライリージャパン) Bryan O'Sullivan John Goerzen Don Stewart(著) 山下 伸夫 伊東 勝利 株式会社タイムインターメディア(翻訳)

9.5(練習問題)、3.

コード(BBEdit)

Sample.hs

{-# OPTIONS -Wall -Werror #-}
import Data.List

main :: IO ()
main = do
    contents <- readFile "pdb2lxe.ent.txt"
    print $ authors contents

authors :: String -> [String]
authors = map trim . concat . map splitComma . map (drop $ length "AUTHOR") .
    filter (isPrefixOf "AUTHOR") . lines

trim :: String -> String
trim = ltrim . rtrim

ltrim :: String -> String
ltrim s = drop (length $ takeWhile (== ' ') s) s

rtrim :: String -> String
rtrim = reverse . ltrim . reverse

splitComma :: String -> [String]
splitComma = fst . foldr  (\x (ys, b) -> if x == ',' then
                                             (ys, False)
                                         else
                                             if b then
                                                 ((x:head ys):tail ys, b)
                                             else
                                                 ([x]:ys, True)) ([], False)

入出力結果(Terminal, runghc)

$ runghc Sample.hs
["P.KRISTIANSEN","M.A.RAHMAN","R.B.AALEN"]
$

{-# OPTIONS -Wall -Werror #-}を記述してるから、細かく型を指定(:: Double)しないと警告がいっぱい出た。慣れるまでは{-# OPTIONS -Wall -Werror #-}の記述を消さずに細かく型を指定していくことに。