nyanmisaka / ffmpeg-rockchip

FFmpeg with async and zero-copy Rockchip MPP & RGA support
Other
326 stars 48 forks source link

HACK: Route v4l2m2m codecs to rkmpp via ENV variable #6

Open hbiyik opened 5 months ago

hbiyik commented 5 months ago

Screenshot from 2024-01-10 22-27-17 Screenshot from 2024-01-10 22-27-30

So this allows firefox (or any other app user v4l2m2m with ffmpeg, vlc?) to import DRMPrime frames from rkmpp* decoders/encoders directly. So it is kind of 0 copy for the browser which is very cool.

I am not proud of the hack since it is in the middle of ffmpeg code but it works. May be there is cleaner way to do this.

How this works: Firefox has enabled v4l2m2m support for raspberry pi, and it uses ffmpeg to communicate with V4l2 interface, Firefox does not ioctl itself, instead just imports the AVCodec, but initially it executes a binary called v4l2test to query capabilites of a target device. With the env flag FFMPEG_RKMPP_DEC_V4L2M2M=1 ffmpeg routes the v4l2 encoders or decoder to rkmpp variants, and with PR https://github.com/nyanmisaka/ffmpeg-rockchip/pull/1 RKMPP codecs provide the DRMPRime frames with HwFramesContext, they do not need to initialized with AVdevice or v4l2device at all. So rkmpp decoders/encoders act as v4l2m2m av codecs.

How to test:

#!/usr/bin/python
import sys

print(f"RKMPP Routing v4l2m2m to mpp with args {sys.argv[1:]}", file=sys.stderr)
print("V4L2_SUPPORTED", file=sys.stdout)
print("TRUE", file=sys.stdout)
print("V4L2_CAPTURE_FMTS", file=sys.stdout)
print("NV12", file=sys.stdout)
print("V4L2_OUTPUT_FMTS", file=sys.stdout)
print("H264 VP9 AV1 HEVC", file=sys.stdout)

Launch the firefox with RDD security sandbox disabled, v4l2 routed to mpp as below.

MOZ_DISABLE_RDD_SANDBOX=1 FFMPEG_RKMPP_DEC_V4L2M2M=1 firefox

Currently only h264 seems to be allowed with that flow, there seems to be VP9 hevc and AV1 or other codecs support but it seems they are blocked somewhere (some config value?). I think this can be tackled somehow and we can have a fully functional browser with HW Decoding / Encoding Enabled.

Above SS is a H264 1080p@60 SS from twitch, there seems to be an offset issue somewhere, see above color displacement, again this can also be tackled.

I am sending this PR not to be merged actually it is quite ugly, but to show the benefits of mainlining, If at least the decoder and encoder of this codec would be mainlined, we could easily PR the firefox to enable other codecs through some configs. Just wanted to show the potential here.