admin管理员组

文章数量:1399474

I'm working on creating an rtp stream that's meant to display live waveform data from Pioneer prolink players. The motivation for sending this video out is to be able to receive it in a flutter frontend. I initially was just sending a base-24 encoding of the raw ARGB packed ints per frame across a Kafka topic to it but processing this data in flutter proved to be untenable and was bogging down the main UI thread. Not sure if this is the most optimal way of going about this but just trying to get anything to work if it means some speedup on the frontend. So the issue the following implementation is experiencing is that when I run vlc --rtsp-timeout=120000 --network-caching=30000 -vvvv stream_1.sdp where

% cat stream_1.sdp
v=0
o=- 0 1 IN IP4 127.0.0.1
s=RTP Stream
c=IN IP4 127.0.0.1
t=0 0
a=tool:libavformat
m=video 5007 RTP/AVP 96
a=rtpmap:96 H264/90000

I see (among other questionable logs) the following:

[0000000144c44d10] live555 demux error: no data received in 10s, aborting
[00000001430ee2f0] main input debug: EOF reached
[0000000144b160c0] main decoder debug: killing decoder fourcc `h264'
[0000000144b160c0] main decoder debug: removing module "videotoolbox"
[0000000144b164a0] main packetizer debug: removing module "h264"
[0000000144c44d10] main demux debug: removing module "live555"
[0000000144c45bb0] main stream debug: removing module "record"
[0000000144a64960] main stream debug: removing module "cache_read"
[0000000144c29c00] main stream debug: removing module "filesystem"
[00000001430ee2f0] main input debug: Program doesn't contain anymore ES
[0000000144806260] main playlist debug: dead input
[0000000144806260] main playlist debug: changing item without a request (current 0/1)
[0000000144806260] main playlist debug: nothing to play
[0000000142e083c0] macosx interface debug: Playback has been ended
[0000000142e083c0] macosx interface debug: Releasing IOKit system sleep blocker (37463)

This is sort of confusing because when I run ffmpeg -protocol_whitelist file,crypto,data,rtp,udp -i stream_1.sdp -vcodec libx264 -f null - I see a number logs about

[h264 @ 0x139304080] non-existing PPS 0 referenced
    Last message repeated 1 times
[h264 @ 0x139304080] decode_slice_header error
[h264 @ 0x139304080] no frame!

After which I see the stream is received and I start getting telemetry on it:

Input #0, sdp, from 'stream_1.sdp':
  Metadata:
    title           : RTP Stream
  Duration: N/A, start: 0.016667, bitrate: N/A
  Stream #0:0: Video: h264 (Constrained Baseline), yuv420p(progressive), 1200x200, 60 fps, 60 tbr, 90k tbn
Stream mapping:
  Stream #0:0 -> #0:0 (h264 (native) -> h264 (libx264))
Press [q] to stop, [?] for help
[libx264 @ 0x107f04f40] using cpu capabilities: ARMv8 NEON
[libx264 @ 0x107f04f40] profile High, level 3.1, 4:2:0, 8-bit
Output #0, null, to 'pipe:':
  Metadata:
    title           : RTP Stream
    encoder         : Lavf61.7.100
  Stream #0:0: Video: h264, yuv420p(tv, progressive), 1200x200, q=2-31, 60 fps, 60 tbn
      Metadata:
        encoder         : Lavc61.19.101 libx264
      Side data:
        cpb: bitrate max/min/avg: 0/0/0 buffer size: 0 vbv_delay: N/A
[out#0/null @ 0x60000069c000] video:144KiB audio:0KiB subtitle:0KiB other streams:0KiB global headers:0KiB muxing overhead: unknown
frame= 1404 fps= 49 q=-1.0 Lsize=N/A time=00:00:23.88 bitrate=N/A speed=0.834x

Not sure why VLC is turning me down like some kind of Berghain bouncer that lets nobody in the entire night.

I initially tried just converting the ARGB ints to a YUV420p buffer and used this to create the Frame objects but I couldn't for the life of me figure out how to properly initialize it as the attempts I made kept spitting out garbled junk.

Please go easy on me, I've made an unhealthy habit of resolving nearly all of my coding questions by simply lurking the internet for answers but that's not really helping me solve this issue.

Here's the Java I'm working on (the meat of the rtp comms occurs within updateWaveformForPlayer()):

package com.bugbytz.prolink;

import .apache.kafka.clients.producer.KafkaProducer;
import .apache.kafka.clients.producer.Producer;
import .apache.kafka.clients.producer.ProducerConfig;
import .apache.kafka.clients.producer.ProducerRecord;
import .bytedeco.ffmpeg.global.avcodec;
import .bytedeco.ffmpeg.global.avutil;
import .bytedeco.javacv.FFmpegFrameGrabber;
import .bytedeco.javacv.FFmpegFrameRecorder;
import .bytedeco.javacv.FFmpegLogCallback;
import .bytedeco.javacv.Frame;
import .bytedeco.javacv.FrameGrabber;
import .deepsymmetry.beatlink.CdjStatus;
import .deepsymmetry.beatlink.DeviceAnnouncement;
import .deepsymmetry.beatlink.DeviceAnnouncementAdapter;
import .deepsymmetry.beatlink.DeviceFinder;
import .deepsymmetry.beatlink.Util;
import .deepsymmetry.beatlink.VirtualCdj;
import .deepsymmetry.beatlink.data.BeatGridFinder;
import .deepsymmetry.beatlink.data.CrateDigger;
import .deepsymmetry.beatlink.data.MetadataFinder;
import .deepsymmetry.beatlink.data.TimeFinder;
import .deepsymmetry.beatlink.data.WaveformDetail;
import .deepsymmetry.beatlink.data.WaveformDetailComponent;
import .deepsymmetry.beatlink.data.WaveformFinder;

import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.nio.ByteBuffer;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;

import static .bytedeco.ffmpeg.global.avutil.AV_PIX_FMT_RGB24;

public class App {
    public static ArrayList<Track> tracks = new ArrayList<>();
    public static boolean dbRead = false;
    public static Properties props = new Properties();
    private static Map<Integer, FFmpegFrameRecorder> recorders = new HashMap<>();
    private static Map<Integer, Integer> frameCount = new HashMap<>();

    private static final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
    private static final int FPS = 60;
    private static final int FRAME_INTERVAL_MS = 1000 / FPS;

    private static Map<Integer, ScheduledFuture> schedules = new HashMap<>();

    private static Set<Integer> streamingPlayers = new HashSet<>();

    public static String byteArrayToMacString(byte[] macBytes) {
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < macBytes.length; i++) {
            sb.append(String.format("%02X%s", macBytes[i], (i < macBytes.length - 1) ? ":" : ""));
        }
        return sb.toString();
    }

    private static void updateWaveformForPlayer(int player) throws Exception {
        Integer frame_for_player = frameCount.get(player);
        if (frame_for_player == null) {
            frame_for_player = 0;
            frameCount.putIfAbsent(player, frame_for_player);
        }

        if (!WaveformFinder.getInstance().isRunning()) {
            WaveformFinder.getInstance().start();
        }
        WaveformDetail detail = WaveformFinder.getInstance().getLatestDetailFor(player);

        if (detail != null) {
            WaveformDetailComponent component = (WaveformDetailComponent) detail.createViewComponent(
                    MetadataFinder.getInstance().getLatestMetadataFor(player),
                    BeatGridFinder.getInstance().getLatestBeatGridFor(player)
            );
            component.setMonitoredPlayer(player);
            component.setPlaybackState(player, TimeFinder.getInstance().getTimeFor(player), true);
            component.setAutoScroll(true);
            int width = 1200;
            int height = 200;
            Dimension dimension = new Dimension(width, height);
            component.setPreferredSize(dimension);
            component.setSize(dimension);
            component.setScale(1);
            component.doLayout();

            // Create a fresh BufferedImage and clear it before rendering
            BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            Graphics2D g = image.createGraphics();
            g.clearRect(0, 0, width, height);  // Clear any old content

            // Draw waveform into the BufferedImage
            component.paint(g);
            g.dispose();

            int port = 5004 + player;
            String inputFile = port + "_" + frame_for_player + ".mp4";
            // Initialize the FFmpegFrameRecorder for YUV420P
            FFmpegFrameRecorder recorder_file = new FFmpegFrameRecorder(inputFile, width, height);
            FFmpegLogCallback.set();  // Enable FFmpeg logging for debugging
            recorder_file.setFormat("mp4");
            recorder_file.setVideoCodec(avcodec.AV_CODEC_ID_H264);
            recorder_file.setPixelFormat(avutil.AV_PIX_FMT_YUV420P);  // Use YUV420P format directly
            recorder_file.setFrameRate(FPS);

            // Set video options
            recorder_file.setVideoOption("preset", "ultrafast");
            recorder_file.setVideoOption("tune", "zerolatency");
            recorder_file.setVideoOption("x264-params", "repeat-headers=1");
            recorder_file.setGopSize(FPS);
            try {
                recorder_file.start();  // Ensure this is called before recording any frames
                System.out.println("Recorder started successfully for player: " + player);
            } catch (.bytedeco.javacv.FFmpegFrameRecorder.Exception e) {
                e.printStackTrace();
            }

            // Get all pixels in one call
            int[] pixels = new int[width * height];
            image.getRGB(0, 0, width, height, pixels, 0, width);
            recorder_file.recordImage(width,height,Frame.DEPTH_UBYTE,1,3 * width, AV_PIX_FMT_RGB24, ByteBuffer.wrap(argbToByteArray(pixels, width, height)));
            recorder_file.stop();
            recorder_file.release();
            final FFmpegFrameRecorder recorder = recorders.get(player);
            FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(inputFile);


            try {
                grabber.start();
            } catch (Exception e) {
                e.printStackTrace();
            }
            if (recorder == null) {
                try {
                    String outputStream = "rtp://127.0.0.1:" + port;
                    FFmpegFrameRecorder initial_recorder = new FFmpegFrameRecorder(outputStream, grabber.getImageWidth(), grabber.getImageHeight());
                    initial_recorder.setFormat("rtp");
                    initial_recorder.setVideoCodec(avcodec.AV_CODEC_ID_H264);
                    initial_recorder.setPixelFormat(avutil.AV_PIX_FMT_YUV420P);
                    initial_recorder.setFrameRate(grabber.getFrameRate());
                    initial_recorder.setGopSize(FPS);
                    initial_recorder.setVideoOption("x264-params", "keyint=60");
                    initial_recorder.setVideoOption("rtsp_transport", "tcp");
                    initial_recorder.start();
                    recorders.putIfAbsent(player, initial_recorder);
                    frameCount.putIfAbsent(player, 0);
                    putToRTP(player, grabber, initial_recorder);
                }
                catch (Exception e) {
                    e.printStackTrace();
                }
            }
            else {
                putToRTP(player, grabber, recorder);
            }
            File file = new File(inputFile);
            if (file.exists() && file.delete()) {
                System.out.println("Successfully deleted file: " + inputFile);
            } else {
                System.out.println("Failed to delete file: " + inputFile);
            }
        }
    }

    public static void putToRTP(int player, FFmpegFrameGrabber grabber, FFmpegFrameRecorder recorder) throws FrameGrabber.Exception {
        final Frame frame = grabber.grabFrame();
        int frameCount_local = frameCount.get(player);
        frame.keyFrame = frameCount_local++ % FPS == 0;
        frameCount.put(player, frameCount_local);
        try {
            recorder.record(frame);
        } catch (FFmpegFrameRecorder.Exception e) {
            throw new RuntimeException(e);
        }
    }
    public static byte[] argbToByteArray(int[] argb, int width, int height) {
        int totalPixels = width * height;
        byte[] byteArray = new byte[totalPixels * 3];  // 4 bytes per pixel (ARGB)

        for (int i = 0; i < totalPixels; i++) {
            int argbPixel = argb[i];

            byteArray[i * 3] = (byte) ((argbPixel >> 16) & 0xFF);  // Red
            byteArray[i * 3 + 1] = (byte) ((argbPixel >> 8) & 0xFF);   // Green
            byteArray[i * 3 + 2] = (byte) (argbPixel & 0xFF);  // Blue
        }

        return byteArray;
    }


    public static void main(String[] args) throws Exception {
        VirtualCdj.getInstance().setDeviceNumber((byte) 4);
        CrateDigger.getInstance().addDatabaseListener(new DBService());
        props.put("bootstrap.servers", "localhost:9092");
        props.put("key.serializer", ".apache.kafkamon.serialization.StringSerializer");
        props.put("value.serializer", "com.bugbytz.prolink.CustomSerializer");
        props.put(ProducerConfig.MAX_REQUEST_SIZE_CONFIG, "20971520");

        VirtualCdj.getInstance().addUpdateListener(update -> {
            if (update instanceof CdjStatus) {
                try (Producer<String, DeviceStatus> producer = new KafkaProducer<>(props)) {
                    DecimalFormat df_obj = new DecimalFormat("#.##");
                    DeviceStatus deviceStatus = new DeviceStatus(
                            update.getDeviceNumber(),
                            ((CdjStatus) update).isPlaying() || !((CdjStatus) update).isPaused(),
                            ((CdjStatus) update).getBeatNumber(),
                            update.getBeatWithinBar(),
                            Double.parseDouble(df_obj.format(update.getEffectiveTempo())),
                            Double.parseDouble(df_obj.format(Util.pitchToPercentage(update.getPitch()))),
                            update.getAddress().getHostAddress(),
                            byteArrayToMacString(DeviceFinder.getInstance().getLatestAnnouncementFrom(update.getDeviceNumber()).getHardwareAddress()),
                            ((CdjStatus) update).getRekordboxId(),
                            update.getDeviceName()
                    );
                    ProducerRecord<String, DeviceStatus> record = new ProducerRecord<>("device-status", "device-" + update.getDeviceNumber(), deviceStatus);
                    try {
                        producer.send(record).get();
                    } catch (InterruptedException ex) {
                        throw new RuntimeException(ex);
                    } catch (ExecutionException ex) {
                        throw new RuntimeException(ex);
                    }
                    producer.flush();
                    if (!WaveformFinder.getInstance().isRunning()) {
                        try {
                            WaveformFinder.getInstance().start();
                        } catch (Exception ex) {
                            throw new RuntimeException(ex);
                        }
                    }
                }
            }
        });
        DeviceFinder.getInstance().addDeviceAnnouncementListener(new DeviceAnnouncementAdapter() {
            @Override
            public void deviceFound(DeviceAnnouncement announcement) {
                if (!streamingPlayers.contains(announcement.getDeviceNumber())) {
                    streamingPlayers.add(announcement.getDeviceNumber());
                    schedules.putIfAbsent(announcement.getDeviceNumber(), scheduler.scheduleAtFixedRate(() -> {
                        try {
                            Runnable task = () -> {
                                try {
                                    updateWaveformForPlayer(announcement.getDeviceNumber());
                                } catch (InterruptedException e) {
                                    System.out.println("Thread interrupted");
                                } catch (Exception e) {
                                    throw new RuntimeException(e);
                                }
                                System.out.println("Lambda thread work completed!");
                            };
                            task.run();
                        } catch (Exception e) {
                            e.printStackTrace();
                        }
                    }, 0, FRAME_INTERVAL_MS, TimeUnit.MILLISECONDS));
                }
            }

            @Override
            public void deviceLost(DeviceAnnouncement announcement) {
                if (streamingPlayers.contains(announcement.getDeviceNumber())) {
                    schedules.get(announcement.getDeviceNumber()).cancel(true);
                    streamingPlayers.remove(announcement.getDeviceNumber());
                }
            }
        });
        BeatGridFinder.getInstance().start();
        MetadataFinder.getInstance().start();
        VirtualCdj.getInstance().start();
        TimeFinder.getInstance().start();
        DeviceFinder.getInstance().start();
        CrateDigger.getInstance().start();

        try {
            LoadCommandConsumer consumer = new LoadCommandConsumer("localhost:9092", "load-command-group");
            Thread consumerThread = new Thread(consumer::startConsuming);
            consumerThread.start();

            Runtime.getRuntime().addShutdownHook(new Thread(() -> {
                consumer.shutdown();
                try {
                    consumerThread.join();
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            }));
            Thread.sleep(60000);
        } catch (InterruptedException e) {
            System.out.println("Interrupted, exiting.");
        }
    }
}

本文标签: javaVLC dead input for RTP streamStack Overflow