X-Git-Url: http://git.cielonegro.org/gitweb.cgi?a=blobdiff_plain;f=Sound%2FEsounD%2FInternals.hs;h=997899a6291e0ae05a100110a5300564e45d0b58;hb=dddadcc3746636546f4010e37aafef211d75cb0d;hp=987dd1e285262891ba79e69a6e985b0c0e7d0bb0;hpb=951367c7c17ddbc417a998fe9a19d6434f63f386;p=EsounD.git

diff --git a/Sound/EsounD/Internals.hs b/Sound/EsounD/Internals.hs
index 987dd1e..997899a 100644
--- a/Sound/EsounD/Internals.hs
+++ b/Sound/EsounD/Internals.hs
@@ -1,27 +1,110 @@
+{-# LANGUAGE
+    EmptyDataDecls
+  , FlexibleInstances
+  , MultiParamTypeClasses
+  , UnicodeSyntax
+  #-}
 module Sound.EsounD.Internals
-    ( wrapSocket
+    ( Frame(..)
+
+    , Channels(..)
+    , Mono
+    , Stereo
+    , interleave
+    , deinterleave
+
+    , toLSV
+    , wrapSocket
     , closeSocket
     , withCStrOrNull
+    , failOnError
     )
     where
-
 import Bindings.EsounD
+import Data.Int
+import Data.StorableVector      as S
+import Data.StorableVector.Lazy as L
 import Foreign.C.String
 import Foreign.C.Types
 import Foreign.Ptr
+import Foreign.Storable
 import System.IO
 import System.Posix.IO
 import System.Posix.Types
 
-wrapSocket :: String -> CInt → IO Handle
+class Storable fr ⇒ Frame fr where
+    frameFmt  ∷ fr → C'esd_format_t
+    frameSize ∷ fr → Int
+
+instance Frame Int8 where
+    frameFmt  _ = c'ESD_BITS8
+    frameSize _ = 1
+
+instance Frame Int16 where
+    frameFmt  _ = c'ESD_BITS16
+    frameSize _ = 2
+
+class Channels ch where
+    channelFmt  ∷ ch → C'esd_format_t
+    numChannels ∷ ch → Int
+
+-- Mono
+data Mono
+
+instance Channels Mono where
+    channelFmt  _ = c'ESD_MONO
+    numChannels _ = 1
+
+-- Stereo
+data Stereo
+
+instance Channels Stereo where
+    channelFmt  _ = c'ESD_STEREO
+    numChannels _ = 2
+
+{-# INLINE interleave #-}
+interleave ∷ Storable α ⇒ L.Vector α → L.Vector α → L.Vector α
+interleave l r
+    -- THINKME: consider using storablevector-streamfusion
+    = let Just (lFr, l') = L.viewL l
+          Just (rFr, r') = L.viewL r
+          lr' = interleave l' r'
+      in
+        L.cons lFr (L.cons rFr lr')
+
+{-# INLINE deinterleave #-}
+deinterleave ∷ Storable α ⇒ L.Vector α → (L.Vector α, L.Vector α)
+deinterleave v
+    -- THINKME: consider using storablevector-streamfusion
+    = let (lr, v') = L.splitAt 2 v
+      in
+        if L.null lr then
+            (L.empty, L.empty)
+        else
+            let Just (lFr, r) = L.viewL lr
+                Just (rFr, _) = L.viewL r
+                (l', r') = deinterleave v'
+            in
+              (L.cons lFr l', L.cons rFr r')
+
+-- Utility functions
+toLSV ∷ Storable α ⇒ S.Vector α → L.Vector α
+toLSV v = L.fromChunks [v]
+
+wrapSocket ∷ String → CInt → IO Handle
 wrapSocket e (-1) = fail e
 wrapSocket _ fd   = fdToHandle (Fd fd)
 
-closeSocket :: Handle → IO ()
+closeSocket ∷ Handle → IO ()
 closeSocket h = do (Fd fd) ← handleToFd h
                    _       ← c'esd_close (fromIntegral fd)
                    return ()
 
-withCStrOrNull :: Maybe String → (CString → IO a) → IO a
+withCStrOrNull ∷ Maybe String → (CString → IO a) → IO a
 withCStrOrNull Nothing  f = f nullPtr
 withCStrOrNull (Just s) f = withCString s f
+
+failOnError ∷ Monad m ⇒ String → (α → Bool) → α → m α
+failOnError msg isErr rv
+    | isErr rv  = fail msg
+    | otherwise = return rv