68 lines
2.1 KiB
Haskell
68 lines
2.1 KiB
Haskell
|
module Util where
|
||
|
|
|
||
|
|
import Codec.Wavefront
|
||
|
|
|
||
|
|
import Control.Monad (sequence)
|
||
|
|
|
||
|
|
import qualified Data.Vector as V
|
||
|
|
|
||
|
|
data LoadedObject = LoadedObject
|
||
|
|
{ loTriangles :: [Float]
|
||
|
|
, loLines :: [Float]
|
||
|
|
, loPoints :: [Float]
|
||
|
|
, loTexTri :: Maybe [Float]
|
||
|
|
}
|
||
|
|
|
||
|
|
loadObj :: WavefrontOBJ -> LoadedObject
|
||
|
|
loadObj obj =
|
||
|
|
LoadedObject ts ls ps tritex
|
||
|
|
where
|
||
|
|
inter = objLocations obj
|
||
|
|
interTex = objTexCoords obj
|
||
|
|
faces = map elValue (V.toList $ objFaces obj)
|
||
|
|
lns = map elValue (V.toList $ objLines obj)
|
||
|
|
points = map elValue (V.toList $ objPoints obj)
|
||
|
|
deface (Face a b c []) =
|
||
|
|
map (\i -> inter V.! (faceLocIndex i -1)) [a, b, c]
|
||
|
|
deface _ =
|
||
|
|
error "loadObj: obj with quads encountered"
|
||
|
|
deline (Line a b) =
|
||
|
|
map (\i -> inter V.! (lineLocIndex i -1)) [a, b]
|
||
|
|
depoint (Point i) = inter V.! (i - 1)
|
||
|
|
tsLocs = concatMap deface faces
|
||
|
|
lsLocs = concatMap deline lns
|
||
|
|
psLocs = map depoint points
|
||
|
|
deLoc (Location x y z w) = [x, y, z, w]
|
||
|
|
deTex (TexCoord r s _) = [r, s]
|
||
|
|
ts = concatMap deLoc tsLocs
|
||
|
|
ls = concatMap deLoc lsLocs
|
||
|
|
ps = concatMap deLoc psLocs
|
||
|
|
defaceTex :: Face -> Maybe [TexCoord]
|
||
|
|
defaceTex (Face a b c []) =
|
||
|
|
mapM
|
||
|
|
(fmap (\x -> interTex V.! (x - 1)) . faceTexCoordIndex)
|
||
|
|
[a, b, c]
|
||
|
|
defaceTex _ =
|
||
|
|
error "loadObj: obj with quads encountered"
|
||
|
|
tritex :: Maybe [Float]
|
||
|
|
tritex = concatMap deTex <$> mftxs
|
||
|
|
mftxs :: Maybe [TexCoord]
|
||
|
|
mftxs = fmap concat (mapM defaceTex faces)
|
||
|
|
|
||
|
|
-- objLocsToPtr :: WavefrontOBJ -> IO (Ptr Float, Int)
|
||
|
|
-- objLocsToPtr obj = do
|
||
|
|
-- let ivs = objLocations obj
|
||
|
|
-- faces = map elValue $ V.toList $ objFaces obj
|
||
|
|
-- vs = concatMap
|
||
|
|
-- (\(Face a b c []) ->
|
||
|
|
-- map (\i -> ivs V.! ((faceLocIndex i) - 1)) [a, b, c])
|
||
|
|
-- faces
|
||
|
|
-- ptr <- newArray $ concatMap (\(Location x y z w) -> [x, y, z, w]) vs
|
||
|
|
-- return (ptr, length vs)
|
||
|
|
--
|
||
|
|
-- objUVsToPtr :: WavefrontOBJ -> IO (Ptr Float, Int)
|
||
|
|
-- objUVsToPtr obj = do
|
||
|
|
-- let uvs= V.toList $ objTexCoords obj
|
||
|
|
-- ptr <- newArray $ concatMap (\(TexCoord r s t) -> [r, s, t]) uvs
|
||
|
|
-- return (ptr, length uvs)
|