2013-02-02 12 views
11

Przetwarzam dźwięk za pomocą portaudio. Wiązania FKI w pakiecie wywołują wywołania zwrotne zdefiniowane przez użytkownika, gdy są przetwarzane dane audio. To wywołanie zwrotne powinno być obsługiwane bardzo szybko i idealnie bez żadnych operacji we/wy. Chciałem zapisać wejście audio i szybko wrócić, ponieważ moja aplikacja nie musi reagować na dźwięk w czasie rzeczywistym (w tej chwili właśnie zapisuję dane audio do pliku, później skonstruuję prosty system rozpoznawania mowy) .Rury i wywołania zwrotne w Haskell

Podoba mi się pomysł pipes i pomyślałem, że mogę użyć tej biblioteki. Problem polega na tym, że nie wiem, jak utworzyć Producer, który zwraca dane przychodzące za pośrednictwem wywołania zwrotnego.

W jaki sposób obchodzić się z przypadkiem użycia?


Oto co ja pracuję teraz, w przypadku, który pomaga (Punkt zerowy MVar nie działa już teraz, ale nie lubię przechowywania wszystkich danych w nast ... bym a przetwarzanie jej jako chodzi, a nie tylko na końcu):

{-# LANGUAGE FlexibleInstances, MultiParamTypeClasses #-} 

module Main where 

import Codec.Wav 

import Sound.PortAudio 
import Sound.PortAudio.Base 
import Sound.PortAudio.Buffer 

import Foreign.Ptr 
import Foreign.ForeignPtr 
import Foreign.C.Types 
import Foreign.Storable 

import qualified Data.StorableVector as SV 
import qualified Data.StorableVector.Base as SVB 

import Control.Exception.Base (evaluate) 

import Data.Int 
import Data.Sequence as Seq 

import Control.Concurrent 

instance Buffer SV.Vector a where 
    fromForeignPtr fp = return . SVB.fromForeignPtr fp 
    toForeignPtr = return . (\(a, b, c) -> (a, c)) . SVB.toForeignPtr 

-- | Wrap a buffer callback into the generic stream callback type. 
buffCBtoRawCB' :: (StreamFormat input, StreamFormat output, Buffer a input, Buffer b output) => 
    BuffStreamCallback input output a b -> StreamCallback input output  
buffCBtoRawCB' func = \a b c d e -> do 
    fpA <- newForeignPtr_ d -- We will not free, as callback system will do that for us 
    fpB <- newForeignPtr_ e -- We will not free, as callback system will do that for us 
    storeInp <- fromForeignPtr fpA (fromIntegral $ 1 * c) 
    storeOut <- fromForeignPtr fpB (fromIntegral $ 0 * c) 
    func a b c storeInp storeOut 

callback :: MVar (Seq.Seq [Int32]) -> PaStreamCallbackTimeInfo -> [StreamCallbackFlag] -> CULong 
      -> SV.Vector Int32 -> SV.Vector Int32 -> IO StreamResult 
callback seqmvar = \timeinfo flags numsamples input output -> do 
    putStrLn $ "timeinfo: " ++ show timeinfo ++ "; flags are " ++ show flags ++ " in callback with " ++ show numsamples ++ " samples." 
    print input 
    -- write data to output 
    --mapM_ (uncurry $ pokeElemOff output) $ zip (map fromIntegral [0..(numsamples-1)]) datum 
    --print "wrote data" 

    input' <- evaluate $ SV.unpack input 
    modifyMVar_ seqmvar (\s -> return $ s Seq.|> input') 

    case flags of 
    [] -> return $ if unPaTime (outputBufferDacTime timeinfo) > 0.2 then Complete else Continue 
    _ -> return Complete 

done doneMVar = do 
    putStrLn "total done dood!" 
    putMVar doneMVar True 
    return() 

main = do 

    let samplerate = 16000 

    Nothing <- initialize 

    print "initialized" 

    m <- newEmptyMVar 
    datum <- newMVar Seq.empty 

    Right s <- openDefaultStream 1 0 samplerate Nothing (Just $ buffCBtoRawCB' (callback datum)) (Just $ done m) 
    startStream s 

    _ <- takeMVar m -- wait until our callbacks decide they are done! 
    Nothing <- terminate 

    print "let's see what we've recorded..." 

    stuff <- takeMVar datum 
    print stuff 

    -- write out wav file 

    -- let datum = 
    --  audio = Audio { sampleRate = samplerate 
    --     , channelNumber = 1 
    --     , sampleData = datum 
    --     } 
    -- exportFile "foo.wav" audio 

    print "main done" 
+0

Czy możesz podać przykładowy kod pozyskiwania danych z oddzwonienia bez rur? – Davorak

+1

Być może warto rozważyć użycie kanałów zamiast "MVar" sekwencji. Działają bardzo dobrze w przypadku problemów producentów i konsumentów. – sabauma

Odpowiedz

13

Najprostszym rozwiązaniem jest użycie MVar S w komunikacji pomiędzy zwrotnego i Producer. Oto jak:

import Control.Proxy 
import Control.Concurrent.MVar 

fromMVar :: (Proxy p) => MVar (Maybe a) ->() -> Producer p a IO() 
fromMVar mvar() = runIdentityP loop where 
    loop = do 
     ma <- lift $ takeMVar mvar 
     case ma of 
      Nothing -> return() 
      Just a -> do 
       respond a 
       loop 

Strumień zwrotna napisze Just input do MVar a finalizacja zwrotna napisze Nothing wypowiedzieć Producer.

Oto ghci przykład pokazujący, jak to działa:

>>> mvar <- newEmptyMVar :: IO (MVar (Maybe Int)) 
>>> forkIO $ runProxy $ fromMVar mvar >-> printD 
>>> putMVar mvar (Just 1) 
1 
>>> putMVar mvar (Just 2) 
2 
>>> putMVar mvar Nothing 
>>> putMVar mvar (Just 3) 
>>> 

Edit: The pipes-concurrency library zapewnia teraz tę funkcję, i to nawet ma section in the tutorial wyjaśniając konkretnie, jak go używać, aby uzyskać dane z wywołań zwrotnych.

+5

Możesz z łatwością użyć 'Chan' zamiast tego, jeśli chcesz przeplatać wykonanie, zamiast blokować' putMVar', gdy konsument nie dotarł do 'take'ing najnowszej wartości. Koszt jest taki, że możesz zużywać więcej pamięci. –

+2

Zgadza się. Zależy to od tego, czy wywołanie zwrotne ma zostać zsynchronizowane z producentem. –

Powiązane problemy