commit 502890dea614256b8926834db8cf3d40f908c380 Author: wangchunlin Date: Wed Sep 6 00:05:10 2023 +0800 发送端基本上正确了,接收端的多线程时间控制还存在问题,计算出来的实际等待时间还是太短了 diff --git a/receiver.cpp b/receiver.cpp new file mode 100644 index 0000000..76d8bd6 --- /dev/null +++ b/receiver.cpp @@ -0,0 +1,103 @@ +#include +#include +#include +#include +#include + +int main() { + // 创建ZeroMQ上下文 + zmq::context_t context(1); + + // 创建套接字并绑定到接收方 + zmq::socket_t socket(context, zmq::socket_type::pull); + socket.bind("ipc:///tmp/video_socket"); + + // 创建窗口用于显示视频 + cv::namedWindow("Received Video", cv::WINDOW_NORMAL); + + // 统计变量 + int frameCount = 0; + double totalTime = 0.0; + double avgFps = 0.0; + double avgInterval = 0.0; + std::chrono::steady_clock::time_point startTime = std::chrono::steady_clock::now(); + + // 接收并实时播放视频帧 + while (true) { + // 接收图像数据 + zmq::message_t message; + socket.recv(&message); + + // 解析接收到的图像数据 + int width, height, type; + memcpy(&width, message.data(), sizeof(int)); + memcpy(&height, static_cast(message.data()) + sizeof(int), sizeof(int)); + memcpy(&type, static_cast(message.data()) + sizeof(int) * 2, sizeof(int)); + + // 计算图像数据的大小 + size_t imageDataSize = message.size() - sizeof(int) * 3 - sizeof(double); + + // 创建图像矩阵 + cv::Mat frame(height, width, type, static_cast(message.data()) + sizeof(int) * 3); + + // 获取帧率信息 + double fps; + memcpy(&fps, static_cast(message.data()) + sizeof(int) * 3 + imageDataSize, sizeof(double)); + + // 显示图像 + cv::imshow("Received Video", frame); + + // 获取当前时间点 + std::chrono::steady_clock::time_point currentTime = std::chrono::steady_clock::now(); + + // 计算已经过去的时间 + std::chrono::duration elapsedSeconds = currentTime - startTime; + + // 计算应该等待的时间间隔 + double expectedInterval = 1000 / fps; + + // 计算实际需要等待的时间间隔 + double actualInterval = expectedInterval - elapsedSeconds.count() * 1000; + + // 更新统计变量 + frameCount++; + totalTime += elapsedSeconds.count(); + + // 每秒钟打印一次统计结果 + if (totalTime >= 1.0) { + avgFps = frameCount / totalTime; + avgInterval = 1000 / avgFps; + std::cout << "Average FPS (last 1 second): " << avgFps << std::endl; + std::cout << "Average Interval (last 1 second): " << avgInterval << " ms" << std::endl; + + // 重置统计变量 + frameCount = 0; + totalTime = 0.0; + } + + // 按下 ESC 键退出循环 + if (cv::waitKey(1) == 27) + break; + + // 打印帧率和时间间隔信息 + std::cout << "Received FPS: " << fps << std::endl; + std::cout << "Expected Interval: " << expectedInterval << " ms" << std::endl; + std::cout << "Elapsed Seconds: " << elapsedSeconds.count() << " s" << std::endl; + std::cout << "Actual Interval: " << actualInterval << " ms" << std::endl; + + // 等待剩余时间,以实现与原视频完全一致的帧率 + if (actualInterval > 0) + std::this_thread::sleep_for(std::chrono::milliseconds(static_cast(std::round(actualInterval)))); + + // 更新起始时间点 + startTime = currentTime; + } + + // 关闭套接字 + socket.close(); + + // 关闭 ZeroMQ 上下文 + context.close(); + + return 0; +} diff --git a/sender.cpp b/sender.cpp new file mode 100644 index 0000000..f5ea35b --- /dev/null +++ b/sender.cpp @@ -0,0 +1,65 @@ +#include +#include +#include + +int main() { + // 创建ZeroMQ上下文 + zmq::context_t context(1); + + // 创建套接字并连接到接收方 + zmq::socket_t socket(context, zmq::socket_type::push); + socket.connect("ipc:///tmp/video_socket"); + + // 打开视频文件 + cv::VideoCapture cap("1.mp4"); + if (!cap.isOpened()) { + std::cerr << "Failed to open video file" << std::endl; + return 1; + } + + // 获取视频帧率 + double fps = cap.get(cv::CAP_PROP_FPS); + std::cout << "Original FPS: " << fps << std::endl; + + // 逐帧读取并发送 + cv::Mat frame; + while (cap.read(frame)) { + // 获取图像尺寸 + int width = frame.cols; + int height = frame.rows; + int type = frame.type(); + + // 创建消息 + zmq::message_t message(sizeof(int) * 3 + frame.total() * frame.elemSize() + sizeof(double)); + + // 将图像尺寸复制到消息 + char* dataPtr = static_cast(message.data()); + memcpy(dataPtr, &width, sizeof(int)); + dataPtr += sizeof(int); + memcpy(dataPtr, &height, sizeof(int)); + dataPtr += sizeof(int); + memcpy(dataPtr, &type, sizeof(int)); + dataPtr += sizeof(int); + + // 将图像数据复制到消息 + memcpy(dataPtr, frame.data, frame.total() * frame.elemSize()); + dataPtr += frame.total() * frame.elemSize(); + + // 将帧率信息复制到消息 + memcpy(dataPtr, &fps, sizeof(double)); + + // 发送消息 + socket.send(message); + + // 打印帧率 + std::cout << "Sent FPS: " << fps << std::endl; + } + + // 关闭套接字 + socket.close(); + + // 关闭 ZeroMQ 上下文 + context.close(); + + return 0; +}