1 files changed, 17 insertions, 2 deletions

diff --git a/src/Graphics/WaveFront/Parse/Common.hs b/src/Graphics/WaveFront/Parse/Common.hs
index bfeb2d8..1554d45 100644
--- a/src/Graphics/WaveFront/Parse/Common.hs
+++ b/src/Graphics/WaveFront/Parse/Common.hs
@@ -33,10 +33,11 @@ module Graphics.WaveFront.Parse.Common where
 -- We'll need these
 --------------------------------------------------------------------------------------------------------------------------------------------
 import           Data.Text (Text, pack)
+import           Data.Maybe
 import qualified Data.Attoparsec.Text as Atto
 
 import Control.Applicative (pure, liftA2, (<$>), (<*>), (<*), (*>), (<|>))
-import Linear (V2(..), V3(..))
+import Linear (V2(..), V3(..),V4(..))
 
 import Graphics.WaveFront.Types
 
@@ -151,6 +152,20 @@ point2D :: Fractional f => Atto.Parser (V2 f)
 point2D = V2 <$> coord <*> coord
 
 
+pointTo3 :: Fractional f => Int -> f -> Atto.Parser (V3 f)
+pointTo3 3 def = V3 <$> coord <*> coord <*> coord
+pointTo3 2 def = V3 <$> coord <*> coord <*> (fromMaybe def <$> optional coord)
+pointTo3 1 def = (\x m -> case m of { Just (y,z) -> V3 x y z; Nothing -> V3 x def def })
+                    <$> coord <*> optional ( do y <- coord
+                                                z <- fromMaybe def <$> optional coord
+                                                return (y,z) )
+pointTo3 0 def = fromMaybe (V3 def def def) <$> optional (pointTo3 1 def)
+
+
+pointTo4 :: Fractional f => Int -> f -> Atto.Parser (V4 f)
+pointTo4 4 def = V4 <$> coord <*> coord <*> coord <*> coord
+pointTo4 n def = (\(V3 x y z) mw -> V4 x y z $ fromMaybe def mw) <$> pointTo3 n def <*> optional coord
+
 -- |
 clamp :: Ord n => n -> n -> n -> Atto.Parser n
 clamp lower upper n
@@ -163,4 +178,4 @@ clamp lower upper n
 -- |
 -- TODO | - Clean up and generalise
 clamped :: Integral i => i -> i -> Atto.Parser i
-clamped lower upper = Atto.decimal >>= clamp lower upper
\ No newline at end of file
+clamped lower upper = Atto.decimal >>= clamp lower upper