Categories: Uncategorized

[Haskell]Input PGM file part2

What I want to do

Yesterday’s result can read header.
Next is reading pixel information.

Program

It’s like following.
import System.IO
import System.Environment (getArgs)

main = do
    args <- getArgs
    if length args <= 1 then do
        print "usage : ./pgm-input input.pgm output.pgm"
    else do
        infile <- openFile (head args) ReadMode

        -- Read PGM Header
        pgm_type <- hGetLine infile
        pgm_comment <- hGetLine infile
        pgm_max_brightness <- hGetLine infile
        pgm_size <- hGetLine infile
        -- Read PGM Pixels-
        src_pixels <- hGetContents infile

        -- Output PGM File
        fh <- openFile (args !! 1) WriteMode
        hPutStrLn fh pgm_type
        hPutStrLn fh pgm_comment
        hPutStrLn fh pgm_max_brightness
        hPutStrLn fh contents
        hPutStrLn fh pgm_size
        hPutStrLn fh src_pixels

        hClose infile
        hClose fh

Input

Output

It seems to work.
zuqqhi2