1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
{-|
Module : PFM
Description : Debevec PFM reader
Copyright : (c) 2019, Yann Herklotz Grave
License : GPL-3
Maintainer : ymherklotz [at] gmail [dot] com
Stability : experimental
Portability : POSIX
Debevec PFM reader
-}
module PFM where
import Control.Applicative ((<|>))
import Data.Attoparsec.ByteString (Parser)
import qualified Data.Attoparsec.ByteString as P
import Data.Binary.Get (runGet)
import Data.Binary.IEEE754 (getFloat32be, getFloat32le,
putFloat32le)
import Data.Binary.Put (runPut)
import Data.ByteString (ByteString)
import qualified Data.ByteString as B
import Data.ByteString.Lazy (fromStrict)
import qualified Data.ByteString.Lazy as BL
import Data.Foldable (fold)
import Data.Text (Text)
import qualified Data.Text as T
import qualified Data.Text.Encoding as T
import Data.Word (Word8)
data PFMImage = PFMImage { pfmWidth :: Int
, pfmHeight :: Int
, pfmColour :: [[PFMColour]]
} deriving (Show)
data PPMImage = PPMImage { ppmWidth :: Int
, ppmHeight :: Int
, ppmColour :: [[PPMColour]]
} deriving (Show)
data PFMColour = PFMColour { r :: Float
, g :: Float
, b :: Float
}
| PFMMono Float
deriving (Show)
data PPMColour = PPMColour { rw :: Word8
, gw :: Word8
, bw :: Word8
}
| PPMMono Word8
deriving (Show)
data Endianness = Big | Little
data ImageType = MonoImage | ColourImage
matchText :: Text -> Parser ByteString
matchText = P.string . T.encodeUtf8
magicNum :: Parser ImageType
magicNum = do
match <- T.decodeUtf8 <$> (matchText "Pf" <|> matchText "PF")
if match == "Pf"
then return MonoImage
else return ColourImage
skipNewline :: Parser ()
skipNewline = P.skip isNewline
where
isNewline w = w == 13 || w == 10
skipSpace :: Parser ()
skipSpace = P.skip (== 32)
decode :: (Read a) => [Word8] -> a
decode = read . T.unpack . T.decodeUtf8 . B.pack
matchMult :: String -> Parser [Word8]
matchMult = P.many1 . P.satisfy . P.inClass
num :: Parser Int
num = decode <$> matchMult "0-9"
endianness :: Parser Endianness
endianness =
getEnd . (<(0.0 :: Float)) . decode <$> matchMult "0-9.-"
where
getEnd True = Little
getEnd False = Big
float :: Endianness -> Parser Float
float e =
runGet conv . fromStrict <$> P.take 4
where
conv = case e of
Big -> getFloat32be
Little -> getFloat32le
header :: Parser (Int, Int, Endianness, ImageType)
header = do
n <- magicNum
skipNewline
n1 <- num
skipSpace
n2 <- num
skipNewline
s <- endianness
skipNewline
return (n1, n2, s, n)
parseColour :: Endianness -> Parser PFMColour
parseColour e = do
ri <- float e
gi <- float e
bi <- float e
return $ PFMColour ri gi bi
parseMono :: Endianness -> Parser PFMColour
parseMono e = PFMMono <$> float e
parser :: Parser PFMImage
parser = do
(w, h, e, i) <- header
c <- P.many1 . P.count w $ fun i e
return $ PFMImage w h c
where
fun i = case i of
ColourImage -> parseColour
MonoImage -> parseMono
magicNumPFM :: [[PFMColour]] -> Text
magicNumPFM ((PFMColour{}:_):_) = "PF"
magicNumPFM ((PFMMono{}:_):_) = "Pf"
magicNumPFM _ = "PF"
tShow :: (Show a) => a -> Text
tShow = T.pack . show
encFloat :: Float -> BL.ByteString
encFloat = runPut . putFloat32le
encodeColourPFM :: PFMColour -> BL.ByteString
encodeColourPFM (PFMColour ri gi bi) =
encFloat ri <> encFloat gi <> encFloat bi
encodeColourPFM (PFMMono m) =
encFloat m
encodeColourPPM :: PPMColour -> BL.ByteString
encodeColourPPM (PPMColour ri gi bi) =
BL.pack [ri, gi, bi]
encodeColourPPM (PPMMono m) =
BL.pack [m, m, m]
encode :: PFMImage -> BL.ByteString
encode (PFMImage w h c) =
fromStrict (T.encodeUtf8 he) <> body
where
he = magicNumPFM c <> "\n" <> tShow w <> " " <> tShow h <> "\n-1.0\n"
body = fold . fold $ fmap encodeColourPFM <$> c
encodePPM :: PPMImage -> BL.ByteString
encodePPM (PPMImage w h c) =
fromStrict (T.encodeUtf8 he) <> body
where
he = "P6" <> "\n" <> tShow w <> " " <> tShow h <> "\n255\n"
body = fold . fold $ fmap encodeColourPPM <$> c
parse :: ByteString -> PFMImage
parse s = case P.parseOnly parser s of
Left str -> error str
Right i -> i
|
