X-Git-Url: http://git.cielonegro.org/gitweb.cgi?a=blobdiff_plain;f=Sound%2FEsounD%2FInternals.hs;h=997899a6291e0ae05a100110a5300564e45d0b58;hb=HEAD;hp=4f57a3b07e694772fc507bd2590041311411e73b;hpb=2c9350d4762d69bbfcdc58212e27506c3d4f7494;p=EsounD.git

diff --git a/Sound/EsounD/Internals.hs b/Sound/EsounD/Internals.hs
index 4f57a3b..997899a 100644
--- a/Sound/EsounD/Internals.hs
+++ b/Sound/EsounD/Internals.hs
@@ -1,5 +1,7 @@
 {-# LANGUAGE
     EmptyDataDecls
+  , FlexibleInstances
+  , MultiParamTypeClasses
   , UnicodeSyntax
   #-}
 module Sound.EsounD.Internals
@@ -8,14 +10,20 @@ module Sound.EsounD.Internals
     , 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
@@ -25,24 +33,63 @@ import System.Posix.IO
 import System.Posix.Types
 
 class Storable fr ⇒ Frame fr where
-    frameFmt ∷ fr → C'esd_format_t
+    frameFmt  ∷ fr → C'esd_format_t
+    frameSize ∷ fr → Int
 
 instance Frame Int8 where
-    frameFmt _ = c'ESD_BITS8
+    frameFmt  _ = c'ESD_BITS8
+    frameSize _ = 1
 
 instance Frame Int16 where
-    frameFmt _ = c'ESD_BITS16
+    frameFmt  _ = c'ESD_BITS16
+    frameSize _ = 2
 
 class Channels ch where
-    channelFmt ∷ ch → C'esd_format_t
+    channelFmt  ∷ ch → C'esd_format_t
+    numChannels ∷ ch → Int
 
+-- Mono
 data Mono
+
 instance Channels Mono where
-    channelFmt _ = c'ESD_MONO
+    channelFmt  _ = c'ESD_MONO
+    numChannels _ = 1
 
+-- Stereo
 data Stereo
+
 instance Channels Stereo where
-    channelFmt _ = c'ESD_STEREO
+    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
@@ -56,3 +103,8 @@ closeSocket h = do (Fd fd) ← handleToFd h
 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