haskell - 如何在 haskell 中使用 ffmpeg-light 查找 mp4 元数据？

Question

我正在使用 ffmpeg-light、JuicyPixels 和光泽度来通过 Haskell 来显示视频。我想找到自动播放的视频的元数据，但我还没有找到方法。

我想访问元数据，例如视频的分辨率和帧速率。

你能帮我吗？

编辑:

我已经尝试过您的解决方案@CRDrost，但视频现在以正常速度的 2 倍播放。我假设函数 imageReaderTime 给出了错误的时间戳。

编辑2:

播放速度异常是ffmpeg-light库的一个bug。我打开了issue在 github 存储库中。

我更新的代码:

import Graphics.Gloss
import Codec.FFmpeg
import Codec.FFmpeg.Juicy
import Codec.Picture
import Control.Applicative
import Data.Maybe
import Graphics.Gloss.Juicy
import Control.Monad
-- import System.IO.Unsafe (unsafePerformIO)-- for debugging purposes

resolution :: (Int,Int)
resolution = (640, 360)

frameCount :: Int
frameCount = 100

main :: IO ()
main = do
    initFFmpeg
    (getFrame, cleanup) <- imageReaderTime "big_buck_bunny.mp4"
    frames <- replicateM frameCount $ nextFrame getFrame
    cleanup
    animate (InWindow "Nice Window" resolution (10,10)) white (frameAt frames)

nextFrame :: IO (Maybe (Image PixelRGB8, Double)) -> IO (Picture, Float)
nextFrame getFrame = mapSnd realToFrac . mapFst fromImageRGB8 . fromJust <$> getFrame

frameAt :: [(Picture, Float)] -> Float -> Picture
frameAt list time = fst . head . dropWhile ((< time) . snd) $ list

mapFst :: (a -> c) -> (a, b) -> (c, b)
mapFst f (a, b) = (f a, b) -- applies f to first element of a 2-tuple

mapSnd :: (b -> c) -> (a, b) -> (a, c)
mapSnd f (a, b) = (a, f b) -- applies f to the second element of a 2-tuple

Answer 1

(a) 我认为 void cleanup 是多余的，只有 cleanup 有效，但我喜欢你不能 100% 确定那是什么IO()值确实如此。

我没有看到读取 FPS 的直接方法，但是 imageReaderTime 会生成以秒为单位的时间戳，这将为您提供一个很好的指示器。要传播时间戳，您需要修改:

nextFrame :: IO (Maybe (Image PixelRGB8, Double)) -> IO (Double, Picture)
nextFrame getFrame = fmap fromImageRGB8 . swap . fromJust <$> getFrame

那么你会说:

stampedFrames <- replicateM frameCount $ nextFrame getFrame
let (tstamps, frames) = unzip stampedFrames
let approx_fps = fromIntegral (length tstamps) / (maximum tstamps - minimum tstamps)

最后，您可以将 approx_fps 作为参数传递给 frameAt，它必须使用 Double 而不是 Float > 或者一些类型强制函数。

但是，对于您正在做的事情，可能更好的是拥有类似的东西:

frameAt :: [(Double, Picture)] -> Double -> Picture
frameAt list time = snd . head . dropWhile ((< time) . fst) $ list

这获取列表，删除第一个元素(时间戳)小于请求时间的所有元素，然后返回此后出现的第一对元素(图片)的第二个元素(图片)。 无需猜测 FPS。